sloth-d2c-mcp 1.0.4-beta86 → 1.0.4-beta89

package/dist/build/server.js

@@ -32,10 +32,11 @@ const upload = multer({
 });
 // 导入默认提示词
 import { chunkOptimizeCodePrompt, aggregationOptimizeCodePrompt, finalOptimizeCodePrompt, chunkOptimizeCodePromptVue, aggregationOptimizeCodePromptVue, finalOptimizeCodePromptVue, noSamplingAggregationPrompt, noSamplingAggregationPromptVue, } from 'sloth-d2c-node/convert';
+import { getParam } from './utils/utils.js';
 // 保存 HTTP 服务器实例
-let httpServer = null;
+export let httpServer = null;
 // 保存 Socket 服务器实例
-let socketServer = null;
+export let socketServer = null;
 // 管理所有活跃的传输对象，按 sessionId 分类
 const transports = {
     streamable: {}, // 流式 HTTP 传输
@@ -196,6 +197,13 @@ export async function startHttpServer(port = PORT, mcpServer, configManagerInsta
         next();
     });
     // app.use(express.json());
+    app.use((req, _res, next) => {
+        // 通过referer取url上面的uuid
+        const uuid = getParam(req.headers.referer || '', 'token') || '';
+        req.uuid = uuid;
+        req.fileManager = fileManager.withUUID(uuid);
+        next();
+    });
     // 处理流式 HTTP 的 POST 请求，支持会话复用和初始化
     app.post('/mcp', async (req, res) => {
         Logger.log('Received StreamableHTTP request');
@@ -333,11 +341,11 @@ export async function startHttpServer(port = PORT, mcpServer, configManagerInsta
                 // 如果提供了 fileKey，返回该 fileKey 的特定配置
                 const globalConfig = await configManager.load();
                 // const fileConfig = globalConfig.fileConfigs?.[fileKey] || {}
-                const fileConfig = (await fileManager.loadConfigSetting(fileKey, nodeId)) || {};
+                const fileConfig = (await req.fileManager.loadConfigSetting(fileKey, nodeId)) || {};
                 // 从 fileManager 按 nodeId 加载 groupsData 和 promptSetting
                 // const fileManager = new FileManager('d2c-mcp')
-                const groupsData = await fileManager.loadGroupsData(fileKey, nodeId);
-                const savedPromptSetting = await fileManager.loadPromptSetting(fileKey, nodeId);
+                const groupsData = await req.fileManager.loadGroupsData(fileKey, nodeId);
+                const savedPromptSetting = await req.fileManager.loadPromptSetting(fileKey, nodeId);
                 let curFramework = fileConfig?.convertSetting?.framework;
                 // 获取框架列表
                 const frameworks = await configManager.getFrameworks();
@@ -681,11 +689,11 @@ export async function startHttpServer(port = PORT, mcpServer, configManagerInsta
             // 使用 fileManager 按 nodeId 保存 groupsData 和 promptSetting
             // const fileManager = new FileManager('d2c-mcp')
             if (fileKey && value.groupsData && Array.isArray(value.groupsData)) {
-                await fileManager.saveGroupsData(fileKey, nodeId, value.groupsData);
+                await req.fileManager.saveGroupsData(fileKey, nodeId, value.groupsData);
                 Logger.log(`已保存 groupsData 到 fileKey "${fileKey}", nodeId "${nodeId}":`, value.groupsData.length, '个分组');
             }
             if (fileKey && value.promptSetting) {
-                await fileManager.savePromptSetting(fileKey, nodeId, value.promptSetting);
+                await req.fileManager.savePromptSetting(fileKey, nodeId, value.promptSetting);
                 Logger.log(`已保存 promptSetting 到 fileKey "${fileKey}", nodeId "${nodeId}"`);
             }
             // 如果有 MCP 配置，更新全局配置
@@ -766,9 +774,9 @@ export async function startHttpServer(port = PORT, mcpServer, configManagerInsta
     /**
      * 获取项目根目录
      */
-    function getProjectRoot() {
+    function getProjectRoot(fs) {
         try {
-            const projectPath = fileManager.getWorkspaceRoot();
+            const projectPath = fs.getWorkspaceRoot();
             return projectPath || './';
         }
         catch (error) {
@@ -782,7 +790,7 @@ export async function startHttpServer(port = PORT, mcpServer, configManagerInsta
     app.get('/getProjectFiles', async (req, res) => {
         try {
             const { directory = '' } = req.query;
-            const projectPath = getProjectRoot();
+            const projectPath = getProjectRoot(req.fileManager);
             const targetPath = path.join(projectPath, directory);
             Logger.log(`获取项目文件树: ${targetPath}`);
             // 排除的文件扩展名（静态资源和配置文件）
@@ -912,12 +920,12 @@ export async function startHttpServer(port = PORT, mcpServer, configManagerInsta
                 return;
             }
             Logger.log(`批量分析 ${filePaths.length} 个项目文件`);
-            const projectPath = getProjectRoot();
+            const projectPath = getProjectRoot(req.fileManager);
             // 尝试加载保存的提示词设置
             let savedPromptSetting = null;
             if (fileKey && nodeId) {
                 try {
-                    savedPromptSetting = await fileManager.loadPromptSetting(fileKey, nodeId);
+                    savedPromptSetting = await req.fileManager.loadPromptSetting(fileKey, nodeId);
                     Logger.log(`已加载提示词设置: fileKey=${fileKey}, nodeId=${nodeId}`);
                 }
                 catch (error) {
@@ -1109,7 +1117,7 @@ export async function startHttpServer(port = PORT, mcpServer, configManagerInsta
             }
             Logger.log(`准备保存 ${components.length} 个组件`);
             // 读取现有组件
-            const existingComponents = await fileManager.loadComponentsDatabase();
+            const existingComponents = await req.fileManager.loadComponentsDatabase();
             const existingMap = new Map(existingComponents.map((c) => [c.id, c]));
             let addedCount = 0;
             let updatedCount = 0;
@@ -1128,7 +1136,7 @@ export async function startHttpServer(port = PORT, mcpServer, configManagerInsta
             }
             const allComponents = Array.from(existingMap.values());
             // 保存到文件（带备份）
-            await fileManager.saveComponentsDatabase(allComponents);
+            await req.fileManager.saveComponentsDatabase(allComponents);
             Logger.log(`✅ 成功保存：新增 ${addedCount} 个，更新 ${updatedCount} 个，共 ${allComponents.length} 个组件`);
             res.json({
                 success: true,
@@ -1154,7 +1162,7 @@ export async function startHttpServer(port = PORT, mcpServer, configManagerInsta
     app.get('/scanComponents', async (req, res) => {
         try {
             const { includePaths, excludePaths } = req.query;
-            const projectPath = getProjectRoot();
+            const projectPath = getProjectRoot(req.fileManager);
             // 读取项目组件（从 .sloth/components.json）
             let projectComponents = [];
             const slothPath = path.join(projectPath, '.sloth', 'components.json');
@@ -1230,7 +1238,7 @@ export async function startHttpServer(port = PORT, mcpServer, configManagerInsta
             const currentScreenshotPath = path.join(tempDir, `suggest_${Date.now()}.png`);
             await fs.promises.writeFile(currentScreenshotPath, currentScreenshotFile.buffer);
             // 从 components.json 加载所有组件
-            const components = await fileManager.loadComponentsDatabase();
+            const components = await req.fileManager.loadComponentsDatabase();
             if (!components || components.length === 0) {
                 Logger.log('未找到已保存的组件，无法生成建议');
                 await fs.promises.unlink(currentScreenshotPath).catch(() => { });
@@ -1250,7 +1258,7 @@ export async function startHttpServer(port = PORT, mcpServer, configManagerInsta
                     continue;
                 }
                 // 根据 signature 搜索截图文件
-                const screenshotPath = await fileManager.findScreenshotByHash(component.signature);
+                const screenshotPath = await req.fileManager.findScreenshotByHash(component.signature);
                 if (!screenshotPath) {
                     Logger.log(`组件 ${component.name} 的截图未找到 (hash: ${component.signature})，跳过`);
                     continue;
@@ -1334,7 +1342,7 @@ export async function startHttpServer(port = PORT, mcpServer, configManagerInsta
             }
             Logger.log(`接收到截图上传请求: fileKey=${fileKey}, nodeId=${nodeId}, hash=${hash}`);
             // 保存截图到 .sloth/{fileKey}/{nodeId}/screenshots/{hash}.png
-            await fileManager.saveScreenshot(fileKey, nodeId, hash, file.buffer);
+            await req.fileManager.saveScreenshot(fileKey, nodeId, hash, file.buffer);
             Logger.log(`截图上传成功: ${hash}.png`);
             res.json({
                 success: true,
@@ -1355,7 +1363,7 @@ export async function startHttpServer(port = PORT, mcpServer, configManagerInsta
     app.get('/listDesignSnapshots', async (req, res) => {
         try {
             Logger.log('获取设计稿快照列表');
-            const workspaceRoot = getProjectRoot();
+            const workspaceRoot = getProjectRoot(req.fileManager);
             Logger.log('workspaceRoot:', workspaceRoot);
             if (!workspaceRoot) {
                 res.json({
@@ -1550,8 +1558,8 @@ export async function startHttpServer(port = PORT, mcpServer, configManagerInsta
                 return;
             }
             // 2. 加载新旧 groupsData
-            const oldGroupsData = (await fileManager.loadGroupsData(oldFileKey, oldNodeId)) || [];
-            const newGroupsData = (await fileManager.loadGroupsData(newFileKey, newNodeId)) || [];
+            const oldGroupsData = (await req.fileManager.loadGroupsData(oldFileKey, oldNodeId)) || [];
+            const newGroupsData = (await req.fileManager.loadGroupsData(newFileKey, newNodeId)) || [];
             // 3. 执行 HTML 差异分析
             // @ts-ignore
             const { diffLines } = await import('diff');
@@ -1770,8 +1778,9 @@ export async function getUserInput(payload) {
                 // 连接到 Socket 服务器
                 await socketClient.connect();
                 Logger.log('Socket 客户端已连接');
-                // 注册 token 并等待响应
-                const responsePromise = socketClient.registerToken(token);
+                const workspaceRoot = fileManager.getWorkspaceRoot();
+                // 注册 token 并等待响应，同时传递 extra 数据给主进程
+                const responsePromise = socketClient.registerToken(token, { workspaceRoot });
                 // 打开浏览器
                 await open(authUrl);
                 // 等待认证响应

package/dist/build/socket-client.js

@@ -115,8 +115,10 @@ export class SocketClient {
     }
     /**
      * 注册 token，等待认证响应
+     * @param token 认证 token
+     * @param extra 额外数据（可选），如 { workspaceRoot: string, ... }
      */
-    registerToken(token) {
+    registerToken(token, extra) {
         return new Promise((resolve, reject) => {
             // 设置超时（可选，根据需求调整）
             const timeout = setTimeout(() => {
@@ -132,9 +134,10 @@ export class SocketClient {
             this.send({
                 type: 'register-token',
                 token,
+                extra,
                 timestamp: Date.now(),
             });
-            Logger.log(`已注册 token: ${token}，等待认证响应...`);
+            Logger.log(`已注册 token: ${token}，extra: ${JSON.stringify(extra)}，等待认证响应...`);
         });
     }
     /**

package/dist/build/socket-server.js

@@ -8,6 +8,7 @@ export class SocketServer {
     server = null;
     connections = new Set();
     tokenSockets = new Map(); // token -> socket 映射
+    tokenExtras = new Map(); // token -> extra 数据映射
     messageBuffers = new Map(); // socket -> 消息缓冲区
     port = 0;
     constructor() {
@@ -42,6 +43,7 @@ export class SocketServer {
             for (const [token, sock] of this.tokenSockets.entries()) {
                 if (sock === socket) {
                     this.tokenSockets.delete(token);
+                    this.tokenExtras.delete(token);
                     Logger.log(`清理 token: ${token}`);
                 }
             }
@@ -95,6 +97,11 @@ export class SocketServer {
                 // 注册 token，建立 token -> socket 映射
                 if (message.token) {
                     this.tokenSockets.set(message.token, socket);
+                    // 保存 extra 数据（如果提供）
+                    if (message.extra && typeof message.extra === 'object') {
+                        this.tokenExtras.set(message.token, message.extra);
+                        Logger.log(`已保存 extra 数据: ${message.token} -> ${JSON.stringify(message.extra)}`);
+                    }
                     Logger.log(`已注册 token: ${message.token} -> ${clientId}`);
                     this.sendMessage(socket, {
                         type: 'token-registered',
@@ -204,6 +211,7 @@ export class SocketServer {
             }
             this.connections.clear();
             this.tokenSockets.clear();
+            this.tokenExtras.clear();
             this.messageBuffers.clear();
             // 关闭服务器
             if (this.server) {
@@ -230,4 +238,23 @@ export class SocketServer {
     getPort() {
         return this.port;
     }
+    /**
+     * 根据 token 获取对应的 extra 数据
+     */
+    getTokenExtra(token) {
+        return this.tokenExtras.get(token);
+    }
+    /**
+     * 根据 token 获取 extra 中的特定字段
+     */
+    getTokenExtraField(token, field) {
+        const extra = this.tokenExtras.get(token);
+        return extra?.[field];
+    }
+    /**
+     * 获取所有已注册的 token extra 映射
+     */
+    getAllTokenExtras() {
+        return new Map(this.tokenExtras);
+    }
 }

package/dist/build/utils/file-manager.js

@@ -2,6 +2,7 @@ import { promises as fs } from 'fs';
 import * as path from 'path';
 import envPaths from 'env-paths';
 import { Logger } from '../utils/logger.js';
+import { httpServer, socketServer } from '../server.js';
 /**
  * 通用文件管理器
  * 可以保存任何类型的文件，使用fileKey和nodeId组织目录结构
@@ -9,11 +10,23 @@ import { Logger } from '../utils/logger.js';
 export class FileManager {
     paths; // 应用路径配置
     baseDir; // 基础存储目录
-    workspaceRoot = null; // MCP 工作目录根路径
+    _workspaceRoot = null; // MCP 工作目录根路径
+    _uuid = '';
     constructor(appName) {
         this.paths = envPaths(appName);
         this.baseDir = path.join(this.paths.data, 'files');
     }
+    get workspaceRoot() {
+        const isMainProcess = httpServer !== null;
+        if (isMainProcess && this._uuid) {
+            const workspaceRoot = socketServer?.getTokenExtraField(this._uuid, 'workspaceRoot');
+            return workspaceRoot || this._workspaceRoot;
+        }
+        return this._workspaceRoot;
+    }
+    set workspaceRoot(rootPath) {
+        this._workspaceRoot = rootPath;
+    }
     /**
      * 设置工作目录根路径（用于 MCP 工作项目）
      * @param rootPath - 工作目录根路径
@@ -786,5 +799,19 @@ export class FileManager {
             return [];
         }
     }
+    /**
+     * 创建一个临时的 FileManager 代理，仅在本次链式调用中使用指定的 uuid
+     * 不会影响原始实例的 _uuid 值
+     * @param uuid - 临时使用的 uuid
+     * @returns 带有临时 uuid 的 FileManager 代理对象
+     */
+    withUUID(uuid) {
+        if (!uuid) {
+            return this;
+        }
+        const proxy = Object.create(this);
+        proxy._uuid = uuid;
+        return proxy;
+    }
 }
 export default FileManager;

package/dist/build/utils/utils.js

@@ -176,3 +176,9 @@ export const formatListRoots = (rootRes) => {
     }
     return root;
 };
+export const getParam = (url, key) => {
+    if (!url)
+        return '';
+    const urlObj = new URL(url);
+    return urlObj.searchParams.get(key);
+};

package/dist/interceptor-web/dist/build-report.json

@@ -1,5 +1,5 @@
 {
-  "buildTime": "2025-12-26T06:57:50.809Z",
+  "buildTime": "2025-12-31T04:53:16.839Z",
   "mode": "build",
   "pages": {
     "main": {