#!/bin/bash
# =============================================================================
# IT智能服务台 — RAGFlow 集成部署脚本
# 目标服务器：10.90.5.110
# 部署路径：/opt/wecom-it-desk
# =============================================================================

set -e

DEPLOY_DIR="/opt/wecom-it-desk"
BACKUP_DIR="/opt/wecom-it-desk-backup-$(date +%Y%m%d_%H%M%S)"

echo "=========================================="
echo "IT智能服务台 — RAGFlow 集成部署"
echo "时间: $(date)"
echo "=========================================="

# --------------------------------------------------------------------------
# 1. 备份当前版本
# --------------------------------------------------------------------------
echo ""
echo ">>> 步骤1: 备份当前版本..."
mkdir -p "$BACKUP_DIR"
cp -r "$DEPLOY_DIR/frontend-admin/dist" "$BACKUP_DIR/frontend-admin-dist" 2>/dev/null || true
cp -r "$DEPLOY_DIR/backend" "$BACKUP_DIR/backend" 2>/dev/null || true
echo "备份完成: $BACKUP_DIR"

# --------------------------------------------------------------------------
# 2. 解压管理后台前端
# --------------------------------------------------------------------------
echo ""
echo ">>> 步骤2: 解压管理后台前端..."
cd "$DEPLOY_DIR"
rm -rf frontend-admin/dist
tar -xf /tmp/deploy-admin.tar -C frontend-admin/
echo "管理后台前端已更新"

# --------------------------------------------------------------------------
# 3. 更新后端代码
# --------------------------------------------------------------------------
echo ""
echo ">>> 步骤3: 更新后端代码..."
cd "$DEPLOY_DIR"

# 备份后端 .env（生产配置，不能覆盖）
cp backend/.env /tmp/backend-env-backup 2>/dev/null || true

# 解压后端代码（覆盖旧文件）
tar -xf /tmp/deploy-backend.tar -C ./

# 恢复后端 .env
cp /tmp/backend-env-backup backend/.env 2>/dev/null || true
echo "后端代码已更新"

# --------------------------------------------------------------------------
# 4. 重建后端 Docker 镜像
# --------------------------------------------------------------------------
echo ""
echo ">>> 步骤4: 重建后端 Docker 镜像..."
cd "$DEPLOY_DIR"
docker compose build --no-cache backend
echo "后端镜像重建完成"

# --------------------------------------------------------------------------
# 5. 重启所有容器
# --------------------------------------------------------------------------
echo ""
echo ">>> 步骤5: 重启所有容器..."
docker compose down
docker compose up -d
echo "容器重启完成"

# --------------------------------------------------------------------------
# 6. 等待服务就绪
# --------------------------------------------------------------------------
echo ""
echo ">>> 步骤6: 等待服务就绪..."
sleep 10

# 检查容器状态
echo ""
echo "容器状态:"
docker compose ps

# 检查后端健康
echo ""
echo "后端健康检查:"
curl -s http://localhost:8000/health || echo "后端未就绪，请稍后重试"

echo ""
echo "=========================================="
echo "部署完成！"
echo "=========================================="
echo ""
echo "下一步：配置 RAGFlow API Key"
echo "  1. 登录管理后台: https://itsupport.servyou.com.cn/itadmin/"
echo "  2. 进入 集成管理 → RAGFlow"
echo "  3. 填入 API 地址: http://10.80.0.85:9380"
echo "  4. 填入 API Key: sk-654e************f7b91ea2b"
echo "  5. 点击 测试连接"
echo ""
echo "如需回滚，执行:"
echo "  cp -r $BACKUP_DIR/frontend-admin-dist $DEPLOY_DIR/frontend-admin/dist"
echo "  cp -r $BACKUP_DIR/backend $DEPLOY_DIR/backend"
echo "  docker compose restart"