79 lines
2.3 KiB
Python
79 lines
2.3 KiB
Python
|
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,
|
||
|
|
}
|