from __future__ import annotations

from yuxi.channels.models import ChatType


def resolve_chat_id(post: dict, channel_data: dict) -> str:
    if post.get("root_id"):
        return build_thread_id(post["channel_id"], post["root_id"])
    ch_type = channel_data.get("type", "")
    if ch_type == "D":
        user_id = post.get("user_id", "") or "unknown"
        return f"dm_{user_id}"
    return f"channel_{post['channel_id']}"


def resolve_chat_type(post: dict, channel_data: dict) -> ChatType:
    if post.get("root_id"):
        return ChatType.THREAD
    ch_type = channel_data.get("type", "")
    if ch_type == "D":
        return ChatType.DIRECT
    return ChatType.GROUP


def build_thread_id(
    chat_type_or_channel_id, root_id_or_user_id=None, *, user_id: str = "", channel_id: str = "", agent_id: str = "main"
) -> str:
    """构建线程 ID 用于会话隔离。

    兼容两种调用方式：
    1. 新式：build_thread_id(channel_id, root_id) — 两个位置参数
    2. 旧式：build_thread_id(ChatType, user_id=..., channel_id=..., agent_id=...)
    """
    if isinstance(chat_type_or_channel_id, ChatType):
        chat_type = chat_type_or_channel_id
        if chat_type == ChatType.DIRECT:
            return f"agent:{agent_id}:mattermost:dm:{user_id}"
        if chat_type in (ChatType.GROUP, ChatType.THREAD):
            return f"agent:{agent_id}:mattermost:channel:{channel_id}"
        return f"agent:{agent_id}:mattermost:channel:{channel_id}"

    channel_id_val = chat_type_or_channel_id
    root_id_val = root_id_or_user_id
    if root_id_val:
        return f"channel_{channel_id_val}:thread_{root_id_val}"
    return f"channel_{channel_id_val}"