Kris
8f352d91bb
1. 新增poll命令支持创建、关闭、列出投票 2. 新增审计日志记录功能 3. 优化远程附件URL安全校验逻辑 4. 修复表格匹配正则，支持包含竖线的表格分隔行 5. 新增媒体AI处理能力，支持图片描述和音频转录 6. 完善配置校验和错误处理 7. 重构文本发送逻辑，增加重试机制 8. 新增投票投票处理逻辑，支持数字快捷投票
274 lines
10 KiB
Python
274 lines
10 KiB
Python
from __future__ import annotations
|
||
|
||
from typing import Any
|
||
|
||
|
||
def describe_imessage_pairing_tool() -> dict:
|
||
return {
|
||
"type": "function",
|
||
"function": {
|
||
"name": "imessage_manage_pairing",
|
||
"description": "管理 iMessage 配对请求：列出待审批的用户、批准或拒绝配对",
|
||
"parameters": {
|
||
"type": "object",
|
||
"properties": {
|
||
"action": {
|
||
"type": "string",
|
||
"enum": ["list", "approve", "reject"],
|
||
"description": "操作类型：list=列出待审批, approve=批准配对, reject=拒绝配对",
|
||
},
|
||
"code": {
|
||
"type": "string",
|
||
"description": "配对码（approve/reject 操作需要）",
|
||
},
|
||
},
|
||
"required": ["action"],
|
||
},
|
||
},
|
||
}
|
||
|
||
|
||
def describe_imessage_allowlist_tool() -> dict:
|
||
return {
|
||
"type": "function",
|
||
"function": {
|
||
"name": "imessage_manage_allowlist",
|
||
"description": "管理 iMessage 白名单：添加或移除联系人/群组白名单",
|
||
"parameters": {
|
||
"type": "object",
|
||
"properties": {
|
||
"action": {
|
||
"type": "string",
|
||
"enum": ["add_dm", "remove_dm", "add_group", "remove_group", "list"],
|
||
"description": "操作类型",
|
||
},
|
||
"handle": {
|
||
"type": "string",
|
||
"description": "电话号码或 chat_guid",
|
||
},
|
||
},
|
||
"required": ["action"],
|
||
},
|
||
},
|
||
}
|
||
|
||
|
||
def describe_imessage_security_status_tool() -> dict:
|
||
return {
|
||
"type": "function",
|
||
"function": {
|
||
"name": "imessage_security_status",
|
||
"description": "查看 iMessage 渠道的安全状态和警告信息",
|
||
"parameters": {
|
||
"type": "object",
|
||
"properties": {},
|
||
"required": [],
|
||
},
|
||
},
|
||
}
|
||
|
||
|
||
def describe_imessage_poll_tool() -> dict:
|
||
return {
|
||
"type": "function",
|
||
"function": {
|
||
"name": "imessage_manage_polls",
|
||
"description": "管理 iMessage 投票：创建、关闭、查看投票",
|
||
"parameters": {
|
||
"type": "object",
|
||
"properties": {
|
||
"action": {
|
||
"type": "string",
|
||
"enum": ["create", "close", "list", "get"],
|
||
"description": "操作类型",
|
||
},
|
||
"chat_guid": {
|
||
"type": "string",
|
||
"description": "聊天 ID（创建/列出时需要）",
|
||
},
|
||
"question": {
|
||
"type": "string",
|
||
"description": "投票问题（创建时需要）",
|
||
},
|
||
"options": {
|
||
"type": "array",
|
||
"items": {"type": "string"},
|
||
"description": "选项列表（创建时需要）",
|
||
},
|
||
"poll_id": {
|
||
"type": "string",
|
||
"description": "投票 ID（关闭/查看时需要）",
|
||
},
|
||
"duration_minutes": {
|
||
"type": "integer",
|
||
"description": "投票持续时间（分钟），默认 5 分钟",
|
||
},
|
||
},
|
||
"required": ["action"],
|
||
},
|
||
},
|
||
}
|
||
|
||
|
||
def describe_imessage_audit_tool() -> dict:
|
||
return {
|
||
"type": "function",
|
||
"function": {
|
||
"name": "imessage_audit_log",
|
||
"description": "查询 iMessage 渠道审计日志",
|
||
"parameters": {
|
||
"type": "object",
|
||
"properties": {
|
||
"limit": {
|
||
"type": "integer",
|
||
"description": "返回记录数上限，默认 20",
|
||
},
|
||
},
|
||
"required": [],
|
||
},
|
||
},
|
||
}
|
||
|
||
|
||
class IMessageAgentToolFactory:
|
||
def __init__(self, adapter: Any):
|
||
self._adapter = adapter
|
||
|
||
def list_tools(self) -> list[dict]:
|
||
return [
|
||
describe_imessage_pairing_tool(),
|
||
describe_imessage_allowlist_tool(),
|
||
describe_imessage_security_status_tool(),
|
||
describe_imessage_poll_tool(),
|
||
describe_imessage_audit_tool(),
|
||
]
|
||
|
||
async def execute_tool(self, tool_name: str, params: dict) -> Any:
|
||
if tool_name == "imessage_manage_pairing":
|
||
return await self._manage_pairing(params)
|
||
if tool_name == "imessage_manage_allowlist":
|
||
return await self._manage_allowlist(params)
|
||
if tool_name == "imessage_security_status":
|
||
return self._security_status()
|
||
if tool_name == "imessage_manage_polls":
|
||
return await self._manage_polls(params)
|
||
if tool_name == "imessage_audit_log":
|
||
return self._audit_log(params)
|
||
raise ValueError(f"Unknown tool: {tool_name}")
|
||
|
||
async def _manage_pairing(self, params: dict) -> dict:
|
||
action = params.get("action", "list")
|
||
|
||
if action == "list":
|
||
pending = self._adapter.list_pending_pairings()
|
||
paired = self._adapter._pairing.get_paired_handles()
|
||
return {"pending": pending, "paired": paired}
|
||
|
||
code = params.get("code", "")
|
||
if action == "approve":
|
||
ok = self._adapter.approve_pairing(code)
|
||
if ok:
|
||
await self._adapter._pairing.save_allowlist()
|
||
return {"success": ok, "message": "Pairing approved" if ok else "Invalid or expired code"}
|
||
|
||
if action == "reject":
|
||
ok = self._adapter.reject_pairing(code)
|
||
return {"success": ok, "message": "Pairing rejected" if ok else "Invalid or expired code"}
|
||
|
||
return {"success": False, "error": f"Unknown action: {action}"}
|
||
|
||
async def _manage_allowlist(self, params: dict) -> dict:
|
||
action = params.get("action", "list")
|
||
|
||
if action == "list":
|
||
return {
|
||
"dm_allowlist": self._adapter._security.allow_list,
|
||
"group_allowlist": self._adapter._security.group_allow_list,
|
||
}
|
||
|
||
handle = params.get("handle", "")
|
||
if not handle:
|
||
return {"success": False, "error": "handle is required"}
|
||
|
||
if action == "add_dm":
|
||
self._adapter._security.add_to_allow_list(handle)
|
||
return {"success": True, "message": f"Added {handle} to DM allowlist"}
|
||
|
||
if action == "remove_dm":
|
||
ok = self._adapter._security.remove_from_allow_list(handle)
|
||
return {"success": ok, "message": f"{'Removed' if ok else 'Not found'}: {handle}"}
|
||
|
||
if action == "add_group":
|
||
self._adapter._security.add_to_group_allow_list(handle)
|
||
return {"success": True, "message": f"Added {handle} to group allowlist"}
|
||
|
||
if action == "remove_group":
|
||
ok = self._adapter._security.remove_from_group_allow_list(handle)
|
||
return {"success": ok, "message": f"{'Removed' if ok else 'Not found'}: {handle}"}
|
||
|
||
return {"success": False, "error": f"Unknown action: {action}"}
|
||
|
||
def _security_status(self) -> dict:
|
||
warnings = self._adapter.get_security_warnings()
|
||
return {
|
||
"dm_policy": self._adapter._security.dm_policy.value,
|
||
"group_policy": self._adapter._security.group_policy.value,
|
||
"require_mention": self._adapter._security.require_mention,
|
||
"warnings": warnings,
|
||
"paired_count": len(self._adapter._pairing.get_paired_handles()),
|
||
}
|
||
|
||
async def _manage_polls(self, params: dict) -> dict:
|
||
action = params.get("action", "list")
|
||
|
||
if action == "list":
|
||
active = [
|
||
{
|
||
"poll_id": p.poll_id,
|
||
"chat_guid": p.chat_guid,
|
||
"question": p.question,
|
||
"closed": p.closed,
|
||
}
|
||
for p in self._adapter._poll_manager._polls.values()
|
||
if not p.closed
|
||
]
|
||
return {"active_polls": active}
|
||
|
||
if action == "create":
|
||
chat_guid = params.get("chat_guid", "")
|
||
question = params.get("question", "")
|
||
options = params.get("options", [])
|
||
duration_minutes = params.get("duration_minutes", 5)
|
||
if not chat_guid or not question or not options:
|
||
return {"success": False, "error": "chat_guid, question, and options are required"}
|
||
result = await self._adapter.create_poll(
|
||
chat_id=chat_guid,
|
||
question=question,
|
||
options=options,
|
||
duration_minutes=duration_minutes,
|
||
)
|
||
return {"success": True, "poll": result}
|
||
|
||
if action == "close":
|
||
poll_id = params.get("poll_id", "")
|
||
if not poll_id:
|
||
return {"success": False, "error": "poll_id is required"}
|
||
result = await self._adapter.close_poll("", poll_id)
|
||
return {"success": True, "result": result}
|
||
|
||
if action == "get":
|
||
poll_id = params.get("poll_id", "")
|
||
if not poll_id:
|
||
return {"success": False, "error": "poll_id is required"}
|
||
result = self._adapter.get_poll(poll_id)
|
||
if result is None:
|
||
return {"success": False, "error": f"Poll not found: {poll_id}"}
|
||
return {"success": True, "poll": result}
|
||
|
||
return {"success": False, "error": f"Unknown action: {action}"}
|
||
|
||
def _audit_log(self, params: dict) -> dict:
|
||
limit = params.get("limit", 20)
|
||
records = self._adapter._audit_logger.get_recent(limit=limit)
|
||
return {"records": records, "count": len(records)}
|