ForcePilot/backend/package/yuxi/channels/adapters/synologychat/probe.py
Kris 3ad218e537 feat(synologychat): 新增群晖Chat适配器完整实现
新增群晖Chat渠道适配器的全套实现,包括:
1. 基础适配器与导出接口定义
2. DSM API认证、探测与会话管理
3. 轮询与Webhook两种消息接收方式
4. 消息去重、格式化与规范化处理
5. 多账号支持与权限安全策略
6. 目录用户/群组发现功能
7. 审批配对与流量控制机制
8. 安全审计与配置检查功能
2026-05-12 00:49:16 +08:00

69 lines
2.0 KiB
Python

"""DSM API probe and version resolution.
Discovers available APIs on the target Synology NAS and resolves the best API version to use.
"""
from __future__ import annotations
from typing import Any
import httpx
from yuxi.utils.logging_config import logger
async def probe_dsm(
http_client: httpx.AsyncClient,
base_url: str,
) -> dict[str, Any]:
try:
response = await http_client.get(
f"{base_url}/webapi/query.cgi",
params={
"api": "SYNO.API.Info",
"version": "1",
"method": "query",
"query": "SYNO.API.Auth,SYNO.Chat.External",
},
)
response.raise_for_status()
data = response.json()
if not data.get("success"):
error_code = data.get("error", {}).get("code", "unknown")
logger.error(f"DSM API probe failed: error code {error_code}")
return {}
api_info = data.get("data", {})
required_apis = ["SYNO.API.Auth", "SYNO.Chat.External"]
missing = [api for api in required_apis if api not in api_info]
if missing:
logger.error(f"Required DSM APIs not available: {missing}")
return {}
logger.info(
f"DSM API probe successful: Auth v{api_info.get('SYNO.API.Auth', {}).get('maxVersion', '?')}, "
f"Chat v{api_info.get('SYNO.Chat.External', {}).get('maxVersion', '?')}"
)
return api_info
except httpx.HTTPError as e:
logger.error(f"DSM API probe HTTP error: {e}")
return {}
except Exception as e:
logger.error(f"DSM API probe failed: {e}")
return {}
def resolve_api_version(api_info: dict[str, Any] | None, api: str) -> int:
if not api_info:
if api == "SYNO.API.Auth":
return 6
if api == "SYNO.Chat.External":
return 1
return 1
api_entry = api_info.get(api, {})
return api_entry.get("maxVersion", api_entry.get("minVersion", 1))