更新一键脚本

2025-12-12 01:00:29 +08:00 · 2024-11-16 05:50:08 +00:00 · 2024-11-16 05:50:08 +00:00 · 69cb9ac950
commit 69cb9ac950
parent 30378211b5
4 changed files with 327 additions and 1 deletions
--- a/4
+++ b/4
@ -285,10 +285,12 @@ run-stage-0:
 run-build-dev:
 	$(DOCKER) buildx build -t registry.cn-hangzhou.aliyuncs.com/silentwind/kvmd:dev \
 		--platform linux/amd64,linux/arm64,linux/arm/v7  \
+		--build-arg CACHEBUST=$(date +%s) \
 		-f build/Dockerfile . \
 		--push
 	$(DOCKER) buildx build -t silentwind0/kvmd:dev \
 		--platform linux/amd64,linux/arm64,linux/arm/v7  \
+		--build-arg CACHEBUST=$(date +%s) \
 		-f build/Dockerfile . \
 		--push

@ -296,11 +298,13 @@ run-build-release:
 	$(DOCKER) buildx build -t registry.cn-hangzhou.aliyuncs.com/silentwind/kvmd \
 		--progress plain \
 		--platform linux/amd64,linux/arm64,linux/arm/v7  \
+		--build-arg CACHEBUST=$(date +%s) \
 		-f build/Dockerfile . \
 		--push
 	$(DOCKER) buildx build -t silentwind0/kvmd \
 		--progress plain \
 		--platform linux/amd64,linux/arm64,linux/arm/v7  \
+		--build-arg CACHEBUST=$(date +%s) \
 		-f build/Dockerfile . \
 		--push

--- a/build/init.sh
+++ b/build/init.sh
@ -120,6 +120,16 @@ EOF
            && echo -e "${GREEN}One-KVM audio device is set to hw:$VIDEONUM.${NC}"
    fi

+    if [ ! -z "$CH9329SPEED" ]; then
+        sed -i "s/speed: 9600/speed: $CH9329SPEED/g" /etc/kvmd/override.yaml \
+            && echo -e "${GREEN}One-KVM CH9329 serial speed is set to $CH9329SPEED.${NC}"
+    fi
+
+    if [ ! -z "$CH9329TIMEOUT" ]; then
+        sed -i "s/read_timeout: 0.3/read_timeout: $CH9329TIMEOUT/g" /etc/kvmd/override.yaml \
+            && echo -e "${GREEN}One-KVM CH9329 timeout is set to $CH9329TIMEOUT s.${NC}"
+    fi
+
    #set htpasswd
    if [ ! -z "$USERNAME" ] && [ ! -z "$PASSWORD" ]; then
        python -m kvmd.apps.htpasswd del admin \
@ -128,7 +138,12 @@ EOF
            && echo "$USERNAME:$PASSWORD -> $USERNAME:$PASSWORD" > /etc/kvmd/ipmipasswd \
            || echo -e "${RED}One-KVM htpasswd init failed.${NC}"
    else
-        echo -e "${YELLOW} USERNAME and PASSWORD environment variables is not set, using defalut(admin/admin).${NC}"
+        echo -e "${YELLOW} USERNAME and PASSWORD environment variables are not set, using defalut(admin/admin).${NC}"
+    fi
+
+    if [ ! -z "$VIDEOFORMAT" ]; then
+        sed -i "s/format=mjpeg/format=$VIDFORMAT/g" /etc/kvmd/override.yaml \
+            && echo -e "${GREEN}One-KVM input video format is set to $VIDFORMAT.${NC}"
    fi
    
    touch /etc/kvmd/.init_flag
--- a/configs/kvmd/override.yaml
+++ b/configs/kvmd/override.yaml
@ -12,6 +12,8 @@ kvmd:
    hid:
        type: ch9329
        device: /dev/ttyUSB0
+        speed: 9600
+        read_timeout: 0.3

        jiggler:
            active: false
--- a/quick_start.sh
+++ b/quick_start.sh
@ -0,0 +1,305 @@
+#!/bin/bash
+#Install Latest Stable One-KVM Dcoker Release
+
+DOCKER_IMAGE_PATH="registry.cn-hangzhou.aliyuncs.com/silentwind/kvmd"
+DOCKER_PORT="-p 8080:8080 -p 4430:4430 -p 5900:5900 -p 623:623"
+DOCKER_NAME="kvmd"
+RED='\033[0;31m'
+GREEN='\033[0;32m'
+YELLOW='\033[0;33m'
+BLUE='\033[0;34m'
+NC='\033[0m'
+
+function check_os_architecture(){
+    osCheck=$(uname -a)
+    if [[ $osCheck =~ 'x86_64' ]];then
+        architecture="amd64"
+    elif [[ $osCheck =~ 'arm64' ]] || [[ $osCheck =~ 'aarch64' ]];then
+        architecture="arm64"
+    elif [[ $osCheck =~ 'armv7l' ]];then
+        architecture="armv7l"
+    else
+        echo "暂不支持的系统架构，请参阅官方文档，选择受支持的系统。\n退出程序"
+        exit 1
+    fi
+}
+
+function check_docker_exists() {
+    if command -v docker &> /dev/null; then
+        echo "$(docker -v)"
+    else
+        echo "Docker 未安装,退出程序"
+        exit 1
+    fi
+}
+
+function check_sudo_exists() {
+    if command -v sudo > /dev/null 2>&1; then
+        sudo_command="sudo"
+    else
+        sudo_command=""
+    fi
+}
+
+function delete_kvmd_container(){
+    if docker ps -a --format '{{.Names}}' | grep -q '^kvmd$'; then
+        $sudo_command docker stop $DOCKER_NAME
+        $sudo_command docker rm $DOCKER_NAME
+    fi
+}
+
+function check_otg_device(){
+    $sudo_command modprobe libcomposite > /dev/null|| echo -e "${YELLOW}libcomposite 内核模块加载失败${NC}"
+    if [[ "$architecture" != "amd64" ]] && [[ -d "/sys/class/udc" ]] && [[ "$(ls -A /sys/class/udc)" ]]; then
+        otg_devices=$(ls -A /sys/class/udc)
+        otg_status=$(cat /sys/class/usb_role/*/role 2>/dev/null | head -n 1)
+        echo -e "${GREEN}当前系统支持 OTG：$otg_devices OTG 状态：$otg_status${NC}"
+    else
+        echo -e "${RED}当前系统不支持 OTG，退出程序${NC}"
+        exit 1
+    fi
+    if [[ ! -d "/sys/kernel/config" ]];then
+        echo -e "${RED}当前系统不支持 configfs 文件系统，退出程序${NC}"
+        exit 1
+    fi
+}
+
+function check_video_device(){
+    if ls /dev/video* 1> /dev/null 2>&1; then
+        video_devices=($(ls /dev/video* 2>/dev/null))
+        video_num_devices=${#video_devices[@]}
+        echo -e ""${GREEN}找到视频设备：$(ls -A /dev/video*)${NC}""
+    else
+        echo -e "${RED}未找到任何视频采集设备，退出程序${NC}"
+        exit 1
+    fi
+}
+
+function check_repeat_install(){
+    if docker ps -a --format '{{.Names}}' | grep -q '^kvmd$'; then
+        echo -e "${YELLOW}检查到 kvmd 容器已存在，是否删除容器重新部署？${NC}"
+        read -p "y/n: " delete_choice
+        case $delete_choice in
+            y|Y)
+                delete_kvmd_container
+                ;;
+            n|N)
+                echo -e "${RED}退出程序${NC}"
+                exit 1
+                ;;
+            *)
+                echo -e "${RED}无效的选择，请输入 y 或者 n，退出程序${NC}"
+                exit 1
+                ;;
+        esac
+    fi
+    if [[ -d "kvmd_config" ]]; then
+        echo -e "${YELLOW}检查到此前配置文件夹已存在，是否删除此前配置文件夹？${NC}"
+        read -p "y/n: " delete_choice
+        case $delete_choice in
+            y|Y)
+                $sudo_command rm -r kvmd_config
+                ;;
+            n|N)
+                echo -e ""
+                ;;
+            *)
+                echo -e "${RED}无效的选择，请输入 y 或者 n，退出程序${NC}"
+                exit 1
+                ;;
+        esac
+    fi
+}
+
+function show_main_menu() {
+    echo -e "${BLUE}==============================${NC}"
+    echo -e "${BLUE}     One-KVM Docker 版管理     ${NC}"
+    echo -e "${BLUE}==============================${NC}"
+    echo " 1. 安装 One-KVM Docker 版"
+    echo " 2. 卸载 One-KVM Docker 版"
+    echo " 3. 拉取 One-KVM 最新镜像"
+    echo " 4. 更多信息"
+    echo -e "${BLUE}==============================${NC}"
+    read -p "请输入数字（1-4）: " choice
+
+    case $choice in
+        1)
+            check_repeat_install
+            get_hid_info
+            get_video_info
+            get_audio_info
+            get_userinfo
+            get_userenv
+            show_install_info
+            get_install_command
+            execute_command
+            ;;
+        2)
+            delete_kvmd_container
+            ;;
+        3)
+            $sudo_command docker pull $DOCKER_IMAGE_PATH
+            ;;
+        4)
+            echo -e "One-KVM 文档：https://one-kvm.mofeng.run/"
+            echo -e "Github：https://github.com/mofeng-git/One-KVM"
+            ;;
+        *)
+            echo -e "${RED}无效的选择，请输入1-4之间的数字，退出程序${NC}"
+            exit 1
+            ;;
+    esac
+}
+
+function get_hid_info() {
+    if [[ "$architecture" == "amd64" ]]; then
+        echo -e "${GREEN}使用的 HID 硬件类型：CH9329${NC}"
+        use_hid="CH9329"
+    else
+        echo -e "${GREEN}请选择使用的 HID 硬件类型：${NC}"
+        echo " 1. OTG"
+        echo " 2. CH9329"
+        read -p "请输入数字（1 或 2）: " hardware_type
+        while [[ "$hardware_type" != "1" && "$hardware_type" != "2" ]]; do
+            echo -e "${RED}无效的选择，请输入1或2。${NC}"
+            read -p "请输入数字（1 或 2）: " hardware_type
+        done
+        if [[ "$hardware_type" == "1" ]]; then
+            use_hid="OTG"
+        else
+            use_hid="CH9329"
+        fi
+    fi
+
+    if [[ "$use_hid" == "CH9329" ]]; then
+        if ls /dev/ttyUSB* 1> /dev/null 2>&1; then
+            echo -e ""${GREEN}找到串口设备：$(ls -A /dev/ttyUSB*)${NC}""
+        else
+            echo -e "${RED}未找到任何 USB 串口设备，退出程序${NC}"
+            exit 1
+        fi
+        read -p "请输入 CH9329 硬件的地址（回车使用默认值 /dev/ttyUSB0）: " ch9329_address
+        read -p "请输入 CH9329 硬件的波特率（回车使用默认值 9600）: " ch9329_serial_rate
+        ch9329_address=${ch9329_address:-/dev/ttyUSB0}
+        ch9329_serial_rate=${ch9329_serial_rate:-9600}
+    fi
+
+    if [[ "$use_hid" == "OTG" ]]; then
+        check_otg_device
+    fi
+}
+
+function get_video_info() {
+    check_video_device
+    if [[ "$video_num_devices" == "3" ]]; then
+        video_default_device="/dev/video1"
+        echo -e "${YELLOW}经检测 /dev/video0 可能不可用，建议使用 /dev/video1${NC}"
+    else
+        video_default_device="/dev/video0"
+    fi
+    read -p "请输入视频设备路径（回车使用默认值 $video_default_device）: " video_device
+    if [[ -z "$video_device" ]]; then
+        video_device=$video_default_device
+    fi
+}
+
+function get_audio_info() {
+    if [[ -d "/dev/snd" ]]; then
+        echo -e ""${GREEN}找到音频设备：$(ls -A /dev/snd)${NC}""
+        read -p "请输入音频设备路径（回车使用默认值 hw:0）: " audio_device
+        if [[ -z "$audio_device" ]]; then
+            audio_device="hw:0"
+        fi
+    else
+        echo -e "${YELLOW}未找到任何音频采集设备${NC}"
+        audio_device="none"
+    fi
+}
+
+function get_userinfo() {
+    read -p "请输入用户名（回车使用默认值 admin）: " username
+    read -s -p "请输入密码（回车使用默认值 admin）: " password
+    if [[ -z "$username" ]]; then
+        username="admin"
+    fi
+    if [[ -z "$password" ]]; then
+        password="admin"
+    fi
+}
+
+function get_userenv() {
+    read -p "\n额外用户环境变量（回车则留空）: " userenv
+}
+
+function show_install_info() {
+    echo -e "\n\n${BLUE}==============================${NC}"
+    echo -e "${BLUE}安装信息如下：${NC}"
+    echo -e "HID 硬件类型: ${GREEN}$use_hid${NC}"
+    if [[ "$use_hid" == "CH9329" ]]; then
+        echo -e "CH9329 地址: \t${GREEN}$ch9329_address${NC} \tCH9329 波特率: \t${GREEN}$ch9329_serial_rate${NC}"
+    fi
+    if [[ "$use_hid" == "OTG" ]]; then
+        echo -e "OTG端口：\t${GREEN}$otg_devices${NC} \tOTG 状态：\t${GREEN}$otg_status${NC}"
+    fi
+    echo -e "视频设备: \t${GREEN}$video_device${NC} \t音频设备: \t${GREEN}$audio_device${NC}"
+    echo -e "用户名: \t${GREEN}$username${NC} \t密码: \t${GREEN}$password${NC}"
+}
+
+function get_install_command(){
+    local docker_init_command="docker run -itd --name $DOCKER_NAME"
+    local append_command=""
+    local append_env=""
+
+    if [[ "$use_hid" == "CH9329" ]]; then
+        append_command="--device $video_device:/dev/video0 --device $ch9329_address:/dev/ttyUSB0 -v ./kvmd_config:/etc/kvmd"
+        
+        if [[ -d "/dev/snd" ]]; then
+            append_command="$append_command --device /dev/snd:/dev/snd -e AUDIONUM=${audio_device:3}"
+        fi
+        append_env="-e USERNAME=$username -e PASSWORD=$password -e CH9329SPEED=$ch9329_serial_rate"
+        docker_command="$sudo_command $docker_init_command $append_command $DOCKER_PORT $append_env $userenv $DOCKER_IMAGE_PATH"
+    else
+        append_command="--privileged=true -v /lib/modules:/lib/modules:ro -v /dev:/dev -v /sys/kernel/config:/sys/kernel/config -v ./kvmd_config:/etc/kvmd"
+        if [[ -d "/dev/snd" ]]; then
+            append_command="$append_command -e AUDIONUM=${audio_device:3}"
+        fi
+        append_env="-e OTG=1 -e USERNAME=$username -e PASSWORD=$password -e VIDEONUM=${video_device:10} -e AUDIONUM=${audio_device:3}"
+        docker_command="$sudo_command $docker_init_command $append_command $DOCKER_PORT $append_env $userenv $DOCKER_IMAGE_PATH"
+    fi
+    echo -e "\n${BLUE}Docker 部署命令：${NC}\n$docker_command"
+    echo -e "${BLUE}==============================${NC}"
+}
+
+function execute_command(){
+    eval "$docker_command"
+    local exit_status=$?
+    if [[ $exit_status -eq 0 ]]; then
+        echo -e "${BLUE}One-KVM 部署成功${NC}"
+        $sudo_command docker update --restart=always $DOCKER_NAME
+        if [[ "$use_hid" == "OTG" ]]; then
+           execute_otg_command
+        fi
+    else
+        echo -e "${RED}One-KVM 部署失败，退出状态码为 $exit_status${NC}"
+    fi
+}
+
+function execute_otg_command(){
+    $sudo_command echo "device" > /sys/class/usb_role/**/role || echo -e "${YELLOW}OTG 端口切换 device 模式失败${NC}"
+    if grep -q "usb_role" /etc/rc.local; then
+        echo -e ""
+    else
+        $sudo_command sed -i '/^exit 0/i echo device > \/sys\/class\/usb_role\/\*\*\/role' /etc/rc.local
+        $sudo_command chmod +x /etc/rc.local
+    fi
+    if grep -q "libcomposite" /etc/modules.conf; then
+        echo -e ""
+    else
+        $sudo_command echo "libcomposite" >> /etc/modules.conf
+    fi
+}
+
+check_os_architecture
+check_docker_exists
+check_sudo_exists
+show_main_menu