"""Top-level entry point for ComfyUI-ImmacStyleMixer."""

import os
import server
from aiohttp import web
import nodes

from comfy_api.latest import ComfyExtension, io

from .src.immac_tools.api import register_routes
from .src.immac_tools.style_mix_node import PickMixNode, StyleMixNode
from .src.immac_tools.style_pick_node import PickStyleNode
from .src.immac_tools.style_weight_node import StyleWeightNode
from .src.immac_tools.style_blend_node import BlendStyleNode
from .src.immac_tools.save_mix_node import SaveMixNode
from .src.immac_tools.save_style_node import SaveStyleNode


class ImmacStyleMixerExtension(ComfyExtension):
    async def get_node_list(self) -> list[type[io.ComfyNode]]:
        return [
            PickMixNode,
            PickStyleNode,
            StyleWeightNode,
            BlendStyleNode,
            SaveMixNode,
            SaveStyleNode,
        ]


async def comfy_entrypoint() -> ImmacStyleMixerExtension:
    return ImmacStyleMixerExtension()

workspace_path = os.path.dirname(__file__)
dist_path = os.path.join(workspace_path, "dist", "immac_style_mixer")
dist_locales_path = os.path.join(workspace_path, "dist", "locales")

if os.path.exists(dist_path):
    server.PromptServer.instance.app.add_routes([
        web.static("/immac_style_mixer/", dist_path),
    ])
    if os.path.exists(dist_locales_path):
        server.PromptServer.instance.app.add_routes([
            web.static("/locales/", dist_locales_path),
        ])

    project_name = os.path.basename(workspace_path)
    try:
        from comfy_config import config_parser
        project_config = config_parser.extract_node_configuration(workspace_path)
        project_name = project_config.project.name
    except Exception as e:
        print(f"[ImmacStyleMixer] Could not load project config, using '{project_name}': {e}")

    nodes.EXTENSION_WEB_DIRS[project_name] = os.path.join(workspace_path, "dist")
else:
    print("[ImmacStyleMixer] Web dist directory not found — skipping static file registration.")

# Always register API routes (data endpoints work even before the UI is built)
register_routes(server.PromptServer.instance.app, workspace_path)