#!/bin/bash
# 完整初始化数据库和管理员账号

set -e  # 遇到错误立即退出

SCRIPT_DIR="$(cd "$(dirname "$0")" && pwd)"
PROJECT_ROOT="$(cd "$SCRIPT_DIR/../../.." && pwd)"
cd "$PROJECT_ROOT/base_platform"

echo "=========================================="
echo "开始初始化 HelloTax AI 基础平台"
echo "=========================================="

# 第一步：初始化数据库表结构
echo ""
echo "第一步：初始化数据库表结构..."
source venv/bin/activate
alembic upgrade head

if [ $? -eq 0 ]; then
    echo "✓ 数据库表结构初始化完成"
else
    echo "✗ 数据库初始化失败"
    exit 1
fi

# 第二步：初始化平台管理员和权限
echo ""
echo "第二步：初始化平台管理员和权限..."
bash "$SCRIPT_DIR/init_platform_admin.sh"

if [ $? -ne 0 ]; then
    echo "✗ 管理员初始化失败"
    exit 1
fi

# 第三步：初始化模型供应商
echo ""
echo "第三步：初始化模型供应商..."
bash "$SCRIPT_DIR/init_model_providers.sh"

if [ $? -ne 0 ]; then
    echo "✗ 模型供应商初始化失败"
    exit 1
fi

# 第四步：初始化训练模型目录
echo ""
echo "第四步：初始化训练模型目录..."
bash "$SCRIPT_DIR/init_trained_models_dir.sh"

if [ $? -eq 0 ]; then
    echo ""
    echo "=========================================="
    echo "✅ HelloTax AI 基础平台初始化完成"
    echo "=========================================="
    echo ""
else
    echo "✗ 训练模型目录初始化失败"
    exit 1
fi