#!/usr/bin/env bash
set -euo pipefail

# 天普零碳园区智慧能源管理平台 - 快速启动脚本

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

log_info()  { echo -e "${GREEN}[INFO]${NC} $1"; }
log_warn()  { echo -e "${YELLOW}[WARN]${NC} $1"; }
log_error() { echo -e "${RED}[ERROR]${NC} $1"; }

# 检查前置依赖
check_prerequisites() {
    log_info "检查前置依赖..."

    if ! command -v docker &> /dev/null; then
        log_error "未找到 Docker，请先安装 Docker: https://docs.docker.com/get-docker/"
        exit 1
    fi

    if ! docker compose version &> /dev/null && ! docker-compose version &> /dev/null; then
        log_error "未找到 Docker Compose，请先安装 Docker Compose"
        exit 1
    fi

    log_info "前置依赖检查通过"
}

# 确定 docker compose 命令
get_compose_cmd() {
    if docker compose version &> /dev/null; then
        echo "docker compose"
    else
        echo "docker-compose"
    fi
}

# 初始化环境变量
init_env() {
    if [ ! -f .env ]; then
        log_warn ".env 文件不存在，从模板创建..."
        if [ -f .env.example ]; then
            cp .env.example .env
            log_warn "已创建 .env 文件，请根据实际情况修改配置"
            log_warn "当前使用默认配置启动，生产环境请务必修改密码和密钥"
        else
            log_error "未找到 .env.example 模板文件"
            exit 1
        fi
    else
        log_info ".env 文件已存在"
    fi
}

# 启动服务
start_services() {
    local compose_cmd
    compose_cmd=$(get_compose_cmd)

    log_info "启动服务..."
    $compose_cmd up -d

    log_info "等待服务就绪..."

    # 等待数据库就绪
    local retries=30
    while [ $retries -gt 0 ]; do
        if docker exec tianpu_db pg_isready -U tianpu -d tianpu_ems &> /dev/null; then
            log_info "数据库已就绪"
            break
        fi
        retries=$((retries - 1))
        sleep 2
    done

    if [ $retries -eq 0 ]; then
        log_error "数据库启动超时"
        exit 1
    fi

    # 等待后端就绪
    retries=30
    while [ $retries -gt 0 ]; do
        if docker exec tianpu_backend curl -sf http://localhost:8000/health &> /dev/null; then
            log_info "后端服务已就绪"
            break
        fi
        retries=$((retries - 1))
        sleep 2
    done

    if [ $retries -eq 0 ]; then
        log_error "后端服务启动超时"
        exit 1
    fi
}

# 初始化数据
init_data() {
    log_info "初始化数据库..."
    docker exec tianpu_backend python scripts/init_db.py || {
        log_warn "数据库初始化跳过（可能已初始化）"
    }

    log_info "写入种子数据..."
    docker exec tianpu_backend python scripts/seed_data.py || {
        log_warn "种子数据写入跳过（可能已存在）"
    }
}

# 打印访问信息
print_info() {
    echo ""
    echo "============================================="
    echo " 天普零碳园区智慧能源管理平台 启动完成"
    echo "============================================="
    echo ""
    echo " 前端页面:     http://localhost:3000"
    echo " 后端 API:     http://localhost:8000"
    echo " API 文档:     http://localhost:8000/docs"
    echo ""
    echo " 默认账号:     admin"
    echo " 默认密码:     admin123"
    echo ""
    echo " 请在首次登录后修改默认密码"
    echo "============================================="
    echo ""
}

# 主流程
main() {
    log_info "天普零碳园区智慧能源管理平台 - 快速启动"
    echo ""

    check_prerequisites
    init_env
    start_services
    init_data
    print_info
}

main "$@"