@gecho-ai/gecho-bridge 1.1.2 → 1.1.3

package/mcp-client.js

@@ -110,8 +110,8 @@ server.setRequestHandler(ListToolsRequestSchema, async () => {
   return {
     tools: [
       {
-        name: "tiktok_search_top_200",
-        description: "在 TikTok 上搜索关键词，自动滚动加载至少 200 条结果，并全部返回。",
+        name: "tiktok_search",
+        description: "在 TikTok 上搜索关键词，自动滚动加载结果并返回。",
         inputSchema: {
           type: "object",
           properties: {
@@ -120,6 +120,18 @@ server.setRequestHandler(ListToolsRequestSchema, async () => {
           },
           required: ["query"]
         }
+      },
+      {
+        name: "tiktok_insight",
+        description: "在 TikTok 搜索的基础上进行商机洞察和趋势分析。",
+        inputSchema: {
+          type: "object",
+          properties: {
+            query: { type: "string", description: "搜索关键词 (例如: '户外野餐垫')" },
+            save_dir: { type: "string", description: "可选的保存目录绝对路径 (例如: '/Users/xxx/data')" }
+          },
+          required: ["query"]
+        }
       }
     ]
   };
@@ -127,9 +139,11 @@ server.setRequestHandler(ListToolsRequestSchema, async () => {
 
 // 2. 转发工具请求到 Service 层
 server.setRequestHandler(CallToolRequestSchema, async (request) => {
-  if (request.params.name === "tiktok_search_top_200") {
-    const { query, save_dir } = request.params.arguments;
+  const toolName = request.params.name;
+  const args = request.params.arguments;
 
+  // 只要是 tiktok_ 开头的工具，都走通用转发逻辑
+  if (toolName.startsWith("tiktok_") || toolName.startsWith("x_") || toolName.startsWith("ins_")) {
     try {
       const requestService = () => new Promise((resolve, reject) => {
         const req = http.request(HTTP_SERVICE_URL, {
@@ -154,7 +168,12 @@ server.setRequestHandler(CallToolRequestSchema, async (request) => {
         });
 
         req.on("error", () => reject(new Error("Service Layer communication error")));
-        req.write(JSON.stringify({ query, save_dir }));
+        
+        // 【核心改动】：直接将工具名作为 action，所有参数作为 payload 透传
+        req.write(JSON.stringify({ 
+          action: toolName, 
+          ...args 
+        }));
         req.end();
       });
 
@@ -177,32 +196,33 @@ server.setRequestHandler(CallToolRequestSchema, async (request) => {
 
       const result = serviceResponse.data;
       if (typeof result === 'object' && result !== null && result.error) {
-        return { content: [{ type: "text", text: `❌ 抓取错误: ${result.error}` }], isError: true };
+        return { content: [{ type: "text", text: `❌ 业务错误: ${result.error}` }], isError: true };
       }
 
       if (!Array.isArray(result)) {
-        return { content: [{ type: "text", text: `❌ 异常: 服务端未返回数组格式的数据` }], isError: true };
+        return { content: [{ type: "text", text: `❌ 异常: 插件未返回数组格式的结果` }], isError: true };
       }
 
       const savePath = serviceResponse.savePath || "";
       const saveLine = savePath
-        ? `📂 完整结果已保存到: ${savePath}\n\n`
-        : `📂 未返回保存路径，请检查服务端日志。\n\n`;
+        ? `📂 数据已存: ${savePath}\n\n`
+        : "";
+      
       const top20 = result.slice(0, 20);
       return {
         content: [
           { 
             type: "text", 
-            text: `✅ 抓取完成！共获取 ${result.length} 条数据。\n` +
+            text: `✅ [${toolName}] 执行成功，共 ${result.length} 条数据。\n` +
                   saveLine +
-                  `以下是点赞最高的前 20 条结果：\n` +
+                  `以下是部分结果展示：\n` +
                   JSON.stringify(top20, null, 2) 
           }
         ]
       };
     } catch (e) {
       return { 
-        content: [{ type: "text", text: `❌ 转发失败: ${e.message}.` }], 
+        content: [{ type: "text", text: `❌ 链路故障: ${e.message}.` }], 
         isError: true 
       };
     }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@gecho-ai/gecho-bridge",
-  "version": "1.1.2",
+  "version": "1.1.3",
   "description": "MCP Bridge for TikTok Extension Relay - Connecting LLMs to TikTok via Chrome Extension",
   "main": "mcp-client.js",
   "bin": {

package/server.js

@@ -149,7 +149,7 @@ const server = http.createServer(async (req, res) => {
     return;
   }
 
-  if (req.method === "POST" && req.url === "/search") {
+  if (req.method === "POST" && (req.url === "/search" || req.url === "/action")) {
     if (shuttingDown) {
       res.statusCode = 503;
       return res.end(JSON.stringify({ error: "Service is shutting down" }));
@@ -160,10 +160,11 @@ const server = http.createServer(async (req, res) => {
     req.on("end", async () => {
       try {
         const payload = JSON.parse(body);
-        const query = normalizeQuery(payload.query, payload.query_b64);
-        if (!query) {
+        const action = payload.action;
+        
+        if (!action) {
           res.statusCode = 400;
-          return res.end(JSON.stringify({ error: "Missing query" }));
+          return res.end(JSON.stringify({ error: "Missing action" }));
         }
 
         if (!extensionSocket || extensionSocket.readyState !== 1) {
@@ -171,20 +172,31 @@ const server = http.createServer(async (req, res) => {
           return res.end(JSON.stringify({ error: "Extension not connected" }));
         }
 
-        console.log(`🔍 Processing search request for: [${query}]`);
+        console.log(`🚀 Dispatching action: [${action}]`);
         const requestId = `svc-${Date.now()}-${requestIdCounter++}`;
         
         const result = await new Promise((resolve) => {
           const timeoutId = setTimeout(() => {
             pendingRequests.delete(requestId);
-            resolve({ error: "Scraping timeout (300s)" });
+            resolve({ error: `Scraping timeout (300s) for action: ${action}` });
           }, 300000);
 
           pendingRequests.set(requestId, { resolve, timeoutId });
 
+          // 通用透传逻辑：将 payload 中的所有参数（除去 action）作为 params 传给插件
+          const { action: _a, ...params } = payload;
+          
+          // 🐷 兼容层：如果插件版本较旧，识别不了 tiktok_search，则映射回 search
+          let finalAction = action;
+          if (action === "tiktok_search") finalAction = "search";
+          if (action === "tiktok_insight") finalAction = "search";
+          
           extensionSocket.send(JSON.stringify({
             method: "execute_action",
-            params: { action: "search", params: { query } },
+            params: { 
+              action: finalAction, 
+              params: params 
+            },
             requestId: requestId
           }));
         });
@@ -193,11 +205,10 @@ const server = http.createServer(async (req, res) => {
         let savePath = "";
         let saveWarning = "";
         if (Array.isArray(result) && result.length > 0) {
-          // 优先级: 1. 请求中传来的 save_dir 2. 环境变量 GECHO_DATA_DIR 3. 默认 data 目录
-          const dataDir = payload.save_dir || process.env.GECHO_DATA_DIR || path.join(__dirname, "..", "data");
+          const dataDir = payload.save_dir || process.env.GECHO_DATA_DIR || path.join(__dirname, "data");
           if (!fs.existsSync(dataDir)) fs.mkdirSync(dataDir, { recursive: true });
-          const safeName = toSafeFileName(query);
-          const fixedPath = path.join(dataDir, `${safeName}_search_results.json`);
+          const safeName = toSafeFileName(params.query || action);
+          const fixedPath = path.join(dataDir, `${safeName}_results.json`);
           try {
             fs.writeFileSync(fixedPath, JSON.stringify(result, null, 2), "utf8");
             savePath = fixedPath;