from __future__ import annotations

from typing import TYPE_CHECKING

if TYPE_CHECKING:
    from yuxi.channel.container import ChannelContainer

from yuxi.channel.container import ChannelContainerFactory, setup_channel

_container: ChannelContainer | None = None


async def init_channel(
    redis_url: str = "",
    agent_port=None,
    *,
    config_yaml_path: str = "channel_config.yaml",
    mq_workers: int = 4,
    mq_max_concurrent: int = 20,
    default_agent_config_id: int = 1,
    use_plugin_registry: bool = True,
) -> ChannelContainer:
    global _container

    plugin_registry_impl = None
    if use_plugin_registry:
        try:
            from yuxi.channel.infrastructure.plugin.registry_impl import PluginRegistryImpl

            plugin_registry_impl = PluginRegistryImpl()

            import yuxi.channel.channels  # noqa: F401

            from yuxi.channel.infrastructure.plugin.bundled.register_builtin import register_builtin_plugins

            await register_builtin_plugins(plugin_registry_impl.registry)
            await plugin_registry_impl.discover_and_register_all()
        except ImportError:
            pass

    _container = await ChannelContainerFactory.create(
        redis_url=redis_url,
        agent_port=agent_port,
        config_yaml_path=config_yaml_path,
        mq_workers=mq_workers,
        mq_max_concurrent=mq_max_concurrent,
        default_agent_config_id=default_agent_config_id,
        plugin_registry_impl=plugin_registry_impl,
    )
    return _container


async def shutdown_channel(container: ChannelContainer | None = None) -> None:
    global _container
    target = container or _container
    if target:
        await target.shutdown()
    _container = None


def get_container() -> ChannelContainer | None:
    return _container


def register_channel_middleware(app) -> None:
    if _container:
        setup_channel(app, _container)
        from yuxi.channel.interfaces.rest.router.registry import register_all_channel_routes

        registered = register_all_channel_routes(app, _container.adapters)
        import logging

        logging.getLogger(__name__).info("registered channel routes: %s", registered)