@vibe-lark/larkpal 0.1.41 → 0.1.42

package/dist/mcp-server.mjs

@@ -0,0 +1,367 @@
+import { randomUUID } from "node:crypto";
+import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
+import { StdioServerTransport } from "@modelcontextprotocol/sdk/server/stdio.js";
+import { z } from "zod";
+//#region src/mcp-server/host-bridge.ts
+/**
+* Host Bridge — 与宿主层的 HTTP 通信桥
+*
+* 负责通过 HTTP 回调与 LarkPal 宿主层通信：
+* - 发送 tool 调用回调到宿主层
+* - 轮询等待宿主层 resolve 结果（用于阻塞式 tool）
+* - fire-and-forget 通知（用于非阻塞式 tool）
+*/
+/** 日志输出到 stderr（stdout 被 MCP 协议占用） */
+function log$2(level, message, meta) {
+	const entry = {
+		timestamp: (/* @__PURE__ */ new Date()).toISOString(),
+		level,
+		component: "mcp-server/host-bridge",
+		message,
+		...meta
+	};
+	process.stderr.write(JSON.stringify(entry) + "\n");
+}
+/** 轮询配置 */
+const POLL_INTERVAL_MS = 2e3;
+const POLL_TIMEOUT_MS = 1800 * 1e3;
+/** 获取宿主层基础 URL */
+function getGatewayBaseUrl() {
+	return `http://localhost:${process.env["LARKPAL_GATEWAY_PORT"] || "3000"}`;
+}
+/** 获取当前 session ID */
+function getSessionId() {
+	const sessionId = process.env["LARKPAL_MCP_SESSION_ID"];
+	if (!sessionId) throw new Error("环境变量 LARKPAL_MCP_SESSION_ID 未设置");
+	return sessionId;
+}
+/**
+* 发送回调请求到宿主层
+*
+* @param tool - tool 名称
+* @param params - tool 参数
+* @returns requestId 用于后续轮询
+*/
+async function sendCallback(tool, params) {
+	const requestId = randomUUID();
+	const sessionId = getSessionId();
+	const url = `${getGatewayBaseUrl()}/api/mcp/callback`;
+	const body = {
+		requestId,
+		tool,
+		params,
+		sessionId
+	};
+	log$2("info", `发送回调请求到宿主层`, {
+		url,
+		requestId,
+		tool,
+		sessionId
+	});
+	const response = await fetch(url, {
+		method: "POST",
+		headers: { "Content-Type": "application/json" },
+		body: JSON.stringify(body)
+	});
+	if (!response.ok) {
+		const text = await response.text().catch(() => "<读取失败>");
+		log$2("error", `回调请求失败`, {
+			requestId,
+			status: response.status,
+			responseBody: text
+		});
+		throw new Error(`宿主层回调请求失败: HTTP ${response.status} - ${text}`);
+	}
+	log$2("info", `回调请求成功`, {
+		requestId,
+		status: response.status
+	});
+	return requestId;
+}
+/**
+* 轮询等待宿主层 resolve 结果
+*
+* @param requestId - 请求 ID
+* @returns resolve 后的结果
+*/
+async function pollForResult(requestId) {
+	const url = `${getGatewayBaseUrl()}/api/mcp/resolve/${requestId}`;
+	const startTime = Date.now();
+	log$2("info", `开始轮询等待结果`, {
+		requestId,
+		url,
+		timeoutMs: POLL_TIMEOUT_MS
+	});
+	while (true) {
+		const elapsed = Date.now() - startTime;
+		if (elapsed >= POLL_TIMEOUT_MS) {
+			log$2("error", `轮询超时`, {
+				requestId,
+				elapsedMs: elapsed
+			});
+			throw new Error(`等待宿主层响应超时 (${POLL_TIMEOUT_MS / 1e3}s): requestId=${requestId}`);
+		}
+		try {
+			const response = await fetch(url, { method: "GET" });
+			if (!response.ok) {
+				const text = await response.text().catch(() => "<读取失败>");
+				log$2("warn", `轮询请求返回非 2xx`, {
+					requestId,
+					status: response.status,
+					responseBody: text
+				});
+				await sleep(POLL_INTERVAL_MS);
+				continue;
+			}
+			const data = await response.json();
+			if (data.status === "resolved") {
+				log$2("info", `轮询获得结果`, {
+					requestId,
+					status: data.status
+				});
+				return data.result;
+			}
+			if (data.status === "rejected") {
+				log$2("error", `请求被拒绝`, {
+					requestId,
+					error: data.error
+				});
+				throw new Error(`宿主层拒绝请求: ${data.error || "未知原因"}`);
+			}
+		} catch (err) {
+			if (err instanceof Error && err.message.startsWith("宿主层拒绝请求")) throw err;
+			log$2("warn", `轮询发生网络错误，将重试`, {
+				requestId,
+				error: err instanceof Error ? err.message : String(err)
+			});
+		}
+		await sleep(POLL_INTERVAL_MS);
+	}
+}
+/**
+* 发送 fire-and-forget 通知（不等待响应结果）
+*
+* @param tool - tool 名称
+* @param params - tool 参数
+*/
+async function sendNotification(tool, params) {
+	const requestId = randomUUID();
+	const sessionId = getSessionId();
+	const url = `${getGatewayBaseUrl()}/api/mcp/callback`;
+	const body = {
+		requestId,
+		tool,
+		params,
+		sessionId
+	};
+	log$2("info", `发送 fire-and-forget 通知`, {
+		url,
+		requestId,
+		tool,
+		sessionId
+	});
+	try {
+		log$2("info", `通知发送完成`, {
+			requestId,
+			status: (await fetch(url, {
+				method: "POST",
+				headers: { "Content-Type": "application/json" },
+				body: JSON.stringify(body)
+			})).status
+		});
+	} catch (err) {
+		log$2("warn", `通知发送失败（已忽略）`, {
+			requestId,
+			error: err instanceof Error ? err.message : String(err)
+		});
+	}
+}
+function sleep(ms) {
+	return new Promise((resolve) => setTimeout(resolve, ms));
+}
+const hostBridge = {
+	sendCallback,
+	pollForResult,
+	sendNotification
+};
+//#endregion
+//#region src/mcp-server/tools.ts
+/** 日志输出到 stderr */
+function log$1(level, message, meta) {
+	const entry = {
+		timestamp: (/* @__PURE__ */ new Date()).toISOString(),
+		level,
+		component: "mcp-server/tools",
+		message,
+		...meta
+	};
+	process.stderr.write(JSON.stringify(entry) + "\n");
+}
+/**
+* 注册所有 tool 到 MCP Server
+*/
+function registerTools(server) {
+	server.tool("ask_user", "向用户提问，等待用户回答。支持多阶段提问，每个阶段可包含选项或自由文本输入。", {
+		question: z.string().describe("主问题描述"),
+		phases: z.array(z.object({
+			id: z.string().describe("阶段唯一标识"),
+			title: z.string().describe("阶段标题"),
+			description: z.string().optional().describe("阶段详细描述"),
+			options: z.array(z.object({
+				label: z.string().describe("选项显示文本"),
+				value: z.string().describe("选项值")
+			})).optional().describe("可选项列表"),
+			allowFreeText: z.boolean().optional().describe("是否允许自由文本输入"),
+			required: z.boolean().optional().describe("是否必填")
+		})).describe("提问阶段列表")
+	}, async (params) => {
+		log$1("info", `ask_user 被调用`, {
+			question: params.question,
+			phaseCount: params.phases.length
+		});
+		try {
+			const toolParams = {
+				question: params.question,
+				phases: params.phases
+			};
+			const requestId = await hostBridge.sendCallback("ask_user", toolParams);
+			log$1("info", `等待用户回答`, { requestId });
+			const result = await hostBridge.pollForResult(requestId);
+			log$1("info", `用户已回答`, {
+				requestId,
+				answers: result
+			});
+			return { content: [{
+				type: "text",
+				text: JSON.stringify(result)
+			}] };
+		} catch (err) {
+			const errorMsg = err instanceof Error ? err.message : String(err);
+			log$1("error", `ask_user 执行失败`, { error: errorMsg });
+			return {
+				content: [{
+					type: "text",
+					text: JSON.stringify({ error: errorMsg })
+				}],
+				isError: true
+			};
+		}
+	});
+	server.tool("request_permission", "请求飞书 API 权限。发送权限申请后等待用户授权。", {
+		scopes: z.array(z.string()).describe("需要的权限列表（如 \"contact:user.base:readonly\"）"),
+		reason: z.string().optional().describe("申请权限的原因说明")
+	}, async (params) => {
+		log$1("info", `request_permission 被调用`, {
+			scopes: params.scopes,
+			reason: params.reason
+		});
+		try {
+			const toolParams = {
+				scopes: params.scopes,
+				reason: params.reason
+			};
+			const requestId = await hostBridge.sendCallback("request_permission", toolParams);
+			log$1("info", `等待权限审批`, { requestId });
+			const result = await hostBridge.pollForResult(requestId);
+			log$1("info", `权限审批完成`, {
+				requestId,
+				granted: result.granted
+			});
+			return { content: [{
+				type: "text",
+				text: JSON.stringify(result)
+			}] };
+		} catch (err) {
+			const errorMsg = err instanceof Error ? err.message : String(err);
+			log$1("error", `request_permission 执行失败`, { error: errorMsg });
+			return {
+				content: [{
+					type: "text",
+					text: JSON.stringify({ error: errorMsg })
+				}],
+				isError: true
+			};
+		}
+	});
+	server.tool("signal_no_reply", "标记本次不需要回复用户。仅用于通知宿主层，立即返回。", { reason: z.string().describe("不回复的原因（仅用于日志记录）") }, async (params) => {
+		log$1("info", `signal_no_reply 被调用`, { reason: params.reason });
+		hostBridge.sendNotification("signal_no_reply", { reason: params.reason }).catch((err) => {
+			log$1("warn", `signal_no_reply 通知发送失败（已忽略）`, { error: err instanceof Error ? err.message : String(err) });
+		});
+		return { content: [{
+			type: "text",
+			text: JSON.stringify({ acknowledged: true })
+		}] };
+	});
+}
+//#endregion
+//#region src/mcp-server/index.ts
+/**
+* MCP Server 入口 — stdio transport
+*
+* LarkPal MCP Server 通过 stdio 与 Claude Code 通信，
+* 提供 ask_user、request_permission、signal_no_reply 三个 tool。
+*
+* 启动方式：由 CC 通过 --mcp-config 参数 spawn 本进程。
+* 环境变量：
+*   - LARKPAL_GATEWAY_PORT: 宿主层 HTTP 端口（默认 3000）
+*   - LARKPAL_MCP_SESSION_ID: 当前 CC session 的 ID
+*/
+/** 日志输出到 stderr（stdout 被 MCP 协议占用） */
+function log(level, message, meta) {
+	const entry = {
+		timestamp: (/* @__PURE__ */ new Date()).toISOString(),
+		level,
+		component: "mcp-server",
+		message,
+		...meta
+	};
+	process.stderr.write(JSON.stringify(entry) + "\n");
+}
+async function main() {
+	log("info", "MCP Server 启动", {
+		sessionId: process.env["LARKPAL_MCP_SESSION_ID"] || "<未设置>",
+		gatewayPort: process.env["LARKPAL_GATEWAY_PORT"] || "3000",
+		pid: process.pid
+	});
+	const server = new McpServer({
+		name: "larkpal",
+		version: "1.0.0"
+	});
+	registerTools(server);
+	const transport = new StdioServerTransport();
+	await server.connect(transport);
+	log("info", "MCP Server 已连接，等待 tool 调用");
+	process.stdin.on("close", () => {
+		log("info", "stdin 已关闭，MCP Server 准备退出");
+		process.exit(0);
+	});
+	process.on("SIGTERM", () => {
+		log("info", "收到 SIGTERM，MCP Server 退出");
+		process.exit(0);
+	});
+	process.on("SIGINT", () => {
+		log("info", "收到 SIGINT，MCP Server 退出");
+		process.exit(0);
+	});
+	process.on("uncaughtException", (err) => {
+		log("error", "未捕获的异常", {
+			error: err.message,
+			stack: err.stack
+		});
+		process.exit(1);
+	});
+	process.on("unhandledRejection", (reason) => {
+		log("error", "未处理的 Promise rejection", { reason: reason instanceof Error ? reason.message : String(reason) });
+		process.exit(1);
+	});
+}
+main().catch((err) => {
+	log("error", "MCP Server 启动失败", {
+		error: err instanceof Error ? err.message : String(err),
+		stack: err instanceof Error ? err.stack : void 0
+	});
+	process.exit(1);
+});
+//#endregion
+export {};

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@vibe-lark/larkpal",
-  "version": "0.1.41",
+  "version": "0.1.42",
   "description": "LarkPal - Lark/Feishu bot service",
   "type": "module",
   "main": "./dist/main.mjs",
@@ -36,6 +36,7 @@
   },
   "dependencies": {
     "@larksuiteoapi/node-sdk": "^1.60.0",
+    "@modelcontextprotocol/sdk": "^1.29.0",
     "@sinclair/typebox": "0.34.48",
     "dotenv": "^17.4.2",
     "express": "^5.1.0",