Kris
8a3601250e
新增 Telegram 渠道扩展，支持在 Yuxi 平台中集成 Telegram 即时通讯渠道。 包含以下功能模块： - config: 渠道配置管理 - gateway: SSE/WebSocket 网关接入 - webhook: Webhook 事件处理 - polling: 长轮询模式 - outbound: 外发消息管理 - streaming: 流式消息处理 - pairing: 用户配对与绑定 - security: 安全校验 - dedupe: 消息去重 - monitor: 渠道状态监控 - status: 会话状态管理 - session: 会话管理 - actions: 动作处理 - inline_keyboard: 内联键盘 - native_commands: 原生指令 - chat: 聊天管理 - delivery: 消息送达确认 - media: 媒体资源处理 - profile: 用户资料 - reactions: 表情反应 - sticker: 贴纸处理 - types: 类型定义
88 lines
3.0 KiB
Python
88 lines
3.0 KiB
Python
from __future__ import annotations
|
|
|
|
import logging
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
SESSION_KEY_SEPARATOR = "::"
|
|
|
|
|
|
class TelegramSession:
|
|
@staticmethod
|
|
def build_dm_key(chat_id: str) -> str:
|
|
return f"dm{SESSION_KEY_SEPARATOR}{chat_id}"
|
|
|
|
@staticmethod
|
|
def build_group_key(chat_id: str) -> str:
|
|
return f"group{SESSION_KEY_SEPARATOR}{chat_id}"
|
|
|
|
@staticmethod
|
|
def build_topic_key(chat_id: str, thread_id: str) -> str:
|
|
return f"topic{SESSION_KEY_SEPARATOR}{chat_id}{SESSION_KEY_SEPARATOR}{thread_id}"
|
|
|
|
@staticmethod
|
|
def parse_session_key(key: str) -> dict:
|
|
parts = key.split(SESSION_KEY_SEPARATOR, 2)
|
|
if len(parts) == 2 and parts[0] == "dm":
|
|
return {"type": "dm", "chat_id": parts[1]}
|
|
if len(parts) == 2 and parts[0] == "group":
|
|
return {"type": "group", "chat_id": parts[1]}
|
|
if len(parts) == 3 and parts[0] == "topic":
|
|
return {"type": "topic", "chat_id": parts[1], "thread_id": parts[2]}
|
|
return {"type": "unknown", "key": key}
|
|
|
|
|
|
class TelegramThreading:
|
|
@staticmethod
|
|
def extract_thread_id(unified_message: dict) -> str | None:
|
|
thread_id = unified_message.get("thread_id")
|
|
if thread_id:
|
|
return str(thread_id)
|
|
|
|
msg_id = unified_message.get("msg_id", "")
|
|
if msg_id.startswith("tg:cb:"):
|
|
return unified_message.get("thread_id")
|
|
|
|
return None
|
|
|
|
@staticmethod
|
|
def resolve_reply_target(unified_message: dict) -> dict:
|
|
group = unified_message.get("group")
|
|
thread_id = TelegramThreading.extract_thread_id(unified_message)
|
|
is_topic = unified_message.get("is_topic_message", False)
|
|
|
|
target_type = "dm"
|
|
target_id = unified_message.get("sender", {}).get("id", "")
|
|
reply_to_id = str(unified_message.get("reply_to_id")) if unified_message.get("reply_to_id") else None
|
|
|
|
if group:
|
|
target_type = "group"
|
|
target_id = group.get("id", "")
|
|
if is_topic and thread_id:
|
|
target_type = "topic"
|
|
|
|
return {
|
|
"target_type": target_type,
|
|
"target_id": target_id,
|
|
"thread_id": thread_id,
|
|
"reply_to_id": reply_to_id,
|
|
"session_key": TelegramThreading._build_session_key(target_type, target_id, thread_id),
|
|
}
|
|
|
|
@staticmethod
|
|
def get_topic_agent_config(account: dict, group_id: str, thread_id: str) -> dict | None:
|
|
groups = account.get("groups", {})
|
|
group_cfg = groups.get(group_id, {})
|
|
topics = group_cfg.get("topics", {})
|
|
if thread_id in topics:
|
|
return topics[thread_id]
|
|
return None
|
|
|
|
@staticmethod
|
|
def _build_session_key(target_type: str, target_id: str, thread_id: str | None) -> str:
|
|
if target_type == "dm":
|
|
return TelegramSession.build_dm_key(target_id)
|
|
if target_type == "topic" and thread_id:
|
|
return TelegramSession.build_topic_key(target_id, thread_id)
|
|
return TelegramSession.build_group_key(target_id)
|