#!/bin/bash

# 主菜单函数
function main_menu() {
    while true; do
        clear
        echo "================================================================"
        echo "脚本由哈哈哈哈编写，推特 @ferdie_jhovie，免费开源，请勿相信收费"
        echo "如有问题，可联系推特，仅此只有一个号"
        echo "================================================================"
        echo "退出脚本，请按键盘 Ctrl + C"
        echo "请选择要执行的操作:"
        echo "1) 安装部署节点"
        echo "2) 查看质押余额"
        echo "3) 查看 broker 日志"
        echo "4) 删除节点"
        echo "q) 退出脚本"
        echo "================================================================"
        read -p "请输入选项 [1/2/3/4/q]: " choice
        case $choice in
            1)
                install_node
                ;;
            2)
                check_stake_balance
                ;;
            3)
                view_broker_logs
                ;;
            4)
                remove_node
                ;;
            q|Q)
                echo "感谢使用，再见！"
                exit 0
                ;;
            *)
                echo "无效选项，请重新选择"
                sleep 2
                ;;
        esac
    done
}

# 安装部署节点函数
function install_node() {
    clear
    echo "开始安装部署节点..."
    
    # 检查是否以 root 权限运行
    if [ "$EUID" -ne 0 ]; then 
        echo "请使用 sudo 运行此脚本"
        exit 1
    fi

    echo "检查 Docker 安装状态..."
    if ! command -v docker &> /dev/null; then
        echo "正在安装 Docker..."
        apt-get update
        apt-get install -y ca-certificates curl gnupg
        install -m 0755 -d /etc/apt/keyrings
        curl -fsSL https://download.docker.com/linux/ubuntu/gpg | gpg --dearmor -o /etc/apt/keyrings/docker.gpg
        chmod a+r /etc/apt/keyrings/docker.gpg
        echo "deb [arch=$(dpkg --print-architecture) signed-by=/etc/apt/keyrings/docker.gpg] https://download.docker.com/linux/ubuntu $(. /etc/os-release && echo "$VERSION_CODENAME") stable" | tee /etc/apt/sources.list.d/docker.list > /dev/null
        apt-get update
        apt-get install -y docker-ce docker-ce-cli containerd.io docker-buildx-plugin docker-compose-plugin
        usermod -aG docker $SUDO_USER
        echo "Docker 安装完成，请注销并重新登录以使组成员身份生效"
    fi

    echo "检查 NVIDIA Docker 支持..."
    if ! command -v nvidia-docker &> /dev/null; then
        echo "正在安装 NVIDIA Container Toolkit..."
        distribution=$(. /etc/os-release;echo $ID$VERSION_ID)
        curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | apt-key add -
        curl -s -L https://nvidia.github.io/nvidia-docker/$distribution/nvidia-docker.list | tee /etc/apt/sources.list.d/nvidia-docker.list
        apt-get update
        apt-get install -y nvidia-container-toolkit
        systemctl restart docker
        echo "NVIDIA Container Toolkit 安装完成"
    fi

    echo "检查 screen 安装状态..."
    if ! command -v screen &> /dev/null; then
        echo "正在安装 screen..."
        apt-get update
        apt-get install -y screen
        if [ $? -ne 0 ]; then
            echo "screen 安装失败，请手动安装"
            exit 1
        fi
        echo "screen 安装完成"
    fi

    echo "检查 just 安装状态..."
    if ! command -v just &> /dev/null; then
        echo "正在安装 just..."
        curl --proto '=https' --tlsv1.2 -sSf https://just.systems/install.sh | bash -s -- --to /usr/local/bin
        if [ $? -ne 0 ]; then
            echo "just 安装失败，请手动安装"
            exit 1
        fi
        echo "just 安装完成"
    fi

    echo "检查 bc 计算器依赖..."
    if ! command -v bc &> /dev/null; then
        echo "正在安装 bc..."
        apt-get update
        apt-get install -y bc
        if [ $? -ne 0 ]; then
            echo "bc 安装失败，请手动安装"
            exit 1
        fi
        echo "bc 安装完成"
    fi

    echo "开始克隆仓库..."
    if [ -d "boundless" ]; then
        echo "检测到已有 boundless 目录，自动删除以保证全新克隆..."
        rm -rf boundless
    fi
    git clone https://github.com/boundless-xyz/boundless
    if [ $? -ne 0 ]; then
        echo "克隆失败，请检查网络连接或仓库地址是否正确"
        exit 1
    fi

    echo "当前工作目录: $(pwd)"
    echo "检查 boundless 目录..."
    if [ ! -d "boundless" ]; then
        echo "错误：未找到 boundless 目录，请确保已正确克隆仓库"
        echo "当前目录内容:"
        ls -la
        exit 1
    fi

    echo "找到 boundless 目录，正在切换..."
    # 切换到 boundless 目录
    cd boundless
    echo "已切换到 boundless 目录: $(pwd)"

    echo "切换到 release-0.10 分支..."
    git checkout release-0.10
    if [ $? -ne 0 ]; then
        echo "切换分支失败，请检查分支名称是否正确"
        exit 1
    fi

    echo "安装 Rust 和相关工具链..."
    echo "正在安装 rustup..."
    curl --proto '=https' --tlsv1.2 -sSf https://sh.rustup.rs | sh -s -- -y
    source "$HOME/.cargo/env"
    if [ $? -ne 0 ]; then
        echo "rustup 安装失败，请检查网络连接或手动安装"
        exit 1
    fi
    echo "rustup 安装完成"

    echo "正在更新 rustup..."
    rustup update
    if [ $? -ne 0 ]; then
        echo "rustup 更新失败，请检查网络连接或手动更新"
        exit 1
    fi
    echo "rustup 更新完成"

    echo "正在安装 Rust 工具链..."
    apt-get update
    apt-get install -y cargo
    if [ $? -ne 0 ]; then
        echo "Rust 工具链安装失败，请手动安装"
        exit 1
    fi
    echo "Rust 工具链安装完成"

    echo "验证 Cargo 安装..."
    cargo --version
    if [ $? -ne 0 ]; then
        echo "Cargo 验证失败，请检查安装"
        exit 1
    fi
    echo "Cargo 验证通过"

    echo "正在安装 rzup..."
    curl -L https://risczero.com/install | bash
    source ~/.bashrc
    export PATH="$HOME/.risc0/bin:$PATH"
    if [ $? -ne 0 ]; then
        echo "rzup 安装失败，请检查网络连接或手动安装"
        exit 1
    fi
    echo "rzup 安装完成"

    echo "验证 rzup 安装..."
    rzup --version
    if [ $? -ne 0 ]; then
        echo "rzup 验证失败，请检查安装"
        exit 1
    fi
    echo "rzup 验证通过"

    echo "正在安装 RISC Zero Rust 工具链..."
    if rzup toolchain list | grep -q 'risc0'; then
        echo "RISC Zero Rust 工具链已安装，跳过安装。"
    else
        rzup install rust
        if [ $? -ne 0 ]; then
            echo "RISC Zero Rust 工具链安装失败，请手动安装"
            echo "如遇 GitHub API rate limit，请在 https://github.com/settings/tokens 生成token并 export GITHUB_TOKEN=你的token 后重试。"
            exit 1
        fi
    fi
    echo "RISC Zero Rust 工具链安装完成"

    echo "正在安装 cargo-risczero..."
    if cargo install --list | grep -q 'cargo-risczero'; then
        echo "cargo-risczero 已安装，跳过安装。"
    else
        cargo install cargo-risczero
        rzup install cargo-risczero
        if [ $? -ne 0 ]; then
            echo "cargo-risczero 安装失败，请检查网络连接或手动安装"
            exit 1
        fi
    fi
    echo "cargo-risczero 安装完成"

    echo "再次更新 rustup..."
    rustup update
    if [ $? -ne 0 ]; then
        echo "rustup 更新失败，请检查网络连接或手动更新"
        exit 1
    fi
    echo "rustup 更新完成"

    echo "正在安装 bento-client..."
    if cargo install --list | grep -q 'bento_cli'; then
        echo "bento-client 已安装，跳过安装。"
    else
        cargo install --git https://github.com/risc0/risc0 bento-client --bin bento_cli
        echo 'export PATH="$HOME/.cargo/bin:$PATH"' >> ~/.bashrc
        source ~/.bashrc
        if [ $? -ne 0 ]; then
            echo "bento-client 安装失败，请检查网络连接或手动安装"
            exit 1
        fi
    fi
    echo "bento-client 安装完成"

    echo "验证 bento-client 安装..."
    bento_cli --version
    if [ $? -ne 0 ]; then
        echo "bento-client 验证失败，请检查安装"
        exit 1
    fi
    echo "bento-client 验证通过"

    echo "执行 setup.sh 脚本..."
    chmod +x scripts/setup.sh
    ./scripts/setup.sh || echo "setup.sh 执行失败，但继续执行后续步骤。"

    echo "正在安装 boundless-cli..."
    if cargo install --list | grep -q 'boundless-cli'; then
        echo "boundless-cli 已安装，跳过安装。"
    else
        cargo install --locked boundless-cli
        export PATH=$PATH:/root/.cargo/bin
        source ~/.bashrc
        if [ $? -ne 0 ]; then
            echo "boundless-cli 安装失败，请检查网络连接或手动安装"
            exit 1
        fi
    fi
    echo "boundless-cli 安装完成"

    echo "验证 boundless-cli 安装..."
    boundless -h
    if [ $? -ne 0 ]; then
        echo "boundless-cli 验证失败，请检查安装"
        exit 1
    fi
    echo "boundless-cli 验证通过"

    echo "所有依赖安装完成！"
    echo "请注销并重新登录以使 Docker 组成员身份生效"

    # 检查物理内存（单位：GB）
    MEM_TOTAL=$(awk '/MemTotal/ {printf "%.0f", $2/1024/1024}' /proc/meminfo)
    # 检查CPU核数
    CPU_TOTAL=$(nproc)

    echo "----------------------------------------"
    echo "本机检测到的最大内存为：${MEM_TOTAL} GB"
    echo "本机检测到的最大CPU核数为：${CPU_TOTAL}"
    echo "请根据实际情况填写要分配给节点的内存和CPU核数（建议不要填满，留给系统部分资源）"
    echo "----------------------------------------"

    # 让用户输入内存和CPU核数
    while true; do
        read -p "请输入分配给节点的内存（单位GB，最大${MEM_TOTAL}）： " MEM_INPUT
        if [[ "$MEM_INPUT" =~ ^[0-9]+$ ]] && [ "$MEM_INPUT" -le "$MEM_TOTAL" ] && [ "$MEM_INPUT" -ge 1 ]; then
            break
        else
            echo "输入无效，请输入1-${MEM_TOTAL}之间的整数"
        fi
    done

    while true; do
        read -p "请输入分配给节点的CPU核数（最大${CPU_TOTAL}）： " CPU_INPUT
        if [[ "$CPU_INPUT" =~ ^[0-9]+$ ]] && [ "$CPU_INPUT" -le "$CPU_TOTAL" ] && [ "$CPU_INPUT" -ge 1 ]; then
            break
        else
            echo "输入无效，请输入1-${CPU_TOTAL}之间的整数"
        fi
    done

    # compose.yml 路径
    COMPOSE_FILE="compose.yml"
    if [ ! -f "$COMPOSE_FILE" ]; then
        COMPOSE_FILE="docker-compose.yml"
    fi

    if [ ! -f "$COMPOSE_FILE" ]; then
        echo "未找到 compose.yml 文件，请确认路径！"
    else
        # 修改 mem_limit
        sed -i "/x-exec-agent-common: &exec-agent-common/,/cpus:/s/mem_limit: .*/mem_limit: ${MEM_INPUT}G/" "$COMPOSE_FILE"
        # 修改 cpus
        sed -i "/x-exec-agent-common: &exec-agent-common/,/environment:/s/cpus: .*/cpus: ${CPU_INPUT}/" "$COMPOSE_FILE"
        echo "compose.yml 已根据您的输入自动修改：内存${MEM_INPUT}G，CPU核数${CPU_INPUT}"
    fi

    # 检查是否有nvidia-smi命令
    if command -v nvidia-smi &> /dev/null; then
        GPU_MEM_TOTAL=$(nvidia-smi --query-gpu=memory.total --format=csv,noheader,nounits | head -n1)
        GPU_NAME=$(nvidia-smi --query-gpu=name --format=csv,noheader | head -n1)
        echo "----------------------------------------"
        echo "检测到GPU: $GPU_NAME"
        echo "本机检测到的GPU 0最大显存为：${GPU_MEM_TOTAL} MB"
        echo "请根据实际情况填写要分配给gpu_prove_agent0的显存和CPU核数（建议不要填满，留给系统部分资源）"
        echo "----------------------------------------"

        # 让用户输入GPU显存（单位GB，四舍五入）
        GPU_MEM_TOTAL_GB=$(( (GPU_MEM_TOTAL + 1023) / 1024 ))
        while true; do
            read -p "请输入分配给gpu_prove_agent0的显存（单位GB，最大${GPU_MEM_TOTAL_GB}）： " GPU_MEM_INPUT
            if [[ "$GPU_MEM_INPUT" =~ ^[0-9]+$ ]] && [ "$GPU_MEM_INPUT" -le "$GPU_MEM_TOTAL_GB" ] && [ "$GPU_MEM_INPUT" -ge 1 ]; then
                break
            else
                echo "输入无效，请输入1-${GPU_MEM_TOTAL_GB}之间的整数"
            fi
        done

        # 让用户输入GPU CPU核数（建议不要填满，通常为1-4）
        GPU_CPU_TOTAL=$(nproc)
        while true; do
            read -p "请输入分配给gpu_prove_agent0的CPU核数（最大${GPU_CPU_TOTAL}，建议不要填满）： " GPU_CPU_INPUT
            if [[ "$GPU_CPU_INPUT" =~ ^[0-9]+$ ]] && [ "$GPU_CPU_INPUT" -le "$GPU_CPU_TOTAL" ] && [ "$GPU_CPU_INPUT" -ge 1 ]; then
                break
            else
                echo "输入无效，请输入1-${GPU_CPU_TOTAL}之间的整数"
            fi
        done

        # compose.yml 路径
        COMPOSE_FILE="compose.yml"
        if [ ! -f "$COMPOSE_FILE" ]; then
            COMPOSE_FILE="docker-compose.yml"
        fi

        if [ ! -f "$COMPOSE_FILE" ]; then
            echo "未找到 compose.yml 文件，请确认路径！"
        else
            # 修改 gpu_prove_agent0 的 mem_limit
            sed -i "/gpu_prove_agent0:/,/cpus:/s/mem_limit: .*/mem_limit: ${GPU_MEM_INPUT}G/" "$COMPOSE_FILE"
            # 修改 gpu_prove_agent0 的 cpus
            sed -i "/gpu_prove_agent0:/,/entrypoint:/s/cpus: .*/cpus: ${GPU_CPU_INPUT}/" "$COMPOSE_FILE"
            echo "compose.yml 已根据您的输入自动修改：gpu_prove_agent0 显存${GPU_MEM_INPUT}G，CPU核数${GPU_CPU_INPUT}"
        fi
    else
        echo "未检测到NVIDIA GPU，跳过GPU相关配置。"
    fi

    # 获取用户输入并写入 .env.base 文件
    echo "请设置您的环境变量："
    echo "提示：请使用 Base 主网的 Alchemy RPC URL"
    echo "您可以在 https://www.alchemy.com/ 注册并获取 API KEY"
    echo "注意：请确保选择 Base 网络"
    echo "----------------------------------------"

    read -p "请输入您的 PRIVATE_KEY: " PRIVATE_KEY
    read -p "请输入您的 Base 主网 RPC_URL: " RPC_URL

    # 检查 RPC_URL 是否包含 base
    if [[ "$RPC_URL" != *"base"* ]]; then
        echo "警告：您使用的不是 Base 网络的 RPC URL，这可能会导致连接问题"
        read -p "是否继续使用当前 RPC URL？(y/n): " confirm
        if [[ "$confirm" != "y" && "$confirm" != "Y" ]]; then
            echo "已取消设置，请重新运行脚本"
            exit 1
        fi
    fi

    # 验证输入是否为空
    if [ -z "$PRIVATE_KEY" ] || [ -z "$RPC_URL" ]; then
        echo "错误：请输入有效的 PRIVATE_KEY 和 RPC_URL"
        exit 1
    fi

    # 避免重复写入，移除旧的 PRIVATE_KEY 和 RPC_URL（如果存在）
    sed -i '/^export PRIVATE_KEY=/d' .env.base 2>/dev/null
    sed -i '/^export RPC_URL=/d' .env.base 2>/dev/null

    # 追加环境变量到 .env.base 文件
    echo "正在将环境变量写入 .env.base..."
    echo "export PRIVATE_KEY=$PRIVATE_KEY" >> .env.base
    echo "export RPC_URL=$RPC_URL" >> .env.base

    # 验证是否写入成功
    if grep -q "export PRIVATE_KEY=$PRIVATE_KEY" .env.base && grep -q "export RPC_URL=$RPC_URL" .env.base; then
        echo ".env.base 文件已成功写入！"
        # 加载环境变量
        echo "正在加载环境变量..."
        source .env.base
        # 验证环境变量是否加载成功
        if [ -z "$PRIVATE_KEY" ] || [ -z "$RPC_URL" ]; then
            echo "错误：加载 .env.base 文件失败，请检查文件内容"
            exit 1
        fi
        echo "环境变量加载成功！"
    else
        echo "错误：写入 .env.base 文件失败，请检查文件权限"
        exit 1
    fi

    # 设置 testnet 环境
    echo "正在设置 testnet 环境..."
    # 环境变量已经通过 .env.base 文件设置
    # source <(just env testnet)
    # if [ $? -ne 0 ]; then
    #     echo "设置 testnet 环境失败，请检查网络连接或手动设置"
    #     exit 1
    # fi
    echo "testnet 环境设置完成！环境变量已通过 .env.base 文件配置"

    echo "----------------------------------------"
    echo "请设置存款数量（USDC）："
    echo "注意：请确保您的账户中有足够的 USDC"
    echo "提示：如果之前已经存过USDC，可以选择跳过此步骤"
    echo "----------------------------------------"

    read -p "是否需要存入USDC? (y/n): " NEED_DEPOSIT
    if [[ "$NEED_DEPOSIT" == "y" || "$NEED_DEPOSIT" == "Y" ]]; then
        while true; do
            read -p "请输入要存入的 USDC 数量（最低5）：" USDC_AMOUNT
            if [[ "$USDC_AMOUNT" =~ ^[0-9]+(\.[0-9]+)?$ ]] && awk "BEGIN{exit !($USDC_AMOUNT >= 5)}"; then
                break
            else
                echo "错误：请输入不少于5的有效数字"
            fi
        done

        echo "正在执行存款操作..."
        echo "存款数量: $USDC_AMOUNT USDC（最低5个，建议多存）"
        boundless account deposit-stake "$USDC_AMOUNT"

        if [ $? -ne 0 ]; then
            echo "存款操作失败，请检查："
            echo "1. 账户余额是否充足"
            echo "2. 网络连接是否正常"
            echo "3. 环境变量是否正确设置"
            exit 1
        fi

        echo "存款操作完成！"
    else
        echo "已跳过USDC存款步骤"
    fi

    # 启动 just broker 前，设置 SEGMENT_SIZE
    echo "----------------------------------------"
    echo "请设置 GPU 的 SEGMENT_SIZE（推荐值：8GB填19，16GB填20，20GB填21，40GB填22）"
    while true; do
        read -p "请输入 SEGMENT_SIZE（19/20/21/22）:" SEGMENT_SIZE
        if [[ "$SEGMENT_SIZE" =~ ^(19|20|21|22)$ ]]; then
            break
        else
            echo "输入无效，请输入 19、20、21 或 22 之一"
        fi
    done

    # compose.yml 路径
    COMPOSE_FILE="compose.yml"
    if [ ! -f "$COMPOSE_FILE" ]; then
        COMPOSE_FILE="docker-compose.yml"
    fi

    if [ ! -f "$COMPOSE_FILE" ]; then
        echo "未找到 compose.yml 文件，请确认路径！"
    else
        # 修改 entrypoint 的 SEGMENT_SIZE
        sed -i "/x-exec-agent-common: &exec-agent-common/,/entrypoint:/s|entrypoint: /app/agent -t exec --segment-po2 \\${SEGMENT_SIZE:-[0-9][0-9]}|entrypoint: /app/agent -t exec --segment-po2 $SEGMENT_SIZE|" "$COMPOSE_FILE"
        echo "compose.yml 已根据您的输入自动修改 SEGMENT_SIZE: $SEGMENT_SIZE"
    fi

    # 确保在 boundless 目录下启动 broker
    just broker &
    echo "broker 服务已在后台启动。"

    echo "脚本执行完成！"

    # 安装完成后返回主菜单
    read -n 1 -s -r -p "按任意键返回主菜单..."
    echo
}

# 查看质押余额函数
function check_stake_balance() {
    clear
    echo "查看质押余额"
    echo "----------------------------------------"
    
    # 检查环境变量是否设置
    if [ ! -f "boundless/.env.base" ]; then
        echo "错误：未找到 .env.base 文件"
        echo "请先运行选项 1 完成安装部署"
        echo "按回车键返回主菜单..."
        read
        return
    fi

    # 加载环境变量
    cd boundless
    source .env.base
    if [ -z "$PRIVATE_KEY" ] || [ -z "$RPC_URL" ]; then
        echo "错误：环境变量未正确加载"
        echo "请检查 .env.base 文件内容"
        echo "按回车键返回主菜单..."
        read
        return
    fi

    # 获取钱包地址
    read -p "请输入要查询的钱包地址: " WALLET_ADDRESS
    
    # 验证钱包地址格式
    if [[ ! "$WALLET_ADDRESS" =~ ^0x[a-fA-F0-9]{40}$ ]]; then
        echo "错误：无效的钱包地址格式"
        echo "请确保输入的是有效的以太坊地址"
        echo "按回车键返回主菜单..."
        read
        return
    fi

    echo "正在查询质押余额..."
    echo "钱包地址: $WALLET_ADDRESS"
    echo "----------------------------------------"
    
    # 执行查询命令
    boundless account stake-balance "$WALLET_ADDRESS"
    
    if [ $? -ne 0 ]; then
        echo "查询失败，请检查："
        echo "1. 钱包地址是否正确"
        echo "2. 网络连接是否正常"
        echo "3. 环境变量是否正确设置"
    fi
    
    echo "----------------------------------------"
    echo "按回车键返回主菜单..."
    read
}

# 查看 broker 日志函数
function view_broker_logs() {
    clear
    echo "查看 broker 日志"
    echo "----------------------------------------"
    
    # 检查是否在 boundless 目录
    if [ ! -d "boundless" ]; then
        echo "错误：未找到 boundless 目录"
        echo "请先运行选项 1 完成安装部署"
        echo "按回车键返回主菜单..."
        read
        return
    fi

    # 切换到 boundless 目录
    cd boundless

    echo "----------------------------------------"
    echo "日志查看说明："
    echo "1. broker 服务已在后台运行"
    echo "2. 使用 Ctrl+C 可以停止查看日志（服务会继续在后台运行）"
    echo "----------------------------------------"
    echo "按回车键开始查看日志..."
    read

    # 查看日志
    docker compose logs -f broker

    echo "----------------------------------------"
    echo "日志查看已结束"
    echo "broker 服务仍在后台运行"
    echo "按回车键返回主菜单..."
    read
}

# 删除节点函数
function remove_node() {
    clear
    echo "删除节点"
    echo "----------------------------------------"
    echo "警告：此操作将完全删除节点，包括："
    echo "1. 停止 broker 服务"
    echo "2. 清理所有节点数据"
    echo "3. 删除整个 boundless 目录"
    echo "----------------------------------------"
    
    read -p "是否确定要删除节点？(y/n): " confirm
    if [[ "$confirm" != "y" && "$confirm" != "Y" ]]; then
        echo "已取消删除操作"
        echo "按回车键返回主菜单..."
        read
        return
    fi

    # 检查是否在 boundless 目录
    if [ ! -d "boundless" ]; then
        echo "错误：未找到 boundless 目录"
        echo "按回车键返回主菜单..."
        read
        return
    fi

    # 切换到 boundless 目录
    cd boundless

    # 停止 broker 服务
    echo "正在停止 broker 服务..."
    if pgrep -f "just broker" > /dev/null; then
        just broker down
        sleep 2
        echo "broker 服务已停止"
    else
        echo "broker 服务未运行"
    fi

    # 清理数据
    echo "正在清理节点数据..."
    just broker clean
    if [ $? -ne 0 ]; then
        echo "警告：清理数据时出现错误，但将继续删除目录"
    else
        echo "节点数据已清理"
    fi

    # 返回上级目录
    cd ..

    # 删除 boundless 目录
    echo "正在删除 boundless 目录..."
    read -p "最后确认：是否要删除整个 boundless 目录？(y/n): " final_confirm
    if [[ "$final_confirm" == "y" || "$final_confirm" == "Y" ]]; then
        rm -rf boundless
        if [ $? -eq 0 ]; then
            echo "boundless 目录已删除"
        else
            echo "错误：删除目录失败，请手动删除"
        fi
    else
        echo "已取消删除目录"
    fi

    echo "----------------------------------------"
    echo "节点删除操作完成"
    echo "按回车键返回主菜单..."
    read
}

# 启动主菜单
main_menu