dingtalk-wiki 1.1.4 → 1.1.6

package/index.js

@@ -19,7 +19,7 @@ const DINGTALK_API_V2 = 'https://api.dingtalk.com';
 const fs = require('fs');
 const path = require('path');
 const os = require('os');
-const CACHE_DIR = path.join(os.homedir(), '.cache', 'dingtalk-wiki-mcp');
+const CACHE_DIR = path.join(os.homedir(), '.cache', 'dingtalk-wiki');
 const UNIONID_CACHE_PATH = path.join(CACHE_DIR, 'unionid-cache.json');
 
 if (!fs.existsSync(CACHE_DIR)) {
@@ -286,7 +286,7 @@ class DingTalkClient {
     return this.operatorId;
   }
 
-  async docRequest(method, pathName, { operatorId = null, data = null } = {}) {
+  async docRequest(method, pathName, { operatorId = null, data = null, extraParams = {} } = {}) {
     const token = await this.getAccessToken();
     const resolvedOperatorId = await this.resolveOperatorId(operatorId);
     const url = `${DINGTALK_API_V2}${pathName}`;
@@ -300,7 +300,8 @@ class DingTalkClient {
           'Content-Type': 'application/json'
         },
         params: {
-          operatorId: resolvedOperatorId
+          operatorId: resolvedOperatorId,
+          ...extraParams
         },
         data
       });
@@ -417,6 +418,10 @@ server.setRequestHandler(ListToolsRequestSchema, async () => {
             workspace_id: {
               type: 'string',
               description: '知识库工作空间 ID'
+            },
+            operator_id: {
+              type: 'string',
+              description: '操作者 unionid（不传则使用默认用户）'
             }
           },
           required: ['workspace_id']
@@ -434,7 +439,11 @@ server.setRequestHandler(ListToolsRequestSchema, async () => {
             },
             parent_node_id: {
               type: 'string',
-              description: '父节点 ID（不传则获取根目录）'
+              description: '父节点 ID（不传则列出根目录节点）'
+            },
+            operator_id: {
+              type: 'string',
+              description: '操作者 unionid（不传则使用默认用户）'
             }
           },
           required: ['workspace_id']
@@ -456,9 +465,11 @@ server.setRequestHandler(ListToolsRequestSchema, async () => {
             },
             doc_type: {
               type: 'string',
-              description: '文档类型: DOC(文字), WORKBOOK(表格), MIND(脑图), FOLDER(文件夹)',
-              enum: ['DOC', 'WORKBOOK', 'MIND', 'FOLDER'],
-              default: 'DOC'
+              description: '文档类型，可选: DOC / WORKBOOK / MIND / FOLDER（默认 DOC）'
+            },
+            operator_id: {
+              type: 'string',
+              description: '操作者 unionid（不传则使用默认用户）'
             },
             parent_node_id: {
               type: 'string',
@@ -481,6 +492,10 @@ server.setRequestHandler(ListToolsRequestSchema, async () => {
             node_id: {
               type: 'string',
               description: '节点 ID'
+            },
+            operator_id: {
+              type: 'string',
+              description: '操作者 unionid（不传则使用默认用户）'
             }
           },
           required: ['node_id']
@@ -488,7 +503,7 @@ server.setRequestHandler(ListToolsRequestSchema, async () => {
       },
       {
         name: 'search_wiki',
-        description: '搜索知识库内容',
+        description: '搜索知识库中的文档和文件夹（遍历目录树按名称匹配）',
         inputSchema: {
           type: 'object',
           properties: {
@@ -498,7 +513,15 @@ server.setRequestHandler(ListToolsRequestSchema, async () => {
             },
             workspace_id: {
               type: 'string',
-              description: '指定知识库 ID（可选）'
+              description: '指定知识库 ID（可选，不传则搜索所有知识库）'
+            },
+            max_results: {
+              type: 'number',
+              description: '返回条数上限（默认 20，最大 50）'
+            },
+            operator_id: {
+              type: 'string',
+              description: '操作者 unionid（不传则使用默认用户）'
             }
           },
           required: ['keyword']
@@ -894,8 +917,10 @@ server.setRequestHandler(CallToolRequestSchema, async (request) => {
       }
 
       case 'get_wiki_workspace': {
-        const { workspace_id } = args;
-        // 通过列表获取详情
+        const { workspace_id, operator_id } = args;
+        if (operator_id) {
+          dingtalk.setOperatorId(operator_id);
+        }
         const result = await dingtalk.wikiRequest('workspaces');
         const workspaces = result.workspaces || [];
         const workspace = workspaces.find(ws => ws.workspaceId === workspace_id);
@@ -1066,24 +1091,122 @@ server.setRequestHandler(CallToolRequestSchema, async (request) => {
       }
 
       case 'get_wiki_node': {
-        const { node_id } = args;
-        // 通过搜索或其他方式获取节点详情
+        const { node_id, operator_id } = args;
+        if (operator_id) {
+          dingtalk.setOperatorId(operator_id);
+        }
+        const result = await dingtalk.docRequest('GET', `/v2.0/wiki/nodes/${node_id}`, {
+          operatorId: operator_id || null
+        });
+        const node = result;
+        let output = `📄 节点详情\n\n`;
+        output += `名称: ${node.name || '-'}\n`;
+        output += `ID: ${node.id || node.nodeId || '-'}\n`;
+        output += `类型: ${node.type || '-'}\n`;
+        output += `知识库 ID: ${node.workspaceId || '-'}\n`;
+        output += `父节点: ${node.parentNodeId || '-'}\n`;
+        output += `创建时间: ${node.createdTime || '-'}\n`;
+        output += `修改时间: ${node.modifiedTime || '-'}\n`;
+        output += `URL: ${node.url || '-'}\n`;
+        if (node.document) {
+          output += `文档信息: ${JSON.stringify(node.document)}\n`;
+        }
         return {
-          content: [{
-            type: 'text',
-            text: `📄 节点 ID: ${node_id}\n\n请使用 list_wiki_nodes 获取节点列表，然后通过节点链接访问详情。`
-          }]
+          content: [{ type: 'text', text: output }]
         };
       }
 
       case 'search_wiki': {
-        const { keyword, workspace_id } = args;
-        // Wiki 搜索 API 需要额外权限
+        const { keyword, workspace_id, max_results = 20, operator_id } = args;
+        if (!keyword) {
+          return {
+            content: [{ type: 'text', text: '⚠️ 请提供搜索关键词 keyword' }],
+            isError: true
+          };
+        }
+
+        const MAX_RESULTS = Math.min(max_results, 50);
+        if (operator_id) {
+          dingtalk.setOperatorId(operator_id);
+        }
+
+        const wsResult = await dingtalk.wikiRequest('workspaces');
+        let workspaces = wsResult.workspaces || [];
+        if (workspace_id) {
+          workspaces = workspaces.filter(ws => ws.workspaceId === workspace_id);
+        }
+
+        if (workspaces.length === 0) {
+          return {
+            content: [{ type: 'text', text: '⚠️ 未找到可搜索的知识库' }],
+            isError: true
+          };
+        }
+
+        const matchedNodes = [];
+
+        for (const ws of workspaces) {
+          if (matchedNodes.length >= MAX_RESULTS) break;
+
+          const queue = [ws.rootNodeId];
+          const visited = new Set();
+
+          while (queue.length > 0 && matchedNodes.length < MAX_RESULTS) {
+            const dentryId = queue.shift();
+            if (!dentryId || visited.has(dentryId)) continue;
+            visited.add(dentryId);
+
+            try {
+              const result = await dingtalk.docRequest('GET', `/v2.0/doc/spaces/${ws.workspaceId}/directories`, {
+                operatorId: operator_id || null,
+                extraParams: dentryId !== ws.rootNodeId ? { dentryId, maxResults: 500 } : { maxResults: 500 }
+              });
+
+              const children = result.children || [];
+              for (const child of children) {
+                const name = child.name || '';
+                if (name.includes(keyword)) {
+                  matchedNodes.push({
+                    name,
+                    nodeId: child.dentryId || child.nodeId || child.id,
+                    workspaceId: ws.workspaceId,
+                    workspaceName: ws.name,
+                    type: child.contentType === 'folder' ? '文件夹' : '文档',
+                    url: child.url || '',
+                    hasChildren: child.hasChildren
+                  });
+                }
+                if (child.hasChildren && matchedNodes.length < MAX_RESULTS) {
+                  const childId = child.dentryId || child.nodeId || child.id;
+                  if (childId && !visited.has(childId)) {
+                    queue.push(childId);
+                  }
+                }
+              }
+            } catch (e) {
+              // 单个节点遍历失败跳过，不中断整体搜索
+            }
+          }
+        }
+
+        let output = `🔍 搜索 "${keyword}" (${matchedNodes.length}条)\n\n`;
+        matchedNodes.slice(0, MAX_RESULTS).forEach((item, i) => {
+          const icon = item.type === '文件夹' ? '📁' : '📄';
+          output += `${i + 1}. ${icon} ${item.name}\n`;
+          output += `   知识库: ${item.workspaceName} (${item.workspaceId})\n`;
+          if (item.url) output += `   链接: ${item.url}\n`;
+          output += '\n';
+        });
+
+        if (matchedNodes.length === 0) {
+          output += '没有找到匹配的文档或文件夹。\n';
+          output += `\n💡 此方式通过遍历目录树按名称匹配，非全文搜索。`;
+          output += `如需全文搜索，请在钉钉客户端中操作：\n`;
+          output += `https://alidocs.dingtalk.com/i/search?keyword=${encodeURIComponent(keyword)}`;
+        }
+
         return {
-          content: [{
-            type: 'text',
-            text: `🔍 搜索知识库: ${keyword}\n\n搜索功能需要 Wiki.Search 权限。\n\n请直接访问知识库网页版进行搜索：\nhttps://alidocs.dingtalk.com/i/spaces/${workspace_id || ''}/search?keyword=${encodeURIComponent(keyword)}`
-          }]
+          content: [{ type: 'text', text: output }]
         };
       }
 
@@ -1165,20 +1288,12 @@ server.setRequestHandler(CallToolRequestSchema, async (request) => {
         if (operator_id) {
           dingtalk.setOperatorId(operator_id);
         }
-        const opId = await dingtalk.resolveOperatorId(operator_id || null);
         await dingtalk.docRequest('POST', `/v1.0/doc/suites/documents/${docKey}/overwriteContent`, {
-          operatorId: opId,
-          data: {
-            operatorId: opId,
-            content,
-            contentType: 'markdown'
-          }
+          operatorId: operator_id || null,
+          data: { content, contentType: 'markdown' }
         });
         return {
-          content: [{
-            type: 'text',
-            text: '✅ 文档内容已更新'
-          }]
+          content: [{ type: 'text', text: '✅ 文档内容已更新' }]
         };
       }
 
@@ -1187,19 +1302,12 @@ server.setRequestHandler(CallToolRequestSchema, async (request) => {
         if (operator_id) {
           dingtalk.setOperatorId(operator_id);
         }
-        const opId = await dingtalk.resolveOperatorId(operator_id || null);
         await dingtalk.docRequest('PATCH', `/v1.0/doc/workspaces/${workspace_id}/docs/${node_id}`, {
-          operatorId: opId,
-          data: {
-            name,
-            operatorId: opId
-          }
+          operatorId: operator_id || null,
+          data: { name }
         });
         return {
-          content: [{
-            type: 'text',
-            text: `✅ 文档已重命名为: ${name}`
-          }]
+          content: [{ type: 'text', text: `✅ 文档已重命名为: ${name}` }]
         };
       }
 
@@ -1208,9 +1316,8 @@ server.setRequestHandler(CallToolRequestSchema, async (request) => {
         if (operator_id) {
           dingtalk.setOperatorId(operator_id);
         }
-        const opId = await dingtalk.resolveOperatorId(operator_id || null);
         await dingtalk.docRequest('DELETE', `/v1.0/doc/workspaces/${workspace_id}/docs/${node_id}`, {
-          operatorId: opId
+          operatorId: operator_id || null
         });
         return {
           content: [{
@@ -1362,7 +1469,7 @@ async function main() {
   const transport = new StdioServerTransport();
   await server.connect(transport);
   console.error('钉钉 Wiki MCP Server 已启动 v2.0');
-  console.error(`Config path: ${CONFIG_PATH}`);
+  console.error(`Config via DINGTALK_WIKI_CONFIG env var`);
 }
 
 main().catch(console.error);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "dingtalk-wiki",
-  "version": "1.1.4",
+  "version": "1.1.6",
   "description": "DingTalk Wiki / Docs read-write MCP server that fills the gap left by DingTalk official MCP.",
   "main": "index.js",
   "bin": {

package/skill/SKILL.md

@@ -100,11 +100,17 @@ mcporter call dingtalk-wiki.delete_wiki_doc \
   workspace_id="your_workspace_id" \
   node_id="your_node_id"
 
-# 搜索知识库内容
+# 搜索知识库
 mcporter call dingtalk-wiki.search_wiki keyword="项目规划"
 
 # 在指定知识库内搜索
 mcporter call dingtalk-wiki.search_wiki keyword="项目规划" workspace_id="your_workspace_id"
+
+# 自定义返回条数
+mcporter call dingtalk-wiki.search_wiki keyword="项目规划" max_results=5
+
+# 分页搜索
+mcporter call dingtalk-wiki.search_wiki keyword="项目规划" next_token="your_next_token"
 ```
 
 ### AI 表格（Notable）