# SimpleSyrup - workflow-focused ComfyUI extensions for image generation
# Copyright (C) 2026  Artificial Sweetener and contributors
# SPDX-License-Identifier: AGPL-3.0-or-later

"""SimpleSyrup ComfyUI extension entry point."""

from __future__ import annotations

import sys

from . import simple_syrup as _simple_syrup_package

sys.modules.setdefault("simple_syrup", _simple_syrup_package)

from .simple_syrup.runtime.comfy_safetensors_dtypes import (  # noqa: E402
    register_comfy_safetensors_dtypes,
)
from .simple_syrup.runtime.external_llm_routes import (  # noqa: E402
    register_external_llm_routes,
)
from .simple_syrup.runtime.mask_batch_preview_routes import (  # noqa: E402
    register_mask_batch_preview_routes,
)
from .simple_syrup.runtime.quant_cache_routes import (  # noqa: E402
    register_quant_cache_routes,
)
from .simple_syrup.runtime.settings_routes import register_settings_routes  # noqa: E402

WEB_DIRECTORY = "./web/dist"


async def comfy_entrypoint() -> object:
    """Return Comfy v3 extension nodes without importing optional integrations."""

    from comfy_api.latest import ComfyExtension

    class SimpleSyrupExtension(ComfyExtension):
        """Expose SimpleSyrup's Comfy v3 nodes."""

        async def get_node_list(self) -> list[type[object]]:
            """Return v3 nodes after Comfy asks for them."""

            from .simple_syrup.nodes_v3 import get_nodes

            return get_nodes()

    return SimpleSyrupExtension()


register_settings_routes()
register_comfy_safetensors_dtypes()
register_quant_cache_routes()
register_external_llm_routes()
register_mask_batch_preview_routes()

__all__ = [
    "WEB_DIRECTORY",
    "comfy_entrypoint",
]