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: 类型定义
142 lines
5.2 KiB
Python
142 lines
5.2 KiB
Python
from __future__ import annotations
|
|
|
|
import logging
|
|
|
|
from yuxi.channel.extensions.mattermost.client import MattermostClient
|
|
from yuxi.channel.extensions.mattermost.errors import MattermostError
|
|
from yuxi.channel.extensions.mattermost.types import MattermostChannel, MattermostUser
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
class MattermostDirectoryAdapter:
|
|
def __init__(self, client: MattermostClient, account_id: str = ""):
|
|
self.client = client
|
|
self.account_id = account_id
|
|
|
|
async def list_channels(self, team_id: str | None = None) -> list[dict]:
|
|
channels: list[dict] = []
|
|
try:
|
|
if team_id:
|
|
raw = await self.client.fetch_team_channels(team_id)
|
|
for item in raw:
|
|
ch = MattermostChannel.from_dict(item)
|
|
channels.append(
|
|
{
|
|
"id": ch.id,
|
|
"name": ch.name,
|
|
"display_name": ch.display_name,
|
|
"type": ch.type,
|
|
"team_id": ch.team_id,
|
|
}
|
|
)
|
|
else:
|
|
me = await self.client.fetch_me()
|
|
teams = await self.client.fetch_user_teams(me["id"])
|
|
for team_data in teams:
|
|
tid = team_data["id"]
|
|
raw = await self.client.fetch_team_channels(tid)
|
|
for item in raw:
|
|
ch = MattermostChannel.from_dict(item)
|
|
channels.append(
|
|
{
|
|
"id": ch.id,
|
|
"name": ch.name,
|
|
"display_name": ch.display_name,
|
|
"type": ch.type,
|
|
"team_id": ch.team_id,
|
|
}
|
|
)
|
|
except MattermostError as e:
|
|
logger.error("Failed to list channels: %s", e)
|
|
return channels
|
|
|
|
async def list_users(self, team_id: str | None = None) -> list[dict]:
|
|
users: list[dict] = []
|
|
try:
|
|
raw = await self.client.fetch_users(page=0, per_page=200, in_team=team_id or "")
|
|
for item in raw:
|
|
user = MattermostUser.from_dict(item)
|
|
users.append(
|
|
{
|
|
"id": user.id,
|
|
"username": user.username,
|
|
"nickname": user.nickname,
|
|
"first_name": user.first_name,
|
|
"last_name": user.last_name,
|
|
"email": user.email,
|
|
"is_bot": user.is_bot,
|
|
}
|
|
)
|
|
except MattermostError as e:
|
|
logger.error("Failed to list users: %s", e)
|
|
return users
|
|
|
|
async def get_user_info(self, user_id: str) -> dict | None:
|
|
try:
|
|
result = await self.client.fetch_user(user_id)
|
|
user = MattermostUser.from_dict(result)
|
|
return {
|
|
"id": user.id,
|
|
"username": user.username,
|
|
"nickname": user.nickname,
|
|
"first_name": user.first_name,
|
|
"last_name": user.last_name,
|
|
"email": user.email,
|
|
"is_bot": user.is_bot,
|
|
}
|
|
except MattermostError:
|
|
return None
|
|
|
|
async def get_channel_info(self, channel_id: str) -> dict | None:
|
|
try:
|
|
result = await self.client.fetch_channel(channel_id)
|
|
ch = MattermostChannel.from_dict(result)
|
|
return {
|
|
"id": ch.id,
|
|
"name": ch.name,
|
|
"display_name": ch.display_name,
|
|
"type": ch.type,
|
|
"team_id": ch.team_id,
|
|
}
|
|
except MattermostError:
|
|
return None
|
|
|
|
async def search_channels(self, query: str) -> list[dict]:
|
|
try:
|
|
me = await self.client.fetch_me()
|
|
teams = await self.client.fetch_user_teams(me["id"])
|
|
results: list[dict] = []
|
|
for team_data in teams:
|
|
tid = team_data["id"]
|
|
try:
|
|
api_results = await self.client.search_channels(tid, query)
|
|
for item in api_results:
|
|
ch = MattermostChannel.from_dict(item)
|
|
results.append({
|
|
"id": ch.id,
|
|
"name": ch.name,
|
|
"display_name": ch.display_name,
|
|
"type": ch.type,
|
|
"team_id": ch.team_id,
|
|
})
|
|
except MattermostError:
|
|
continue
|
|
return results
|
|
except MattermostError:
|
|
pass
|
|
|
|
all_channels = await self.list_channels()
|
|
q = query.lower()
|
|
return [ch for ch in all_channels if q in ch.get("name", "").lower() or q in ch.get("display_name", "").lower()]
|
|
|
|
async def search_posts(
|
|
self,
|
|
team_id: str,
|
|
terms: str,
|
|
*,
|
|
page: int = 0,
|
|
per_page: int = 60,
|
|
) -> dict:
|
|
return await self.client.search_posts(team_id, terms, page=page, per_page=per_page)
|