"""
FastAPI application factory for Sanguo Quant API
"""
from fastapi import FastAPI
from .routes import router, set_orchestrator, set_auth_config
from .routes_paper import router as paper_router, set_db_path
from .auth import set_jwt_config
from .ws import manager
from sanguo_orchestrator.runner import Orchestrator


def create_app(db_path: str, file_dir=None, auth_config=None, max_workers: int = 2) -> FastAPI:
    """Create FastAPI application with orchestrator and optional authentication"""
    app = FastAPI(title="Sanguo Quant API")

    # Initialize orchestrator
    orch = Orchestrator(db_path=db_path, file_dir=file_dir, max_workers=max_workers)

    # Set up WebSocket stage callback
    async def _on_stage(task_id, stage):
        """Broadcast stage updates to WebSocket subscribers"""
        await manager.broadcast(task_id, {"task_id": task_id, "stage": stage})

    orch.set_on_stage(_on_stage)
    set_orchestrator(orch)

    # Configure authentication if provided
    if auth_config:
        set_auth_config(auth_config)
        set_jwt_config(
            auth_config.get("jwt_secret", "x"),
            auth_config.get("expire_minutes", 60)
        )

    # Include routes
    app.include_router(router, prefix="/api/v1")
    app.include_router(paper_router, prefix="/api/v1")
    set_db_path(db_path)

    @app.on_event("startup")
    def _register_live_step():
        """容器启动：注册每日 20:30 全局实走 step job（C-S3，遍历 live accounts）。"""
        try:
            from sanguo_trader.scheduler import register_live_step_job
            register_live_step_job(db_path)
        except Exception as e:  # noqa: BLE001  本机无 apscheduler 时跳过
            import logging
            logging.getLogger(__name__).warning("register_live_step_job 失败: %s", e)

    return app