#!/bin/bash

# ============================================
# 灵越智报 - 发布到 sygpudev 服务器
# 本地编译 → 上传 jar → 重启服务
# ============================================

# 不使用 set -e，因为 ssh 远程命令的退出码会导致误退出

# 颜色定义
RED='\033[0;31m'
GREEN='\033[0;32m'
YELLOW='\033[1;33m'
BLUE='\033[0;34m'
NC='\033[0m'

# ========== 配置区域 ==========
SERVER_USER="root"
SERVER_HOST="sygpudev"
SERVER_PORT="${SERVER_PORT:-28529}"

# 路径配置
LOCAL_PROJECT_DIR="/home/hws/workspace/GitLab/ay/lingyue-zhibao"
LOCAL_BACKEND_DIR="${LOCAL_PROJECT_DIR}/backend"
LOCAL_JAR_PATH="${LOCAL_BACKEND_DIR}/lingyue-starter/target/lingyue-starter.jar"

REMOTE_APP_DIR="/data/application/lingyue-zhibao"
REMOTE_JAR="${REMOTE_APP_DIR}/lingyue-starter.jar"
REMOTE_LOG_DIR="${REMOTE_APP_DIR}/logs"

# 应用配置
JAR_NAME="lingyue-starter.jar"
# ==============================

log_info()  { echo -e "${GREEN}[INFO]${NC} $1"; }
log_warn()  { echo -e "${YELLOW}[WARN]${NC} $1"; }
log_error() { echo -e "${RED}[ERROR]${NC} $1"; }
log_title() { echo -e "\n${BLUE}========== $1 ==========${NC}\n"; }

ssh_cmd() {
    ssh -p ${SERVER_PORT} ${SERVER_USER}@${SERVER_HOST} "$@"
}

# 检查服务器连接
check_connection() {
    log_title "检查服务器连接"

    if ssh_cmd "echo 'OK'" > /dev/null 2>&1; then
        log_info "服务器连接成功: ${SERVER_USER}@${SERVER_HOST}"
    else
        log_error "无法连接服务器: ${SERVER_USER}@${SERVER_HOST}"
        log_warn "请检查："
        log_warn "  1. 服务器地址是否正确（可在 /etc/hosts 或 ~/.ssh/config 中配置 sygpudev）"
        log_warn "  2. SSH 密钥是否配置"
        log_warn "  3. 可以通过 SERVER_HOST=<IP> 指定服务器地址"
        exit 1
    fi
}

# 本地编译
build_local() {
    log_title "本地编译项目"

    cd ${LOCAL_BACKEND_DIR}
    log_info "清理并编译..."
    mvn clean package -DskipTests -q

    if [ ! -f "${LOCAL_JAR_PATH}" ]; then
        log_error "编译产物不存在: ${LOCAL_JAR_PATH}"
        exit 1
    fi

    local jar_size=$(du -h "${LOCAL_JAR_PATH}" | awk '{print $1}')
    log_info "编译完成，jar 大小: ${jar_size}"
}

# 停止远程应用
stop_app() {
    log_title "停止远程应用"

    # 优先使用服务器上的 start.sh stop（如果支持）
    ssh_cmd "test -f ${REMOTE_APP_DIR}/start.sh && cd ${REMOTE_APP_DIR} && bash start.sh stop; true"
    sleep 2

    # 确保进程已停止
    ssh_cmd "pkill -f '${JAR_NAME}'; true"
    sleep 2

    if ssh_cmd "pgrep -f '${JAR_NAME}'" > /dev/null 2>&1; then
        log_warn "进程仍在运行，强制终止..."
        ssh_cmd "pkill -9 -f '${JAR_NAME}'; true"
        sleep 2
    fi

    log_info "应用已停止"
}

# 上传 jar
upload_jar() {
    log_title "上传 JAR 到服务器"

    # 备份旧 jar
    ssh_cmd "cd ${REMOTE_APP_DIR} && \
             [ -f ${JAR_NAME} ] && cp ${JAR_NAME} ${JAR_NAME}.bak || true"
    log_info "已备份旧版本为 ${JAR_NAME}.bak"

    # 上传新 jar
    log_info "正在上传 ${JAR_NAME}..."
    scp -P ${SERVER_PORT} "${LOCAL_JAR_PATH}" "${SERVER_USER}@${SERVER_HOST}:${REMOTE_JAR}"

    local remote_size=$(ssh_cmd "du -h ${REMOTE_JAR}" | awk '{print $1}')
    log_info "上传完成，远程文件大小: ${remote_size}"
}

# 启动远程应用
start_app() {
    log_title "启动远程应用"

    # 优先使用服务器上的 start.sh
    if ssh_cmd "test -f ${REMOTE_APP_DIR}/start.sh" 2>/dev/null; then
        log_info "使用服务器 start.sh 启动..."
        ssh_cmd "cd ${REMOTE_APP_DIR} && bash start.sh start"
    else
        log_info "直接启动 jar..."
        ssh_cmd "cd ${REMOTE_APP_DIR} && \
                 mkdir -p ${REMOTE_LOG_DIR} && \
                 nohup java -Xms1g -Xmx2g -XX:+UseG1GC \
                       -jar ${JAR_NAME} \
                       --spring.config.location=file:./application.yml \
                       > ${REMOTE_LOG_DIR}/app.log 2>&1 &"
    fi

    log_info "应用启动中..."
    sleep 5

    if ssh_cmd "pgrep -f '${JAR_NAME}'" > /dev/null 2>&1; then
        local pid=$(ssh_cmd "pgrep -f '${JAR_NAME}'")
        log_info "应用启动成功 (PID: ${pid})"
    else
        log_error "应用启动失败，查看日志："
        ssh_cmd "tail -50 ${REMOTE_LOG_DIR}/app.log 2>/dev/null || tail -50 /var/log/lingyue.log 2>/dev/null || echo '未找到日志文件'"
        exit 1
    fi
}

# 健康检查
health_check() {
    log_title "健康检查"

    local max_attempts=12
    local attempt=1

    while [ $attempt -le $max_attempts ]; do
        log_info "尝试 ${attempt}/${max_attempts}..."

        local response=$(ssh_cmd "curl -s -o /dev/null -w '%{http_code}' http://localhost:8001/api/v1/projects 2>/dev/null" || echo "000")

        if [ "$response" != "000" ] && [ "$response" != "404" ]; then
            log_info "健康检查通过 ✓ (HTTP ${response})"
            echo ""
            return 0
        fi

        sleep 5
        attempt=$((attempt + 1))
    done

    log_warn "健康检查超时，应用可能仍在启动中"
    log_info "可手动检查: ssh ${SERVER_USER}@${SERVER_HOST} 'curl http://localhost:8001/actuator/health'"
}

# 回滚到上一版本
rollback() {
    log_title "回滚到上一版本"

    if ssh_cmd "test -f ${REMOTE_APP_DIR}/${JAR_NAME}.bak" 2>/dev/null; then
        stop_app
        ssh_cmd "cd ${REMOTE_APP_DIR} && mv ${JAR_NAME}.bak ${JAR_NAME}"
        log_info "已恢复上一版本"
        start_app
        health_check
    else
        log_error "没有找到备份文件 ${JAR_NAME}.bak，无法回滚"
        exit 1
    fi
}

# 查看远程日志
show_logs() {
    log_title "应用日志"
    ssh_cmd "tail -100 ${REMOTE_LOG_DIR}/app.log 2>/dev/null || tail -100 /var/log/lingyue.log 2>/dev/null || echo '未找到日志文件'"
}

tail_logs() {
    log_info "实时日志 (Ctrl+C 退出)"
    ssh_cmd "tail -f ${REMOTE_LOG_DIR}/app.log 2>/dev/null || tail -f /var/log/lingyue.log 2>/dev/null || echo '未找到日志文件'"
}

show_status() {
    log_title "应用状态"

    if ssh_cmd "pgrep -f '${JAR_NAME}'" > /dev/null 2>&1; then
        local pid=$(ssh_cmd "pgrep -f '${JAR_NAME}'")
        log_info "应用运行中 (PID: ${pid})"
        ssh_cmd "ps -p ${pid} -o pid,rss,vsz,etime,args --no-headers" 2>/dev/null || true
    else
        log_warn "应用未运行"
    fi
}

# 完整部署: 编译 → 上传 → 重启
full_deploy() {
    check_connection
    build_local
    stop_app
    upload_jar
    start_app
    health_check

    log_title "部署完成 🎉"
    echo -e "${GREEN}服务器: ${SERVER_USER}@${SERVER_HOST}:${REMOTE_APP_DIR}${NC}"
    echo ""
}

# 快速部署: 仅上传 jar 并重启（跳过编译，使用已有的 target）
quick_deploy() {
    if [ ! -f "${LOCAL_JAR_PATH}" ]; then
        log_error "本地 jar 不存在，请先编译或使用 deploy 命令"
        exit 1
    fi

    check_connection
    stop_app
    upload_jar
    start_app
    health_check

    log_title "快速部署完成 🎉"
}

# 帮助
show_help() {
    cat <<EOF
灵越智报 - 发布到 sygpudev 服务器

用法: ./deploy-sygpudev.sh [命令]

命令:
    deploy          完整部署（本地编译 + 上传 + 重启）
    quick           快速部署（上传已编译的 jar + 重启）
    build           仅本地编译
    upload          仅上传 jar（不重启）
    restart         重启远程应用（不编译不上传）
    stop            停止远程应用
    rollback        回滚到上一版本
    status          查看应用状态
    logs            查看最近日志
    tail            实时查看日志
    help            显示此帮助

示例:
    ./deploy-sygpudev.sh deploy       # 完整部署
    ./deploy-sygpudev.sh quick        # 快速部署（已编译过）
    ./deploy-sygpudev.sh rollback     # 回滚到上一版本
    ./deploy-sygpudev.sh tail         # 实时看日志

EOF
}

# 主函数
main() {
    case "${1:-deploy}" in
        deploy)
            full_deploy
            ;;
        quick)
            quick_deploy
            ;;
        build)
            build_local
            ;;
        upload)
            check_connection
            upload_jar
            log_info "上传完成（未重启）"
            ;;
        restart)
            check_connection
            stop_app
            start_app
            health_check
            ;;
        stop)
            check_connection
            stop_app
            ;;
        rollback)
            check_connection
            rollback
            ;;
        status)
            check_connection
            show_status
            ;;
        logs)
            check_connection
            show_logs
            ;;
        tail)
            check_connection
            tail_logs
            ;;
        help|--help|-h)
            show_help
            ;;
        *)
            log_error "未知命令: $1"
            show_help
            exit 1
            ;;
    esac
}

main "$@"