from __future__ import annotations

from typing import Any

from yuxi.channels.models import ChatType, EventType

WECHAT_TO_UNIFIED_MESSAGE_TYPE: dict[str, str] = {
    "text": "text",
    "image": "image",
    "voice": "voice",
    "video": "video",
    "file": "file",
    "link": "link",
    "location": "location",
    "event": "event",
    "news": "news",
}

WECHAT_TO_UNIFIED_EVENT_TYPE: dict[str, str] = {
    "subscribe": "subscribe",
    "unsubscribe": "unsubscribe",
    "CLICK": "click",
    "VIEW": "view",
    "LOCATION": "location_event",
    "scancode_push": "scancode",
    "scancode_waitmsg": "scancode",
    "enter_agent": "enter_agent",
}

WECHAT_MODE_LABEL: dict[str, str] = {
    "wecom": "WeCom",
    "mp": "Official Account",
    "personal": "Personal Bridge",
}


def map_message_type(wechat_msg_type: str) -> str:
    return WECHAT_TO_UNIFIED_MESSAGE_TYPE.get(wechat_msg_type, wechat_msg_type)


def map_event_type(wechat_event_type: str) -> str:
    return WECHAT_TO_UNIFIED_EVENT_TYPE.get(wechat_event_type, wechat_event_type)


def map_chat_type(from_user: str, to_user: str, chat_id: str = "") -> ChatType:
    if chat_id and chat_id != from_user:
        return ChatType.GROUP
    return ChatType.DIRECT


def map_event(wechat_event_type: str) -> EventType | None:
    mapping: dict[str, EventType] = {
        "subscribe": EventType.USER_JOIN,
        "unsubscribe": EventType.USER_LEAVE,
        "CLICK": EventType.MESSAGE,
        "enter_agent": EventType.MESSAGE,
    }
    return mapping.get(wechat_event_type)


def map_wechat_error(errcode: int, errmsg: str) -> dict[str, Any]:
    error_map: dict[int, tuple[str, str]] = {
        40001: ("invalid_credential", "Invalid access token"),
        40014: ("invalid_token", "Invalid access token"),
        41001: ("access_token_missing", "Access token missing"),
        42001: ("token_expired", "Access token expired"),
        48001: ("api_unauthorized", "API not authorized"),
    }
    code, message = error_map.get(errcode, ("unknown", errmsg))
    return {"code": code, "message": message}


def build_channel_identity(mode: str, from_user: str, chat_id: str) -> dict[str, Any]:
    return {
        "channel_user_id": f"wx:{from_user}",
        "channel_chat_id": f"wx:{chat_id}" if chat_id else f"wx:{from_user}",
        "wechat_mode": mode,
    }