from __future__ import annotations

from yuxi.channels.models import ChatType


def resolve_chat_type(source_type: str) -> ChatType:
    if source_type == "user":
        return ChatType.DIRECT
    return ChatType.GROUP


def resolve_channel_chat_id(source_type: str, source_id: str) -> str:
    prefix_map = {"user": "user_", "group": "group_", "room": "room_"}
    return f"{prefix_map.get(source_type, 'unknown_')}{source_id}"


def resolve_thread_key(agent_id: str, source_type: str, source_id: str, account_id: str = "default") -> str:
    if source_type == "user":
        return f"agent:{agent_id}:line:{account_id}:dm:{source_id}"
    return f"agent:{agent_id}:line:{account_id}:{source_type}:{source_id}"