import os
from typing import Annotated

import typer
import uvicorn
from alembic.config import Config
from fastapi import FastAPI

from alembic import command
from app.common.enums import EnvironmentEnum
from app.config.setting import settings
from app.core.logger import logger
from app.utils.banner import worship

fastapiadmin_cli = typer.Typer()
alembic_cfg = Config("alembic.ini")


def create_app() -> FastAPI:
 """
 创建 FastAPI 应用实例并完成日志、中间件、路由与静态资源注册。

 返回:
 - FastAPI: 已配置生命周期的应用对象。
 """
 from app.init_app import lifespan, register_exceptions, register_files, register_middlewares, register_routers, reset_api_docs
 # 创建FastAPI应用
 app = FastAPI(**settings.FASTAPI_CONFIG, lifespan=lifespan)
 # 注册异常处理器
 register_exceptions(app)
 # 注册中间件
 register_middlewares(app)
 # 注册路由
 register_routers(app)
 # 注册静态文件
 register_files(app)
 # 重设API文档
 reset_api_docs(app)
 return app


# typer.Option是非必填;typer.Argument是必填
@fastapiadmin_cli.command(
 name="run",
 help="启动 FastapiAdmin 服务, 运行 uv run main.py run --env=dev 不加参数默认 dev 环境",
)
def run(
 env: Annotated[
 EnvironmentEnum, typer.Option("--env", help="运行环境 (dev, prod)")
 ] = EnvironmentEnum.DEV,
) -> None:
 """
 按指定环境加载配置并启动 Uvicorn(开发环境开启 reload)。

 参数:
 - env (EnvironmentEnum): 运行环境,对应 `--env`。

 返回:
 - None
 """

 # 设置环境变量(必须在 import settings 之前,确保加载正确环境)
 os.environ["ENVIRONMENT"] = env.value

 typer.secho(
 message="FastapiAdmin 服务启动",
 fg=typer.colors.GREEN,
 )
 logger.info(worship(env.value))

 # 启动uvicorn服务
 uvicorn.run(
 app="main:create_app",
 host=settings.SERVER_HOST,
 port=settings.SERVER_PORT,
 reload=env.value == EnvironmentEnum.DEV.value,
 factory=True,
 log_config=None,
 )


@fastapiadmin_cli.command(
 name="revision",
 help="生成新的 Alembic 迁移脚本, 运行 python main.py revision --env=dev",
)
def revision(
 env: Annotated[
 EnvironmentEnum, typer.Option("--env", help="运行环境 (dev, prod)")
 ] = EnvironmentEnum.DEV,
) -> None:
 """
 使用 Alembic 自动生成迁移脚本(autogenerate)。

 参数:
 - env (EnvironmentEnum): 运行环境,用于加载对应数据库模型元数据。

 返回:
 - None
 """
 os.environ["ENVIRONMENT"] = env.value
 from app.config.setting import get_settings

 get_settings.cache_clear()
 command.revision(alembic_cfg, autogenerate=True, message="迁移脚本")
 typer.echo("迁移脚本已生成")


@fastapiadmin_cli.command(
 name="upgrade",
 help="应用最新的 Alembic 迁移, 运行 python main.py upgrade --env=dev",
)
def upgrade(
 env: Annotated[
 EnvironmentEnum, typer.Option("--env", help="运行环境 (dev, prod)")
 ] = EnvironmentEnum.DEV,
) -> None:
 """
 将数据库升级到 Alembic 最新版本(head)。

 参数:
 - env (EnvironmentEnum): 运行环境。

 返回:
 - None
 """
 os.environ["ENVIRONMENT"] = env.value
 from app.config.setting import get_settings

 get_settings.cache_clear()
 command.upgrade(alembic_cfg, "head")
 typer.echo("所有迁移已应用。")


if __name__ == "__main__":
 fastapiadmin_cli()