@wings006/agent-link 0.1.3 → 0.1.5

package/dist/mcp-server.js

@@ -2,12 +2,19 @@ import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
 import { StdioServerTransport } from "@modelcontextprotocol/sdk/server/stdio.js";
 import { z } from "zod";
 import http from "node:http";
+import { readFileSync } from "node:fs";
+import { fileURLToPath } from "node:url";
+import path from "node:path";
 import { sendNotification } from "./notify.js";
+import { loadConfig, saveConfig, CONFIG_PATH } from "./config.js";
+const __dirname = path.dirname(fileURLToPath(import.meta.url));
+const pkg = JSON.parse(readFileSync(path.join(__dirname, "..", "package.json"), "utf-8"));
 export class AgentLinkMcpServer {
     mcp;
     daemonUrl;
     agentName;
     pendingTasks = new Map();
+    hasShownPollTip = false;
     constructor(agentName, daemonApiPort) {
         this.agentName = agentName;
         this.daemonUrl = `http://127.0.0.1:${daemonApiPort}`;
@@ -15,6 +22,7 @@ export class AgentLinkMcpServer {
             name: "agent-link",
             version: "0.1.0",
         });
+        this.registerPrompts();
         this.registerTools();
     }
     async daemonFetch(path, options) {
@@ -25,9 +33,486 @@ export class AgentLinkMcpServer {
         }
         return res;
     }
+    registerPrompts() {
+        this.mcp.prompt("check-messages", "检查收件箱，播报新消息并询问是否回复", async () => {
+            try {
+                const res = await this.daemonFetch("/api/tasks/inbox?unread_only=true");
+                const tasks = (await res.json());
+                if (tasks.length === 0) {
+                    return { messages: [{ role: "user", content: { type: "text", text: "[Agent Link] 没有新消息。" } }] };
+                }
+                const summary = tasks.map((t) => `- 来自 ${t.from}（${t.createdAt}）: ${t.message}（ID: ${t.id}）`).join("\n");
+                return {
+                    messages: [{
+                            role: "user",
+                            content: {
+                                type: "text",
+                                text: `[Agent Link] 收到 ${tasks.length} 条未读消息:\n\n${summary}\n\n请简洁地播报这些消息给用户，并询问是否需要回复。如需回复，使用 agent_reply 工具。`,
+                            },
+                        }],
+                };
+            }
+            catch {
+                return { messages: [{ role: "user", content: { type: "text", text: "[Agent Link] 无法连接 daemon，跳过本次检查。" } }] };
+            }
+        });
+    }
+    async checkMuted(type, name) {
+        try {
+            const res = await this.daemonFetch(`/api/mute/check?type=${encodeURIComponent(type)}&name=${encodeURIComponent(name)}`);
+            const data = (await res.json());
+            return data.muted;
+        }
+        catch {
+            return false;
+        }
+    }
     registerTools() {
+        // 发送好友请求
+        this.mcp.tool("agent_friend_request", "向目标 Agent 发送好友请求", {
+            to: z.string().describe("目标 Agent 名称"),
+            message: z.string().optional().describe("附言"),
+        }, async ({ to, message }) => {
+            try {
+                const res = await this.daemonFetch("/api/friends/request", {
+                    method: "POST", headers: { "Content-Type": "application/json" },
+                    body: JSON.stringify({ to, message }),
+                });
+                const data = (await res.json());
+                return { content: [{ type: "text", text: `好友请求已发送给 ${to}（请求ID: ${data.requestId}）` }] };
+            }
+            catch (err) {
+                return { content: [{ type: "text", text: `发送失败: ${err instanceof Error ? err.message : String(err)}` }], isError: true };
+            }
+        });
+        // 查看好友请求
+        this.mcp.tool("agent_friend_requests", "查看收到的好友请求", {}, async () => {
+            try {
+                const res = await this.daemonFetch("/api/friends/requests");
+                const reqs = (await res.json());
+                return {
+                    content: [{
+                            type: "text",
+                            text: reqs.length === 0 ? "没有待处理的好友请求" : JSON.stringify(reqs, null, 2),
+                        }],
+                };
+            }
+            catch (err) {
+                return { content: [{ type: "text", text: `查询失败: ${err instanceof Error ? err.message : String(err)}` }], isError: true };
+            }
+        });
+        // 回复好友请求
+        this.mcp.tool("agent_friend_respond", "接受或拒绝好友请求", {
+            request_id: z.string().describe("好友请求 ID"),
+            accept: z.boolean().describe("true=接受, false=拒绝"),
+        }, async ({ request_id, accept }) => {
+            try {
+                await this.daemonFetch("/api/friends/respond", {
+                    method: "POST", headers: { "Content-Type": "application/json" },
+                    body: JSON.stringify({ request_id, accept }),
+                });
+                return { content: [{ type: "text", text: accept ? "已接受好友请求" : "已拒绝好友请求" }] };
+            }
+            catch (err) {
+                return { content: [{ type: "text", text: `操作失败: ${err instanceof Error ? err.message : String(err)}` }], isError: true };
+            }
+        });
+        // 查看好友列表
+        this.mcp.tool("agent_friends", "查看好友列表（含在线状态）", {
+            category: z.string().optional().describe("按分组筛选"),
+        }, async ({ category }) => {
+            try {
+                const url = category ? `/api/friends?category=${encodeURIComponent(category)}` : "/api/friends";
+                const res = await this.daemonFetch(url);
+                const friends = (await res.json());
+                return {
+                    content: [{
+                            type: "text",
+                            text: friends.length === 0 ? "好友列表为空" : JSON.stringify(friends, null, 2),
+                        }],
+                };
+            }
+            catch (err) {
+                return { content: [{ type: "text", text: `查询失败: ${err instanceof Error ? err.message : String(err)}` }], isError: true };
+            }
+        });
+        // 删除好友
+        this.mcp.tool("agent_friend_remove", "删除好友", {
+            name: z.string().describe("好友名称"),
+        }, async ({ name }) => {
+            try {
+                await this.daemonFetch("/api/friends/remove", {
+                    method: "POST", headers: { "Content-Type": "application/json" },
+                    body: JSON.stringify({ name }),
+                });
+                return { content: [{ type: "text", text: `已删除好友 ${name}` }] };
+            }
+            catch (err) {
+                return { content: [{ type: "text", text: `操作失败: ${err instanceof Error ? err.message : String(err)}` }], isError: true };
+            }
+        });
+        // 设置好友分组
+        this.mcp.tool("agent_friend_category", "设置好友分组", {
+            friend: z.string().describe("好友名称"),
+            category: z.string().describe("分组名称"),
+        }, async ({ friend, category }) => {
+            try {
+                await this.daemonFetch("/api/friends/category", {
+                    method: "POST", headers: { "Content-Type": "application/json" },
+                    body: JSON.stringify({ friend, category }),
+                });
+                return { content: [{ type: "text", text: `已将 ${friend} 移至分组「${category}」` }] };
+            }
+            catch (err) {
+                return { content: [{ type: "text", text: `操作失败: ${err instanceof Error ? err.message : String(err)}` }], isError: true };
+            }
+        });
+        // 静音/取消静音
+        this.mcp.tool("agent_mute", "静音指定 Agent 或群组，静音后不弹出系统通知", {
+            type: z.enum(["agent", "group"]).describe("类型：agent 或 group"),
+            name: z.string().describe("Agent 名称或群组 ID"),
+        }, async ({ type, name }) => {
+            try {
+                await this.daemonFetch("/api/mute", {
+                    method: "POST",
+                    headers: { "Content-Type": "application/json" },
+                    body: JSON.stringify({ type, name }),
+                });
+                return { content: [{ type: "text", text: `已静音 ${type === "agent" ? "Agent" : "群组"} "${name}"` }] };
+            }
+            catch (err) {
+                return { content: [{ type: "text", text: `操作失败: ${err instanceof Error ? err.message : String(err)}` }], isError: true };
+            }
+        });
+        this.mcp.tool("agent_unmute", "取消静音指定 Agent 或群组", {
+            type: z.enum(["agent", "group"]).describe("类型：agent 或 group"),
+            name: z.string().describe("Agent 名称或群组 ID"),
+        }, async ({ type, name }) => {
+            try {
+                await this.daemonFetch("/api/unmute", {
+                    method: "POST",
+                    headers: { "Content-Type": "application/json" },
+                    body: JSON.stringify({ type, name }),
+                });
+                return { content: [{ type: "text", text: `已取消静音 ${type === "agent" ? "Agent" : "群组"} "${name}"` }] };
+            }
+            catch (err) {
+                return { content: [{ type: "text", text: `操作失败: ${err instanceof Error ? err.message : String(err)}` }], isError: true };
+            }
+        });
+        // 群组功能
+        this.mcp.tool("agent_group_create", "创建群组", {
+            name: z.string().describe("群组名称"),
+        }, async ({ name }) => {
+            try {
+                const res = await this.daemonFetch("/api/groups", {
+                    method: "POST", headers: { "Content-Type": "application/json" },
+                    body: JSON.stringify({ name }),
+                });
+                const data = (await res.json());
+                return { content: [{ type: "text", text: `群组「${name}」已创建（ID: ${data.id}）` }] };
+            }
+            catch (err) {
+                return { content: [{ type: "text", text: `创建失败: ${err instanceof Error ? err.message : String(err)}` }], isError: true };
+            }
+        });
+        this.mcp.tool("agent_group_list", "获取我加入的群组列表", {}, async () => {
+            try {
+                const res = await this.daemonFetch("/api/groups");
+                const groups = (await res.json());
+                return {
+                    content: [{
+                            type: "text",
+                            text: groups.length === 0 ? "没有加入任何群组" : JSON.stringify(groups, null, 2),
+                        }],
+                };
+            }
+            catch (err) {
+                return { content: [{ type: "text", text: `查询失败: ${err instanceof Error ? err.message : String(err)}` }], isError: true };
+            }
+        });
+        this.mcp.tool("agent_group_members", "获取群组成员列表", {
+            group_id: z.string().describe("群组 ID"),
+        }, async ({ group_id }) => {
+            try {
+                const res = await this.daemonFetch(`/api/groups/${group_id}/members`);
+                const members = (await res.json());
+                return {
+                    content: [{
+                            type: "text",
+                            text: members.length === 0 ? "群组无成员" : JSON.stringify(members, null, 2),
+                        }],
+                };
+            }
+            catch (err) {
+                return { content: [{ type: "text", text: `查询失败: ${err instanceof Error ? err.message : String(err)}` }], isError: true };
+            }
+        });
+        this.mcp.tool("agent_group_invite", "邀请 Agent 加入群组", {
+            group_id: z.string().describe("群组 ID"),
+            to: z.string().describe("目标 Agent 名称"),
+        }, async ({ group_id, to }) => {
+            try {
+                const res = await this.daemonFetch("/api/groups/invite", {
+                    method: "POST", headers: { "Content-Type": "application/json" },
+                    body: JSON.stringify({ groupId: group_id, to }),
+                });
+                const data = (await res.json());
+                if (data.error) {
+                    return { content: [{ type: "text", text: data.error }], isError: true };
+                }
+                return { content: [{ type: "text", text: `已邀请 ${to} 加入群组（邀请ID: ${data.id}）` }] };
+            }
+            catch (err) {
+                return { content: [{ type: "text", text: `邀请失败: ${err instanceof Error ? err.message : String(err)}` }], isError: true };
+            }
+        });
+        this.mcp.tool("agent_group_invites", "查看收到的群组邀请", {}, async () => {
+            try {
+                const res = await this.daemonFetch("/api/groups/invites");
+                const invites = (await res.json());
+                return {
+                    content: [{
+                            type: "text",
+                            text: invites.length === 0 ? "没有待处理的群组邀请" : JSON.stringify(invites, null, 2),
+                        }],
+                };
+            }
+            catch (err) {
+                return { content: [{ type: "text", text: `查询失败: ${err instanceof Error ? err.message : String(err)}` }], isError: true };
+            }
+        });
+        this.mcp.tool("agent_group_invite_respond", "接受或拒绝群组邀请", {
+            invite_id: z.string().describe("邀请 ID"),
+            accept: z.boolean().describe("true=接受, false=拒绝"),
+        }, async ({ invite_id, accept }) => {
+            try {
+                await this.daemonFetch("/api/groups/invites/respond", {
+                    method: "POST", headers: { "Content-Type": "application/json" },
+                    body: JSON.stringify({ inviteId: invite_id, accept }),
+                });
+                return { content: [{ type: "text", text: accept ? "已接受群组邀请" : "已拒绝群组邀请" }] };
+            }
+            catch (err) {
+                return { content: [{ type: "text", text: `操作失败: ${err instanceof Error ? err.message : String(err)}` }], isError: true };
+            }
+        });
+        this.mcp.tool("agent_group_message", "发送群组消息", {
+            group_id: z.string().describe("群组 ID"),
+            message: z.string().describe("消息内容"),
+        }, async ({ group_id, message }) => {
+            try {
+                const res = await this.daemonFetch("/api/groups/message", {
+                    method: "POST", headers: { "Content-Type": "application/json" },
+                    body: JSON.stringify({ groupId: group_id, message }),
+                });
+                const data = (await res.json());
+                if (data.error) {
+                    return { content: [{ type: "text", text: data.error }], isError: true };
+                }
+                return { content: [{ type: "text", text: `消息已发送到群组（消息ID: ${data.id}）` }] };
+            }
+            catch (err) {
+                return { content: [{ type: "text", text: `发送失败: ${err instanceof Error ? err.message : String(err)}` }], isError: true };
+            }
+        });
+        this.mcp.tool("agent_group_messages", "获取群组消息列表", {
+            group_id: z.string().describe("群组 ID"),
+            limit: z.number().optional().describe("获取消息数量（默认 50）"),
+        }, async ({ group_id, limit }) => {
+            try {
+                const url = `/api/groups/${group_id}/messages${limit ? `?limit=${limit}` : ""}`;
+                const res = await this.daemonFetch(url);
+                const messages = (await res.json());
+                return {
+                    content: [{
+                            type: "text",
+                            text: messages.length === 0 ? "群组暂无消息" : JSON.stringify(messages, null, 2),
+                        }],
+                };
+            }
+            catch (err) {
+                return { content: [{ type: "text", text: `查询失败: ${err instanceof Error ? err.message : String(err)}` }], isError: true };
+            }
+        });
+        this.mcp.tool("agent_group_mark_read", "标记群组消息已读", {
+            group_id: z.string().describe("群组 ID"),
+        }, async ({ group_id }) => {
+            try {
+                await this.daemonFetch(`/api/groups/${group_id}/read`, { method: "POST" });
+                return { content: [{ type: "text", text: `群组 ${group_id} 的消息已标记为已读` }] };
+            }
+            catch (err) {
+                return { content: [{ type: "text", text: `操作失败: ${err instanceof Error ? err.message : String(err)}` }], isError: true };
+            }
+        });
+        this.mcp.tool("agent_group_leave", "离开群组", {
+            group_id: z.string().describe("群组 ID"),
+        }, async ({ group_id }) => {
+            try {
+                const res = await this.daemonFetch(`/api/groups/${group_id}/leave`, { method: "POST" });
+                const data = (await res.json());
+                return { content: [{ type: "text", text: data.ok ? `已离开群组 ${group_id}` : `不在该群组中` }] };
+            }
+            catch (err) {
+                return { content: [{ type: "text", text: `操作失败: ${err instanceof Error ? err.message : String(err)}` }], isError: true };
+            }
+        });
+        // 设置在线状态
+        this.mcp.tool("agent_set_status", "设置本 Agent 的在线状态。offline 时其他人无法发送消息给你。", {
+            status: z.enum(["online", "offline", "busy"]).describe("在线状态"),
+        }, async ({ status }) => {
+            try {
+                await this.daemonFetch("/api/status", {
+                    method: "POST",
+                    headers: { "Content-Type": "application/json" },
+                    body: JSON.stringify({ status }),
+                });
+                const desc = { online: "在线", offline: "离线", busy: "忙碌" };
+                return {
+                    content: [
+                        { type: "text", text: `状态已设为「${desc[status]}」` },
+                    ],
+                };
+            }
+            catch (err) {
+                return {
+                    content: [
+                        { type: "text", text: `设置失败: ${err instanceof Error ? err.message : String(err)}` },
+                    ],
+                    isError: true,
+                };
+            }
+        });
+        // 配置管理
+        this.mcp.tool("agent_config", "查看或修改 agent-link 配置（名称、中继服务器等）。修改后需重启 Claude Code 生效。", {
+            name: z.string().optional().describe("设置 Agent 名称"),
+            relay_url: z.string().optional().describe("设置中继服务器地址，如 ws://IP:3458。设为空字符串则清除"),
+        }, async ({ name, relay_url }) => {
+            // 无参数时显示当前配置和引导
+            if (name === undefined && relay_url === undefined) {
+                const fileConfig = loadConfig();
+                let health = null;
+                try {
+                    const res = await this.daemonFetch("/api/health");
+                    health = (await res.json());
+                }
+                catch { }
+                const activeName = health?.name || fileConfig.name || "(未知)";
+                const activeRelay = health?.relayUrl || "(未配置)";
+                const lines = [
+                    `== 当前生效配置 ==`,
+                    `  名称: ${activeName}`,
+                    `  中继: ${activeRelay}`,
+                    `  本机发现: ${health?.noLocal ? "禁用" : "启用"}`,
+                    `  mDNS: ${health?.noMdns ? "禁用" : "启用"}`,
+                    ``,
+                    `== 配置文件 (${CONFIG_PATH}) ==`,
+                    `  名称: ${fileConfig.name || "(未设置)"}`,
+                    `  中继: ${fileConfig.relayUrl || "(未设置)"}`,
+                    ``,
+                    `== 配置方式（任选其一） ==`,
+                    `方式1: 通过本工具修改配置文件`,
+                    `  agent_config(name: "你的名称")`,
+                    `  agent_config(relay_url: "ws://中继服务器IP:3458")`,
+                    ``,
+                    `方式2: 通过 MCP 启动参数（优先级更高）`,
+                    `  "args": ["--name", "名称", "--relay", "ws://IP:3458"]`,
+                    ``,
+                    `配置优先级: MCP 启动参数 > 配置文件 > 环境变量 > 默认值`,
+                    `修改后需重启 Claude Code 生效。`,
+                ];
+                return { content: [{ type: "text", text: lines.join("\n") }] };
+            }
+            // 修改配置
+            const updates = {};
+            if (name !== undefined)
+                updates.name = name;
+            if (relay_url !== undefined)
+                updates.relayUrl = relay_url || undefined;
+            const saved = saveConfig(updates);
+            const changes = [];
+            if (name !== undefined)
+                changes.push(`名称 → ${name}`);
+            if (relay_url !== undefined)
+                changes.push(relay_url ? `中继 → ${relay_url}` : `中继 → 已清除`);
+            return {
+                content: [{
+                        type: "text",
+                        text: `配置已保存到 ${CONFIG_PATH}\n${changes.join("\n")}\n\n⚠ 需要重启 Claude Code 使配置生效。`,
+                    }],
+            };
+        });
+        // 状态总览
+        this.mcp.tool("agent_status", "查看当前状态总览：Daemon 信息、在线 Agent、好友、群组等", {}, async () => {
+            try {
+                const [healthRes, agentsRes, friendsRes, groupsRes, muteRes] = await Promise.all([
+                    this.daemonFetch("/api/health"),
+                    this.daemonFetch("/api/agents"),
+                    this.daemonFetch("/api/friends"),
+                    this.daemonFetch("/api/groups"),
+                    this.daemonFetch("/api/mute"),
+                ]);
+                const health = (await healthRes.json());
+                const agents = (await agentsRes.json());
+                const friends = (await friendsRes.json());
+                const groups = (await groupsRes.json());
+                const muted = (await muteRes.json());
+                const uptimeMin = Math.floor(health.uptime / 60000);
+                const uptimeStr = uptimeMin < 60
+                    ? `${uptimeMin} 分钟`
+                    : `${Math.floor(uptimeMin / 60)} 小时 ${uptimeMin % 60} 分钟`;
+                const discoveryModes = [
+                    !health.noLocal ? "本机" : null,
+                    !health.noMdns ? "mDNS" : null,
+                    health.relayUrl ? `中继(${health.relayUrl})` : null,
+                ].filter(Boolean).join("、") || "无";
+                const lines = [
+                    `== Daemon ==`,
+                    `  名称: ${health.name}`,
+                    `  版本: ${pkg.version}`,
+                    `  PID: ${health.pid}`,
+                    `  运行时间: ${uptimeStr}`,
+                    `  A2A 端口: ${health.port} | API 端口: ${health.apiPort}`,
+                    `  发现模式: ${discoveryModes}`,
+                    ``,
+                    `== 在线 Agent (${agents.length}) ==`,
+                    ...(agents.length === 0
+                        ? ["  (无)"]
+                        : agents.map((a) => `  ${a.name} — ${a.status}${a.skills?.length ? ` [${a.skills.join(",")}]` : ""}`)),
+                    ``,
+                    `== 好友 (${friends.length}) ==`,
+                    ...(friends.length === 0
+                        ? ["  (无)"]
+                        : friends.map((f) => `  ${f.friendName} — ${f.status}${f.category !== "default" ? ` (${f.category})` : ""}`)),
+                    ``,
+                    `== 群组 (${groups.length}) ==`,
+                    ...(groups.length === 0
+                        ? ["  (无)"]
+                        : groups.map((g) => `  ${g.name} (${g.id.slice(0, 8)}...) — 创建者: ${g.creator}`)),
+                ];
+                if (muted.length > 0) {
+                    lines.push(``, `== 静音 (${muted.length}) ==`);
+                    for (const m of muted) {
+                        lines.push(`  ${m.targetType}: ${m.targetName}`);
+                    }
+                }
+                lines.push(``, `== 提示 ==`);
+                if (!health.relayUrl) {
+                    lines.push(`  ⚠ 未配置中继服务器，仅限局域网/本机通信。使用 agent_config 配置中继以支持跨网络通信。`);
+                }
+                lines.push(`  开启自动查收: /loop 1m 使用 agent_inbox(unread_only=true) 检查未读消息，有新消息则播报并询问是否回复，没有则不输出`);
+                return { content: [{ type: "text", text: lines.join("\n") }] };
+            }
+            catch (err) {
+                return {
+                    content: [{ type: "text", text: `无法连接 daemon: ${err instanceof Error ? err.message : String(err)}` }],
+                    isError: true,
+                };
+            }
+        });
         // 列出在线 Agent
-        this.mcp.tool("agent_list", "列出局域网内所有在线的 Agent", {}, async () => {
+        this.mcp.tool("agent_list", "列出所有在线的 Agent", {}, async () => {
             try {
                 const res = await this.daemonFetch("/api/agents");
                 const agents = (await res.json());
@@ -317,11 +802,16 @@ export class AgentLinkMcpServer {
             reconnect(5000);
         });
     }
-    handleSSEEvent(event) {
+    async handleSSEEvent(event) {
         const task = event.task;
         if (event.type === "task:created" && task.to === this.agentName) {
-            // 收到新消息通知
-            sendNotification(`Agent Link: 来自 ${task.from}`, task.message);
+            // 收到新消息，检查是否静音
+            const muted = await this.checkMuted("agent", task.from);
+            if (!muted) {
+                const tip = this.hasShownPollTip ? "" : "\n\n提示: 执行 /loop 1m /agent-link:check-messages 开启自动查收";
+                sendNotification(`Agent Link: 来自 ${task.from}`, task.message + tip);
+                this.hasShownPollTip = true;
+            }
             return;
         }
         if (event.type === "task:updated") {
@@ -329,10 +819,13 @@ export class AgentLinkMcpServer {
             const pending = this.pendingTasks.get(task.id);
             if (pending && (task.status === "completed" || task.status === "rejected" || task.status === "canceled")) {
                 pending.task = task;
-                const title = task.status === "completed"
-                    ? `${pending.to} 回复了`
-                    : `${pending.to} ${task.status === "rejected" ? "拒绝了任务" : "取消了任务"}`;
-                sendNotification(title, task.result || "(无内容)");
+                const muted = await this.checkMuted("agent", pending.to);
+                if (!muted) {
+                    const title = task.status === "completed"
+                        ? `${pending.to} 回复了`
+                        : `${pending.to} ${task.status === "rejected" ? "拒绝了任务" : "取消了任务"}`;
+                    sendNotification(title, task.result || "(无内容)");
+                }
             }
         }
     }