"""FastAPI 应用入口"""
from fastapi import FastAPI
from fastapi.middleware.cors import CORSMiddleware
from fastapi.staticfiles import StaticFiles
import os

from database import init_db
from config import UPLOAD_DIR
import models.system_config  # 确保 system_configs 表被创建
import models.category  # 确保 categories 表被创建
import models.nav_category  # 确保 nav_categories 表被创建
import models.nav_link  # 确保 nav_links 表被创建
import models.project  # 确保 projects 表被创建
import models.shared_api  # 确保 shared_api 相关表被创建
import models.knowledge_base  # 确保 kb 相关表被创建
import models.attachment  # 确保 attachments 表被创建
from routers import auth, requirement, architecture, posts, search, ai_models, bookmarks, users, notifications, upload, admin, nav, projects, shared_api, knowledge_base, web_search, ai_format

# 确保上传目录存在
os.makedirs(UPLOAD_DIR, exist_ok=True)

app = FastAPI(
    title="极码 GeekCode",
    description="极码 GeekCode - 开发者社区、AI工具库、经验知识库",
    version="2.0.0",
)

# CORS中间件
app.add_middleware(
    CORSMiddleware,
    allow_origins=["http://localhost:5173", "http://127.0.0.1:5173"],
    allow_credentials=True,
    allow_methods=["*"],
    allow_headers=["*"],
)

# 静态文件（上传的图片等）
app.mount("/uploads", StaticFiles(directory=UPLOAD_DIR), name="uploads")

# 注册路由
app.include_router(auth.router, prefix="/api/auth", tags=["认证"])
app.include_router(requirement.router, prefix="/api/requirement", tags=["需求理解助手"])
app.include_router(architecture.router, prefix="/api/architecture", tags=["架构选型助手"])
app.include_router(posts.router, prefix="/api/posts", tags=["经验知识库"])
app.include_router(search.router, prefix="/api/search", tags=["搜索"])
app.include_router(ai_models.router)
app.include_router(ai_models.public_router)
app.include_router(bookmarks.router, prefix="/api/bookmarks", tags=["网站收藏"])
app.include_router(users.router, prefix="/api/users", tags=["用户"])
app.include_router(notifications.router, prefix="/api/notifications", tags=["消息通知"])
app.include_router(upload.router, prefix="/api/upload", tags=["文件上传"])
app.include_router(admin.router, prefix="/api/admin", tags=["后台管理"])
app.include_router(nav.router, prefix="/api/nav", tags=["导航站"])
app.include_router(projects.router, prefix="/api/projects", tags=["开源项目"])
app.include_router(shared_api.router, prefix="/api/api-hub", tags=["API Hub"])
app.include_router(knowledge_base.router, prefix="/api/kb", tags=["团队知识库"])
app.include_router(web_search.router, prefix="/api/web-search", tags=["联网搜索"])
app.include_router(ai_format.router)


@app.on_event("startup")
async def startup():
    """应用启动时初始化数据库"""
    init_db()
    _init_default_categories()
    _migrate_user_is_approved()
    _migrate_project_collect_count()
    _migrate_web_search_enabled()
    _migrate_web_search_count()

def _init_default_categories():
    """如果分类表为空，插入默认分类"""
    from database import SessionLocal
    from models.category import Category
    db = SessionLocal()
    try:
        if db.query(Category).count() == 0:
            defaults = ['前端', '后端', '部署', '踩坑', '最佳实践', '工具']
            for i, name in enumerate(defaults):
                db.add(Category(name=name, sort_order=i))
            db.commit()
    finally:
        db.close()


def _migrate_user_is_approved():
    """迁移：给 users 表添加 is_approved 字段（已有用户自动设为已审核）"""
    from database import SessionLocal
    from sqlalchemy import text
    db = SessionLocal()
    try:
        # 检查字段是否存在
        result = db.execute(text(
            "SELECT COUNT(*) FROM information_schema.COLUMNS "
            "WHERE TABLE_SCHEMA = DATABASE() AND TABLE_NAME = 'users' AND COLUMN_NAME = 'is_approved'"
        ))
        exists = result.scalar()
        if not exists:
            # 添加字段，先默认值1让已有用户自动通过审核
            db.execute(text("ALTER TABLE users ADD COLUMN is_approved TINYINT(1) NOT NULL DEFAULT 1"))
            # 再把默认值改为0，新用户需审核
            db.execute(text("ALTER TABLE users ALTER COLUMN is_approved SET DEFAULT 0"))
            db.commit()
    except Exception as e:
        db.rollback()
        print(f"[migrate] is_approved: {e}")
    finally:
        db.close()


@app.get("/api/health")
async def health_check():
    """健康检查"""
    return {"status": "ok", "message": "极码 GeekCode API 运行中"}


def _migrate_project_collect_count():
    """迁移：给 projects 表添加 collect_count 字段"""
    from database import SessionLocal
    from sqlalchemy import text
    db = SessionLocal()
    try:
        result = db.execute(text(
            "SELECT COUNT(*) FROM information_schema.COLUMNS "
            "WHERE TABLE_SCHEMA = DATABASE() AND TABLE_NAME = 'projects' AND COLUMN_NAME = 'collect_count'"
        ))
        if not result.scalar():
            db.execute(text("ALTER TABLE projects ADD COLUMN collect_count INT NOT NULL DEFAULT 0"))
            db.commit()
    except Exception as e:
        db.rollback()
        print(f"[migrate] project collect_count: {e}")
    finally:
        db.close()


def _migrate_web_search_enabled():
    """迁移：给 ai_model_configs 表添加 web_search_enabled 字段"""
    from database import SessionLocal
    from sqlalchemy import text
    db = SessionLocal()
    try:
        result = db.execute(text(
            "SELECT COUNT(*) FROM information_schema.COLUMNS "
            "WHERE TABLE_SCHEMA = DATABASE() AND TABLE_NAME = 'ai_model_configs' AND COLUMN_NAME = 'web_search_enabled'"
        ))
        if not result.scalar():
            db.execute(text("ALTER TABLE ai_model_configs ADD COLUMN web_search_enabled TINYINT(1) NOT NULL DEFAULT 0"))
            db.commit()
    except Exception as e:
        db.rollback()
        print(f"[migrate] web_search_enabled: {e}")
    finally:
        db.close()


def _migrate_web_search_count():
    """迁移：给 ai_model_configs 表添加 web_search_count 字段"""
    from database import SessionLocal
    from sqlalchemy import text
    db = SessionLocal()
    try:
        result = db.execute(text(
            "SELECT COUNT(*) FROM information_schema.COLUMNS "
            "WHERE TABLE_SCHEMA = DATABASE() AND TABLE_NAME = 'ai_model_configs' AND COLUMN_NAME = 'web_search_count'"
        ))
        if not result.scalar():
            db.execute(text("ALTER TABLE ai_model_configs ADD COLUMN web_search_count INT NOT NULL DEFAULT 5"))
            db.commit()
    except Exception as e:
        db.rollback()
        print(f"[migrate] web_search_count: {e}")
    finally:
        db.close()