#!/bin/bash
# IoT Platform 一键部署脚本
# 用法:
#   ./deploy.sh --build              # 本地构建后部署
#   ./deploy.sh --jar path/to/jar    # 使用指定 jar 部署
#   ./deploy.sh --build --jar path   # 构建并使用指定 jar（覆盖）
#
# 环境变量:
#   SERVER_HOST    默认: 47.104.204.180
#   SERVER_USER    默认: root
#   REMOTE_DIR     默认: /opt/iot-platform

set -e

# 配置
SERVER_HOST="${SERVER_HOST:-47.104.204.180}"
SERVER_USER="${SERVER_USER:-root}"
REMOTE_DIR="${REMOTE_DIR:-/opt/iot-platform}"
APP_NAME="iot-platform"
LOCAL_JAR=""
DO_BUILD=false

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

# 解析参数
while [[ $# -gt 0 ]]; do
    case $1 in
        --build)
            DO_BUILD=true
            shift
            ;;
        --jar)
            LOCAL_JAR="$2"
            shift 2
            ;;
        --help|-h)
            echo "用法: ./deploy.sh [--build] [--jar <path>]"
            echo "  --build     本地执行 mvn clean package"
            echo "  --jar       指定要部署的 jar 文件路径"
            exit 0
            ;;
        *)
            echo "未知参数: $1"
            exit 1
            ;;
    esac
done

# 确定 jar 路径
if [ "$DO_BUILD" = true ]; then
    echo -e "${YELLOW}[deploy] 开始本地构建...${NC}"
    cd "$(dirname "$0")/.."
    mvn clean package -pl iot-platform -am -Dmaven.test.skip=true
    LOCAL_JAR="iot-platform/target/iot-platform.jar"
    echo -e "${GREEN}[deploy] 构建完成: ${LOCAL_JAR}${NC}"
fi

if [ -z "$LOCAL_JAR" ]; then
    # 默认使用本地 target 下的 jar
    cd "$(dirname "$0")/.."
    LOCAL_JAR="iot-platform/target/iot-platform.jar"
fi

if [ ! -f "$LOCAL_JAR" ]; then
    echo -e "${RED}[deploy] 错误: 找不到 jar 文件: ${LOCAL_JAR}${NC}"
    exit 1
fi

JAR_NAME=$(basename "$LOCAL_JAR")
TIMESTAMP=$(date +%Y%m%d-%H%M%S)
BACKUP_NAME="${APP_NAME}-${TIMESTAMP}.jar"

echo "========================================"
echo "IoT Platform 部署"
echo "========================================"
echo "目标服务器: ${SERVER_USER}@${SERVER_HOST}"
echo "远程目录:   ${REMOTE_DIR}"
echo "本地 JAR:   ${LOCAL_JAR}"
echo "========================================"

# 步骤 1: 上传 JAR 到临时目录
echo -e "${YELLOW}[1/6] 上传 JAR 到服务器...${NC}"
ssh "${SERVER_USER}@${SERVER_HOST}" "mkdir -p ${REMOTE_DIR}/tmp ${REMOTE_DIR}/backup"
scp "${LOCAL_JAR}" "${SERVER_USER}@${SERVER_HOST}:${REMOTE_DIR}/tmp/${JAR_NAME}"
echo -e "${GREEN}[1/6] 上传完成${NC}"

# 步骤 2: 备份当前版本
echo -e "${YELLOW}[2/6] 备份当前版本...${NC}"
ssh "${SERVER_USER}@${SERVER_HOST}" "
    if [ -f ${REMOTE_DIR}/${APP_NAME}.jar ]; then
        cp -a ${REMOTE_DIR}/${APP_NAME}.jar ${REMOTE_DIR}/backup/${BACKUP_NAME}
        echo '  已备份: ${BACKUP_NAME}'
        ls -lh ${REMOTE_DIR}/backup/ | tail -5
    else
        echo '  无现有版本，跳过备份'
    fi
"
echo -e "${GREEN}[2/6] 备份完成${NC}"

# 步骤 3: 停止服务
echo -e "${YELLOW}[3/6] 停止服务...${NC}"
ssh "${SERVER_USER}@${SERVER_HOST}" "systemctl stop ${APP_NAME} || true"
sleep 2
echo -e "${GREEN}[3/6] 服务已停止${NC}"

# 步骤 4: 替换 JAR
echo -e "${YELLOW}[4/6] 替换 JAR...${NC}"
ssh "${SERVER_USER}@${SERVER_HOST}" "
    cp ${REMOTE_DIR}/tmp/${JAR_NAME} ${REMOTE_DIR}/${APP_NAME}.jar
    chmod 644 ${REMOTE_DIR}/${APP_NAME}.jar
    rm -f ${REMOTE_DIR}/tmp/${JAR_NAME}
    echo '  新 JAR:'
    ls -lh ${REMOTE_DIR}/${APP_NAME}.jar
"
echo -e "${GREEN}[4/6] 替换完成${NC}"

# 步骤 5: 启动服务
echo -e "${YELLOW}[5/6] 启动服务...${NC}"
ssh "${SERVER_USER}@${SERVER_HOST}" "systemctl start ${APP_NAME}"
sleep 3

echo -e "${GREEN}[5/6] 服务已启动${NC}"

# 步骤 6: 健康检查
echo -e "${YELLOW}[6/6] 健康检查...${NC}"
if ssh "${SERVER_USER}@${SERVER_HOST}" "bash ${REMOTE_DIR}/bin/health-check.sh localhost 8887 30"; then
    echo -e "${GREEN}[6/6] 健康检查通过，部署成功!${NC}"
    echo ""
    echo "========================================"
    echo "部署摘要"
    echo "========================================"
    echo "时间:     ${TIMESTAMP}"
    echo "JAR:      ${LOCAL_JAR}"
    echo "备份:     ${BACKUP_NAME}"
    echo "状态:     成功"
    echo ""
    echo "服务状态:"
    ssh "${SERVER_USER}@${SERVER_HOST}" "systemctl status ${APP_NAME} --no-pager"
    exit 0
else
    echo -e "${RED}[6/6] 健康检查失败! 执行回滚...${NC}"

    # 回滚
    echo -e "${YELLOW}[rollback] 停止服务...${NC}"
    ssh "${SERVER_USER}@${SERVER_HOST}" "systemctl stop ${APP_NAME} || true"

    if ssh "${SERVER_USER}@${SERVER_HOST}" "test -f ${REMOTE_DIR}/backup/${BACKUP_NAME}"; then
        echo -e "${YELLOW}[rollback] 恢复旧版本...${NC}"
        ssh "${SERVER_USER}@${SERVER_HOST}" "
            cp ${REMOTE_DIR}/backup/${BACKUP_NAME} ${REMOTE_DIR}/${APP_NAME}.jar
            systemctl start ${APP_NAME}
        "
        sleep 3

        if ssh "${SERVER_USER}@${SERVER_HOST}" "bash ${REMOTE_DIR}/bin/health-check.sh localhost 8887 30"; then
            echo -e "${GREEN}[rollback] 回滚成功，旧版本已恢复${NC}"
        else
            echo -e "${RED}[rollback] 回滚后健康检查仍失败，请手动排查!${NC}"
        fi
    else
        echo -e "${RED}[rollback] 无备份可回滚，请手动修复!${NC}"
    fi

    exit 1
fi