@douyinfe/semi-mcp 1.0.1 → 1.0.2

package/dist/index.js

@@ -2,9 +2,21 @@
 import { Server } from "@modelcontextprotocol/sdk/server/index.js";
 import { StdioServerTransport } from "@modelcontextprotocol/sdk/server/stdio.js";
 import { CallToolRequestSchema, ListResourcesRequestSchema, ListToolsRequestSchema, ReadResourceRequestSchema } from "@modelcontextprotocol/sdk/types.js";
+import { mkdir, writeFile } from "fs/promises";
+import { join } from "path";
+import { tmpdir } from "os";
 const UNPKG_BASE_URL = 'https://unpkg.com';
 const NPMMIRROR_BASE_URL = 'https://registry.npmmirror.com';
-async function fetchFromSource(baseUrl, packageName, version, path, isNpmMirror = false) {
+function flattenDirectoryStructure(item, result = []) {
+    result.push({
+        path: item.path,
+        type: item.type,
+        size: item.size
+    });
+    if (item.files && Array.isArray(item.files)) for (const file of item.files)flattenDirectoryStructure(file, result);
+    return result;
+}
+async function fetchDirectoryListFromSource(baseUrl, packageName, version, path, isNpmMirror = false) {
     const url = isNpmMirror ? `${baseUrl}/${packageName}/${version}/files/${path}/?meta` : `${baseUrl}/${packageName}@${version}/${path}/?meta`;
     const response = await fetch(url, {
         headers: {
@@ -35,15 +47,62 @@ async function fetchFromSource(baseUrl, packageName, version, path, isNpmMirror
         };
     };
     if (Array.isArray(data)) return data.map(normalizeType);
-    if (data && 'object' == typeof data && 'files' in data && Array.isArray(data.files)) return data.files.map(normalizeType);
-    if (data && 'object' == typeof data && 'path' in data) return [
-        normalizeType(data)
-    ];
+    if (data && 'object' == typeof data && 'files' in data) {
+        if (Array.isArray(data.files)) {
+            const flattened = [];
+            for (const item of data.files)flattenDirectoryStructure(item, flattened);
+            return flattened.map(normalizeType);
+        }
+        return [];
+    }
+    if (data && 'object' == typeof data && 'path' in data) {
+        const singleItem = data;
+        if (singleItem.files && Array.isArray(singleItem.files)) {
+            const flattened = [];
+            flattenDirectoryStructure(singleItem, flattened);
+            return flattened.map(normalizeType);
+        }
+        return [
+            normalizeType(singleItem)
+        ];
+    }
     throw new Error('无法解析目录列表数据格式');
 }
 async function fetchDirectoryList(packageName, version, path) {
-    const unpkgPromise = fetchFromSource(UNPKG_BASE_URL, packageName, version, path, false);
-    const npmmirrorPromise = fetchFromSource(NPMMIRROR_BASE_URL, packageName, version, path, true);
+    const unpkgPromise = fetchDirectoryListFromSource(UNPKG_BASE_URL, packageName, version, path, false);
+    const npmmirrorPromise = fetchDirectoryListFromSource(NPMMIRROR_BASE_URL, packageName, version, path, true);
+    const unpkgWithFallback = unpkgPromise.catch(()=>new Promise(()=>{}));
+    const npmmirrorWithFallback = npmmirrorPromise.catch(()=>new Promise(()=>{}));
+    const raceResult = await Promise.race([
+        unpkgWithFallback,
+        npmmirrorWithFallback
+    ]).catch(()=>null);
+    if (raceResult) return raceResult;
+    const results = await Promise.allSettled([
+        unpkgPromise,
+        npmmirrorPromise
+    ]);
+    const errors = [];
+    for (const result of results)if ('rejected' === result.status) errors.push(result.reason instanceof Error ? result.reason : new Error(String(result.reason)));
+    throw new Error(`所有数据源都失败了: ${errors.map((e)=>e.message).join('; ')}`);
+}
+const fetch_file_content_UNPKG_BASE_URL = 'https://unpkg.com';
+const fetch_file_content_NPMMIRROR_BASE_URL = 'https://registry.npmmirror.com';
+async function fetchFileContentFromSource(baseUrl, packageName, version, filePath, isNpmMirror = false) {
+    const url = isNpmMirror ? `${baseUrl}/${packageName}/${version}/files/${filePath}` : `${baseUrl}/${packageName}@${version}/${filePath}`;
+    const response = await fetch(url, {
+        headers: {
+            Accept: 'text/plain, application/json, */*'
+        }
+    });
+    if (!response.ok) throw new Error(`获取文件失败: ${response.status} ${response.statusText}`);
+    const content = await response.text();
+    if (content.trim().startsWith('<!DOCTYPE html>') || content.includes('npmmirror 镜像站')) throw new Error('返回了 HTML 错误页面');
+    return content;
+}
+async function fetchFileContent(packageName, version, filePath) {
+    const unpkgPromise = fetchFileContentFromSource(fetch_file_content_UNPKG_BASE_URL, packageName, version, filePath, false);
+    const npmmirrorPromise = fetchFileContentFromSource(fetch_file_content_NPMMIRROR_BASE_URL, packageName, version, filePath, true);
     const unpkgWithFallback = unpkgPromise.catch(()=>new Promise(()=>{}));
     const npmmirrorWithFallback = npmmirrorPromise.catch(()=>new Promise(()=>{}));
     const raceResult = await Promise.race([
@@ -91,6 +150,11 @@ const getSemiDocumentTool = {
             version: {
                 type: 'string',
                 description: '版本号，例如 2.89.1。如果不提供，默认使用 latest'
+            },
+            get_path: {
+                type: 'boolean',
+                description: '如果为 true，将文档写入操作系统临时目录并返回路径，而不是在响应中返回文档内容。默认为 false',
+                default: false
             }
         },
         required: []
@@ -117,18 +181,36 @@ async function getComponentDocuments(componentName, version) {
     }
     if (-1 === categoryIndex || categoryIndex >= pathParts.length) return null;
     const category = pathParts[categoryIndex];
-    const documents = componentFiles.map((file)=>{
+    const documentPromises = componentFiles.map(async (file)=>{
+        const filePath = file.path.startsWith('/') ? file.path.slice(1) : file.path;
         const parts = file.path.split('/');
-        return parts[parts.length - 1].toLowerCase();
-    }).filter((name)=>name);
+        const fileName = parts[parts.length - 1];
+        try {
+            const content = await fetchFileContent(packageName, version, filePath);
+            return {
+                name: fileName,
+                path: file.path,
+                content: content
+            };
+        } catch (error) {
+            const errorMessage = error instanceof Error ? error.message : String(error);
+            return {
+                name: fileName,
+                path: file.path,
+                content: `获取文档内容失败: ${errorMessage}`
+            };
+        }
+    });
+    const documents = await Promise.all(documentPromises);
     return {
         category,
-        documents: Array.from(new Set(documents)).sort()
+        documents: documents.sort((a, b)=>a.name.localeCompare(b.name))
     };
 }
 async function handleGetSemiDocument(args) {
     const componentName = args?.componentName;
     const version = args?.version || 'latest';
+    const getPath = args?.get_path || false;
     try {
         if (componentName) {
             const result = await getComponentDocuments(componentName, version);
@@ -149,6 +231,60 @@ async function handleGetSemiDocument(args) {
                     }
                 ]
             };
+            const documentsWithLines = result.documents.map((doc)=>({
+                    ...doc,
+                    lines: doc.content.split('\n').length
+                }));
+            const hasLargeDocument = documentsWithLines.some((doc)=>doc.lines > 888);
+            const userExplicitlySetGetPath = 'get_path' in args;
+            const shouldUsePath = getPath || hasLargeDocument && !userExplicitlySetGetPath;
+            if (shouldUsePath) {
+                const baseTempDir = tmpdir();
+                const tempDirName = `semi-docs-${componentName.toLowerCase()}-${version}-${Date.now()}`;
+                const tempDir = join(baseTempDir, tempDirName);
+                await mkdir(tempDir, {
+                    recursive: true
+                });
+                const filePaths = [];
+                for (const doc of result.documents){
+                    const filePath = join(tempDir, doc.name);
+                    await writeFile(filePath, doc.content, 'utf-8');
+                    filePaths.push(filePath);
+                }
+                const largeDocs = documentsWithLines.filter((doc)=>doc.lines > 888);
+                let message = `文档已保存到临时目录: ${tempDir}\n请使用文件读取工具查看文档内容。`;
+                if (hasLargeDocument && !userExplicitlySetGetPath) {
+                    const largeDocNames = largeDocs.map((doc)=>`${doc.name} (${doc.lines.toLocaleString()} 行)`).join(', ');
+                    message = `文档已保存到临时目录: ${tempDir}\n注意：以下文档文件较大，已自动保存到临时目录：${largeDocNames}\n请使用文件读取工具查看文档内容。`;
+                } else if (hasLargeDocument) {
+                    const largeDocNames = largeDocs.map((doc)=>`${doc.name} (${doc.lines.toLocaleString()} 行)`).join(', ');
+                    message = `文档已保存到临时目录: ${tempDir}\n注意：以下文档文件较大：${largeDocNames}\n请使用文件读取工具查看文档内容。`;
+                }
+                return {
+                    content: [
+                        {
+                            type: 'text',
+                            text: JSON.stringify({
+                                componentName: componentName.toLowerCase(),
+                                version,
+                                category: result.category,
+                                tempDirectory: tempDir,
+                                files: documentsWithLines.map((doc)=>({
+                                        name: doc.name,
+                                        path: join(tempDir, doc.name),
+                                        contentLength: doc.content.length,
+                                        lines: doc.lines
+                                    })),
+                                count: result.documents.length,
+                                message,
+                                autoGetPath: hasLargeDocument && !userExplicitlySetGetPath,
+                                allComponents,
+                                allComponentsCount: allComponents.length
+                            }, null, 2)
+                        }
+                    ]
+                };
+            }
             return {
                 content: [
                     {
@@ -157,7 +293,16 @@ async function handleGetSemiDocument(args) {
                             componentName: componentName.toLowerCase(),
                             version,
                             category: result.category,
-                            documents: result.documents,
+                            documents: result.documents.map((doc)=>({
+                                    name: doc.name,
+                                    path: doc.path,
+                                    contentLength: doc.content.length
+                                })),
+                            contents: result.documents.map((doc)=>({
+                                    name: doc.name,
+                                    path: doc.path,
+                                    content: doc.content
+                                })),
                             count: result.documents.length,
                             allComponents,
                             allComponentsCount: allComponents.length

package/dist/utils/fetch-directory-list.d.ts

@@ -2,6 +2,16 @@
  * 从 unpkg 或 npmmirror 获取目录列表
  * 同时向两个数据源发送请求，使用第一个成功返回的结果
  */
+export declare const UNPKG_BASE_URL = "https://unpkg.com";
+export declare const NPMMIRROR_BASE_URL = "https://registry.npmmirror.com";
+/**
+ * 从单个源获取目录列表
+ * 导出用于测试
+ */
+export declare function fetchDirectoryListFromSource(baseUrl: string, packageName: string, version: string, path: string, isNpmMirror?: boolean): Promise<Array<{
+    path: string;
+    type: string;
+}>>;
 /**
  * 从 unpkg 或 npmmirror 获取目录列表
  * 同时向两个数据源发送请求，使用第一个成功返回的结果

package/dist/utils/fetch-file-content.d.ts

@@ -2,6 +2,13 @@
  * 从 unpkg 或 npmmirror 获取具体文件内容
  * 同时向两个数据源发送请求，使用第一个成功返回的结果
  */
+export declare const UNPKG_BASE_URL = "https://unpkg.com";
+export declare const NPMMIRROR_BASE_URL = "https://registry.npmmirror.com";
+/**
+ * 从单个源获取文件内容
+ * 导出用于测试
+ */
+export declare function fetchFileContentFromSource(baseUrl: string, packageName: string, version: string, filePath: string, isNpmMirror?: boolean): Promise<string>;
 /**
  * 从 unpkg 或 npmmirror 获取具体文件内容
  * 同时向两个数据源发送请求，使用第一个成功返回的结果

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@douyinfe/semi-mcp",
-  "version": "1.0.1",
+  "version": "1.0.2",
   "description": "Semi Design MCP Server - Model Context Protocol server for Semi Design components and documentation",
   "type": "module",
   "main": "./dist/index.js",