Kris
05abecc02b
该提交实现了支持企业微信、微信公众号、个人微信桥接三种模式的完整微信渠道适配器，包含以下核心模块： 1. 基础认证与配置相关：auth_adapter、config_reload、setup_contract等 2. 消息处理与格式转换：format、attachment_adapter、outbound_adapter等 3. 多模式客户端支持：wecom/mp子模块，包含加解密、消息收发能力 4. 辅助能力：限速器、防抖、会话绑定、事件映射、模板渲染等 5. 扩展能力：二维码登录、消息读取、特权用户、心跳监控等 实现了完整的微信生态对接能力，支持消息收发、事件处理、API调用限流、配置热重载等功能。
58 lines
2.3 KiB
Python
58 lines
2.3 KiB
Python
from __future__ import annotations
|
|
|
|
from typing import Any
|
|
|
|
from yuxi.utils.logging_config import logger
|
|
|
|
|
|
class WeChatElevatedAdapter:
|
|
def __init__(self):
|
|
self._elevated_users: set[str] = set()
|
|
self._elevation_enabled: bool = False
|
|
|
|
def load_elevated(self, config: dict[str, Any]) -> None:
|
|
elevated = config.get("elevated_users", [])
|
|
if isinstance(elevated, list):
|
|
self._elevated_users = {f"wx:{u}" if not u.startswith("wx:") else u for u in elevated}
|
|
self._elevation_enabled = config.get("elevation_enabled", False)
|
|
logger.info(
|
|
f"[WeChat/Elevated] Loaded {len(self._elevated_users)} elevated user(s), enabled={self._elevation_enabled}"
|
|
)
|
|
|
|
def is_elevated(self, channel_user_id: str) -> bool:
|
|
normalized = channel_user_id if channel_user_id.startswith("wx:") else f"wx:{channel_user_id}"
|
|
return self._elevation_enabled and normalized in self._elevated_users
|
|
|
|
def adjust_routing_priority(self, channel_user_id: str, base_priority: int = 0) -> int:
|
|
if self.is_elevated(channel_user_id):
|
|
return max(base_priority, 10)
|
|
return base_priority
|
|
|
|
def resolve_elevated_agents(self, channel_user_id: str, config: dict[str, Any]) -> list[str]:
|
|
if not self.is_elevated(channel_user_id):
|
|
return []
|
|
agents = config.get("elevated_agents", [])
|
|
if isinstance(agents, list):
|
|
return agents
|
|
return config.get("agent_id", "1") if isinstance(config.get("agent_id"), str) else ["1"]
|
|
|
|
def collect_elevation_audit(self) -> list[dict[str, Any]]:
|
|
findings: list[dict[str, Any]] = []
|
|
if self._elevation_enabled and len(self._elevated_users) > 100:
|
|
findings.append(
|
|
{
|
|
"severity": "warning",
|
|
"category": "elevated_at_scale",
|
|
"message": f"{len(self._elevated_users)} elevated users (>100)，建议审计特权用户列表",
|
|
}
|
|
)
|
|
if self._elevation_enabled and not self._elevated_users:
|
|
findings.append(
|
|
{
|
|
"severity": "info",
|
|
"category": "elevated_empty",
|
|
"message": "特权模式已启用但 elevated_users 为空",
|
|
}
|
|
)
|
|
return findings
|