#!/bin/bash

# ============================================
# 灵越智报 - 前端部署到 ayt 服务器 (139.196.72.192)
# 本地构建 → 上传 dist → 配置 Nginx
# ============================================

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

# ========== 配置区域 ==========
SERVER_USER="root"
SERVER_HOST="139.196.72.192"
SERVER_PORT="28529"

# 路径配置
LOCAL_PROJECT_DIR="/home/hws/workspace/GitLab/ay/lingyue-zhibao"
LOCAL_FRONTEND_DIR="${LOCAL_PROJECT_DIR}/frontend/vue-demo"
LOCAL_DIST_DIR="${LOCAL_FRONTEND_DIR}/dist"

REMOTE_FRONTEND_DIR="/data/web/lingyuezhibao"
REMOTE_NGINX_CONF="/etc/nginx/conf.d/lingyue-frontend-ayt.conf"

# 后端 API 地址（前端请求代理目标）
BACKEND_API="http://127.0.0.1:8001"  # ayt 服务后端通常在 8080 (gateway)
# 前端监听端口
FRONTEND_PORT=28082
# ==============================

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}"
        exit 1
    fi
}

# 安装依赖
install_deps() {
    log_title "安装前端依赖"
    cd ${LOCAL_FRONTEND_DIR}
    if [ ! -d "node_modules" ]; then
        log_info "安装 npm 依赖..."
        yarn install || npm install
    else
        log_info "依赖已存在，跳过安装"
    fi
}

# 本地构建
build_local() {
    log_title "本地构建前端"
    cd ${LOCAL_FRONTEND_DIR}
    rm -rf dist
    log_info "执行 yarn build..."
    yarn build || npm run build
    if [ ! -d "${LOCAL_DIST_DIR}" ]; then
        log_error "构建失败: dist 目录不存在"
        exit 1
    fi
}

# 上传 dist
upload_dist() {
    log_title "上传前端文件到服务器"
    ssh_cmd "mkdir -p ${REMOTE_FRONTEND_DIR}"
    
    # 备份旧版本
    log_info "备份旧版本..."
    ssh_cmd "cd ${REMOTE_FRONTEND_DIR} && [ -d assets ] && tar -czf ../lingyuezhibao_bak_\$(date +%Y%m%d%H%M%S).tar.gz . || true"

    log_info "正在上传 dist 目录..."
    scp -P ${SERVER_PORT} -r "${LOCAL_DIST_DIR}/"* "${SERVER_USER}@${SERVER_HOST}:${REMOTE_FRONTEND_DIR}/"

    log_info "上传完成"
}

# 配置 Nginx
setup_nginx() {
    log_title "配置 Nginx"
    
    local nginx_conf="server {
    listen ${FRONTEND_PORT};
    server_name _;

    root ${REMOTE_FRONTEND_DIR};
    index index.html;

    # Gzip 压缩
    gzip on;
    gzip_types text/plain text/css application/json application/javascript text/xml application/xml;

    # API 代理到后端
    location /api {
        proxy_pass ${BACKEND_API};
        proxy_set_header Host \$host;
        proxy_set_header X-Real-IP \$remote_addr;
        proxy_set_header X-Forwarded-For \$proxy_add_x_forwarded_for;
        proxy_set_header X-Forwarded-Proto \$scheme;

        # WebSocket 支持
        proxy_http_version 1.1;
        proxy_set_header Upgrade \$http_upgrade;
        proxy_set_header Connection \"upgrade\";

        # 超时设置
        proxy_connect_timeout 60s;
        proxy_send_timeout 60s;
        proxy_read_timeout 60s;
    }

    # SPA 路由支持
    location / {
        try_files \$uri \$uri/ /index.html;
    }

    # 文件上传大小限制
    client_max_body_size 100M;
}"

    echo "${nginx_conf}" | ssh_cmd "cat > ${REMOTE_NGINX_CONF}"
    
    if ssh_cmd "nginx -t" 2>&1 | grep -q "successful"; then
        log_info "Nginx 配置测试通过"
        ssh_cmd "systemctl reload nginx"
        log_info "Nginx 已重载"
    else
        log_error "Nginx 配置有误"
        ssh_cmd "nginx -t"
    fi
}

# 完整部署
full_deploy() {
    check_connection
    # install_deps # 假设已经安装
    build_local
    upload_dist
    setup_nginx
    log_title "前端部署完成 🎉"
}

full_deploy