#!/usr/bin/env python3
"""
数据库初始化脚本
用于创建数据库表结构
"""

import asyncio
import sys
import os

# 添加项目根目录到Python路径
sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))

from app.database import init_db, engine
from app.config import settings
from sqlalchemy import text


async def create_database():
    """创建数据库（如果不存在）"""
    try:
        # 从DATABASE_URL中提取数据库名
        db_url = settings.database_url
        if "postgresql+asyncpg://" in db_url:
            # 解析数据库URL获取数据库名
            # 格式: postgresql+asyncpg://user:password@host:port/dbname
            import re
            match = re.match(r'.+/([^/]+)$', db_url)
            if match:
                db_name = match.group(1)
                # 创建不包含数据库名的连接URL
                base_url = db_url.rsplit('/', 1)[0]
                
                # 连接到postgres默认数据库创建新数据库
                from sqlalchemy.ext.asyncio import create_async_engine
                temp_engine = create_async_engine(f"{base_url}/postgres")
                
                async with temp_engine.begin() as conn:
                    # 检查数据库是否已存在
                    result = await conn.execute(
                        text("SELECT 1 FROM pg_database WHERE datname = :db_name"),
                        {"db_name": db_name}
                    )
                    exists = result.scalar()
                    
                    if not exists:
                        await conn.execute(text(f"CREATE DATABASE {db_name}"))
                        print(f"✅ 数据库 '{db_name}' 创建成功")
                    else:
                        print(f"✅ 数据库 '{db_name}' 已存在")
                
                await temp_engine.dispose()
        
    except Exception as e:
        print(f"❌ 创建数据库失败: {e}")
        return False
    
    return True


async def init_tables():
    """初始化数据库表结构"""
    try:
        print("🔄 开始初始化数据库表结构...")
        
        # 初始化所有表
        await init_db()
        
        print("✅ 数据库表结构初始化成功")
        
        # 验证表是否创建成功
        async with engine.begin() as conn:
            result = await conn.execute(text("""
                SELECT table_name 
                FROM information_schema.tables 
                WHERE table_schema = 'public' 
                ORDER BY table_name
            """))
            tables = [row[0] for row in result.fetchall()]
            
            print(f"📋 已创建的表: {', '.join(tables)}")
            
            # 检查必要的表
            required_tables = ['callback_failure_logs', 'callback_failure_data', 'external_api_logs']
            missing_tables = [table for table in required_tables if table not in tables]
            
            if missing_tables:
                print(f"❌ 缺少必要的表: {', '.join(missing_tables)}")
                return False
            else:
                print("✅ 所有必要的表都已创建")
            
            # 检查手机号索引
            result = await conn.execute(text("""
                SELECT indexname 
                FROM pg_indexes 
                WHERE tablename = 'callback_failure_data' 
                AND indexname = 'idx_phone_number'
            """))
            phone_index = result.fetchone()
            
            if phone_index:
                print("✅ 手机号索引 'idx_phone_number' 已创建")
            else:
                print("⚠️ 手机号索引 'idx_phone_number' 未找到")
        
        return True
        
    except Exception as e:
        print(f"❌ 初始化数据库表结构失败: {e}")
        return False


async def verify_connection():
    """验证数据库连接"""
    try:
        print("🔄 验证数据库连接...")
        
        async with engine.begin() as conn:
            result = await conn.execute(text("SELECT version()"))
            version = result.scalar()
            print(f"✅ 数据库连接成功")
            print(f"📊 PostgreSQL版本: {version}")
        
        return True
        
    except Exception as e:
        print(f"❌ 数据库连接失败: {e}")
        print("\n💡 请检查以下配置:")
        print(f"   - DATABASE_URL: {settings.database_url}")
        print("   - PostgreSQL服务是否运行")
        print("   - 用户名密码是否正确")
        print("   - 网络连接是否正常")
        return False


async def main():
    """主函数"""
    print("🚀 开始数据库初始化...")
    print(f"📋 配置信息:")
    print(f"   - 数据库URL: {settings.database_url}")
    print(f"   - Redis URL: {settings.redis_url}")
    print(f"   - 应用名称: {settings.app_name}")
    print(f"   - 外部API URL: {settings.external_api_url}")
    print()
    
    # 验证连接
    if not await verify_connection():
        sys.exit(1)
    
    print()
    
    # 创建数据库（如果需要）
    if not await create_database():
        sys.exit(1)
    
    print()
    
    # 初始化表结构
    if not await init_tables():
        sys.exit(1)
    
    print()
    print("🎉 数据库初始化完成！")
    print("\n📝 下一步:")
    print("   1. 配置 .env 文件中的数据库连接信息")
    print("   2. 确保 Redis 服务正在运行")
    print("   3. 运行应用: python main.py")
    print(f"   4. 访问API文档: http://localhost:8000/docs")
    print(f"   5. 接口地址: POST /ai-talk/callback/{{siteId}}/failure")


if __name__ == "__main__":
    try:
        asyncio.run(main())
    except KeyboardInterrupt:
        print("\n⚠️  用户中断操作")
        sys.exit(1)
    except Exception as e:
        print(f"\n❌ 初始化过程中发生错误: {e}")
        sys.exit(1)