Kris
4e9c6dd8ab
新增 Matrix 渠道扩展，支持在 Yuxi 平台中集成 Matrix 去中心化通讯协议。 包含以下功能模块： - config: 渠道配置管理 - gateway: SSE/WebSocket 网关接入 - outbound: 外发消息管理 - streaming: 流式消息处理 - pairing: 用户配对与绑定 - security: 安全校验 - crypto: 端到端加密 - dedupe: 消息去重 - monitor: 渠道状态监控 - status: 会话状态管理 - session: 会话管理 - room_resolver: 房间解析 - dm_tracker: 私聊追踪 - rate_limiter: 速率限制 - actions: 动作处理 - constants: 常量定义 - utils: 工具函数 - types: 类型定义
59 lines
1.9 KiB
Python
59 lines
1.9 KiB
Python
from __future__ import annotations
|
|
|
|
import logging
|
|
import secrets
|
|
import time
|
|
from collections import OrderedDict
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
id_label = "matrixUserId"
|
|
|
|
_pairing_codes: OrderedDict[str, tuple[str, float]] = OrderedDict()
|
|
_PAIRING_MAX_SIZE = 10000
|
|
_PAIRING_TTL_SECONDS = 600
|
|
|
|
|
|
async def generate_code(peer_id: str, config: dict = None, account_id: str = None) -> str:
|
|
alphabet = "ABCDEFGHJKLMNPQRSTUVWXYZ23456789"
|
|
code = "".join(secrets.choice(alphabet) for _ in range(8))
|
|
_evict_expired()
|
|
_pairing_codes[peer_id] = (code, time.monotonic() + _PAIRING_TTL_SECONDS)
|
|
if len(_pairing_codes) > _PAIRING_MAX_SIZE:
|
|
_pairing_codes.popitem(last=False)
|
|
logger.info("Generated pairing code for peer %s", peer_id)
|
|
return code
|
|
|
|
|
|
async def verify_code(peer_id: str, code: str, config: dict = None, account_id: str = None) -> bool:
|
|
_evict_expired()
|
|
stored = _pairing_codes.get(peer_id)
|
|
if stored is None:
|
|
logger.info("Pairing verify failed: no code for peer %s", peer_id)
|
|
return False
|
|
stored_code, expires_at = stored
|
|
if time.monotonic() > expires_at:
|
|
del _pairing_codes[peer_id]
|
|
logger.info("Pairing verify failed: code expired for peer %s", peer_id)
|
|
return False
|
|
if code.upper() != stored_code:
|
|
logger.info("Pairing verify failed: code mismatch for peer %s", peer_id)
|
|
return False
|
|
del _pairing_codes[peer_id]
|
|
logger.info("Pairing verified for peer %s", peer_id)
|
|
return True
|
|
|
|
|
|
def _evict_expired() -> None:
|
|
now = time.monotonic()
|
|
expired = [k for k, (_, exp) in _pairing_codes.items() if now > exp]
|
|
for k in expired:
|
|
del _pairing_codes[k]
|
|
|
|
|
|
def normalize_allow_entry(entry: str) -> str:
|
|
return entry.strip().lower()
|
|
|
|
|
|
async def notify_approval(config: dict, peer_id: str, account_id: str | None = None) -> None:
|
|
logger.info("Matrix pairing approved for %s", peer_id) |