utils.sh

#!/bin/bash
# 通用工具脚本

init_vim() {
    cat <<EOF > ~/.vimrc
" 禁用鼠标模式
set mouse=
set number
set nopaste
EOF
}

alias_ll() {
    if [ ! -f ~/.profile ]; then
        touch ~/.profile
    fi
    if ! grep -q "alias ll='ls -lrhta'" ~/.profile; then
        echo "alias ll='ls -lrhta'" >> ~/.profile
    fi
}

add_x_to_script() {
    chmod +x *.sh
}

send_telegram_message() {
    if [ "$#" -ne 3 ]; then
        echo "Usage: $0 telegram chat_id token msg"
        exit 1
    fi

    local chat_id="$1"
    local bot_token="$2"
    local message="$3"
    local api_url="https://api.telegram.org/bot${bot_token}/sendMessage"

    curl -s -X POST "$api_url" -d chat_id="$chat_id" -d text="$message"
}

pushplus_notify() {
    if [ "$#" -ne 3 ]; then
        echo "Usage: $0 pushplus token title msg"
        exit 1
    fi

    local api_token="$1"
    local title="$2"
    local msg="$3"
    local api_url="http://www.pushplus.plus/send"

    local json_data
    json_data=$(cat <<EOF
{
    "token": "${api_token}",
    "title": "${title}",
    "content": "${msg}"
}
EOF
    )

    curl -X POST "$api_url" \
         -H "Content-Type: application/json" \
         -d "$json_data"
}

gen_ed25519() {
    local user_name=$(whoami)
    local ssh_dir="/home/${user_name}/.ssh"
    local pub_key="$ssh_dir/id_ed25519.pub"
    local private_key="$ssh_dir/id_ed25519"
    local authorized_keys="$ssh_dir/authorized_keys"

    if [ ! -d "$ssh_dir" ]; then
        mkdir -p "$ssh_dir"
        chmod 700 "$ssh_dir"
    fi

    if [ ! -f "$private_key" ]; then
        ssh-keygen -t ed25519 -C "vfhky@qq.com" -f "$private_key" -N ""
    fi

    if [ -f "$pub_key" ]; then
        cat "$pub_key" > "$authorized_keys"
    fi

    find "$ssh_dir" -type f -exec chmod 600 {} \;
    for file in "$pub_key" "$private_key" "$authorized_keys"; do
        if [ ! -f "$file" ]; then
            echo "Error: $file does not exist."
            exit 1
        fi
    done
}

rename_config_files() {
    local dir="$1"

    if [ ! -d "$dir" ]; then
        echo "目录 $dir 不存在"
        return 1
    fi

    for file in "$dir"/*.eg; do
        if [ -e "$file" ]; then
            local new_file="${file%.eg}.conf"

            if [ -e "$new_file" ]; then
                local backup_file="${new_file}.$(date +%Y_%m_%d_%H_%M)"
                \cp -f "$new_file" "$backup_file"
            else
                \cp -f "$file" "$new_file"
                continue
            fi

            local keys_file="./keys_file_temp.txt"
            touch "$keys_file"

            awk -F= '!/^#/ {print $1}' "$new_file" > "$keys_file"

            while IFS= read -r line; do
                if [[ "$line" =~ ^# ]]; then
                    if ! grep -Fxq "$line" "$new_file"; then
                        echo "$line" >> "$new_file"
                    fi
                else
                    key=$(echo "$line" | awk -F= '{print $1}')
                    
                    if ! grep -q "^${key}" "$keys_file"; then
                        echo "$line" >> "$new_file"
                    fi
                fi
            done < "$file"

            rm -f "$keys_file"
        fi
    done
}

modify_config() {
    if [ "$#" -ne 1 ]; then
        echo "Usage: $0 modify_config v0/v1"
        exit 1
    fi

    # v0 or v1
    local version="$1"

    user_name=$(whoami)
    nz_app_path="/home/${user_name}/nezha_app"
    script_dir=$(dirname "$(readlink -f "$0")")
    if [ "${version}" == "v0" ]; then
      download_nezha_sh="${script_dir}/download_nezha.sh"
      echo "==> 即将修改 v0 版本的配置"
    else
      download_nezha_sh="${script_dir}/download_nezha_v1.sh"
      echo "==> 即将修改 v1 版本的配置"
    fi
    heart_beat_entry_sh="${script_dir}/heart_beat_entry.sh"

    # 执行下载配置脚本
    "${download_nezha_sh}" config "${nz_app_path}"
    if [[ $? -ne 0 ]]; then
        echo "Error: 执行 ${download_nezha_sh} 失败."
    fi

    # 启动进程
    echo "==> 修改完毕，开始重启探针进程"
    "${heart_beat_entry_sh}"
}

user_pkill() {
    user_name=$(whoami)
    echo "==> 停止用户 ${user_name} 所有应用"
    pkill -kill -u "${user_name}"
    echo "==> 停止用户 ${user_name} 完成"
}

restore() {
    echo "确定要重装系统吗？会删除整个用户目录的文件。确定重装请输入Y/y"
    read -r input_value
    if [[ "${input_value}" != "Y" && "${input_value}" != "y" ]]; then
        echo "操作已取消。"
        exit 1
    fi

    user_pkill

    script_dir=$(dirname "$(readlink -f "$0")")
    find ~ -type f ! -path "$script_dir/*" -exec chmod 644 {} + 2>/dev/null
    find ~ -type d ! -path "$script_dir/*" -exec chmod 755 {} + 2>/dev/null
    find ~ ! -path "$script_dir/*" ! -path "$script_dir" -mindepth 1 -exec rm -rf {} + 2>/dev/null
}

init_all() {
    init_vim
    alias_ll
    add_x_to_script
}

gen_monitor_config() {
    if [ "$#" -ne 5 ]; then
        echo "Usage: $0 monitor 配置文件的完整路径  新增的进程路径  新增的进程名  进程的启动命令  新增的进程运行方式(background-前台 foreground-后台)"
        exit 1
    fi
    monitor_config=$1
    process_dir=$2
    process_name=$3
    process_run=$4
    process_run_mode=$5

    if [ ! -f "${monitor_config}" ]; then
        touch "${monitor_config}"
    fi

    config="${process_dir}|${process_name}|${process_run}|${process_run_mode}"
    if grep -q "${config}" "${monitor_config}"; then
        echo "监控配置 [${config}] 已经存在于 [${monitor_config}] 中，本次不予写入"
    else
        echo "${config}" >> "${monitor_config}"
    fi
}

gen_heart_beat_config() {
    if [ "$#" -ne 4 ]; then
        echo "Usage: $0 heart 配置文件的完整路径  serv00_ct8_host  serv00_ct8_port  serv00_ct8_username"
        exit 1
    fi
    heart_beat_config=$1
    serv00_ct8_host=$2
    serv00_ct8_port=$3
    serv00_ct8_username=$4

    if [ ! -f "${heart_beat_config}" ]; then
        touch "${heart_beat_config}"
    fi

    config="${serv00_ct8_host}|${serv00_ct8_port}|${serv00_ct8_username}"
    if grep -q "${config}" "${heart_beat_config}"; then
        echo "心跳配置 [${config}] 已经存在于 [${heart_beat_config}] 中，本次不予写入"
    else
        echo "${config}" >> "${heart_beat_config}"
    fi
}

add_cron_job() {
    if [ "$#" -lt 2 ]; then
        echo "Usage: $0 cron '定时时间' '脚本路径' [脚本参数...]"
        exit 1
    fi

    cron_time=$1
    script_path=$2
    shift 2
    script_params="$@"

    new_cron_job="$cron_time $script_path $script_params"
    existing_cron=$(crontab -l | grep -F "$script_path")

    if [ -n "$existing_cron" ]; then
        updated_cron=$(crontab -l | sed "s|^.*$script_path.*$|$new_cron_job|")
        echo "$updated_cron" | crontab -
        echo "定时任务已更新: $new_cron_job"
    else
        (crontab -l; echo "$new_cron_job") | crontab -
        echo "定时任务已添加: $new_cron_job"
    fi
}

update_check_cfg() {
    if [ "$#" -lt 2 ]; then
        echo "Usage: $0 <1-开启本机监控 | 0-关闭本机监控> <配置文件的完整路径>"
        exit 1
    fi

    opt=$1
    monitor_config_file=$2

    if [ ! -f "$monitor_config_file" ]; then
        echo "配置文件不存在: $monitor_config_file"
        exit 1
    fi

    if [ "$(uname)" = "FreeBSD" ]; then
        sed_command="sed -i ''"
    else
        sed_command="sed -i"
    fi

    if grep -q '^CHECK_MONITOR_URL_DNS=' "$monitor_config_file"; then
        eval "$sed_command 's/^CHECK_MONITOR_URL_DNS=.*/CHECK_MONITOR_URL_DNS=${opt}/' \"$monitor_config_file\""
        echo "更新了CHECK_MONITOR_URL_DNS=${opt}在配置文件中"
    else
        echo "CHECK_MONITOR_URL_DNS=${opt}" >> "$monitor_config_file"
        echo "追加了CHECK_MONITOR_URL_DNS=${opt}到配置文件中"
    fi
}

kill_process() {
    local process_name=$1
    local pids=$(pgrep -f "${process_name}")
    if [[ -n "$pids" ]]; then
        echo "====> 正在关闭进程 [${process_name}]"
        for pid in $pids; do
            kill -15 "$pid" && sleep 2
            if kill -0 "$pid" 2>/dev/null; then
                kill -9 "$pid"
            fi
        done
        echo "====> 关闭进程 [${process_name}] 成功"
    else
        echo "====> 进程 [${process_name}] 不存在"
    fi
}

restart() {
    local script_dir=$(dirname "$(readlink -f "$0")")
    local heart_beat_entry_sh="${script_dir}/heart_beat_entry.sh"

    echo "是否要重启 dashboard 面板？[Y/n]"
    read -r input_value
    if [[ "${input_value}" =~ ^[Yy]$ ]]; then
        kill_process "nezha-dashboard"
    fi

    echo "是否要重启 agent 客户端？[Y/n]"
    read -r input_value
    if [[ "${input_value}" =~ ^[Yy]$ ]]; then
        kill_process "nezha-agent"
    fi

    echo "正在重启服务..."
    if [[ -x "${heart_beat_entry_sh}" ]]; then
        "${heart_beat_entry_sh}"
    else
        echo "错误：${heart_beat_entry_sh} 不存在或不可执行"
        exit 1
    fi
}

show_agent_key() {
    if [ "$#" -ne 1 ]; then
        echo "Usage: $0 modify_config v0/v1"
        exit 1
    fi

    local dashboard_config_file="$1"
    local agent_secret_key=$(grep -E '^agentsecretkey:' "$dashboard_config_file" | awk -F ': ' '{print $2}' | sed 's/^\\s*//;s/\\s*$//')
    if [[ -n "$agent_secret_key" ]]; then
        echo "====> 已经找到用于agent连接的密钥: $agent_secret_key"
    else
        echo "====> 未找到用于agent连接的密钥, 请手工执行命令获取: grep agentsecretkey $dashboard_config_file"
    fi
}

uninstall() {
    local script_dir=$(cd "$(dirname "$0")" && pwd)
    local config_dir="${script_dir}/config"
    local config_file="${config_dir}/monitor.conf"
    if [[ ! -f "$config_file" ]]; then
        echo "配置文件 $config_file 不存在，退出。"
        return 1
    fi

    declare -a process_list
    while IFS='|' read -r app_path process_name script_command run_mode; do
        [[ "$app_path" =~ ^#.*$ ]] && continue
        process_list+=("$app_path|$process_name|$script_command|$run_mode")
    done < "$config_file"

    for entry in "${process_list[@]}"; do
        IFS='|' read -ra process_info <<< "$entry"
        local app_path="${process_info[0]}"
        local process_name="${process_info[1]}"

        read -r -p "是否要停止进程 [${process_name}] 并删除目录 [${app_path}]？[Y/n] " input_value
        input_value=${input_value:-Y}
        if [[ "${input_value,,}" == "y" ]]; then
            pkill -f "$process_name" && echo "已停止进程 ${process_name}" || echo "停止进程 ${process_name} 失败"
            if \rm -rf "$app_path"; then
                echo "已删除目录 ${app_path}"
            else
                echo "删除目录 ${app_path} 失败，请检查权限。"
            fi
        else
            echo "跳过进程 ${process_name} 和目录 ${app_path}。"
        fi
    done

    # 删除定时任务
    local cron_tab_serv00_ct8_nezha=$(crontab -l | grep -a "serv00_ct8_nezha")
    if [[ -n "$cron_tab_serv00_ct8_nezha" ]]; then
        crontab -l | grep -v "serv00_ct8_nezha" | crontab -
        echo "已删除定时任务 ${cron_tab_serv00_ct8_nezha}"
    else
        echo "未找到定时任务 serv00_ct8_nezha"
    fi

    ## 删除配置文件
    if [ -d "$config_dir" ]; then
        \rm -rf "${config_dir}"/*.conf*
    fi

    echo "==== 操作结束 ===="
}

case "$1" in
    "init")
        init_all
        ;;
    "kill")
        user_pkill
        ;;
    "key")
        gen_ed25519
        ;;
    "monitor")
        shift 1
        gen_monitor_config "$@"
        ;;
    "heart")
        shift 1
        gen_heart_beat_config "$@"
        ;;
    "cron")
        shift 1
        add_cron_job "$@"
        ;;
    "check")
        shift 1
        update_check_cfg "$@"
        ;;
    "rename_config")
        shift 1
        rename_config_files "$@"
        ;;
    "modify_config")
        shift 1
        modify_config "$@"
        ;;
    "telegram")
        shift 1
        send_telegram_message "$@"
        ;;
    "pushplus")
        shift 1
        pushplus_notify "$@"
        ;;
    "restore")
        restore
        ;;
    "restart")
        restart
        ;;
    "show_agent_key")
        shift 1
        show_agent_key "$@"
        ;;
    "uninstall")
        uninstall
        ;;
    *)
        echo "====== 用法 ====="
        echo "$0 init - 优化使用环境"
        echo "$0 kill - 停止用户所有应用"
        echo "$0 key - 生成 ed25519 公私钥"
        echo "$0 monitor - 写入进程监控配置, 参数: 配置文件的完整路径  新增的进程路径  新增的进程名 进程的启动命令 新增的进程运行方式(background-前台 foreground-后台)"
        echo "$0 heart - 写入心跳监控配置, 参数: 配置文件的完整路径  serv00_ct8_host  serv00_ct8_port  serv00_ct8_username"
        echo "$0 cron - 添加定时任务, 参数: '定时时间' '脚本路径' [脚本参数...]"
        echo "$0 check - [1-增加本机监控 0-关闭本机监控] 配置文件的完整路径"
        echo "$0 rename_config - 从配置模板文件中生成具体配置文件"
        echo "$0 modify_config - 修改哪吒dashboard或者agent的配置并重启服务 参数: v0 或者 v1"
        echo "$0 telegram - 发送telegram通知 chat_id token msg"
        echo "$0 pushplus - 发送pushplus通知 token title msg"
        echo "$0 restore - 重装系统"
        echo "$0 restart - 重启面板和agent"
        echo "$0 show_agent_key - 查看面板生成的 agentsecretkey 参数: 面板config.yaml配置文件路径"
        echo "$0 uninstall - 卸载哪吒dashboard和agent"
        exit 1
        ;;
esac