One-KVM/build/init.sh

#!/bin/bash
# ========================================================================== #
#                                                                            #
#    KVMD - The main PiKVM daemon.                                           #
#                                                                            #
#    Copyright (C) 2023-2025  SilentWind <mofeng654321@hotmail.com>         #
#                                                                            #
#    This program is free software: you can redistribute it and/or modify    #
#    it under the terms of the GNU General Public License as published by    #
#    the Free Software Foundation, either version 3 of the License, or       #
#    (at your option) any later version.                                     #
#                                                                            #
#    This program is distributed in the hope that it will be useful,         #
#    but WITHOUT ANY WARRANTY; without even the implied warranty of          #
#    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the           #
#    GNU General Public License for more details.                            #
#                                                                            #
#    You should have received a copy of the GNU General Public License       #
#    along with this program.  If not, see <https://www.gnu.org/licenses/>.  #
#                                                                            #
# ========================================================================== #

# 定义颜色代码
RED='\033[0;31m'
GREEN='\033[0;32m'
YELLOW='\033[0;33m'
BLUE='\033[0;34m'
NC='\033[0m'

# 输出日志的函数
log_info() {
    echo -e "${GREEN}[INFO] $1${NC}"
}

log_warn() {
    echo -e "${YELLOW}[WARN] $1${NC}"
}

log_error() {
    echo -e "${RED}[ERROR] $1${NC}"
}

# 初始化检查
log_info "One-KVM 正在启动..."

# 首次初始化配置
if [ ! -f /etc/kvmd/.init_flag ]; then
    log_info "首次初始化配置..."
    
    # 创建必要目录并移动配置文件
    if mkdir -p /etc/kvmd/ && \
       mv /etc/kvmd_backup/* /etc/kvmd/ && \
       touch /etc/kvmd/.docker_flag && \
       sed -i 's/localhost.localdomain/docker/g' /etc/kvmd/meta.yaml && \
       sed -i 's/localhost/localhost:4430/g' /etc/kvmd/kvm_input.sh; then
        log_info "移动配置文件完成"
    else
        log_error "移动配置文件失败"
        exit 1
    fi

    # SSL证书配置
    if ! /usr/share/kvmd/kvmd-gencert --do-the-thing; then
        log_error "Nginx SSL 证书生成失败"
        exit 1
    fi
    if ! /usr/share/kvmd/kvmd-gencert --do-the-thing --vnc; then
        log_error "VNC SSL 证书生成失败"
        exit 1
    fi

    # 设置用户名和密码
    if [ ! -z "$USERNAME" ] && [ ! -z "$PASSWORD" ]; then
        if python -m kvmd.apps.htpasswd del admin \
            && echo "$PASSWORD" | python -m kvmd.apps.htpasswd set -i "$USERNAME" \
            && echo "$PASSWORD -> $USERNAME:$PASSWORD" > /etc/kvmd/vncpasswd \
            && echo "$USERNAME:$PASSWORD -> $USERNAME:$PASSWORD" > /etc/kvmd/ipmipasswd; then
            log_info "用户凭据设置成功"
        else
            log_error "用户凭据设置失败"
            exit 1
        fi
    else
        log_warn "未设置 USERNAME 和 PASSWORD 环境变量，使用默认值(admin/admin)"
    fi

    # SSL开关配置
    if [ "$NOSSL" == 1 ]; then
        log_info "已禁用SSL"
        if ! python -m kvmd.apps.ngxmkconf /etc/kvmd/nginx/nginx.conf.mako /etc/kvmd/nginx/nginx.conf -o nginx/https/enabled=false; then
            log_error "Nginx 配置失败"
            exit 1
        fi
    else
        if ! python -m kvmd.apps.ngxmkconf /etc/kvmd/nginx/nginx.conf.mako /etc/kvmd/nginx/nginx.conf; then
            log_error "Nginx 配置失败"
            exit 1
        fi
    fi

    # 认证配置
    if [ "$NOAUTH" == "1" ]; then
        sed -i "s/enabled: true/enabled: false/g" /etc/kvmd/override.yaml
        log_info "已禁用认证"
    fi

    #add supervisord conf
    if [ "$NOWEBTERM" == "1" ]; then
        log_info "已禁用 WebTerm 功能"
        rm -r /usr/share/kvmd/extras/webterm
    else
        cat >> /etc/supervisord.conf  << EOF

[program:kvmd-webterm]
command=/usr/local/bin/ttyd --interface=/run/kvmd/ttyd.sock --port=0 --writable /bin/bash -c '/etc/kvmd/armbain-motd; bash'
directory=/
autostart=true
autorestart=true
priority=14
stopasgroup=true
stdout_logfile=/dev/stdout
stdout_logfile_maxbytes = 0
redirect_stderr=true
EOF
    fi

    if [ "$NOWEBTERMWRITE" == "1" ]; then
        sed -i "s/--writable//g" /etc/supervisord.conf
    fi

    if [  "$NOVNC" == "1" ]; then
        log_info "已禁用 VNC 功能"
        rm -r /usr/share/kvmd/extras/vnc
    else
        cat >> /etc/supervisord.conf << EOF

[program:kvmd-vnc]
command=python -m kvmd.apps.vnc --run
directory=/
autostart=true
autorestart=true
priority=11
stopasgroup=true
stdout_logfile=/dev/stdout
stdout_logfile_maxbytes = 0
redirect_stderr=true
EOF
    fi

    if [  "$NOIPMI" == "1" ]; then
        log_info "已禁用IPMI功能"
        rm -r /usr/share/kvmd/extras/ipmi
    else
        cat >> /etc/supervisord.conf << EOF

[program:kvmd-ipmi]
command=python -m kvmd.apps.ipmi --run
directory=/
autostart=true
autorestart=true
priority=12
stopasgroup=true
stdout_logfile=/dev/stdout
stdout_logfile_maxbytes = 0
redirect_stderr=true
EOF
    fi

    #switch OTG config
    if [ "$OTG" == "1" ]; then
        log_info "已启用 OTG 功能"
        sed -i "s/ch9329/otg/g" /etc/kvmd/override.yaml
        sed -i "s/device: \/dev\/ttyUSB0//g" /etc/kvmd/override.yaml
        if [ "$NOMSD" == 1 ]; then
            log_info "已禁用 MSD 功能"
        else
            sed -i "s/#type: otg/type: otg/g" /etc/kvmd/override.yaml
        fi
    fi

    if [ ! -z "$VIDEONUM" ]; then
        if sed -i "s/\/dev\/video0/\/dev\/video$VIDEONUM/g" /etc/kvmd/override.yaml && \
           sed -i "s/\/dev\/video0/\/dev\/video$VIDEONUM/g" /etc/kvmd/janus/janus.plugin.ustreamer.jcfg; then
            log_info "视频设备已设置为 /dev/video$VIDEONUM"
        fi
    fi

    if [ ! -z "$AUDIONUM" ]; then
        if sed -i "s/hw:0/hw:$AUDIONUM/g" /etc/kvmd/janus/janus.plugin.ustreamer.jcfg; then
            log_info "音频设备已设置为 hw:$AUDIONUM"
        fi
    fi

    if [ ! -z "$CH9329SPEED" ]; then
        if sed -i "s/speed: 9600/speed: $CH9329SPEED/g" /etc/kvmd/override.yaml; then
            log_info "CH9329 串口速率已设置为 $CH9329SPEED"
        fi
    fi

    if [ ! -z "$CH9329TIMEOUT" ]; then
        if sed -i "s/read_timeout: 0.3/read_timeout: $CH9329TIMEOUT/g" /etc/kvmd/override.yaml; then
            log_info "CH9329 超时已设置为 $CH9329TIMEOUT 秒"
        fi
    fi

    if [ ! -z "$H264PRESET" ]; then
        if sed -i "s/ultrafast/$H264PRESET/g" /etc/kvmd/override.yaml; then
            log_info "H264 预设已设置为 $H264PRESET"
        fi
    fi

    if [ ! -z "$VIDEOFORMAT" ]; then
        if sed -i "s/format=mjpeg/format=$VIDFORMAT/g" /etc/kvmd/override.yaml; then
            log_info "视频输入格式已设置为 $VIDFORMAT"
        fi
    fi
    
    touch /etc/kvmd/.init_flag
    log_info "初始化配置完成"
fi

# OTG设备配置
if [ "$OTG" == "1" ]; then
    log_info "正在配置 OTG 设备..."
    rm -r /run/kvmd/otg &> /dev/null
    
    if ! modprobe libcomposite; then
        log_error "加载 libcomposite 模块失败"
        exit 1
    fi

    if python -m kvmd.apps.otg start; then
        ln -s /dev/hidg1 /dev/kvmd-hid-mouse
        ln -s /dev/hidg0 /dev/kvmd-hid-keyboard
        ln -s /dev/hidg2 /dev/kvmd-hid-mouse-alt
        log_info "OTG 设备配置完成"
    else
        log_warn "OTG 设备挂载失败"
        #exit 1
    fi
fi

log_info "One-KVM 配置文件准备完成，正在启动服务..."
exec supervisord -c /etc/supervisord.conf