sloth-d2c-mcp 1.0.4-beta96 → 1.0.4-beta98

package/cli/run.js

@@ -198,6 +198,133 @@ if (args[0] === 'config') {
     console.log(`[sloth] 共停止 ${killedCount} 个进程`);
   }
   process.exit(0);
+} else if (args[0] === 'update') {
+  // update 命令：先停止进程，然后更新 sloth
+  const { execSync, spawnSync } = await import('node:child_process');
+  const os = await import('node:os');
+  const { fileURLToPath } = await import('node:url');
+  const path = await import('node:path');
+  
+  console.log('[sloth] 正在停止运行中的进程...');
+  
+  // 复用 stop 命令的逻辑
+  const ports = [3100, 3101];
+  const pidsToKill = new Set();
+  
+  for (const port of ports) {
+    try {
+      let output = '';
+      if (os.platform() === 'win32') {
+        output = execSync(`netstat -ano | findstr :${port}`, { encoding: 'utf-8', stdio: ['pipe', 'pipe', 'pipe'] });
+        const lines = output.trim().split('\n');
+        for (const line of lines) {
+          const match = line.match(/LISTENING\s+(\d+)/);
+          if (match) {
+            pidsToKill.add(match[1]);
+          }
+        }
+      } else {
+        output = execSync(`lsof -i :${port} -t`, { encoding: 'utf-8', stdio: ['pipe', 'pipe', 'pipe'] });
+        const pids = output.trim().split('\n').filter(Boolean);
+        pids.forEach(pid => pidsToKill.add(pid.trim()));
+      }
+    } catch {
+      // 端口未被占用，忽略错误
+    }
+  }
+  
+  let killedCount = 0;
+  for (const pid of pidsToKill) {
+    try {
+      let cmdline = '';
+      if (os.platform() === 'win32') {
+        cmdline = execSync(`wmic process where processid=${pid} get commandline`, { encoding: 'utf-8', stdio: ['pipe', 'pipe', 'pipe'] });
+      } else {
+        cmdline = execSync(`ps -p ${pid} -o command=`, { encoding: 'utf-8', stdio: ['pipe', 'pipe', 'pipe'] });
+      }
+      
+      const isSlothProcess = cmdline.includes('sloth') || cmdline.includes('d2c-mcp');
+      
+      if (isSlothProcess) {
+        process.kill(Number(pid), 'SIGTERM');
+        console.log(`[sloth] 已停止进程 PID: ${pid}`);
+        killedCount++;
+      }
+    } catch (e) {
+      // 进程可能已经退出
+    }
+  }
+  
+  if (killedCount > 0) {
+    console.log(`[sloth] 共停止 ${killedCount} 个进程`);
+  }
+  
+  // 检测包管理器：通过检查 sloth 命令的安装路径来判断
+  let packageManager = 'npm'; // 默认使用 npm
+  
+  try {
+    // 获取当前脚本的实际路径
+    const scriptPath = fileURLToPath(import.meta.url);
+    const scriptDir = path.dirname(scriptPath);
+    
+    // 检查路径中是否包含 pnpm 相关目录
+    if (scriptDir.includes('pnpm') || scriptDir.includes('.pnpm')) {
+      packageManager = 'pnpm';
+    } else {
+      // 尝试通过 which/where 命令检查 sloth 的安装位置
+      try {
+        const whichCmd = os.platform() === 'win32' ? 'where sloth' : 'which sloth';
+        const slothPath = execSync(whichCmd, { encoding: 'utf-8', stdio: ['pipe', 'pipe', 'pipe'] }).trim();
+        
+        if (slothPath.includes('pnpm') || slothPath.includes('.pnpm')) {
+          packageManager = 'pnpm';
+        }
+      } catch {
+        // 忽略错误，使用默认的 npm
+      }
+      
+      // 额外检查：尝试检测全局 pnpm 目录
+      if (packageManager === 'npm') {
+        try {
+          const pnpmRoot = execSync('pnpm root -g', { encoding: 'utf-8', stdio: ['pipe', 'pipe', 'pipe'] }).trim();
+          if (scriptDir.startsWith(pnpmRoot) || scriptDir.includes(path.dirname(pnpmRoot))) {
+            packageManager = 'pnpm';
+          }
+        } catch {
+          // pnpm 未安装或命令失败，使用 npm
+        }
+      }
+    }
+  } catch {
+    // 检测失败，使用默认的 npm
+  }
+  
+  console.log(`[sloth] 检测到包管理器: ${packageManager}`);
+  console.log('[sloth] 正在更新 sloth-d2c-mcp...');
+  
+  // 执行全局安装更新
+  const installCmd = packageManager === 'pnpm' 
+    ? 'pnpm install -g sloth-d2c-mcp' 
+    : 'npm install -g sloth-d2c-mcp';
+  
+  try {
+    const result = spawnSync(installCmd, {
+      shell: true,
+      stdio: 'inherit',
+      env: process.env
+    });
+    
+    if (result.status === 0) {
+      console.log('[sloth] 更新完成!');
+    } else {
+      console.error('[sloth] 更新失败，退出码:', result.status);
+      process.exit(result.status || 1);
+    }
+  } catch (e) {
+    console.error('[sloth] 更新失败:', e.message);
+    process.exit(1);
+  }
+  process.exit(0);
 } else if (args[0] === '--server') {
   process.env.SLOTH_COMMAND = 'server';
 } else {

package/dist/build/plugin/loader.js

@@ -3,6 +3,7 @@ import { promises as fs } from 'fs';
 import * as path from 'path';
 import { fileURLToPath } from 'url';
 import envPaths from 'env-paths';
+import { PLUGIN_HOOKS } from './types.js';
 import { createRequire } from 'module';
 const __filename = fileURLToPath(import.meta.url);
 const __dirname = path.dirname(__filename);
@@ -235,7 +236,12 @@ export async function loadWorkspacePlugins(workspaceRoot) {
                 const pluginModule = await import(localPluginPath);
                 const plugin = {
                     name: pluginName,
-                    ...(pluginModule.default || pluginModule),
+                    // 只取出PluginHooks
+                    ...PLUGIN_HOOKS.reduce((acc, hookName) => {
+                        acc[hookName] = (pluginModule.default || pluginModule)[hookName];
+                        return acc;
+                    }, {}),
+                    // ...(pluginModule.default || pluginModule),
                     _config: pluginConfig,
                 };
                 plugins.push(plugin);
@@ -265,7 +271,10 @@ export async function loadWorkspacePlugins(workspaceRoot) {
                 console.log('pluginModule', pluginModule);
                 const plugin = {
                     name: pluginName,
-                    ...(pluginModule.default || pluginModule),
+                    ...PLUGIN_HOOKS.reduce((acc, hookName) => {
+                        acc[hookName] = (pluginModule.default || pluginModule)[hookName];
+                        return acc;
+                    }, {}),
                     _config: pluginConfig,
                 };
                 plugins.push(plugin);
@@ -301,7 +310,10 @@ export async function loadAllPlugins() {
             console.log('pluginConfig', pluginConfig);
             const plugin = {
                 name: packageName,
-                ...(pluginModule.default || pluginModule),
+                ...PLUGIN_HOOKS.reduce((acc, hookName) => {
+                    acc[hookName] = (pluginModule.default || pluginModule)[hookName];
+                    return acc;
+                }, {}),
                 _config: pluginConfig,
             };
             plugins.push(plugin);
@@ -322,7 +334,10 @@ export async function loadPlugin(packageName) {
         const pluginModule = await import(packageName);
         const plugin = {
             name: packageName,
-            ...(pluginModule.default || pluginModule),
+            ...PLUGIN_HOOKS.reduce((acc, hookName) => {
+                acc[hookName] = (pluginModule.default || pluginModule)[hookName];
+                return acc;
+            }, {}),
         };
         console.log(`[sloth] 插件 ${packageName} 加载成功`);
         return plugin;

package/dist/build/plugin/types.js

@@ -1 +1,6 @@
-export {};
+export const PLUGIN_HOOKS = [
+    'afterSubmitConfigData',
+    'onFetchImage',
+    'onSamplingComplete',
+    'beforeFinalPrompt',
+];

package/dist/build/server.js

@@ -18,7 +18,7 @@ import multer from 'multer';
 import { ImageMatcher } from './utils/image-matcher.js';
 import { initOpenCV } from './utils/opencv-loader.js';
 import { extractJson } from './utils/extract.js';
-import { componentAnalysisPrompt, componentAnalysisPromptVue, chunkOptimizeCodePromptKuikly, aggregationOptimizeCodePromptKuikly, finalOptimizeCodePromptKuikly, chunkOptimizeCodePrompt, aggregationOptimizeCodePrompt, finalOptimizeCodePrompt, chunkOptimizeCodePromptVue, aggregationOptimizeCodePromptVue, finalOptimizeCodePromptVue, frameworkPromptKuikly, noSamplingAggregationPrompt, noSamplingAggregationPromptVue, noSamplingAggregationPromptKuikly, chunkOptimizeCodePromptIosOc, aggregationOptimizeCodePromptIosOc, finalOptimizeCodePromptIosOc, noSamplingAggregationPromptIosOc, chunkOptimizeCodePromptIosSwift, aggregationOptimizeCodePromptIosSwift, finalOptimizeCodePromptIosSwift, noSamplingAggregationPromptIosSwift, } from 'sloth-d2c-node/convert';
+import { componentAnalysisPrompt, componentAnalysisPromptVue, chunkOptimizeCodePromptKuikly, aggregationOptimizeCodePromptKuikly, finalOptimizeCodePromptKuikly, chunkOptimizeCodePrompt, aggregationOptimizeCodePrompt, finalOptimizeCodePrompt, chunkOptimizeCodePromptVue, aggregationOptimizeCodePromptVue, finalOptimizeCodePromptVue, frameworkPromptKuikly, noSamplingAggregationPrompt, noSamplingAggregationPromptVue, noSamplingAggregationPromptKuikly, chunkOptimizeCodePromptIosOc, aggregationOptimizeCodePromptIosOc, finalOptimizeCodePromptIosOc, noSamplingAggregationPromptIosOc, chunkOptimizeCodePromptIosSwift, aggregationOptimizeCodePromptIosSwift, finalOptimizeCodePromptIosSwift, noSamplingAggregationPromptIosSwift, chunkOptimizeCodePromptTaro, aggregationOptimizeCodePromptTaro, finalOptimizeCodePromptTaro, noSamplingAggregationPromptTaro, chunkOptimizeCodePromptUniapp, aggregationOptimizeCodePromptUniapp, finalOptimizeCodePromptUniapp, noSamplingAggregationPromptUniapp, } from 'sloth-d2c-node/convert';
 import { SocketServer } from './socket-server.js';
 import { getParam } from './utils/utils.js';
 import { loadPluginFromGlobalDirAsync } from './plugin/loader.js';
@@ -83,6 +83,8 @@ export async function loadConfig(mcpServer, configManagerInstance) {
                 { value: 'ios-oc', label: 'iOS OC' },
                 { value: 'ios-swift', label: 'iOS Swift' },
                 { value: 'kuikly', label: 'Kuikly' },
+                { value: 'taro', label: 'Taro' },
+                { value: 'uniapp', label: 'Uniapp' },
             ];
             // 初始化默认框架配置
             if (!config.frameworks || config.frameworks.length === 0) {
@@ -212,6 +214,54 @@ export async function loadConfig(mcpServer, configManagerInstance) {
                     ...(kuiklyConfig || {}),
                 });
             }
+            // 检查 taro.json 是否存在，如果不存在则创建
+            const taroConfigExists = await configManager.frameworkConfigExists('taro');
+            if (!taroConfigExists) {
+                await configManager.saveFrameworkConfig('taro', {
+                    chunkOptimizePrompt: chunkOptimizeCodePromptTaro,
+                    aggregationOptimizePrompt: aggregationOptimizeCodePromptTaro,
+                    finalOptimizePrompt: finalOptimizeCodePromptTaro,
+                    componentAnalysisPrompt: '',
+                    noSamplingAggregationPrompt: noSamplingAggregationPromptTaro,
+                });
+                const taroConfigPath = configManager.getFrameworkConfigPath('taro');
+                Logger.log(`已创建默认 taro 配置文件: ${taroConfigPath}`);
+            }
+            else {
+                const taroConfig = await configManager.loadFrameworkConfig('taro');
+                await configManager.saveFrameworkConfig('taro', {
+                    chunkOptimizePrompt: chunkOptimizeCodePromptTaro,
+                    aggregationOptimizePrompt: aggregationOptimizeCodePromptTaro,
+                    finalOptimizePrompt: finalOptimizeCodePromptTaro,
+                    componentAnalysisPrompt: '',
+                    noSamplingAggregationPrompt: noSamplingAggregationPromptTaro,
+                    ...(taroConfig || {}),
+                });
+            }
+            // 检查 uniapp.json 是否存在，如果不存在则创建
+            const uniappConfigExists = await configManager.frameworkConfigExists('uniapp');
+            if (!uniappConfigExists) {
+                await configManager.saveFrameworkConfig('uniapp', {
+                    chunkOptimizePrompt: chunkOptimizeCodePromptUniapp,
+                    aggregationOptimizePrompt: aggregationOptimizeCodePromptUniapp,
+                    finalOptimizePrompt: finalOptimizeCodePromptUniapp,
+                    componentAnalysisPrompt: '',
+                    noSamplingAggregationPrompt: noSamplingAggregationPromptUniapp,
+                });
+                const uniappConfigPath = configManager.getFrameworkConfigPath('uniapp');
+                Logger.log(`已创建默认 uniapp 配置文件: ${uniappConfigPath}`);
+            }
+            else {
+                const uniappConfig = await configManager.loadFrameworkConfig('uniapp');
+                await configManager.saveFrameworkConfig('uniapp', {
+                    chunkOptimizePrompt: chunkOptimizeCodePromptUniapp,
+                    aggregationOptimizePrompt: aggregationOptimizeCodePromptUniapp,
+                    finalOptimizePrompt: finalOptimizeCodePromptUniapp,
+                    componentAnalysisPrompt: '',
+                    noSamplingAggregationPrompt: noSamplingAggregationPromptUniapp,
+                    ...(uniappConfig || {}),
+                });
+            }
         }
         catch (error) {
             Logger.error(`获取配置信息时出错: ${error}`);
@@ -657,7 +707,9 @@ export async function startHttpServer(port = PORT, mcpServer, configManagerInsta
             const isIosOc = framework.toLowerCase() === 'ios-oc';
             const isIosSwift = framework.toLowerCase() === 'ios-swift';
             const isKuikly = framework.toLowerCase() === 'kuikly';
-            const promptSetting = {
+            const isTaro = framework.toLowerCase() === 'taro';
+            const isUniapp = framework.toLowerCase() === 'uniapp';
+            let promptSetting = {
                 enableFrameworkGuide: isKuikly ? true : false,
                 frameworkGuidePrompt: isKuikly ? frameworkPromptKuikly : '',
                 chunkOptimizePrompt: isKuikly
@@ -698,6 +750,24 @@ export async function startHttpServer(port = PORT, mcpServer, configManagerInsta
                                 ? noSamplingAggregationPromptVue
                                 : noSamplingAggregationPrompt,
             };
+            if (isTaro) {
+                promptSetting.enableFrameworkGuide = false;
+                promptSetting.frameworkGuidePrompt = '';
+                promptSetting.chunkOptimizePrompt = chunkOptimizeCodePromptTaro;
+                promptSetting.aggregationOptimizePrompt = aggregationOptimizeCodePromptTaro;
+                promptSetting.finalOptimizePrompt = finalOptimizeCodePromptTaro;
+                promptSetting.noSamplingAggregationPrompt = noSamplingAggregationPromptTaro;
+                promptSetting.componentAnalysisPrompt = componentAnalysisPrompt;
+            }
+            if (isUniapp) {
+                promptSetting.enableFrameworkGuide = false;
+                promptSetting.frameworkGuidePrompt = '';
+                promptSetting.chunkOptimizePrompt = chunkOptimizeCodePromptUniapp;
+                promptSetting.aggregationOptimizePrompt = aggregationOptimizeCodePromptUniapp;
+                promptSetting.finalOptimizePrompt = finalOptimizeCodePromptUniapp;
+                promptSetting.noSamplingAggregationPrompt = noSamplingAggregationPromptUniapp;
+                promptSetting.componentAnalysisPrompt = componentAnalysisPromptVue;
+            }
             res.json({
                 success: true,
                 data: promptSetting,
@@ -948,8 +1018,6 @@ export async function startHttpServer(port = PORT, mcpServer, configManagerInsta
             return;
         }
         const renderer = pluginRenderer.default.renderConfigForm;
-        console.log('pluginRenderer.default', pluginRenderer.default);
-        console.log('renderer', renderer);
         res.type('application/javascript; charset=utf-8').send(renderer);
     });
     // ==================== 组件映射相关接口 ====================
@@ -1871,6 +1939,30 @@ ${truncatedDiff}
             });
         }
     });
+    app.get('/getVersion', async (req, res) => {
+        try {
+            // 计算 package.json 的正确路径
+            // 从 dist/build/server.js 到根目录的 package.json
+            const packagePath = path.join(__dirname, '../../package.json');
+            const packageContent = await fs.promises.readFile(packagePath, 'utf-8');
+            const packageData = JSON.parse(packageContent);
+            res.json({
+                success: true,
+                data: {
+                    version: packageData.version,
+                    name: packageData.name,
+                },
+            });
+        }
+        catch (error) {
+            Logger.error('获取版本信息失败:', error);
+            res.status(500).json({
+                success: false,
+                message: '获取版本信息失败',
+                error: error instanceof Error ? error.message : String(error),
+            });
+        }
+    });
     // 启动 HTTP 服务器，监听端口
     httpServer = app.listen(port, async (err) => {
         if (err) {
@@ -1962,7 +2054,9 @@ export async function getUserInput(payload) {
                 Logger.log('Socket 客户端已连接');
                 const workspaceRoot = fileManager.getWorkspaceRoot();
                 // 注册 token 并等待响应，同时传递 extra 数据给主进程
-                const responsePromise = socketClient.registerToken(token, { workspaceRoot, plugins: pluginManager.getPlugins() });
+                const socketParams = { workspaceRoot, plugins: pluginManager.getPlugins() };
+                Logger.log('socketParams', socketParams);
+                const responsePromise = socketClient.registerToken(token, socketParams);
                 // 打开浏览器
                 await open(authUrl);
                 // 等待认证响应

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

@@ -1,5 +1,5 @@
 {
-  "buildTime": "2026-01-25T18:16:09.150Z",
+  "buildTime": "2026-01-26T14:18:48.853Z",
   "mode": "build",
   "pages": {
     "main": {