#!/bin/bash

# 开发者信息
DEVELOPER="ZK.Work 的 DC 社区中 KarenL"

# 检查是否以 root 用户运行脚本
if [ "$(id -u)" != "0" ]; then
    echo "此脚本需要以 root 用户权限运行。"
    echo "请尝试使用 'sudo -i' 命令切换到 root 用户，然后再次运行此脚本。"
    exit 1
fi
function install_and_run_miner() {
    version=$1
    type=$2
    base_url="https://github.com/6block/zkwork_aleo_gpu_worker/releases/download"

    # 切换到 /root 目录
    cd /root
    # 下载 zkwork 矿工
    wget "${base_url}/${version}/aleo_prover-${version}${type}.tar.gz"
    # 获取 Aleo 钱包地址
    read -p "输入你的 Aleo 钱包地址: " aleo_address
    # 解压和进入目录
    tar -zvxf "aleo_prover-${version}${type}.tar.gz" && cd aleo_prover

    # 更新 Aleo 地址并设置自定义矿工名称
    read -p "输入自定义矿工名称: " miner_name
    sed -i "s/^reward_address=.*$/reward_address=$aleo_address/" inner_prover.sh
    sed -i "s/^custom_name=\".*\"$/custom_name=\"$miner_name\"/" inner_prover.sh

    # 启动矿工
    chmod +x run_prover.sh
    ./run_prover.sh &
    wait  # 等待进程结束

    # 提示用户矿工已启动
    echo "矿工已启动！"
    echo "你可以使用 'tail -f /root/aleo_prover/prover.log' 命令来查看日志。"
}

function install_and_run_miner_v0_2_2_lite(){
  # 切换到 /root 目录
    cd /root
    # 下载 zkwork 矿工（标准版 v0.1.2）
   wget https://github.com/6block/zkwork_aleo_gpu_worker/releases/download/v0.2.2/aleo_prover-v0.2.2_lite.tar.gz
    # 获取 Aleo 钱包地址
    read -p "输入你的 Aleo 钱包地址: " aleo_address
    # 解压和进入目录
    tar -zvxf aleo_prover-v0.2.2_lite.tar.gz && cd aleo_prover

    # 更新 Aleo 地址并设置自定义矿工名称
    read -p "输入自定义矿工名称: " miner_name
    sed -i "s/^reward_address=.*$/reward_address=$aleo_address/" inner_prover.sh
    sed -i "s/^custom_name=\".*\"$/custom_name=\"$miner_name\"/" inner_prover.sh

    # 启动矿工
    chmod +x run_prover.sh
    ./run_prover.sh &
    wait  # 等待进程结束
    # 提示用户矿工已启动
    # 提示用户矿工已启动
    echo "矿工已启动！"
    echo "你可以使用 'tail -f /root/aleo_prover/prover.log' 命令来查看日志。" 
}
function install_and_run_miner_v0_2_2_full(){
  # 切换到 /root 目录
    cd /root
    # 下载 zkwork 矿工（标准版 v0.1.2）
   wget https://github.com/6block/zkwork_aleo_gpu_worker/releases/download/v0.2.2/aleo_prover-v0.2.2_full.tar.gz
    # 获取 Aleo 钱包地址
    read -p "输入你的 Aleo 钱包地址: " aleo_address
    # 解压和进入目录
    tar -zvxf aleo_prover-v0.2.2_full.tar.gz && cd aleo_prover

    # 更新 Aleo 地址并设置自定义矿工名称
    read -p "输入自定义矿工名称: " miner_name
    sed -i "s/^reward_address=.*$/reward_address=$aleo_address/" inner_prover.sh
    sed -i "s/^custom_name=\".*\"$/custom_name=\"$miner_name\"/" inner_prover.sh

    # 启动矿工
    chmod +x run_prover.sh
    ./run_prover.sh &
    wait  # 等待进程结束
    # 提示用户矿工已启动
    # 提示用户矿工已启动
    echo "矿工已启动！"
    echo "你可以使用 'tail -f /root/aleo_prover/prover.log' 命令来查看日志。" 
}
function install_and_run_miner_v0_2_1(){
  # 切换到 /root 目录
    cd /root
    # 下载 zkwork 矿工（标准版 v0.1.2）
   wget https://github.com/6block/zkwork_aleo_gpu_worker/releases/download/v0.2.1/aleo_prover-v0.2.1.tar.gz
    # 获取 Aleo 钱包地址
    read -p "输入你的 Aleo 钱包地址: " aleo_address
    # 解压和进入目录
    tar -zvxf aleo_prover-v0.2.1.tar.gz && cd aleo_prover

    # 更新 Aleo 地址并设置自定义矿工名称
    read -p "输入自定义矿工名称: " miner_name
    sed -i "s/^reward_address=.*$/reward_address=$aleo_address/" inner_prover.sh
    sed -i "s/^custom_name=\".*\"$/custom_name=\"$miner_name\"/" inner_prover.sh

    # 启动矿工
    chmod +x run_prover.sh
    ./run_prover.sh &
    wait  # 等待进程结束
    # 提示用户矿工已启动
    # 提示用户矿工已启动
    echo "矿工已启动！"
    echo "你可以使用 'tail -f /root/aleo_prover/prover.log' 命令来查看日志。" 
}
function install_and_run_miner_v0_2_0(){
  # 切换到 /root 目录
    cd /root
    # 下载 zkwork 矿工（标准版 v0.1.2）
   wget https://github.com/6block/zkwork_aleo_gpu_worker/releases/download/v0.2.0/aleo_prover-v0.2.0.tar.gz
    # 获取 Aleo 钱包地址
    read -p "输入你的 Aleo 钱包地址: " aleo_address
    # 解压和进入目录
    tar -zvxf aleo_prover-v0.2.0.tar.gz && cd aleo_prover

    # 更新 Aleo 地址并设置自定义矿工名称
    read -p "输入自定义矿工名称: " miner_name
    sed -i "s/^reward_address=.*$/reward_address=$aleo_address/" inner_prover.sh
    sed -i "s/^custom_name=\".*\"$/custom_name=\"$miner_name\"/" inner_prover.sh

    # 启动矿工
    chmod +x run_prover.sh
    ./run_prover.sh &
    wait  # 等待进程结束
    # 提示用户矿工已启动
    # 提示用户矿工已启动
    echo "矿工已启动！"
    echo "你可以使用 'tail -f /root/aleo_prover/prover.log' 命令来查看日志。" 
}
function install_and_run_miner_pre_v0_1_3(){
  # 切换到 /root 目录
    cd /root
    # 下载 zkwork 矿工（标准版 v0.1.2）
   wget https://github.com/6block/zkwork_aleo_gpu_worker/releases/download/v0.1.3/aleo_prover-v0.1.3_boost.tar.gz
    # 获取 Aleo 钱包地址
    read -p "输入你的 Aleo 钱包地址: " aleo_address
    # 解压和进入目录
    tar -zvxf aleo_prover-v0.1.3_boost.tar.gz && cd aleo_prover

    # 更新 Aleo 地址并设置自定义矿工名称
    read -p "输入自定义矿工名称: " miner_name
    sed -i "s/^reward_address=.*$/reward_address=$aleo_address/" inner_prover.sh
    sed -i "s/^custom_name=\".*\"$/custom_name=\"$miner_name\"/" inner_prover.sh

    # 启动矿工
    chmod +x run_prover.sh
    ./run_prover.sh &
    wait  # 等待进程结束
    # 提示用户矿工已启动
    # 提示用户矿工已启动
    echo "矿工已启动！"
    echo "你可以使用 'tail -f /root/aleo_prover/prover.log' 命令来查看日志。" 
}
function install_and_run_miner_v0_1_3(){
  # 切换到 /root 目录
    cd /root
    # 下载 zkwork 矿工（标准版 v0.1.2）
   wget https://github.com/6block/zkwork_aleo_gpu_worker/releases/download/v0.1.3/aleo_prover-v0.1.3_stable.tar.gz
    # 获取 Aleo 钱包地址
    read -p "输入你的 Aleo 钱包地址: " aleo_address
    # 解压和进入目录
    tar -zvxf aleo_prover-v0.1.3_stable.tar.gz && cd aleo_prover

    # 更新 Aleo 地址并设置自定义矿工名称
    read -p "输入自定义矿工名称: " miner_name
    sed -i "s/^reward_address=.*$/reward_address=$aleo_address/" inner_prover.sh
    sed -i "s/^custom_name=\".*\"$/custom_name=\"$miner_name\"/" inner_prover.sh

    # 启动矿工
    chmod +x run_prover.sh
    ./run_prover.sh &
    wait  # 等待进程结束
    # 提示用户矿工已启动
    # 提示用户矿工已启动
    echo "矿工已启动！"
    echo "你可以使用 'tail -f /root/aleo_prover/prover.log' 命令来查看日志。" 
}
function  install_and_run_miner_pre_v0_1_2(){
    # 切换到 /root 目录
    cd /root
    # 下载 zkwork 矿工（标准版 v0.1.2）
   wget https://github.com/6block/zkwork_aleo_gpu_worker/releases/download/v0.1.2/aleo_prover-v0.1.2_pre.tar.gz
    # 获取 Aleo 钱包地址
    read -p "输入你的 Aleo 钱包地址: " aleo_address
    # 解压和进入目录
    tar -zvxf aleo_prover-v0.1.2_pre.tar.gz && cd aleo_prover

    # 更新 Aleo 地址并设置自定义矿工名称
    read -p "输入自定义矿工名称: " miner_name
    sed -i "s/^reward_address=.*$/reward_address=$aleo_address/" inner_prover.sh
    sed -i "s/^custom_name=\".*\"$/custom_name=\"$miner_name\"/" inner_prover.sh

    # 启动矿工
    chmod +x run_prover.sh
    ./run_prover.sh &
    wait  # 等待进程结束
    # 提示用户矿工已启动
    # 提示用户矿工已启动
    echo "矿工已启动！"
    echo "你可以使用 'tail -f /root/aleo_prover/prover.log' 命令来查看日志。" 
}
function install_and_run_miner_v0_1_2(){
  

    # 切换到 /root 目录
    cd /root

    # 下载 zkwork 矿工（标准版 v0.1.2）
    wget https://github.com/6block/zkwork_aleo_gpu_worker/releases/download/v0.1.2/aleo_prover-v0.1.2_stable.tar.gz
    # 获取 Aleo 钱包地址
    read -p "输入你的 Aleo 钱包地址: " aleo_address
    # 解压和进入目录
    tar -zvxf aleo_prover-v0.1.2_stable.tar.gz && cd aleo_prover

    # 更新 Aleo 地址并设置自定义矿工名称
    read -p "输入自定义矿工名称: " miner_name
    sed -i "s/^reward_address=.*$/reward_address=$aleo_address/" inner_prover.sh
    sed -i "s/^custom_name=\".*\"$/custom_name=\"$miner_name\"/" inner_prover.sh

    # 启动矿工
    chmod +x run_prover.sh
    ./run_prover.sh &
    wait

    # 提示用户矿工已启动
    echo "矿工已启动！"
    echo "你可以使用 'tail -f /root/aleo_prover/prover.log' 命令来查看日志。" 
}
# 安装并运行 zkwork_aleo 挖矿程序（标准版 v0.1.1）
function install_and_run_miner_v0_1_1() {
    # 获取 Aleo 钱包地址
    read -p "输入你的 Aleo 钱包地址: " aleo_address

    # 切换到 /root 目录
    cd /root

    # 下载 zkwork 矿工（标准版 v0.1.1）
    wget https://github.com/6block/zkwork_aleo_gpu_worker/releases/download/v0.1.1/aleo_prover-v0.1.1.tar.gz

    # 解压和进入目录
    tar -zvxf aleo_prover-v0.1.1.tar.gz && cd aleo_prover

    # 更新 Aleo 地址并设置自定义矿工名称
    read -p "输入自定义矿工名称: " miner_name
    sed -i "s/^reward_address=.*$/reward_address=$aleo_address/" run_prover
    sed -i "s/^custom_name=\".*\"$/custom_name=\"$miner_name\"/" run_prover

    # 启动矿工
    chmod +x run_prover.sh
    ./run_prover.sh &

    # 提示用户矿工已启动
    echo "矿工已启动！"
    echo "你可以使用 'tail -f /root/aleo_prover/prover.log' 命令来查看日志。"
    view_logs
}

# 安装并运行 zkwork_aleo 挖矿程序（hot 版本 v0.1.1）
function install_and_run_hot_miner_v0_1_1() {
    # 获取 Aleo 钱包地址
    read -p "输入你的 Aleo 钱包地址: " aleo_address

    # 切换到 /root 目录
    cd /root

    # 下载 zkwork 矿工（hot 版本 v0.1.1）
    wget https://github.com/6block/zkwork_aleo_gpu_worker/releases/download/v0.1.1-hot/aleo_prover-v0.1.1_hot.tar.gz

    # 解压和进入目录
    tar -zvxf aleo_prover-v0.1.1_hot.tar.gz && cd aleo_prover

    # 更新 Aleo 地址并设置自定义矿工名称
    read -p "输入自定义矿工名称: " miner_name
    sed -i "s/^reward_address=.*$/reward_address=$aleo_address/" run_prover.sh
    sed -i "s/^custom_name=\".*\"$/custom_name=\"$miner_name\"/" run_prover.sh

    # 启动矿工
    chmod +x run_prover.sh
    ./run_prover.sh &

    # 提示用户矿工已启动
    echo "矿工已启动！"
    echo "你可以使用 'tail -f /root/aleo_prover/prover.log' 命令来查看日志。"
    view_logs
}

# 安装并运行 zkwork_aleo 挖矿程序（hot 版本 v0.1.0）
function install_and_run_hot_miner_v0_1_0() {
    # 获取 Aleo 钱包地址
    read -p "输入你的 Aleo 钱包地址: " aleo_address

    # 切换到 /root 目录
    cd /root

    # 下载 zkwork 矿工（hot 版本 v0.1.0）
    wget https://github.com/6block/zkwork_aleo_gpu_worker/releases/download/v0.1.1-hot/aleo_prover-v0.1.0_hot.tar.gz

    # 解压和进入目录
    tar -zvxf aleo_prover-v0.1.0_hot.tar.gz && cd aleo_prover

    # 更新 Aleo 地址并设置自定义矿工名称
    read -p "输入自定义矿工名称: " miner_name
    sed -i "s/^reward_address=.*$/reward_address=$aleo_address/" run_prover.sh
    sed -i "s/^custom_name=\".*\"$/custom_name=\"$miner_name\"/" run_prover.sh

    # 启动矿工
    chmod +x run_prover.sh
    ./run_prover.sh &

    # 提示用户矿工已启动
    echo "矿工已启动！"
    echo "你可以使用 'tail -f /root/aleo_prover/prover.log' 命令来查看日志。"
    view_logs
}

# 安装并运行 zkwork_aleo 挖矿程序（标准版 v0.1.0）
function install_and_run_miner_v0_1_0() {
    # 获取 Aleo 钱包地址
    read -p "输入你的 Aleo 钱包地址: " aleo_address

    # 切换到 /root 目录
    cd /root

    # 下载 zkwork 矿工（标准版 v0.1.0）
    wget https://github.com/6block/zkwork_aleo_gpu_worker/releases/download/v0.1.0/aleo_prover-v0.1.0.tar.gz

    # 解压和进入目录
    tar -zvxf aleo_prover-v0.1.0.tar.gz && cd aleo_prover

    # 更新 Aleo 地址并设置自定义矿工名称
    read -p "输入自定义矿工名称: " miner_name
    sed -i "s/^reward_address=.*$/reward_address=$aleo_address/" run_prover.sh
    sed -i "s/^custom_name=\".*\"$/custom_name=\"$miner_name\"/" run_prover.sh

    # 启动矿工
    chmod +x run_prover.sh
    ./run_prover.sh &

    # 提示用户矿工已启动
    echo "矿工已启动！"
    echo "你可以使用 'tail -f /root/aleo_prover/prover.log' 命令来查看日志。"
    view_logs
}

# 查看挖矿日志
function view_logs() {
    tail -f /root/aleo_prover/prover.log
}

# 重启 zkwork_aleo 挖矿程序
function restart_miner() {
    echo "正在重启 zkwork_aleo 挖矿程序..."

    # 重新执行 run_prover.sh
    cd /root/aleo_prover || { echo "无法进入目录 /root/aleo_prover"; exit 1; }
    ./run_prover.sh &

    # 确认矿工是否已成功启动
    echo "矿工已成功重启！"
}

# 卸载 zkwork_aleo 挖矿程序
function uninstall_miner() {
    # 停止矿工
    pkill -f run_prover.sh

    # 删除安装文件
    rm -rf /root/aleo_prover
    rm -f /root/aleo_prover-v0.1.1_hot.tar.gz
    rm -f /root/aleo_prover-v0.1.0_hot.tar.gz
    rm -f /root/aleo_prover-v0.1.0.tar.gz

    echo "矿工程序已卸载。"
}
 
# 主菜单
function main_menu() {
    while true; do
        clear
        echo "zk.work社区一键 zkwork_aleo 脚本"
        echo "开发者: $DEVELOPER"
        echo "==========================="
        
        echo "1. 安装并运行 zkwork_aleo 挖矿程序（v0_2_3_lite_fix)）"
        echo "2. 安装并运行 zkwork_aleo 挖矿程序（v0_2_3_full_fix)）"
        echo "3. 安装并运行 zkwork_aleo 挖矿程序（v0_2_3_lite)）"
        echo "4. 安装并运行 zkwork_aleo 挖矿程序（v0_2_3_full)）"
        echo "5. 安装并运行 zkwork_aleo 挖矿程序（v0_2_2_full)）"
        echo "6. 安装并运行 zkwork_aleo 挖矿程序（v0_2_2_lite)）"
        echo "7. 安装并运行 zkwork_aleo 挖矿程序（v0.2.1)）"
        echo "8. 安装并运行 zkwork_aleo 挖矿程序（v0.2.0)）"
        echo "9. 安装并运行 zkwork_aleo 挖矿程序（v0.1.3-stable)）"
        echo "10. 安装并运行 zkwork_aleo 挖矿程序（v0.1.3-boost,(cpu好选这个）"
        echo "11. 安装并运行 zkwork_aleo 挖矿程序（v0.1.2-stable)）"
        echo "12. 安装并运行 zkwork_aleo 挖矿程序（v0.1.2-pre,(cpu好选这个）"
        echo "13. 安装并运行 zkwork_aleo 挖矿程序（标准版 v0.1.1）"
        echo "14. 安装并运行 zkwork_aleo 挖矿程序（hot 版本 v0.1.1）"
        echo "15. 安装并运行 zkwork_aleo 挖矿程序（hot 版本 v0.1.0）"
        echo "16. 安装并运行 zkwork_aleo 挖矿程序（标准版 v0.1.0）"
        echo "17. 查看挖矿日志"
        echo "18. 重启 zkwork_aleo 挖矿程序"
        echo "19. 卸载 zkwork_aleo 挖矿程序"
        echo "20. 退出脚本"
        read -p "请输入选项（1-18）: " OPTION

        case $OPTION in
        1) install_and_run_miner "v0.2.3" "_lite_fix" ;;
        2) install_and_run_miner "v0.2.3" "_full_fix";;
        3) install_and_run_miner "v0.2.3" "_lite" ;;
        4) install_and_run_miner "v0.2.3" "_full";;
        5) install_and_run_miner_v0_2_2_full ;;
        6) install_and_run_miner_v0_2_2_lite ;;
        7) install_and_run_miner_v0_2_1 ;;
        8) install_and_run_miner_v0_2_0 ;;
        9) install_and_run_miner_v0_1_3 ;;
        10) install_and_run_miner_pre_v0_1_3 ;;
        11) install_and_run_miner_v0_1_2 ;;
        12) install_and_run_miner_pre_v0_1_2 ;;
        13) install_and_run_miner_v0_1_1 ;;
        14) install_and_run_hot_miner_v0_1_1 ;;
        15) install_and_run_hot_miner_v0_1_0 ;;
        16) install_and_run_miner_v0_1_0 ;;
        17) view_logs ;;
        18) restart_miner ;;
        19) uninstall_miner ;;
        20) exit 0 ;;
        *) echo "无效选项。" ;;
        esac
        echo "按任意键返回主菜单..."
        read -n 1
    done
}

# 显示主菜单
main_menu