1
#!/bin/bash
2
PATH=/bin:/sbin:/usr/bin:/usr/sbin:/usr/local/bin:/usr/local/sbin:~/bin
3
export PATH
4

5
# 颜色选择
6
red='\033[0;31m'
7
green='\033[0;32m'
8
yellow='\033[0;33m'
9
font="\033[0m"
10

11
ffmpeg_install(){
12
# 安装FFMPEG
13
read -p "你的机器内是否已经安装过FFmpeg4.x?安装FFmpeg才能正常推流,是否现在安装FFmpeg?(yes/no):" Choose
14
if [ $Choose = "yes" ];then
15
    yum -y install wget
16
    wget --no-check-certificate https://www.johnvansickle.com/ffmpeg/old-releases/ffmpeg-4.0.3-64bit-static.tar.xz
17
    tar -xJf ffmpeg-4.0.3-64bit-static.tar.xz
18
    cd ffmpeg-4.0.3-64bit-static
19
    mv ffmpeg /usr/bin && mv ffprobe /usr/bin && mv qt-faststart /usr/bin && mv ffmpeg-10bit /usr/bin
20
fi
21
if [ $Choose = "no" ]
22
then
23
    echo -e "${yellow} 你选择不安装FFmpeg,请确定你的机器内已经自行安装过FFmpeg,否则程序无法正常工作! ${font}"
24
    sleep 2
25
fi
26
    }
27

28
stream_start(){
29
# 定义推流地址和推流码
30
read -p "输入你的推流地址和推流码(rtmp协议):" rtmp
31

32
# 判断用户输入的地址是否合法
33
if [[ $rtmp =~ "rtmp://" ]];then
34
    echo -e "${green} 推流地址输入正确,程序将进行下一步操作. ${font}"
35
      sleep 2
36
    else
37
      echo -e "${red} 你输入的地址不合法,请重新运行程序并输入! ${font}"
38
      exit 1
39
fi
40

41
# 定义视频存放目录
42
read -p "输入你的视频存放目录 (格式仅支持mp4,并且要绝对路径,例如/opt/video):" folder
43

44
# 判断是否需要添加水印
45
read -p "是否需要为视频添加水印?水印位置默认在右上方,需要较好CPU支持(yes/no):" watermark
46
if [ $watermark = "yes" ];then
47
    read -p "输入你的水印图片存放绝对路径,例如/opt/image/watermark.jpg (格式支持jpg/png/bmp):" image
48
    echo -e "${yellow} 添加水印完成,程序将开始推流. ${font}"
49
    # 循环
50
    while true
51
    do
52
        cd $folder
53
        for video in $(ls *.mp4)
54
        do
55
        ffmpeg -re -i "$video" -i "$image" -filter_complex overlay=W-w-5:5 -c:v libx264 -c:a aac -b:a 192k -strict -2 -f flv ${rtmp}
56
        done
57
    done
58
fi
59
if [ $watermark = "no" ]
60
then
61
    echo -e "${yellow} 你选择不添加水印,程序将开始推流. ${font}"
62
    # 循环
63
    while true
64
    do
65
        cd $folder
66
        for video in $(ls *.mp4)
67
        do
68
        ffmpeg -re -i "$video" -c:v copy -c:a aac -b:a 192k -strict -2 -f flv ${rtmp}
69
        done
70
    done
71
fi
72
    }
73

74
# 停止推流
75
stream_stop(){
76
    screen -S stream -X quit
77
    killall ffmpeg
78
    }
79

80
# 开始菜单设置
81
echo -e "${yellow} CentOS7 X86_64 FFmpeg无人值守循环推流 For LALA.IM ${font}"
82
echo -e "${red} 请确定此脚本目前是在screen窗口内运行的! ${font}"
83
echo -e "${green} 1.安装FFmpeg (机器要安装FFmpeg才能正常推流) ${font}"
84
echo -e "${green} 2.开始无人值守循环推流 ${font}"
85
echo -e "${green} 3.停止推流 ${font}"
86
start_menu(){
87
    read -p "请输入数字(1-3),选择你要进行的操作:" num
88
    case "$num" in
89
        1)
90
        ffmpeg_install
91
        ;;
92
        2)
93
        stream_start
94
        ;;
95
        3)
96
        stream_stop
97
        ;;
98
        *)
99
        echo -e "${red} 请输入正确的数字 (1-3) ${font}"
100
        ;;
101
    esac
102
    }
103

104
# 运行开始菜单
105
start_menu

1
*   **功能：**
2
    *   安装 FFmpeg (如果未安装)。
3
    *   配置推流地址和码流。
4
    *   判断是否需要添加水印。
5
    *   循环推送视频文件。
6
*   **关键参数：**
7
    *   `rtmp`: 推流地址和码流 (例如：`rtmp://xxxxx.xxxxx.xxx/live/xxxx-xxxx`)。
8
    *   `folder`: 视频文件存放目录 (绝对路径)。
9
    *   `watermark`: 是否添加水印 (yes/no)。
10
    *   `image`: 水印图片路径 (绝对路径)。

1
ffmpeg -re -i input.mp4 -c:v libx264 -c:a aac -f flv rtmp://your-server-url/app/stream