Kris
1c590097be
新增 Twitter 和 Viber 两个渠道扩展。 Twitter 渠道扩展功能模块： - auth: OAuth 认证管理 - config: 渠道配置管理 - gateway: SSE/WebSocket 网关接入 - webhook: Webhook 事件处理 - outbound: 外发消息管理 - streaming: 流式消息处理 - pairing: 用户配对与绑定 - security: 安全校验 - dedupe: 消息去重 - monitor: 渠道状态监控 - status: 会话状态管理 - session: 会话管理 - tweets: 推文管理 - social: 社交互动 - reactions: 表情反应 - media: 媒体资源处理 Viber 渠道扩展功能模块： - config: 渠道配置管理 - gateway: SSE/WebSocket 网关接入 - webhook: Webhook 事件处理 - outbound: 外发消息管理 - streaming: 流式消息处理 - pairing: 用户配对与绑定 - security: 安全校验 - dedupe: 消息去重 - monitor: 渠道状态监控 - status: 会话状态管理 - rate_limiter: 速率限制 - media: 媒体资源处理
49 lines
1.5 KiB
Python
49 lines
1.5 KiB
Python
from __future__ import annotations
|
|
|
|
import logging
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
class ViberSecurityAdapter:
|
|
DM_POLICY_OPTIONS = ["pairing", "allowlist", "open", "disabled"]
|
|
|
|
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:
|
|
return True
|
|
|
|
def is_allowed_dm(self, peer_id: str, allow_from: list[str], policy: str) -> bool:
|
|
if "*" in allow_from:
|
|
return True
|
|
if policy == "open":
|
|
return True
|
|
if policy == "disabled":
|
|
return False
|
|
if policy == "allowlist":
|
|
return self._match_peer(peer_id, allow_from)
|
|
if policy == "pairing":
|
|
return self._match_peer(peer_id, allow_from)
|
|
return False
|
|
|
|
@staticmethod
|
|
def _match_peer(peer_id: str, allow_from: list[str]) -> bool:
|
|
for entry in allow_from:
|
|
entry = entry.strip()
|
|
if entry == "*":
|
|
return True
|
|
if entry.startswith("viber:user:"):
|
|
target = entry[11:]
|
|
elif entry.startswith("viber:"):
|
|
target = entry[6:]
|
|
else:
|
|
target = entry
|
|
if target == peer_id:
|
|
return True
|
|
return False
|