Kris
1f78c44b03
这是一个批量整理提交，包含以下主要改动： 1. 删除多处冗余的空行和未使用的导入 2. 修复文件末尾缺少换行符的问题 3. 调整部分模块的导入顺序与代码排版 4. 修复部分配置默认值与策略逻辑 5. 新增多个功能模块与辅助工具 6. 完善异常处理与日志记录 7. 修复速率限制、消息缓存、权限校验等逻辑bug 8. 废弃部分旧有API与配置项并添加警告提示
120 lines
3.5 KiB
Python
120 lines
3.5 KiB
Python
from __future__ import annotations
|
|
|
|
from typing import Any
|
|
|
|
TWITCH_ROLES = frozenset({"moderator", "owner", "vip", "subscriber", "all"})
|
|
|
|
|
|
def normalize_twitch_channel(channel: str) -> str:
|
|
return channel.lstrip("#@").strip().lower()
|
|
|
|
|
|
def resolve_channel_name(channel_chat_id: str) -> str:
|
|
return normalize_twitch_channel(channel_chat_id)
|
|
|
|
|
|
def resolve_route(
|
|
agent_id: str,
|
|
channel_name: str,
|
|
) -> str:
|
|
channel_name = normalize_twitch_channel(channel_name)
|
|
return f"agent:{agent_id}:twitch:channel:{channel_name}"
|
|
|
|
|
|
def resolve_agent_id_for_channel(
|
|
config: dict[str, Any],
|
|
channel_name: str,
|
|
default_agent_id: str | None = None,
|
|
) -> str:
|
|
channel_name = normalize_twitch_channel(channel_name)
|
|
channels_config = config.get("channels_config", {})
|
|
channel_config = channels_config.get(channel_name, {})
|
|
return channel_config.get("agent_id") or default_agent_id or ""
|
|
|
|
|
|
def check_require_mention(
|
|
config: dict[str, Any],
|
|
content: str,
|
|
) -> bool:
|
|
bot_name = config.get("bot_username", "").lower()
|
|
require_mention = config.get("require_mention", True)
|
|
if not require_mention:
|
|
return True
|
|
return bool(bot_name) and bot_name in content.lower()
|
|
|
|
|
|
def check_allowed_roles(
|
|
config: dict[str, Any],
|
|
channel: str,
|
|
badges: dict[str, str],
|
|
) -> bool:
|
|
allowed_roles: list[str] = config.get("allowedRoles", [])
|
|
|
|
if "all" in allowed_roles:
|
|
return True
|
|
|
|
channels_config = config.get("channels_config", {})
|
|
channel_name = normalize_twitch_channel(channel)
|
|
per_channel_roles: list[str] = channels_config.get(channel_name, {}).get("allowedRoles", [])
|
|
if "all" in per_channel_roles:
|
|
return True
|
|
|
|
effective_roles = list(dict.fromkeys(allowed_roles + per_channel_roles))
|
|
if not effective_roles:
|
|
return True
|
|
|
|
for role in effective_roles:
|
|
role_lower = role.lower()
|
|
if role_lower not in TWITCH_ROLES:
|
|
continue
|
|
if role_lower == "all":
|
|
return True
|
|
if role_lower == "owner" and "broadcaster" in badges:
|
|
return True
|
|
if role_lower == "moderator" and "moderator" in badges:
|
|
return True
|
|
if role_lower == "vip" and "vip" in badges:
|
|
return True
|
|
if role_lower == "subscriber" and ("subscriber" in badges or "founder" in badges):
|
|
return True
|
|
|
|
return False
|
|
|
|
|
|
def check_group_policy(
|
|
config: dict[str, Any],
|
|
channel: str,
|
|
user_id: str,
|
|
content: str,
|
|
user_name: str = "",
|
|
) -> bool:
|
|
channel_name = normalize_twitch_channel(channel)
|
|
group_policy = config.get("group_policy", "open")
|
|
bot_name = config.get("bot_username", "").lower()
|
|
|
|
match group_policy:
|
|
case "open":
|
|
return True
|
|
case "disabled":
|
|
return False
|
|
case "mention_only":
|
|
require_mention = config.get("require_mention", True)
|
|
if not require_mention:
|
|
return True
|
|
return bool(bot_name) and bot_name in content.lower()
|
|
case "allowlist":
|
|
allowlist = config.get("group_allow_from", [])
|
|
channels_config = config.get("channels_config", {})
|
|
per_channel_allow = channels_config.get(channel_name, {}).get("allow_from", [])
|
|
|
|
candidates = [
|
|
f"twitch:{user_id}",
|
|
user_id,
|
|
]
|
|
if user_name:
|
|
candidates.append(user_name.lower())
|
|
|
|
return any(candidate in allowlist or candidate in per_channel_allow for candidate in candidates)
|
|
case _:
|
|
return False
|