"""Fake `fastapi` / `open_webui.*` modules, installed into `sys.modules`."""

import pathlib
import sys
import time
import types
from dataclasses import dataclass, field
from types import SimpleNamespace
from typing import Any


@dataclass
class FakeEnv:
    """Stand-in for the Open WebUI configuration the tool reads."""

    comfyui_url: str
    workflow: str
    api_key: str = ""
    nodes: list[dict[str, Any]] = field(default_factory=list)
    image_size: str = "1024x1024"
    image_steps: int = 25
    output_dir: pathlib.Path = pathlib.Path("test_out")
    stored: list[pathlib.Path] = field(default_factory=list)


def _module(name: str, **attrs: Any) -> types.ModuleType:
    mod = types.ModuleType(name)
    for key, value in attrs.items():
        setattr(mod, key, value)
    return mod


def install(env: FakeEnv) -> None:
    """Register fake modules. Call before importing `comfier`."""

    class Request: ...

    class Config:
        @staticmethod
        async def get(key: str, default: Any = None) -> Any:
            return default

    class UsersTable:
        async def get_user_by_id(self, id: str, db: Any = None) -> Any:
            return SimpleNamespace(id=id, name="dev")

    Users = UsersTable()

    async def get_image_config() -> Any:
        return SimpleNamespace(
            COMFYUI_BASE_URL=env.comfyui_url,
            COMFYUI_API_KEY=env.api_key,
            COMFYUI_WORKFLOW=env.workflow,
            COMFYUI_WORKFLOW_NODES=env.nodes,
            IMAGE_SIZE=env.image_size,
            IMAGE_STEPS=env.image_steps,
        )

    async def upload_image(
        request: Any,
        image_data: bytes,
        content_type: str,
        metadata: dict[str, Any],
        user: Any,
    ) -> Any:
        env.output_dir.mkdir(parents=True, exist_ok=True)
        ext = ".png" if "png" in content_type else ".jpg"
        file_id = f"seed_{metadata.get('seed', 'unknown')}_{time.monotonic_ns()}"
        out_file = env.output_dir / f"{file_id}{ext}"
        out_file.write_bytes(image_data)
        env.stored.append(out_file)
        print(f"  [Storage] Saved {len(image_data)} bytes -> {out_file}")
        # Open WebUI 0.12 returns (file_item, {id, url, name, content_type})
        return SimpleNamespace(id=file_id), {
            "id": file_id,
            "url": f"/api/v1/files/{file_id}/content",
        }

    modules = {
        "fastapi": _module("fastapi", Request=Request),
        "open_webui": _module("open_webui"),
        "open_webui.models": _module("open_webui.models"),
        "open_webui.routers": _module("open_webui.routers"),
        "open_webui.models.config": _module("open_webui.models.config", Config=Config),
        "open_webui.models.users": _module("open_webui.models.users", Users=Users),
        "open_webui.routers.images": _module(
            "open_webui.routers.images",
            get_image_config=get_image_config,
            upload_image=upload_image,
        ),
    }
    sys.modules.pop("comfier", None)
    sys.modules.update(modules)


__all__ = ["FakeEnv", "install"]