#!/bin/bash
# =============================================================================
# kgrag 迁移统一配置文件
# 所有脚本从这里读取配置，只需修改此文件
# =============================================================================

# ───────────────────────────────────────────────
# 原数据机（源服务器）
# ───────────────────────────────────────────────
SRC_HOST="192.168.0.164"
SRC_USER="root"
SRC_SSH_PORT=22
SRC_APP_DIR="/home/workspace/kgrag"

# 源服务器各服务端口
SRC_KGRAG_PORT=9085
SRC_NEO4J_PORT=7687
SRC_MINERU_PORT=9988
SRC_EMBED_PORT=9700
SRC_LLM_PORT=9800

# ───────────────────────────────────────────────
# 迁移目标机（新服务器）
# ───────────────────────────────────────────────
DST_HOST="192.168.1.108"
KGRAG_CONTAINER="kgrag"
CONTAINER_APP_DIR="/app"

# kgrag 代码目录（容器挂载的宿主机路径，对应容器内 /app）
# 示例：/home/workspace/kgrag
DST_CODE_DIR="/home/workspace/kgrag"

# 文件目录（原始文档+高亮PDF，对应容器内 /app/files 的挂载路径）
# 若与代码目录相同（即没有单独挂载 /app/files），填与 DST_CODE_DIR/files 相同
# 示例：/home/zdht/wx/data/an-webui/uploads
DST_FILES_DIR="/home/workspace/kgrag/files"

# 目标服务器各服务端口
DST_KGRAG_PORT=9085
DST_NEO4J_PORT=7687
DST_MINERU_PORT=9988
DST_EMBED_PORT=9700
DST_LLM_PORT=9800

# ───────────────────────────────────────────────
# Neo4j 连接（目标机）
# ───────────────────────────────────────────────
NEO4J_URI="bolt://localhost:${DST_NEO4J_PORT}"
NEO4J_USER="neo4j"
NEO4J_PASSWORD="zdht123@"

# ───────────────────────────────────────────────
# 按舰迁移配置
# 留空则全量迁移，多个用空格分隔
# 已知舰船号：101  122  163
# ───────────────────────────────────────────────
TARGET_SHIPS="163"       # 01_copy_files_by_ship.sh 使用（拉哪些舰的文件）
IMPORT_SHIP="163舰"      # 02_load_neo4j_by_ship.sh 使用（导入哪艘舰的快照）

# ───────────────────────────────────────────────
# rsync 额外选项（如限速：--bwlimit=50000）
# ───────────────────────────────────────────────
RSYNC_EXTRA=""

# =============================================================================
# 以下为派生变量，无需修改
# =============================================================================
SRC_SERVER="${SRC_USER}@${SRC_HOST}"
OLD_BASE_URL="http://${SRC_HOST}:${SRC_KGRAG_PORT}"
NEW_BASE_URL="http://${DST_HOST}:${DST_KGRAG_PORT}"
OLD_NEO4J_URI="bolt://${SRC_HOST}:${SRC_NEO4J_PORT}"
OLD_MINERU_URL="http://${SRC_HOST}:${SRC_MINERU_PORT}"
OLD_EMBED_URL="http://${SRC_HOST}:${SRC_EMBED_PORT}"
OLD_LLM_URL="http://${SRC_HOST}:${SRC_LLM_PORT}"
NEW_MINERU_URL="http://${DST_HOST}:${DST_MINERU_PORT}"
NEW_EMBED_URL="http://${DST_HOST}:${DST_EMBED_PORT}"
NEW_LLM_URL="http://${DST_HOST}:${DST_LLM_PORT}"