ForcePilot/backend/package/yuxi/channel/channels/hooks/translator.py

from __future__ import annotations

from yuxi.channel.channels.hooks.config import HookMapping
from yuxi.channel.domain.model.message.peer import Peer
from yuxi.channel.domain.model.message.unified_message import UnifiedMessage
from yuxi.channel.domain.model.shared.channel_type import ChannelType


class HooksTranslator:
    @staticmethod
    def translate(raw: dict, mapping: HookMapping) -> UnifiedMessage:
        agent_id = raw.get("agent_id", mapping.default_agent_id)
        if not mapping.allowed_agent_ids or agent_id not in mapping.allowed_agent_ids:
            agent_id = mapping.default_agent_id

        session_key = mapping.default_session_key
        if mapping.allow_request_session_key:
            req_key = raw.get("session_key", "")
            if req_key:
                if not mapping.allowed_session_key_prefixes or any(
                    req_key.startswith(p) for p in mapping.allowed_session_key_prefixes
                ):
                    session_key = req_key

        strategy = mapping.session_key_strategy

        return UnifiedMessage(
            message_id=raw.get("id", ""),
            channel_type=ChannelType.HOOKS,
            sender=Peer(id="hooks", name="Hooks", kind="system"),
            content=raw.get("content", raw.get("text", "")),
            metadata={
                "source": "hooks",
                "match_path": mapping.match_path,
                "session_key_strategy": strategy,
                "session_key": session_key,
                "session_key_prefix": mapping.match_path,
                "deliver": mapping.deliver,
                "account_id": raw.get("account_id", mapping.match_path),
            },
            agent_config_id=agent_id,
            raw_payload=raw,
        )