import logging

from yuxi.channel.gateway.routes import webhook_registry
from yuxi.channel.gateway.webhook_security import WebhookGuardConfig

logger = logging.getLogger("yuxi.channel.generic_webhook.gateway")


class GenericWebhookGateway:
    def __init__(self, config_mgr, deduplicator, mapping_engine):
        self._config_mgr = config_mgr
        self._deduplicator = deduplicator
        self._mapping_engine = mapping_engine
        self._running = False

    async def start(self, ctx) -> object:
        from yuxi.channel.extensions.generic_webhook.webhook import set_runtime_state

        set_runtime_state(
            config_mgr=self._config_mgr,
            deduplicator=self._deduplicator,
            mapping_engine=self._mapping_engine,
        )

        endpoints = self._config_mgr.list_endpoint_ids()
        self._register_webhook_handler()

        if not endpoints:
            logger.info("Gateway 已启动（无预注册端点，等待动态添加）")
        else:
            logger.info(
                "Gateway 已启动，已注册 %d 个端点: %s",
                len(endpoints),
                ", ".join(endpoints),
            )

        self._running = True
        return self

    def _register_webhook_handler(self):
        from yuxi.channel.extensions.generic_webhook.webhook import generic_webhook_handler

        guard_config = WebhookGuardConfig(
            channel_type="generic-webhook",
            allowed_methods=("POST",),
            allowed_content_types=("application/json",),
            body_size_limit=1_048_576,
        )
        webhook_registry.register(
            "generic-webhook",
            generic_webhook_handler,
            guard_config=guard_config,
        )

    async def stop(self, ctx=None):
        webhook_registry.remove("generic-webhook")
        self._running = False
        logger.info("Gateway 已停止")

    @property
    def is_running(self) -> bool:
        return self._running