"""
Copyright (c) 2025-2025 Huawei Technologies Co., Ltd.

sysHAX is licensed under Mulan PSL v2.
You can use this software according to the terms and conditions of the Mulan PSL v2.
You may obtain a copy of Mulan PSL v2 at:
    http://license.coscl.org.cn/MulanPSL2
THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND, EITHER EXPRESS OR
IMPLIED, INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT, MERCHANTABILITY OR FIT FOR A PARTICULAR
PURPOSE.
See the Mulan PSL v2 for more details.
Created: 2025-05-23
Desc:sysHAX 主程序
"""

import sys
from collections.abc import AsyncGenerator
from contextlib import asynccontextmanager
from pathlib import Path
import uvicorn
from fastapi import FastAPI

from src import routes
from src.core.engine import Engine
from src.core.metrics import MetricsService
from src.core.monitor import SystemMonitor
from src.core.scheduler import Scheduler
from src.core.runner import Runner
from src.utils.logger import Logger
from src.utils.config import load_syshax_config

try:
    syshax_config = load_syshax_config()
except Exception as e:
    Logger.error(f"配置加载失败，sysHAX启动失败，失败原因: {e}")
    sys.exit(1)

@asynccontextmanager
async def lifespan(app: FastAPI) -> AsyncGenerator[None, None]:
    """管理应用生命周期：初始化监控、调度器、解码器和性能测试，并在关闭时清理资源"""
    Path("logs").mkdir(parents=True, exist_ok=True)

    app.state.config = syshax_config
    app.state.metrics_service = MetricsService()
    app.state.monitor = SystemMonitor(
        metrics_service=app.state.metrics_service,
        syshax_config=syshax_config
    )
    app.state.runner = Runner(
        metrics_service=app.state.metrics_service,
        syshax_config=syshax_config
    )
    app.state.scheduler = Scheduler(
        system_monitor=app.state.monitor,
        runner=app.state.runner,
        metrics_service=app.state.metrics_service,
        syshax_config=syshax_config
    )

    app.state.engine = Engine(
        scheduler=app.state.scheduler,
        metrics_service=app.state.metrics_service
    )
    app.state.engine.start()
    Logger.info("应用启动完成，API接口已就绪")
    try:
        yield
    finally:
        Logger.info("应用关闭，正在清理资源...")
        await app.state.engine.stop()

# 将 FastAPI 实例化并使用 lifespan 管理生命周期
app = FastAPI(lifespan=lifespan)

# 注册路由
app.include_router(routes.router)

def run() -> None:
    """主程序入口"""
    # 从配置读取主机和端口
    host = syshax_config.syshax_host
    port = syshax_config.syshax_port

    # 启动FastAPI应用
    uvicorn.run(app, host=host, port=port, timeout_graceful_shutdown=1)

if __name__ == "__main__":
    run()