"""Conversation id helpers for Douyin IM."""


def parse_conversation_parts(conversation_id: str) -> tuple[int, int] | None:
    parts = (conversation_id or "").split(":")
    if len(parts) >= 4 and parts[0] == "0" and parts[1] == "1":
        try:
            return int(parts[2]), int(parts[3])
        except (TypeError, ValueError):
            return None
    return None


def resolve_peer_uid(conversation_id: str, my_uid: int) -> int | None:
    """Resolve peer user id from conversation id or bare numeric id."""
    raw = (conversation_id or "").strip()
    if not raw:
        return None

    parts = parse_conversation_parts(raw)
    if parts:
        uid1, uid2 = parts
        if my_uid and uid1 == my_uid:
            return uid2
        if my_uid and uid2 == my_uid:
            return uid1
        # 0:1:{my}:{peer} — 若 my_uid 与首段不一致，仍取末段为对方
        return uid2

    if raw.isdigit():
        peer = int(raw)
        if my_uid and peer == my_uid:
            return None
        return peer
    return None


def build_conversation_id(my_uid: int, peer_uid: int) -> str:
    return f"0:1:{int(my_uid)}:{int(peer_uid)}"


def normalize_conversation_id(conversation_id: str, my_uid: int) -> str:
    """Ensure conversation id uses current account uid as first participant."""
    peer_uid = resolve_peer_uid(conversation_id, my_uid)
    if peer_uid and my_uid:
        return build_conversation_id(my_uid, peer_uid)
    return (conversation_id or "").strip()