momo-ai 1.0.17 → 1.0.19

package/package.json

@@ -1,6 +1,6 @@
 {
     "name": "momo-ai",
-    "version": "1.0.17",
+    "version": "1.0.19",
     "description": "Rachel Momo ( OpenSpec )",
     "main": "src/momo.js",
     "bin": {

package/src/_template/PROMPT/trae/momo-module-req.ejs

@@ -4,7 +4,7 @@
 详细解读 specification/requirement.md 需求文档中的模块和每个模块的功能列表，在 specification/changes/ 目录之下创建模块的核心文档，每个模块用 MXX- 前缀，其中 XX 是模块编号。
 
 1. 每个模块中包括 tasks/ 目录、任务明细文档存放位置。
-2. 每个模块中包括 proposal.md，此文档中用来存放模块的详细需求。
+2. 每个模块中包括 proposal.md，此文档中用来存放模块的详细需求，记得头部使用带有需求编号的注释：`<!-- CODE: XXX -->`。
 3. 每个模块中包括 tasks.md，此文档中是描述所有任务的总清单。
 4. 模块所需的数据模型可参考 specification/project-model.md 文档。
 5. 有多少模块在于你的模块的整体设计，记得模块内部业务要形成完整闭环。

package/src/_template/PROMPT/trae/momo-task.ejs

@@ -8,4 +8,5 @@
 3. 任务文件名统一成 MXX-TNNN.md，其中 NN 是任务编号，NNN 是任务编号，编号长度不足用 0 填充。
 4. 每个任务文件中要写出任务实施过程的详细步骤。
 5. 系统相关的数据模型参考 specification/data-model.md 文件。
+4. 更新 specification/changes/MXX-模块/tasks.md 文件时，每个任务必须使用 `- [ ]` 的前缀做任务标记，后边跟上任务编号和标题。
 <!-- END -->

package/src/commander/repo.json

@@ -9,8 +9,8 @@
             "description": "当前项目的远程 Git 仓库地址！"
         },
         {
-            "name": "i",
-            "alias": "instance",
+            "name": "instance",
+            "alias": "i",
             "description": "需要 Clone 的远程仓库的实例副本数量！",
             "default": 10
         }

package/src/executor/executeConsole.js

@@ -4,65 +4,92 @@
 
 const readline = require('readline');
 const colors = require('colors');
-const Ec = require('../epic');
+const path = require('path');
+const fs = require('fs');
+const terminalCommands = require('../terminal');
+
+// 设置颜色主题
+colors.setTheme({
+    silly: 'rainbow',
+    input: 'grey',
+    verbose: 'cyan',
+    prompt: 'red',
+    info: 'green',
+    data: 'blue',
+    help: 'cyan',
+    warn: 'yellow',
+    debug: 'magenta',
+    error: 'red'
+});
 
 // 显示欢迎界面和菜单
 const showMenu = () => {
     // 清屏
     process.stdout.write('\x1Bc');
-    
+
     // 显示标准头部信息
-    Ec.executeHeader("Rachel Momo / SDD");
-    
+    showHeader();
+
     // 使用96个字符宽度
     const width = 96;
     const headerBorder = '='.repeat(width).blue;
     const footerBorder = '-'.repeat(width).blue;
-    
+
     console.log('');
     console.log(headerBorder);
     const title = 'Momo AI / Lain Console';
     const padding = ' '.repeat(Math.floor((width - title.length) / 2) - 1);
     console.log(`${padding}${title}`.bold.brightCyan);
     console.log(headerBorder);
-    
+
     console.log('');
     console.log('欢迎使用 Momo AI / Lain 控制台！'.green);
     console.log('这是一个交互式命令行界面。'.yellow);
     console.log('');
-    
+
     console.log('可用命令：'.bold);
     console.log('  help     - 显示帮助信息'.white);
-    console.log('  exit     - 退出控制台'.white);
+    console.log('  llm      - 查看大模型配置信息'.white);
     console.log('  quit     - 退出控制台'.white);
     console.log('');
-    
+
     console.log('请在提示符后输入命令。'.gray);
     console.log(footerBorder);
 };
 
+// 显示标准头部信息
+const showHeader = () => {
+    const appInfo = JSON.parse(fs.readFileSync(path.resolve(__dirname, '../../package.json'), 'utf8'));
+
+    console.log(`[Momo AI]`.green.bold + ` ----------------- Rachel Momo / AI工具项  ------------------`.rainbow);
+    console.log(`[Momo AI]`.green.bold + ' 应用名称: '.bold + 'Rachel Momo / SDD');
+    console.log(`[Momo AI]`.green.bold + ' 工具主页: '.bold + appInfo.homepage.blue);
+    console.log(`[Momo AI]`.green.bold + ` 工具版本: ` + `${appInfo.version}`.red + '  ' + `( Node >= 22.x )`.yellow);
+    console.log(`[Momo AI]`.green.bold);
+    console.log(`[Momo AI]`.green.bold + ` ----------------- AI 系统启动…… ----------------------------`.rainbow);
+};
+
 // 处理用户输入
-const handleInput = (input) => {
+const handleInput = (input, commands) => {
     const command = input.trim().toLowerCase();
-    
+
+    // 创建命令上下文
+    const context = {
+        commands: commands
+    };
+
     switch (command) {
         case '':
             // 空命令，不处理
             break;
         case 'help':
-            console.log('');
-            console.log('帮助信息：'.bold.brightYellow);
-            console.log('  help     - 显示此帮助信息'.white);
-            console.log('  exit     - 退出控制台'.white);
-            console.log('  quit     - 退出控制台'.white);
-            console.log('');
+            commands.help(context);
+            break;
+        case 'llm':
+            commands.llm(context);
             break;
-        case 'exit':
         case 'quit':
-            console.log('');
-            console.log('感谢使用 Momo AI / Lain 控制台，再见！'.brightGreen);
-            console.log('');
-            process.exit(0);
+            commands.quit(context);
             break;
         default:
             console.log('');
@@ -97,7 +124,7 @@ const executeConsole = async () => {
     
     rl.on('line', (line) => {
         const input = line.trim();
-        handleInput(input);
+        handleInput(input, terminalCommands);
         rl.prompt();
     }).on('close', () => {
         console.log('\n' + '感谢使用 Momo AI / Lain 控制台，再见！'.brightGreen + '\n');

package/src/executor/executeList.js

@@ -5,10 +5,17 @@ const Ec = require('../epic');
 /**
  * 从 Markdown 文件中提取编号信息
  * @param {string} filePath 文件路径
+ * @param {string} dirName 目录名（用于从目录名中提取编号）
  * @returns {string} 编号信息或 'N/A'
  */
-const _extractId = (filePath) => {
+const _extractId = (filePath, dirName) => {
     try {
+        // 首先尝试从目录名中提取编号（MXX格式）
+        const dirMatch = dirName.match(/^M\d{2}/);
+        if (dirMatch) {
+            return dirMatch[0];
+        }
+        
         if (!fs.existsSync(filePath)) {
             return 'N/A';
         }
@@ -21,8 +28,17 @@ const _extractId = (filePath) => {
             return codeMatch[1].trim();
         }
         
-        // 如果没有找到CODE格式，尝试查找 "- 编号：" 格式
+        // 然后尝试从文件内容的标题中提取编号（MXX格式）
         const lines = content.split('\n');
+        for (const line of lines) {
+            // 匹配 # MXX 格式的标题（M后跟两位数字）
+            const titleMatch = line.match(/^#\s+(M\d{2})/);
+            if (titleMatch) {
+                return titleMatch[1];
+            }
+        }
+        
+        // 如果没有找到CODE格式，尝试查找 "- 编号：" 格式
         for (const line of lines) {
             const idMatch = line.match(/-\s*编号[：:]\s*(.+)/);
             if (idMatch) {
@@ -51,10 +67,11 @@ const _countTasks = (tasksFile) => {
         const lines = content.split('\n');
         let count = 0;
         
-        // 使用更直接的方法匹配任务
+        // 匹配任务列表格式:
+        // - [ ] 任务描述
+        // - [xxx] 任务描述
         for (const line of lines) {
-            // 匹配任务格式，例如: - [ ] [任务1简要描述](tasks/M01-T001-01.md)
-            if (line.trim().startsWith('- [') && line.includes('] [') && line.includes('](') && line.includes('tasks/')) {
+            if (line.trim().match(/^-\s*\[.*\]/)) {
                 count++;
             }
         }
@@ -89,17 +106,22 @@ module.exports = (options) => {
             process.exit(0);
         }
         
-        // 显示表头
-        Ec.waiting('编号\t\t名称\t\t\t任务数量');
-        Ec.waiting('------------------------------------------------');
+        // 显示表头（调整列宽和间距）
+        const header = '编号'.padEnd(12) + '名称'.padEnd(32) + '任务数量';
+        Ec.waiting(header);
+        
+        // 显示分隔线
+        const separator = '-'.repeat(10) + '  ' + '-'.repeat(30) + '  ' + '-'.repeat(8);
+        Ec.waiting(separator);
         
         // 显示需求列表
         requirements.forEach((requirement) => {
             const proposalFile = path.join(changesDir, requirement, 'proposal.md');
             const tasksFile = path.join(changesDir, requirement, 'tasks.md');
-            const id = _extractId(proposalFile);
+            const id = _extractId(proposalFile, requirement);
             const taskCount = _countTasks(tasksFile);
-            Ec.waiting(`${id}\t\t${requirement}\t\t\t${taskCount}`);
+            const line = id.padEnd(12) + requirement.padEnd(32) + String(taskCount);
+            Ec.waiting(line);
         });
         
         Ec.info(`✅ 共找到 ${requirements.length} 个需求`);

package/src/executor/executeRepo.js

@@ -176,8 +176,8 @@ module.exports = async (options) => {
     const parsed = Ec.parseArgument(options);
     
     // 获取参数
-    const repoUrl = parsed.address || parsed.a;
-    const instanceCount = parsed.i || parsed.instance || 10; // 默认值为10
+    const repoUrl = parsed.address;
+    const instanceCount = parsed.instance;
     
     // 验证参数
     if (!repoUrl) {

package/src/executor/executeRun.js

@@ -121,11 +121,14 @@ const _findTaskInstances = (taskName) => {
 
             // 检查任务文件是否存在
             if (fs.existsSync(taskPath)) {
+                // 提取任务标题
+                const title = _extractTitleFromMarkdown(taskPath);
                 taskInstances.push({
                     name: taskName,
                     path: taskPath,
                     requirement: requirement,
-                    relativePath: path.relative(process.cwd(), taskPath)
+                    relativePath: path.relative(process.cwd(), taskPath),
+                    title: title
                 });
             }
             
@@ -134,11 +137,14 @@ const _findTaskInstances = (taskName) => {
             const donePath = path.resolve(tasksDir, doneFile);
             
             if (!fs.existsSync(taskPath) && fs.existsSync(donePath)) {
+                // 提取任务标题（从 .done 文件中）
+                const title = _extractTitleFromMarkdown(donePath);
                 taskInstances.push({
                     name: taskName,
                     path: donePath,
                     requirement: requirement,
-                    relativePath: path.relative(process.cwd(), donePath)
+                    relativePath: path.relative(process.cwd(), donePath),
+                    title: title
                 });
             }
         }
@@ -347,7 +353,9 @@ const _saveAssignmentRecord = async (taskName, workspaceName, actorName, require
 
         // 创建文件名：{任务编号}-develop-xx-时间戳.txt
         const timestamp = new Date().toISOString().replace(/[:.]/g, '-').replace('T', '_').slice(0, -5);
-        const fileName = `${taskName}-${workspaceName}-${timestamp}.txt`;
+        // 只使用工作空间名称，而不是完整路径
+        const workspaceDirName = path.basename(workspaceName);
+        const fileName = `${taskName}-${workspaceDirName}-${timestamp}.txt`;
         const filePath = path.join(activitiesDir, fileName);
 
         // 写入分派记录
@@ -355,12 +363,12 @@ const _saveAssignmentRecord = async (taskName, workspaceName, actorName, require
 =====================
 
 Task Name: ${taskName}
-Workspace: ${workspaceName}
+Workspace: ${workspaceDirName}
 Actor: ${actorName}
 Requirement: ${requirementName}
 Assigned At: ${new Date().toISOString()}
 
-This task has been assigned to workspace ${workspaceName} for actor ${actorName}.
+This task has been assigned to workspace ${workspaceDirName} for actor ${actorName}.
 `;
         
         await fsAsync.writeFile(filePath, recordContent);
@@ -421,16 +429,26 @@ module.exports = async (options) => {
                     const coloredName = task.name.cyan;
                     Ec.waiting(`${index + 1}. [${coloredStatus}] ${task.display} (${coloredName})`);
                 });
+                
+                // 添加退出选项
+                Ec.waiting(`${allTasks.length + 1}. 退出`);
 
                 // 获取用户选择
                 const answer = await Ec.ask('请输入选项编号: ');
                 const selectedIndex = parseInt(answer) - 1;
 
                 // 验证选择
-                if (isNaN(selectedIndex) || selectedIndex < 0 || selectedIndex >= allTasks.length) {
+                if (isNaN(selectedIndex) || selectedIndex < 0 || selectedIndex > allTasks.length) {
                     Ec.error('❌ 无效的选择');
                     continue;
                 }
+                
+                // 检查是否选择退出
+                if (selectedIndex === allTasks.length) {
+                    Ec.info("已退出任务选择");
+                    Ec.askClose();
+                    process.exit(0);
+                }
 
                 // 检查选中的任务是否正在进行中
                 const tasksDir = path.resolve(process.cwd(), 'specification', 'changes', allTasks[selectedIndex].requirement, 'tasks');
@@ -592,15 +610,15 @@ module.exports = async (options) => {
                 });
 
                 // 打印额外信息
-                Ec.waiting(`[Momo AI] 执行任务: ${selectedTask.name} - ${selectedTask.title}`);
-                Ec.waiting(`[Momo AI] 使用工作空间: ${selectedWorkspace.path}`);
-                Ec.waiting('[Momo AI] 提示词内容:');
+                Ec.waiting(`执行任务: ${selectedTask.name} - ${selectedTask.title || '未知任务'}`);
+                Ec.waiting(`使用工作空间: ${selectedWorkspace.path}`);
+                Ec.waiting('提示词内容:');
                 Ec.waiting('--------------------------------------------------');
                 
                 // 按行打印提示词内容，保持原有格式
                 const lines = renderedContent.split('\n');
                 lines.forEach(line => {
-                    Ec.waiting(`[Momo AI] ${line}`);
+                    Ec.waiting(line);
                 });
                 
                 Ec.waiting('--------------------------------------------------');