from __future__ import annotations

import asyncio
import os

import quart
import quart_cors

from ....core import app, entities as core_entities
from .groups import logs, system, settings, plugins, stats, user
from . import group


class HTTPController:

    ap: app.Application

    quart_app: quart.Quart

    def __init__(self, ap: app.Application) -> None:
        self.ap = ap
        self.quart_app = quart.Quart(__name__)
        quart_cors.cors(self.quart_app, allow_origin="*")

    async def initialize(self) -> None:
        await self.register_routes()

    async def run(self) -> None:
        if self.ap.system_cfg.data["http-api"]["enable"]:

            async def shutdown_trigger_placeholder():
                while True:
                    await asyncio.sleep(1)

            async def exception_handler(*args, **kwargs):
                try:
                    await self.quart_app.run_task(
                        *args, **kwargs
                    )
                except Exception as e:
                    self.ap.logger.error(f"启动 HTTP 服务失败： {e}")

            self.ap.task_mgr.create_task(
                exception_handler(
                    host=self.ap.system_cfg.data["http-api"]["host"],
                    port=self.ap.system_cfg.data["http-api"]["port"],
                    shutdown_trigger=shutdown_trigger_placeholder,
                ),
                name="http-api-quart",
                scopes=[core_entities.LifecycleControlScope.APPLICATION],
            )

            # await asyncio.sleep(5)

    async def register_routes(self) -> None:

        @self.quart_app.route("/healthz")
        async def healthz():
            return {"code": 0, "msg": "ok"}

        for g in group.preregistered_groups:
            ginst = g(self.ap, self.quart_app)
            await ginst.initialize()

        frontend_path = "web/dist"

        @self.quart_app.route("/")
        async def index():
            return await quart.send_from_directory(
                frontend_path,
                "index.html",
                mimetype="text/html"
            )

        @self.quart_app.route("/<path:path>")
        async def static_file(path: str):

            mimetype = None

            if path.endswith(".html"):
                mimetype = "text/html"
            elif path.endswith(".js"):
                mimetype = "application/javascript"
            elif path.endswith(".css"):
                mimetype = "text/css"
            elif path.endswith(".png"):
                mimetype = "image/png"
            elif path.endswith(".jpg"):
                mimetype = "image/jpeg"
            elif path.endswith(".jpeg"):
                mimetype = "image/jpeg"
            elif path.endswith(".gif"):
                mimetype = "image/gif"
            elif path.endswith(".svg"):
                mimetype = "image/svg+xml"
            elif path.endswith(".ico"):
                mimetype = "image/x-icon"
            elif path.endswith(".json"):
                mimetype = "application/json"
            elif path.endswith(".txt"):
                mimetype = "text/plain"

            return await quart.send_from_directory(
                frontend_path,
                path,
                mimetype=mimetype
            )