from fastapi import FastAPI
from fastapi.staticfiles import StaticFiles
from fastapi.middleware.cors import CORSMiddleware
from contextlib import asynccontextmanager
import os

from .core.database_loader import init_database, load_database_pool, close_database_pool
from app.api import cards as cards_router
from app.api import images as images_router
from app.api import labelme as labelme_router
from app.api import formate_xy as formate_xy_router
from app.api import config_proxy as config_proxy_router
from app.api import rating_report as rating_report_router
from app.api import auto_import as auto_import_router

from .core.config import settings
from .core.logger import setup_logging, get_logger

setup_logging()
logger = get_logger(__name__)

settings.set_config()


@asynccontextmanager
async def lifespan(main_app: FastAPI):
    print("--- 应用启动 ---")

    init_database()
    load_database_pool()
    yield
    print("--- 应用关闭 ---")
    close_database_pool()


app = FastAPI(title="卡片分数数据存储服务", lifespan=lifespan)

app.add_middleware(
    CORSMiddleware,
    allow_origins=["*"],
    allow_credentials=True,
    allow_methods=["*"],
    allow_headers=["*"]
)

app.include_router(cards_router.router, prefix=f"{settings.API_PREFIX}/cards", tags=["Cards"])
app.include_router(images_router.router, prefix=f"{settings.API_PREFIX}/images", tags=["Images"])
app.include_router(labelme_router.router, prefix=f"{settings.API_PREFIX}/labelme", tags=["Labelme"])
app.include_router(formate_xy_router.router, prefix=f"{settings.API_PREFIX}/formate_xy", tags=["Formate"])
app.include_router(auto_import_router.router, prefix=f"{settings.API_PREFIX}/import", tags=["AutoImport"])
app.include_router(config_proxy_router.router, prefix=f"{settings.API_PREFIX}/config", tags=["ConfigProxy"])
app.include_router(rating_report_router.router, prefix=f"{settings.API_PREFIX}/cards/rating", tags=["Rating"])