#!/bin/bash
#  启动/停止/重启脚本

# ===================== 配置区 =====================
# 程序路径
AGENT_PATH="/bin/monitor-agent"
# 日志文件路径
LOG_FILE="/root/monitor/agent.log"
# PID文件路径（记录进程ID）
PID_FILE="/root/monitor/agent.pid"
# 启动参数（根据实际需求调整）
START_ARGS=""
# ==================== 配置区结束 ====================

# 检查程序文件是否存在
check_agent_exists() {
    if [ ! -f "${AGENT_PATH}" ]; then
        echo "错误：程序文件 ${AGENT_PATH} 不存在！"
        exit 1
    fi
    if [ ! -x "${AGENT_PATH}" ]; then
        echo "错误：程序文件 ${AGENT_PATH} 没有执行权限，正在尝试添加..."
        chmod +x "${AGENT_PATH}"
        if [ $? -ne 0 ]; then
            echo "错误：添加执行权限失败，请手动执行 chmod +x ${AGENT_PATH}"
            exit 1
        fi
    fi
}

# 检查进程是否运行
check_running() {
    if [ -f "${PID_FILE}" ]; then
        PID=$(cat "${PID_FILE}")
        if ps -p "${PID}" > /dev/null 2>&1; then
            return 0  # 运行中
        else
            rm -f "${PID_FILE}"  # PID文件存在但进程已死，清理PID文件
        fi
    fi
    return 1  # 未运行
}

# 启动程序
start_agent() {
    if check_running; then
        echo "✅ monitor-agent 已在运行（PID: $(cat ${PID_FILE})）"
        return 0
    fi

    echo "🚀 正在启动 monitor-agent..."
    # 创建日志目录（如果不存在）
    mkdir -p "$(dirname ${LOG_FILE})"
    # 后台启动程序，重定向日志，记录PID
    nohup "${AGENT_PATH}" ${START_ARGS} > "${LOG_FILE}" 2>&1 &
    AGENT_PID=$!
    echo "${AGENT_PID}" > "${PID_FILE}"

    # 等待2秒检查是否启动成功
    sleep 2
    if check_running; then
        echo "✅ monitor-agent 启动成功（PID: ${AGENT_PID}）"
        echo "日志文件：${LOG_FILE}"
    else
        echo "❌ monitor-agent 启动失败！请查看日志：${LOG_FILE}"
        rm -f "${PID_FILE}"
        exit 1
    fi
}

# 停止程序
stop_agent() {
    if ! check_running; then
        echo "ℹ️ monitor-agent 未运行"
        return 0
    fi

    PID=$(cat "${PID_FILE}")
    echo "🛑 正在停止 monitor-agent（PID: ${PID}）..."
    # 优雅停止（先尝试TERM信号，失败则强制KILL）
    kill "${PID}" > /dev/null 2>&1
    sleep 3

    if ps -p "${PID}" > /dev/null 2>&1; then
        echo "⚠️ 优雅停止失败，强制杀死进程..."
        kill -9 "${PID}" > /dev/null 2>&1
        sleep 1
    fi

    # 清理PID文件
    rm -f "${PID_FILE}"
    echo "✅ monitor-agent 已停止"
}

# 查看状态
status_agent() {
    if check_running; then
        echo "✅ monitor-agent 运行中（PID: $(cat ${PID_FILE})）"
    else
        echo "ℹ️ monitor-agent 未运行"
    fi
}

# 重启程序
restart_agent() {
    echo "🔄 正在重启 monitor-agent..."
    stop_agent
    sleep 2
    start_agent
}

# 帮助信息
show_help() {
    echo "使用方法：$0 [start|stop|restart|status|help]"
    echo "  start   - 启动 monitor-agent"
    echo "  stop    - 停止 monitor-agent"
    echo "  restart - 重启 monitor-agent"
    echo "  status  - 查看 monitor-agent 运行状态"
    echo "  help    - 显示帮助信息"
}

# 主逻辑
main() {
    # 检查是否为root用户（可选，根据需求调整）
    if [ "$(id -u)" -ne 0 ]; then
        echo "警告：建议使用root用户运行此脚本（当前用户：$(whoami)）"
        # exit 1  # 如果强制要求root，取消注释
    fi

    check_agent_exists

    case "$1" in
        start)
            start_agent
            ;;
        stop)
            stop_agent
            ;;
        restart)
            restart_agent
            ;;
        status)
            status_agent
            ;;
        help|--help|-h)
            show_help
            ;;
        *)
            echo "错误：无效参数 '$1'"
            show_help
            exit 1
            ;;
    esac
}

# 执行主逻辑
main "$@"