#!/bin/bash

# 定义服务名称和文件路径
SERVICE_NAME="nexus"
SERVICE_FILE="/etc/systemd/system/nexus.service"  # 更新服务文件路径

# 脚本保存路径
SCRIPT_PATH="$HOME/nexus.sh"

# 检查是否以root用户运行脚本
if [ "$(id -u)" != "0" ]; then
    echo "此脚本需要以root用户权限运行。"
    echo "请尝试使用 'sudo -i' 命令切换到root用户，然后再次运行此脚本。"
    exit 1
fi

# 主菜单函数
function main_menu() {
    while true; do
        clear
        echo "脚本由大赌社区哈哈哈哈编写，推特 @ferdie_jhovie，免费开源，请勿相信收费"
        echo "如有问题，可联系推特，仅此只有一个号"
        echo "新建了一个电报群，方便大家交流：t.me/Sdohua"
        echo "================================================================"
        echo "退出脚本，请按键盘 ctrl + C 退出即可"
        echo "请选择要执行的操作:"
        echo "1. 启动节点"
        echo "2. 查看 Prover 状态"
        echo "3. 查看日志"
        echo "4. 删除节点"
        echo "5. 显示 ID"  # 新增选项
        echo "6) 退出"
        
        read -p "请输入选项 (1-7): " choice
        
        case $choice in
            1)
                start_node  # 调用启动节点函数
                ;;
            2)
                check_prover_status  # 调用查看 Prover 状态函数
                ;;
            3)
                view_logs  # 调用查看日志函数
                ;;
            4)
                delete_node  # 调用删除节点函数
                ;;
            5)
                show_id  # 调用显示 ID 函数
                ;;
            6)
                echo "退出脚本。"
                exit 0
                ;;
            *)
                echo "无效选项，请重新选择。"
                ;;
        esac
    done
}

# 显示 ID 的函数
function show_id() {
    if [ -f /root/.nexus/prover-id ]; then
        echo "Prover ID 内容:"
        echo "$(</root/.nexus/prover-id)"  # 使用 echo 显示文件内容
    else
        echo "文件 /root/.nexus/prover-id 不存在。"
    fi

    # 等待用户按任意键返回主菜单
    read -p "按任意键返回主菜单"
}

# 启动节点的函数
function start_node() {
    # 检查服务是否正在运行
    if systemctl is-active --quiet nexus.service; then
        echo "nexus.service 当前正在运行。正在停止并禁用它..."
        sudo systemctl stop nexus.service
        sudo systemctl disable nexus.service
    else
        echo "nexus.service 当前未运行。"
    fi

    # 确保目录存在
    mkdir -p /root/.nexus  # 创建目录（如果不存在）

    # 更新系统并安装必要的软件包
    echo "更新系统并安装必要的软件包..."
    if ! sudo apt update && sudo apt upgrade -y && sudo apt install curl iptables build-essential git wget lz4 jq make gcc nano automake autoconf tmux htop nvme-cli pkg-config libssl-dev libleveldb-dev tar clang bsdmainutils ncdu unzip -y; then
        echo "安装软件包失败。"  # 错误信息
        exit 1
    fi
    
    # 检查并安装 Git
    if ! command -v git &> /dev/null; then
        echo "Git 未安装。正在安装 Git..."
        if ! sudo apt install git -y; then
            echo "安装 Git 失败。"  # 错误信息
            exit 1
        fi
    else
        echo "Git 已安装。"  # 成功信息
    fi

    # 检查 Rust 是否已安装
    if command -v rustc &> /dev/null; then
        echo "Rust 已安装，版本为: $(rustc --version)"
    else
        echo "Rust 未安装，正在安装 Rust..."
        # 使用 rustup 安装 Rust
        curl --proto '=https' --tlsv1.2 -sSf https://sh.rustup.rs | sh
        echo "Rust 安装完成。"
        
        # 加载 Rust 环境
        source $HOME/.cargo/env
        echo "Rust 环境已加载。"
    fi

    if [ -d "$HOME/network-api" ]; then
    show "正在删除现有的仓库..." "progress"
    rm -rf "$HOME/network-api"
    fi
    
    # 克隆指定的 GitHub 仓库
    echo "正在克隆仓库..."
    cd
    git clone https://github.com/nexus-xyz/network-api.git

    # 安装依赖项
    cd $HOME/network-api/clients/cli
    echo "安装所需的依赖项..." 
    if ! sudo apt install pkg-config libssl-dev -y; then
        echo "安装依赖项失败。"  # 错误信息
        exit 1
    fi
    
    # 创建 systemd 服务文件
    echo "创建 systemd 服务..." 
    if ! sudo bash -c "cat > $SERVICE_FILE <<EOF
[Unit]
Description=Nexus XYZ Prover Service
After=network.target

[Service]
User=$USER
WorkingDirectory=$HOME/network-api/clients/cli
Environment=NONINTERACTIVE=1
Environment=PATH=/root/.cargo/bin:$PATH
ExecStart=$HOME/.cargo/bin/cargo run --release --bin prover -- beta.orchestrator.nexus.xyz
Restart=always
RestartSec=10

[Install]
WantedBy=multi-user.target
EOF"; then
        echo "创建 systemd 服务文件失败。" 
        exit 1
    fi

    # 重新加载 systemd 并启动服务
    echo "重新加载 systemd 并启动服务..." 
    if ! sudo systemctl daemon-reload; then
        echo "重新加载 systemd 失败。"
        exit 1
    fi

    if ! sudo systemctl start nexus.service; then
        echo "启动服务失败。" 
        exit 1
    fi

    if ! sudo systemctl enable nexus.service; then
        echo "启用服务失败。" 
        exit 1
    fi

    echo "节点启动成功！"
    
    # 等待用户按任意键返回主菜单
    read -p "按任意键返回主菜单"
}

# 查看 Prover 状态的函数
function check_prover_status() {
    echo "查看 Prover 状态..."
    systemctl status nexus.service
}

# 查看日志的函数
function view_logs() {
    echo "查看 Prover 日志..."
    journalctl -u nexus.service -f -n 50
}

# 删除节点的函数
function delete_node() {
    echo "正在删除节点..."
    sudo systemctl stop nexus.service
    sudo systemctl disable nexus.service
    rm -rf /root/network-api
    rm -rf /etc/systemd/system/nexus.service
    echo "成功删除节点，按任意键返回主菜单。"
    
    # 等待用户按任意键返回主菜单
    read -p "按任意键返回主菜单"
}

# 显示状态的函数
function show_status() {
    echo "\$1"
}

# 调用主菜单函数
main_menu