Kris
ebab14660a
新增 Mattermost 渠道扩展，支持在 Yuxi 平台中集成 Mattermost 团队协作平台。 包含以下功能模块： - client: Mattermost API 客户端封装 - config: 渠道配置管理 - gateway: SSE/WebSocket 网关接入 - websocket: WebSocket 实时连接 - outbound: 外发消息管理 - streaming: 流式消息处理 - pairing: 用户配对与绑定 - security: 安全校验 - dedup: 消息去重 - monitor: 渠道状态监控 - status: 会话状态管理 - session: 会话管理 - interactions: 交互处理 - slash_commands: 斜杠指令 - actions: 动作处理 - approval: 审批流程 - delivery: 消息送达确认 - directory: 目录管理 - threading: 线程管理 - gating: 门控管理 - reconnect: 重连机制 - reactions: 表情反应 - media: 媒体资源处理 - model_picker: 模型选择 - types: 类型定义
85 lines
3.6 KiB
Python
85 lines
3.6 KiB
Python
from __future__ import annotations
|
|
|
|
import logging
|
|
|
|
from yuxi.channel.extensions.mattermost.client import MattermostClient
|
|
from yuxi.channel.extensions.mattermost.errors import MattermostError
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
DEFAULT_COMMAND_SPECS = [
|
|
{"trigger": "oc_status", "method": "P", "auto_complete": True,
|
|
"auto_complete_hint": "", "auto_complete_desc": "Show bot status"},
|
|
{"trigger": "oc_model", "method": "P", "auto_complete": True,
|
|
"auto_complete_hint": "[model_name]", "auto_complete_desc": "Switch AI model"},
|
|
{"trigger": "oc_models", "method": "P", "auto_complete": True,
|
|
"auto_complete_hint": "", "auto_complete_desc": "List available models"},
|
|
{"trigger": "oc_new", "method": "P", "auto_complete": True,
|
|
"auto_complete_hint": "", "auto_complete_desc": "Start a new conversation"},
|
|
{"trigger": "oc_help", "method": "P", "auto_complete": True,
|
|
"auto_complete_hint": "", "auto_complete_desc": "Show help information"},
|
|
{"trigger": "oc_think", "method": "P", "auto_complete": True,
|
|
"auto_complete_hint": "[level]", "auto_complete_desc": "Set reasoning level"},
|
|
{"trigger": "oc_reasoning", "method": "P", "auto_complete": True,
|
|
"auto_complete_hint": "", "auto_complete_desc": "Toggle reasoning mode"},
|
|
{"trigger": "oc_verbose", "method": "P", "auto_complete": True,
|
|
"auto_complete_hint": "", "auto_complete_desc": "Toggle verbose mode"},
|
|
]
|
|
|
|
|
|
class SlashCommandManager:
|
|
def __init__(self, client: MattermostClient, account_id: str):
|
|
self.client = client
|
|
self.account_id = account_id
|
|
self._registered: list[dict] = []
|
|
|
|
async def register_commands(
|
|
self,
|
|
callback_url: str,
|
|
team_ids: list[str],
|
|
) -> list[dict]:
|
|
registered: list[dict] = []
|
|
for team_id in team_ids:
|
|
try:
|
|
existing = await self.client.list_commands(team_id)
|
|
existing_triggers = {c["trigger"] for c in existing}
|
|
|
|
for spec in DEFAULT_COMMAND_SPECS:
|
|
trigger = spec["trigger"]
|
|
if trigger in existing_triggers:
|
|
logger.debug("Command %s already registered for team %s", trigger, team_id)
|
|
continue
|
|
|
|
payload = {
|
|
"team_id": team_id,
|
|
"trigger": trigger,
|
|
"method": spec["method"],
|
|
"url": f"{callback_url}/{self.account_id}",
|
|
"auto_complete": spec.get("auto_complete", True),
|
|
"auto_complete_desc": spec.get("auto_complete_desc", ""),
|
|
"auto_complete_hint": spec.get("auto_complete_hint", ""),
|
|
}
|
|
cmd = await self.client.create_command(payload)
|
|
cmd["managed"] = True
|
|
registered.append(cmd)
|
|
logger.info("Registered command %s for team %s", trigger, team_id)
|
|
except MattermostError as e:
|
|
logger.warning("Failed to register commands for team %s: %s", team_id, e)
|
|
|
|
self._registered = registered
|
|
return registered
|
|
|
|
async def cleanup_commands(self) -> None:
|
|
for cmd in self._registered:
|
|
if cmd.get("managed"):
|
|
try:
|
|
await self.client.delete_command(cmd["id"])
|
|
logger.info("Deleted command %s", cmd.get("trigger", cmd["id"]))
|
|
except MattermostError:
|
|
pass
|
|
self._registered.clear()
|
|
|
|
def get_managed_commands(self) -> list[dict]:
|
|
return list(cmd for cmd in self._registered if cmd.get("managed"))
|