Kris
002d601d1b
新增 Mattermost 渠道完整实现，包含适配器核心、消息处理、交互回调、命令支持、安全校验、多账号管理等功能，支持机器人消息发送、交互按钮、命令注册、投票功能以及配置动态修改等特性。
94 lines
3.0 KiB
Python
94 lines
3.0 KiB
Python
from __future__ import annotations
|
|
|
|
from typing import Any
|
|
|
|
from yuxi.utils.logging_config import logger
|
|
|
|
|
|
class ConfigWritesManager:
|
|
"""配置写入管理 — 允许通过命令动态修改 Mattermost 渠道配置。
|
|
|
|
支持的写操作：
|
|
- dm_policy、group_policy 安全策略切换
|
|
- allow_from / group_allow_from 白名单管理
|
|
- chatmode / require_mention 行为配置
|
|
- reply_to_mode 回复模式
|
|
- auto_register_commands 命令注册开关
|
|
"""
|
|
|
|
ALLOWED_WRITE_KEYS = frozenset(
|
|
{
|
|
"dm_policy",
|
|
"group_policy",
|
|
"allow_from",
|
|
"group_allow_from",
|
|
"chatmode",
|
|
"require_mention",
|
|
"onchar_prefixes",
|
|
"reply_to_mode",
|
|
"auto_register_commands",
|
|
"dangerously_allow_name_matching",
|
|
}
|
|
)
|
|
|
|
def __init__(self, config: dict[str, Any] | None = None):
|
|
self._config = config or {}
|
|
|
|
def is_config_writes_enabled(self) -> bool:
|
|
return bool(self._config.get("config_writes", False))
|
|
|
|
def write_config(self, key: str, value: Any) -> bool:
|
|
if not self.is_config_writes_enabled():
|
|
logger.warning("[Mattermost] Config writes are disabled")
|
|
return False
|
|
|
|
if key not in self.ALLOWED_WRITE_KEYS:
|
|
logger.warning(f"[Mattermost] Config write denied for key '{key}'")
|
|
return False
|
|
|
|
self._config[key] = value
|
|
logger.info(f"[Mattermost] Config updated: {key} = {value}")
|
|
return True
|
|
|
|
def add_to_allowlist(self, target: str, list_type: str = "dm") -> bool:
|
|
key = "allow_from" if list_type == "dm" else "group_allow_from"
|
|
if not self.is_config_writes_enabled():
|
|
return False
|
|
|
|
current = self._config.get(key, [])
|
|
if not isinstance(current, list):
|
|
current = []
|
|
|
|
from .security import normalize_allow_entry
|
|
|
|
normalized = normalize_allow_entry(target)
|
|
if not normalized:
|
|
return False
|
|
|
|
if normalized not in current:
|
|
current.append(normalized)
|
|
self._config[key] = current
|
|
logger.info(f"[Mattermost] Added '{target}' to {key}")
|
|
return True
|
|
|
|
def remove_from_allowlist(self, target: str, list_type: str = "dm") -> bool:
|
|
key = "allow_from" if list_type == "dm" else "group_allow_from"
|
|
if not self.is_config_writes_enabled():
|
|
return False
|
|
|
|
current = self._config.get(key, [])
|
|
if not isinstance(current, list):
|
|
return False
|
|
|
|
from .security import normalize_allow_entry
|
|
|
|
normalized = normalize_allow_entry(target)
|
|
if normalized in current:
|
|
current.remove(normalized)
|
|
self._config[key] = current
|
|
logger.info(f"[Mattermost] Removed '{target}' from {key}")
|
|
return True
|
|
|
|
def get_config(self) -> dict[str, Any]:
|
|
return {k: v for k, v in self._config.items() if k in self.ALLOWED_WRITE_KEYS}
|