#!/bin/bash

# 卸载 Agent 的函数
uninstall_agent() {
    # 停止并卸载服务
    if [ -d "/opt/nezha/agent" ]; then
        cd /opt/nezha/agent/
        ./nezha-agent service uninstall
        # 删除 Agent 文件夹
        cd /
        rm -rf /opt/nezha/agent/
        echo "Agent 服务已成功卸载并删除相关文件。"
    else
        echo "Agent 未安装在默认路径，跳过卸载步骤。"
    fi

    # 如果使用了 agent.sh 安装，可以尝试使用其卸载功能
    if [ -f "./agent.sh" ]; then
        ./agent.sh uninstall
        echo "使用 agent.sh 成功卸载 Agent。"
    else
        echo "agent.sh 脚本不存在，无法使用其卸载功能。"
    fi
}

# 安装 Agent 的函数
install_agent() {
    local server="nz.337.plus:443"
    local tls="true"
    local client_secret="kDxAXwry5760XON89Z7r7xXrojvLBr1T"

    curl -L https://raw.githubusercontent.com/nezhahq/scripts/main/agent/install.sh -o agent.sh && chmod +x agent.sh && \
    env NZ_SERVER=$server NZ_TLS=$tls NZ_CLIENT_SECRET=$client_secret ./agent.sh

    echo "Agent 已成功安装。"
}

echo "请选择操作："
echo "1. 卸载 Agent"
echo "2. 安装 Agent"
read -p "输入你的选择 (1/2): " choice

case $choice in
    1)
        uninstall_agent
        ;;
    2)
        install_agent
        ;;
    *)
        echo "无效的选择，请输入 1 或 2。"
        ;;
esac