ForcePilot/backend/package/yuxi/channel/extensions/whatsapp/security.py
Kris 7215418610 feat(channel): 添加企业微信、微博、WhatsApp 和 Workplace 渠道扩展
新增企业微信、微博、WhatsApp、Workplace 四个渠道扩展。

企业微信渠道扩展主要模块:config, gateway, webhook, webhook_bot, outbound, streaming, pairing, security, crypto, dedupe, persistent_dedupe, card, directory, events, externalcontact, media, mentions, menu, message, oauth, status

微博渠道扩展主要模块:config, gateway, webhook, outbound, streaming, pairing, security, dedupe, passive_reply, broadcast, message, menu, media, subscription, template, status

WhatsApp 渠道扩展主要模块:config, gateway, webhook, outbound, streaming, pairing, security, dedupe, actions, monitor, status

Workplace 渠道扩展主要模块:config, gateway, webhook, outbound, streaming, pairing, security, dedupe, actions, challenge, groups, media, mentions, menu, monitor, persona, quick_reply, signature, subscriptions, template, threading, users, status
2026-05-21 12:01:56 +08:00

67 lines
2.2 KiB
Python

"""WhatsApp Security 适配器 — DM/群组安全策略"""
import logging
logger = logging.getLogger(__name__)
class WhatsAppSecurity:
def resolve_dm_policy(self) -> dict:
return {"mode": "pairing", "allow_from": []}
def resolve_dm_policy_for_account(self, account: dict) -> dict:
mode = account.get("dm_policy", "pairing")
return {"mode": mode, "allow_from": account.get("allow_from", [])}
async def check_allowlist(self, peer_id: str, channel_type: str) -> bool:
from yuxi.channel.extensions.whatsapp.config import WhatsAppConfigAdapter
adapter = WhatsAppConfigAdapter()
account = await adapter.resolve_account("default")
if not account.get("configured"):
return True
dm_policy = account.get("dm_policy", "pairing")
if dm_policy == "open":
return True
if dm_policy == "disabled":
return False
allow_from = account.get("allow_from", [])
normalized_peer = peer_id.strip().lstrip("+")
return any(entry.strip().lstrip("+") == normalized_peer for entry in allow_from)
def resolve_require_mention(self, ctx) -> bool | None:
config = getattr(ctx, "config", {}) if ctx else {}
group_id = getattr(ctx, "group_id", None) if ctx else None
if group_id:
groups = config.get("groups", {})
group_cfg = groups.get(group_id, {})
if "require_mention" in group_cfg:
return group_cfg["require_mention"]
return True
def resolve_group_intro_hint(self, ctx) -> str | None:
return None
def resolve_tool_policy(self, ctx) -> dict | None:
return None
def apply_config_fixes(self, config: dict) -> dict:
return config
def collect_warnings(
self, config: dict, account_id: str | None = None, account: dict | None = None
) -> list[str]:
warnings = []
group_policy = account.get("group_policy", "") if account else config.get("group_policy", "")
if group_policy == "open" and not config.get("groups"):
warnings.append("groupPolicy is 'open' without groups allowlist")
return warnings