ForcePilot/backend/package/yuxi/channel/channels/web/adapter.py

from __future__ import annotations

import logging

from yuxi.channel.channels.web.outbound import WEB_CAPABILITIES, WebOutbound
from yuxi.channel.channels.web.translator import WebTranslator
from yuxi.channel.domain.model.message.dispatch_result import SendResult
from yuxi.channel.domain.model.message.unified_message import UnifiedMessage
from yuxi.channel.domain.model.shared.channel_capabilities import ChannelCapabilities
from yuxi.channel.domain.model.shared.channel_type import ChannelType
from yuxi.channel.domain.port.channel_route_contributor_port import ChannelRouteContributorPort
from yuxi.channel.domain.port.sse_push_port import SsePushPort
from yuxi.channel.domain.port.ws_connection_port import WsConnectionPort

logger = logging.getLogger(__name__)


class _WebRouteContributor:
    @property
    def router(self) -> object:
        from yuxi.channel.channels.web.routes import router

        return router


class WebAdapter:
    def __init__(self, *, sse_push: SsePushPort | None = None) -> None:
        self._outbound = WebOutbound(sse_push=sse_push)
        self._opened = False

    @property
    def capabilities(self) -> ChannelCapabilities:
        return WEB_CAPABILITIES

    @property
    def channel_type(self) -> str:
        return ChannelType.WEB.value

    @property
    def ws_connection(self) -> WsConnectionPort | None:
        return None

    @property
    def route_contributor(self) -> ChannelRouteContributorPort | None:
        return _WebRouteContributor()

    @classmethod
    def get_default_config(cls) -> dict:
        return {
            "max_connections": 1000,
        }

    async def open(self) -> None:
        self._opened = True

    async def close(self) -> None:
        self._opened = False

    async def receive_message(self, raw: dict) -> UnifiedMessage:
        return WebTranslator.translate_message(raw)

    async def send_message(self, session_id: str, content: str, *, channel_type: str, metadata: dict) -> SendResult:
        try:
            ok = await self._outbound.send_text(session_id, content, metadata=metadata)
            return SendResult(success=ok, error="" if ok else "sse_push_failed")
        except Exception as exc:
            logger.error("SSE push failed for session %s: %s", session_id, exc)
            return SendResult(success=False, error=str(exc))

    async def send_typing(self, session_id: str) -> None:
        await self._outbound.send_typing(session_id)

    async def send_media(self, session_id: str, *, url: str, media_type: str, metadata: dict) -> bool:
        return await self._outbound.send_media(session_id, url=url, media_type=media_type, metadata=metadata)

    async def is_healthy(self) -> bool:
        return self._opened