momo-ai 1.0.20 → 1.0.22

package/src/executor/executeOpen.js

@@ -1,159 +1,178 @@
+const fs = require('fs');
+const path = require('path');
 const Ec = require('../epic');
-const {spawn} = require('child_process');
-
-module.exports = async (options) => {
-    // 参数提取
-    const parsed = Ec.parseArgument(options);
-
-    try {
-        // 定义可用的AI工具
-        const aiTools = [
-            {name: 'Trae', command: 'trae'},
-            {name: 'Cursor', command: 'cursor'},
-            {name: 'Lingma', command: 'lingma'},
-            {name: 'Kiro', command: 'kiro'}
-        ];
-
-        // 检查哪些工具可用
-        const availableTools = [];
-        for (const tool of aiTools) {
-            if (await _isCommandAvailable(tool.command)) {
-                availableTools.push(tool);
-            }
-        }
-
-        // 如果没有可用工具，提示用户安装
-        if (availableTools.length === 0) {
-            Ec.error('❌ 未找到可用的AI工具，请确保已安装以下工具之一：Trae、Cursor、Lingma、Kiro');
-            Ec.askClose();
-            process.exit(1);
-        }
-
-        // 如果只有一个工具可用，直接使用它
-        if (availableTools.length === 1) {
-            const tool = availableTools[0];
-            let toolDisplayName = tool.name;
-            if (tool.name === 'Trae' || tool.name === 'Cursor') {
-                toolDisplayName = `${tool.name}（${'Multi-Agent'.cyan}）`;
-            }
-            if (tool.name === 'Trae') {
-                Ec.waiting(`🔍 检测到可用工具: ${toolDisplayName.green}`);
-            } else {
-                Ec.waiting(`🔍 检测到可用工具: ${toolDisplayName}`);
-            }
-            await _openWithTool(tool.command);
-            Ec.askClose();
-            process.exit(0);
-        }
-
-        // 显示交互式选择菜单
-        Ec.waiting('🔍 检测到多个可用的AI工具，请选择要使用的工具:');
-        const choices = availableTools.map((tool, index) => {
-            let toolDisplayName = tool.name;
-            if (tool.name === 'Trae' || tool.name === 'Cursor') {
-                toolDisplayName = `${tool.name}（${'Multi-Agent'.cyan}）`;
-            }
-            if (tool.name === 'Trae') {
-                return `${index + 1}. ${toolDisplayName} ${'推荐'.green}`;
-            } else {
-                return `${index + 1}. ${toolDisplayName}`;
-            }
-        });
-
-        // 添加退出选项
-        choices.push(`${choices.length + 1}. 退出`);
-
-        for (const choice of choices) {
-            Ec.waiting(choice);
-        }
-
-        // 获取用户选择
-        const answer = await Ec.ask('请输入选项编号: ');
-        const selectedIndex = parseInt(answer) - 1;
+const { spawn } = require('child_process');
+const { selectSingle } = require('../utils/momo-menu');
+
+// AI 工具配置
+const AI_TOOLS = [
+    {
+        name: 'Antigravity',
+        command: 'antigravity',
+        description: 'Claude Code 官方客户端',
+        openMethod: 'spawn'  // 使用 spawn 直接执行
+    },
+    {
+        name: 'Trae',
+        command: 'trae',
+        description: '字节跳动 AI IDE',
+        openMethod: 'spawn'
+    },
+    {
+        name: 'Cursor',
+        command: 'cursor',
+        description: 'Cursor AI IDE',
+        openMethod: 'spawn'
+    }
+];
 
-        // 检查用户选择
-        if (isNaN(selectedIndex) || selectedIndex < 0 || selectedIndex >= availableTools.length) {
-            if (selectedIndex === availableTools.length) {
-                Ec.waiting('👋 已取消操作');
-                Ec.askClose();
-                process.exit(0);
+/**
+ * 解析 -d 参数
+ * @returns {string|null} 目录路径
+ */
+const _parseDirArg = () => {
+    const args = process.argv.slice(3);
+    
+    for (let i = 0; i < args.length; i++) {
+        if (args[i] === '-d' || args[i] === '--dir') {
+            const next = args[i + 1];
+            if (next && !next.startsWith('-')) {
+                return next;
             }
-            Ec.error('❌ 无效的选择');
-            Ec.askClose();
-            process.exit(1);
         }
-
-        // 执行选择的工具
-        const selectedTool = availableTools[selectedIndex];
-        let toolDisplayName = selectedTool.name;
-        if (selectedTool.name === 'Trae' || selectedTool.name === 'Cursor') {
-            toolDisplayName = `${selectedTool.name}（${'Multi-Agent'.cyan}）`;
-        }
-        Ec.waiting(`🚀 正在使用 ${toolDisplayName} 打开项目...`);
-        await _openWithTool(selectedTool.command);
-        Ec.askClose();
-        process.exit(0);
-
-    } catch (error) {
-        console.error(error);
-        Ec.error(`❌ 执行过程中发生错误: ${error.message}`);
-        if (process.platform === 'win32') {
-            Ec.waiting('💡 Windows 用户提示: 请确保您在具有足够权限的命令行中运行此命令');
-        }
-        Ec.askClose();
-        process.exit(1);
     }
+    
+    return null;
 };
 
 /**
  * 检查命令是否可用
  * @param {string} command 命令名称
- * @returns {Promise<boolean>} 命令是否可用
+ * @returns {Promise<boolean>}
  */
 const _isCommandAvailable = async (command) => {
     return new Promise((resolve) => {
-        // 在 Windows 上使用 where 命令，在其他系统上使用 which 命令
         const whereCmd = process.platform === 'win32' ? 'where' : 'which';
-        const childProcess = spawn(whereCmd, [command]);
+        const childProcess = spawn(whereCmd, [command], { stdio: 'pipe' });
         childProcess.on('close', (code) => {
             resolve(code === 0);
         });
+        childProcess.on('error', () => {
+            resolve(false);
+        });
     });
 };
 
 /**
- * 使用指定工具打开当前目录
- * @param {string} tool 工具命令
+ * 使用 spawn 方式打开目录
+ * @param {string} command 命令
+ * @param {string} targetDir 目标目录
  */
-const _openWithTool = async (tool) => {
+const _openWithSpawn = async (command, targetDir) => {
     return new Promise((resolve, reject) => {
-        // 使用工具命令打开当前目录
-        const child = spawn(tool, ['.'], {
+        const child = spawn(command, [targetDir], {
             stdio: 'inherit',
-            cwd: process.cwd()
+            detached: true
         });
 
-        child.on('close', (code) => {
-            if (code === 0) {
-                Ec.info(`✅ 项目已成功在 ${tool} 中打开`);
-                resolve();
-            } else {
-                Ec.error(`❌ ${tool} 执行失败，退出码: ${code}`);
-                reject(new Error(`${tool} 执行失败，退出码: ${code}`));
-            }
-        });
+        // 分离子进程，让它独立运行
+        child.unref();
+
+        // 给一点时间让进程启动
+        setTimeout(() => {
+            resolve();
+        }, 500);
 
         child.on('error', (error) => {
             if (error.code === 'ENOENT') {
-                Ec.error(`❌ 未找到命令: ${tool}，请确保已正确安装`);
-                if (process.platform === 'win32') {
-                    Ec.waiting('💡 Windows 用户提示: 请确保工具已正确安装并在 PATH 环境变量中');
-                }
-                reject(new Error(`未找到命令: ${tool}，请确保已正确安装`));
+                reject(new Error(`未找到命令: ${command}`));
             } else {
-                Ec.error(`❌ 执行 ${tool} 时发生错误: ${error.message}`);
                 reject(error);
             }
         });
     });
-};
+};
+
+/**
+ * 打开目录
+ * @param {Object} tool 工具配置
+ * @param {string} targetDir 目标目录
+ */
+const _openDirectory = async (tool, targetDir) => {
+    Ec.waiting(`🚀 正在使用 ${tool.name.cyan} 打开目录...`);
+    
+    switch (tool.openMethod) {
+        case 'spawn':
+        default:
+            await _openWithSpawn(tool.command, targetDir);
+            break;
+    }
+    
+    Ec.info(`✅ 已启动 ${tool.name}`);
+};
+
+module.exports = async (options) => {
+    try {
+        // 1. 解析目录参数
+        const dirArg = _parseDirArg();
+        let targetDir = dirArg ? path.resolve(dirArg) : process.cwd();
+        
+        // 验证目录是否存在
+        if (!fs.existsSync(targetDir)) {
+            Ec.error(`❌ 目录不存在: ${targetDir}`);
+            process.exit(1);
+        }
+        
+        if (!fs.statSync(targetDir).isDirectory()) {
+            Ec.error(`❌ 路径不是目录: ${targetDir}`);
+            process.exit(1);
+        }
+        
+        // 2. 检查可用工具
+        const availableTools = [];
+        
+        for (const tool of AI_TOOLS) {
+            if (await _isCommandAvailable(tool.command)) {
+                availableTools.push(tool);
+            }
+        }
+        
+        if (availableTools.length === 0) {
+            Ec.error('❌ 未找到可用的 AI 工具');
+            console.log('');
+            console.log('  支持的工具:'.gray);
+            AI_TOOLS.forEach(tool => {
+                console.log(`    - ${tool.name}: ${tool.description}`.gray);
+            });
+            process.exit(1);
+        }
+        
+        // 3. 显示目标目录
+        console.log('');
+        Ec.info(`📂 目标目录: ${targetDir.cyan}`);
+        console.log('');
+        
+        // 4. 交互式选择工具
+        const menuItems = availableTools.map(tool => ({
+            name: tool.name,
+            description: tool.description,
+            _tool: tool
+        }));
+        
+        const selected = await selectSingle(menuItems, '选择 AI 工具');
+        
+        if (!selected) {
+            Ec.waiting('已取消');
+            process.exit(0);
+        }
+        
+        // 5. 打开目录
+        await _openDirectory(selected._tool, targetDir);
+        
+        process.exit(0);
+        
+    } catch (error) {
+        Ec.error(`❌ 执行失败: ${error.message}`);
+        process.exit(1);
+    }
+};

package/src/executor/index.js

@@ -1,50 +1,28 @@
 const executeHelp = require('./executeHelp');
 const executeInit = require('./executeInit');
-const executeRepo = require('./executeRepo');
-const executeAdd = require('./executeAdd');
-const executePlan = require('./executePlan');
-const executeArchive = require('./executeArchive');
 const executeEnv = require('./executeEnv');
 const executeOpen = require('./executeOpen');
-const executeShow = require('./executeShow');
-const executeList = require('./executeList');
-const executeValidate = require('./executeValidate');
-const executeActor = require('./executeActor');
-const executeActors = require('./executeActors');
-const executeRun = require('./executeRun');
-const executeTasks = require('./executeTasks');
-const executeUnlock = require('./executeUnlock');
-const executeProject = require('./executeProject');
-const executeAgentCfg = require('./executeAgentCfg');
-const executeAgent = require('./executeAgent');
-const executeConsole = require('./executeConsole');
-const executePull = require('./executePull');
-const executeCommit = require('./executeCommit');
-const executePush = require('./executePush');
+const executeMcp = require('./executeMcp');
+const executeApp = require('./executeApp');
+const executeApply = require('./executeApply');
+const executeDomain = require('./executeDomain');
+const executeAsk = require('./executeAsk');
+const executeMmr2 = require('./executeMmr2');
+const executeMmr0 = require('./executeMmr0');
+const executeDocs = require('./executeDocs');
 
 const exported = {
     executeHelp,
     executeInit,
-    executeRepo,
-    executeAdd,
-    executePlan,
-    executeArchive,
     executeEnv,
     executeOpen,
-    executeShow,
-    executeList,
-    executeValidate,
-    executeActor,
-    executeActors,
-    executeRun,
-    executeTasks,
-    executeUnlock,
-    executeProject,
-    executeAgentCfg,
-    executeAgent,
-    executeConsole,
-    executePull,
-    executeCommit,
-    executePush
+    executeMcp,
+    executeApp,
+    executeApply,
+    executeDomain,
+    executeAsk,
+    executeMmr2,
+    executeMmr0,
+    executeDocs
 };
-module.exports = exported;
+module.exports = exported;

package/src/momo.js

@@ -28,7 +28,8 @@ if (_isVersionMode()) {
 }
 
 // 输出头部
-Ec.executeHeader("Rachel Momo / SDD");
+Ec.info("SDD / Spec Driven Development ...")
+// Ec.executeHeader("Rachel Momo / SDD");
 
 // 读取配置文件
 const configArr = Ec.parseMetadata();