from __future__ import annotations

from yuxi.channel.protocols import SessionResolution


class GoogleChatSessionAdapter:
    def resolve_session(self, msg) -> SessionResolution:
        from yuxi.channel.routing.models import PeerKind

        if hasattr(msg, "sender") and hasattr(msg.sender, "kind"):
            if msg.sender.kind == PeerKind.DIRECT:
                space_id = _extract_space_id(msg)
                return SessionResolution(
                    kind="direct",
                    conversation_id=space_id or msg.sender.id,
                )

        space_id = _extract_space_id(msg)
        if space_id:
            return SessionResolution(kind="group", conversation_id=space_id)

        if msg.group and msg.group.id:
            return SessionResolution(kind="group", conversation_id=msg.group.id)

        return SessionResolution(kind="group", conversation_id="unknown")

    @staticmethod
    def build_dm_session_key(agent_id: str, account_id: str, space_id: str) -> str:
        return f"agent:{agent_id}:googlechat:direct:{space_id}"

    @staticmethod
    def build_group_session_key(agent_id: str, account_id: str, space_id: str) -> str:
        return f"agent:{agent_id}:googlechat:group:{space_id}"


def _extract_space_id(msg) -> str | None:
    metadata = getattr(msg, "metadata", {}) or {}
    space_name = metadata.get("space_name", "")
    if space_name:
        return space_name

    raw = getattr(msg, "raw_payload", {}) or {}
    event = raw.get("event", raw)
    space = event.get("space", {})
    if space:
        return space.get("name")

    return None