momo-ai 1.0.20 → 1.0.22

package/src/executor/executeAdd.js

@@ -1,307 +0,0 @@
-const fs = require('fs');
-const path = require('path');
-const util = require('util');
-const Ec = require('../epic');
-const fsAsync = require('fs').promises;
-const { outCopy } = require('../epic/momo.fn.out');
-
-/**
- * 读取模板文件并提取 <!-- BEGIN --> 到 <!-- END --> 之间的内容
- * @param {string} templatePath 模板文件路径
- * @returns {Promise<string>} 提取的模板内容
- */
-const _readTemplate = async (templatePath) => {
-    try {
-        const content = await fsAsync.readFile(templatePath, 'utf8');
-        const lines = content.split('\n');
-        let beginIndex = -1;
-        let endIndex = -1;
-        
-        for (let i = 0; i < lines.length; i++) {
-            if (lines[i].includes('<!-- BEGIN -->')) {
-                beginIndex = i;
-            } else if (lines[i].includes('<!-- END -->')) {
-                endIndex = i;
-                break;
-            }
-        }
-        
-        if (beginIndex !== -1 && endIndex !== -1) {
-            // 提取 BEGIN 和 END 之间的内容（不包括 BEGIN 和 END 行）
-            return lines.slice(beginIndex + 1, endIndex).join('\n');
-        }
-        
-        return content;
-    } catch (error) {
-        Ec.waiting(`读取模板文件失败: ${error.message}`);
-        if (process.platform === 'win32') {
-            Ec.waiting('💡 Windows 用户提示: 请确保您在具有足够权限的命令行中运行此命令');
-        }
-        throw error;
-    }
-};
-
-/**
- * 渲染 EJS 模板
- * @param {string} template 模板内容
- * @param {Object} data 数据对象
- * @returns {string} 渲染后的内容
- */
-const _renderTemplate = (template, data) => {
-    let result = template;
-    for (const key in data) {
-        if (data.hasOwnProperty(key)) {
-            const regex = new RegExp(`<%=\\s*${key}\\s*%>`, 'g');
-            result = result.replace(regex, data[key]);
-        }
-    }
-    return result;
-};
-
-/**
- * 检查需求目录是否已存在
- * @param {string} changesDir changes目录路径
- * @param {string} requirementName 需求名称
- * @returns {boolean} 是否存在
- */
-const _isRequirementExists = (changesDir, requirementName) => {
-    const requirementDir = path.join(changesDir, requirementName);
-    return fs.existsSync(requirementDir);
-};
-
-/**
- * 从 Markdown 文件中提取第一个标题作为需求名称
- * @param {string} filePath 文件路径
- * @returns {Promise<string>} 第一个标题内容
- */
-const _extractTitleFromMarkdown = async (filePath) => {
-    try {
-        const content = await fsAsync.readFile(filePath, 'utf8');
-        const lines = content.split('\n');
-        
-        for (const line of lines) {
-            // 匹配 # 开头的标题行
-            const titleMatch = line.match(/^#\s+(.+)$/);
-            if (titleMatch) {
-                return titleMatch[1].trim();
-            }
-        }
-        
-        // 如果没有找到标题，使用文件名（不含扩展名）作为需求名称
-        const fileName = path.basename(filePath);
-        return fileName.replace(/\.[^/.]+$/, ""); // 移除扩展名
-    } catch (error) {
-        Ec.waiting(`读取文件失败: ${error.message}`);
-        throw error;
-    }
-};
-
-/**
- * 创建需求目录并拷贝模板文件
- * @param {string} templateDir 模板目录
- * @param {string} targetDir 目标目录
- * @param {string} sourceFilePath 源文件路径（可选）
- */
-const _copyTemplateFiles = async (templateDir, targetDir, sourceFilePath = null) => {
-    try {
-        // 确保目标目录存在
-        if (!fs.existsSync(targetDir)) {
-            await fsAsync.mkdir(targetDir, { recursive: true });
-        }
-        
-        let proposalSource = path.join(templateDir, 'proposal.md');
-        const proposalTarget = path.join(targetDir, 'proposal.md');
-        
-        // 如果提供了源文件路径，则使用源文件内容覆盖 proposal.md
-        if (sourceFilePath && fs.existsSync(sourceFilePath)) {
-            proposalSource = sourceFilePath;
-            Ec.waiting(`从文件 "${sourceFilePath}" 拷贝内容到 proposal.md`);
-            await fsAsync.copyFile(sourceFilePath, proposalTarget);
-        } else if (fs.existsSync(proposalSource)) {
-            Ec.waiting(`拷贝文件: ${proposalTarget}`);
-            await fsAsync.copyFile(proposalSource, proposalTarget);
-        } else {
-            // 如果模板文件不存在，则创建空文件
-            Ec.waiting(`创建空文件: ${proposalTarget}`);
-            await fsAsync.writeFile(proposalTarget, '');
-        }
-        
-        // 拷贝 tasks.md
-        const tasksSource = path.join(templateDir, 'tasks.md');
-        const tasksTarget = path.join(targetDir, 'tasks.md');
-        if (fs.existsSync(tasksSource)) {
-            Ec.waiting(`拷贝文件: ${tasksTarget}`);
-            await fsAsync.copyFile(tasksSource, tasksTarget);
-        } else {
-            // 如果模板文件不存在，则创建空文件
-            Ec.waiting(`创建空文件: ${tasksTarget}`);
-            await fsAsync.writeFile(tasksTarget, '');
-        }
-        
-        // 创建 tasks 目录
-        const tasksDir = path.join(targetDir, 'tasks');
-        if (!fs.existsSync(tasksDir)) {
-            Ec.waiting(`创建目录: ${tasksDir}`);
-            await fsAsync.mkdir(tasksDir, { recursive: true });
-        }
-        
-        // 拷贝 task-detail.md 到 tasks 目录
-        const taskDetailSource = path.join(templateDir, 'tasks', 'task-detail.md');
-        const taskDetailTarget = path.join(tasksDir, 'M01-T001-01.md'); // 示例任务文件名
-        if (fs.existsSync(taskDetailSource)) {
-            Ec.waiting(`拷贝文件: ${taskDetailTarget}`);
-            await fsAsync.copyFile(taskDetailSource, taskDetailTarget);
-        } else {
-            // 如果模板文件不存在，则创建空文件
-            Ec.waiting(`创建空文件: ${taskDetailTarget}`);
-            await fsAsync.writeFile(taskDetailTarget, '');
-        }
-    } catch (error) {
-        Ec.waiting(`拷贝模板文件失败: ${error.message}`);
-        if (process.platform === 'win32') {
-            Ec.waiting('💡 Windows 用户提示: 请确保您在具有足够权限的命令行中运行此命令');
-        }
-        throw error;
-    }
-};
-
-/**
- * 将内容复制到剪贴板（去除换行符）
- * @param {string} content 要复制的内容
- */
-const _copyToClipboard = async (content) => {
-    try {
-        // 去除换行符，将内容合并为一行
-        const contentWithoutNewlines = content.replace(/\r?\n|\r/g, ' ');
-        
-        // 使用统一的剪贴板函数
-        await outCopy(contentWithoutNewlines);
-        Ec.waiting('✅ 提示词已复制到剪贴板');
-    } catch (error) {
-        Ec.waiting(`复制到剪贴板失败: ${error.message}`);
-        // 提供备选方案
-        Ec.waiting('提示词内容:');
-        console.log(content);
-        if (process.platform === 'win32') {
-            Ec.waiting('💡 Windows 用户提示: 请确保您在具有足够权限的命令行中运行此命令');
-        }
-    }
-};
-
-module.exports = async (options) => {
-    // 参数提取
-    const parsed = Ec.parseArgument(options);
-    
-    // 获取需求名称
-    let requirementName = parsed.name || parsed.n;
-    
-    // 验证参数
-    if (!requirementName) {
-        Ec.error("❌ 请提供需求名称或需求文件路径 (-n, --name)");
-        process.exit(1);
-    }
-    
-    // 检查是否包含扩展名
-    const hasExtension = path.extname(requirementName) !== '';
-    
-    // 如果包含扩展名，视为文件路径处理
-    let sourceFilePath = null;
-    if (hasExtension) {
-        // 检查文件是否存在
-        if (!fs.existsSync(requirementName)) {
-            // 检查是否在当前目录下存在
-            const fullPath = path.resolve(process.cwd(), requirementName);
-            if (!fs.existsSync(fullPath)) {
-                Ec.error(`❌ 文件 "${requirementName}" 不存在`);
-                process.exit(1);
-            }
-            sourceFilePath = fullPath;
-        } else {
-            sourceFilePath = requirementName;
-        }
-        
-        // 从文件中提取标题作为需求名称
-        requirementName = await _extractTitleFromMarkdown(sourceFilePath);
-    }
-    
-    // 检查需求名称是否包含非法字符（Windows 上特别检查）
-    if (process.platform === 'win32') {
-        const illegalChars = /[<>:"/\\|?*\x00-\x1F]/g;
-        if (illegalChars.test(requirementName)) {
-            Ec.error(`❌ 需求名称 "${requirementName}" 包含非法字符，请避免使用以下字符: < > : " / \\ | ? * 以及控制字符`);
-            process.exit(1);
-        }
-    }
-    
-    // 检查需求名称是否包含点号（除了允许的独立需求）
-    if (!hasExtension && requirementName.includes('.')) {
-        Ec.error("❌ 独立需求名称不能包含点号(.)，以防止与文件扩展名混淆");
-        process.exit(1);
-    }
-    
-    Ec.waiting(`准备添加新需求: ${requirementName}`);
-    
-    try {
-        // 获取项目路径
-        const projectDir = process.cwd();
-        const changesDir = path.join(projectDir, 'specification', 'changes');
-        const templateDir = path.join(__dirname, '../_template/LAIN/changes');
-        const promptTemplatePath = path.join(__dirname, '../_template/PROMPT/add.md.ejs');
-        
-        // 检查需求是否已存在
-        if (_isRequirementExists(changesDir, requirementName)) {
-            Ec.error(`❌ 需求 "${requirementName}" 已存在`);
-            process.exit(1);
-        }
-        
-        // 确保 changes 目录存在
-        if (!fs.existsSync(changesDir)) {
-            Ec.waiting(`创建目录: ${changesDir}`);
-            try {
-                await fsAsync.mkdir(changesDir, { recursive: true });
-            } catch (mkdirError) {
-                Ec.error(`❌ 创建目录失败: ${mkdirError.message}`);
-                if (process.platform === 'win32') {
-                    Ec.waiting('💡 Windows 用户提示: 请确保您在具有足够权限的命令行中运行此命令');
-                }
-                process.exit(1);
-            }
-        }
-        
-        // 创建需求目录
-        const requirementDir = path.join(changesDir, requirementName);
-        Ec.waiting(`创建需求目录: ${requirementDir}`);
-        try {
-            await fsAsync.mkdir(requirementDir, { recursive: true });
-        } catch (mkdirError) {
-            Ec.error(`❌ 创建目录失败: ${mkdirError.message}`);
-            if (process.platform === 'win32') {
-                Ec.waiting('💡 Windows 用户提示: 请确保您在具有足够权限的命令行中运行此命令');
-                // 检查是否是由于特殊字符导致的问题
-                if (mkdirError.message.includes('Invalid argument') || mkdirError.message.includes('EINVAL')) {
-                    Ec.waiting('💡 可能是由于需求名称包含非法字符，请尝试使用更简单的名称');
-                }
-            }
-            process.exit(1);
-        }
-        
-        // 拷贝模板文件
-        await _copyTemplateFiles(templateDir, requirementDir, sourceFilePath);
-        
-        // 读取并处理提示词模板
-        const templateContent = await _readTemplate(promptTemplatePath);
-        const renderedContent = _renderTemplate(templateContent, { NAME: requirementName });
-        
-        // 将提示词复制到剪贴板
-        await _copyToClipboard(renderedContent);
-        
-        Ec.info(`✅ 成功添加新需求 "${requirementName}"`);
-        process.exit(0);
-    } catch (error) {
-        Ec.error(`❌ 执行过程中发生错误: ${error.message}`);
-        if (process.platform === 'win32') {
-            Ec.waiting('💡 Windows 用户提示: 请确保您在具有足够权限的命令行中运行此命令');
-        }
-        process.exit(1);
-    }
-};

package/src/executor/executeAgent.js

@@ -1,299 +0,0 @@
-const fs = require('fs');
-const path = require('path');
-const util = require('util');
-const Ec = require('../epic');
-
-// 将 fs 方法转换为 Promise 版本
-const fsAsync = {
-    readFile: util.promisify(fs.readFile),
-    readdir: util.promisify(fs.readdir),
-    stat: util.promisify(fs.stat)
-};
-
-/**
- * 获取所有 agents 信息
- * @returns {Array} agents 信息数组
- */
-const _getAgents = () => {
-    const agentsDir = path.resolve(__dirname, '../_agent');
-    const agents = [];
-    
-    if (!fs.existsSync(agentsDir)) {
-        return agents;
-    }
-    
-    // 遍历 _agent 目录下的所有子目录
-    const types = fs.readdirSync(agentsDir).filter(file => 
-        fs.statSync(path.join(agentsDir, file)).isDirectory()
-    );
-    
-    // 遍历每个类型目录
-    types.forEach(type => {
-        const typeDir = path.join(agentsDir, type);
-        const files = fs.readdirSync(typeDir).filter(file => file.endsWith('.json'));
-        
-        // 遍历每个 json 文件
-        files.forEach(file => {
-            try {
-                const filePath = path.join(typeDir, file);
-                const content = fs.readFileSync(filePath, 'utf8');
-                const agentInfo = JSON.parse(content);
-                
-                agents.push({
-                    argName: file.replace('momo-', '').replace('.json', ''),
-                    type: type,
-                    id: agentInfo.id,
-                    name: agentInfo.name,
-                    uri: agentInfo.uri,
-                    uriCn: agentInfo['uri-cn'], // 添加CN链接
-                    filePath: filePath
-                });
-            } catch (error) {
-                Ec.waiting(`⚠️ 无法解析文件: ${file} - ${error.message}`);
-            }
-        });
-    });
-    
-    return agents;
-};
-
-/**
- * 从 EJS 模板中提取内容
- * @param {string} templatePath 模板路径
- * @returns {Promise<string>} 提取的模板内容
- */
-const _extractTemplateContent = async (templatePath) => {
-    try {
-        const content = await fsAsync.readFile(templatePath, 'utf8');
-        const lines = content.split('\n');
-        let beginIndex = -1;
-        let endIndex = -1;
-        
-        for (let i = 0; i < lines.length; i++) {
-            if (lines[i].includes('<!-- BEGIN -->')) {
-                beginIndex = i;
-            } else if (lines[i].includes('<!-- END -->')) {
-                endIndex = i;
-                break;
-            }
-        }
-        
-        if (beginIndex !== -1 && endIndex !== -1) {
-            // 提取 BEGIN 和 END 之间的内容（不包括 BEGIN 和 END 行）
-            return lines.slice(beginIndex + 1, endIndex).join('\n');
-        }
-        
-        return content;
-    } catch (error) {
-        throw new Error(`无法读取模板文件: ${error.message}`);
-    }
-};
-
-/**
- * 将内容复制到剪贴板
- * @param {string} content 要复制的内容
- */
-const _copyToClipboard = async (content) => {
-    const { outCopy } = require('../epic/momo.fn.out');
-    try {
-        // 先打印内容，按行打印并保留前缀
-        Ec.waiting('📄 提示词内容:');
-        Ec.waiting('----------------------------------------');
-        const lines = content.split('\n');
-        lines.forEach(line => {
-            Ec.waiting(line);
-        });
-        Ec.waiting('----------------------------------------');
-        
-        // 再复制到剪贴板
-        await outCopy(content);
-        Ec.waiting('✅ 提示词已复制到剪贴板');
-    } catch (error) {
-        Ec.waiting(`复制到剪贴板失败: ${error.message}`);
-        // 提供备选方案
-        Ec.waiting('提示词内容:');
-        const lines = content.split('\n');
-        lines.forEach(line => {
-            Ec.waiting(line);
-        });
-    }
-};
-
-module.exports = async (options) => {
-    // 参数提取
-    const parsed = Ec.parseArgument(options);
-    const agentName = parsed.agent || parsed.a;
-
-    try {
-        // 获取所有 agents
-        const agents = _getAgents();
-        
-        if (agents.length === 0) {
-            Ec.error("❌ 未找到任何 agents");
-            Ec.askClose();
-            process.exit(1);
-        }
-        
-        // 如果没有提供 agent 参数，显示可用的 agents 并让用户选择
-        if (!agentName) {
-            while (true) {
-                Ec.waiting('可用的 Agents:');
-                
-                // 计算列宽以适应内容
-                const typeColumnWidth = 12;
-                const idColumnWidth = 20;
-                const uriColumnWidth = 35;
-                const uriCnColumnWidth = 35;
-                const nameColumnWidth = 35;
-                
-                // 表头
-                const header = "编号".padEnd(6) + 
-                              "Type".padEnd(typeColumnWidth) + 
-                              "ID".padEnd(idColumnWidth) + 
-                              "URI".padEnd(uriColumnWidth) + 
-                              "URI-CN".padEnd(uriCnColumnWidth) + 
-                              "Name".padEnd(nameColumnWidth);
-                Ec.waiting(header);
-                
-                // 分隔线
-                const separator = "-".repeat(5) + " " +
-                                 "-".repeat(typeColumnWidth - 1) + " " +
-                                 "-".repeat(idColumnWidth - 1) + " " +
-                                 "-".repeat(uriColumnWidth - 1) + " " +
-                                 "-".repeat(uriCnColumnWidth - 1) + " " +
-                                 "-".repeat(nameColumnWidth - 1);
-                Ec.waiting(separator);
-                
-                // 数据行
-                agents.forEach((agent, index) => {
-                    const name = `${agent.argName} (${agent.name})`;
-                    const row = `${(index + 1).toString().padEnd(6)}` +
-                               `${agent.type.padEnd(typeColumnWidth)}` +
-                               `${agent.id.padEnd(idColumnWidth).red.bold}` +
-                               `${(agent.uri || '').padEnd(uriColumnWidth)}` +
-                               `${(agent.uriCn || '').padEnd(uriCnColumnWidth)}` +
-                               name.padEnd(nameColumnWidth);
-                    Ec.waiting(row);
-                });
-                
-                Ec.waiting(separator);
-                Ec.waiting(`${agents.length + 1}. 退出`);
-                
-                // 获取用户选择
-                const answer = await Ec.ask('请选择要使用的 Agent (输入编号): ');
-                const selectedIndex = parseInt(answer) - 1;
-                
-                // 验证选择
-                if (isNaN(selectedIndex) || selectedIndex < 0 || selectedIndex > agents.length) {
-                    Ec.error("❌ 无效的选择");
-                    continue;
-                }
-                
-                // 检查是否选择退出
-                if (selectedIndex === agents.length) {
-                    Ec.info("已退出 Agent 选择");
-                    Ec.askClose();
-                    process.exit(0);
-                }
-                
-                // 使用选择的 agent
-                const selectedAgent = agents[selectedIndex];
-                return await _processAgent(selectedAgent);
-            }
-        }
-        
-        // 查找指定的 agent
-        const agent = agents.find(a => a.argName === agentName);
-        if (!agent) {
-            Ec.error(`❌ 未找到名为 "${agentName}" 的 Agent`);
-            
-            // 显示所有 agents（表格形式）
-            Ec.waiting('可用的 Agents:');
-            
-            // 计算列宽以适应内容
-            const typeColumnWidth = 12;
-            const idColumnWidth = 20;
-            const uriColumnWidth = 35;
-            const uriCnColumnWidth = 35;
-            const nameColumnWidth = 35;
-            
-            // 表头
-            const header = "编号".padEnd(6) + 
-                          "Type".padEnd(typeColumnWidth) + 
-                          "ID".padEnd(idColumnWidth) + 
-                          "URI".padEnd(uriColumnWidth) + 
-                          "URI-CN".padEnd(uriCnColumnWidth) + 
-                          "Name".padEnd(nameColumnWidth);
-            Ec.waiting(header);
-            
-            // 分隔线
-            const separator = "-".repeat(5) + " " +
-                             "-".repeat(typeColumnWidth - 1) + " " +
-                             "-".repeat(idColumnWidth - 1) + " " +
-                             "-".repeat(uriColumnWidth - 1) + " " +
-                             "-".repeat(uriCnColumnWidth - 1) + " " +
-                             "-".repeat(nameColumnWidth - 1);
-            Ec.waiting(separator);
-            
-            // 数据行
-            agents.forEach((agent, index) => {
-                const name = `${agent.argName} (${agent.name})`;
-                const row = `${(index + 1).toString().padEnd(6)}` +
-                           `${agent.type.padEnd(typeColumnWidth)}` +
-                           `${agent.id.padEnd(idColumnWidth).red.bold}` +
-                           `${(agent.uri || '').padEnd(uriColumnWidth)}` +
-                           `${(agent.uriCn || '').padEnd(uriCnColumnWidth)}` +
-                           name.padEnd(nameColumnWidth);
-                Ec.waiting(row);
-            });
-            
-            Ec.waiting(separator);
-            Ec.waiting(`${agents.length + 1}. 退出`);
-            Ec.askClose();
-            process.exit(1);
-        }
-        
-        // 处理选中的 agent
-        await _processAgent(agent);
-        
-    } catch (error) {
-        Ec.error(`❌ 执行过程中发生错误: ${error.message}`);
-        Ec.askClose();
-        process.exit(1);
-    }
-};
-
-/**
- * 处理选中的 agent
- * @param {Object} agent 选中的 agent 对象
- */
-const _processAgent = async (agent) => {
-    try {
-        // 构建模板路径
-        const templatePath = path.resolve(__dirname, `../_template/PROMPT/${agent.type}/${agent.id}.ejs`);
-        
-        // 检查模板文件是否存在
-        if (!fs.existsSync(templatePath)) {
-            Ec.error(`❌ 未找到模板文件: ${templatePath}`);
-            Ec.askClose();
-            process.exit(1);
-        }
-        
-        // 提取模板内容
-        const content = await _extractTemplateContent(templatePath);
-        
-        // 复制到剪贴板
-        await _copyToClipboard(content);
-        
-        // 显示警告提示
-        Ec.waiting('');
-        Ec.warn('⚠️  为了获得最佳效果，请在您的 IDE 工具窗口中选择对应的 Agent 来配合使用此提示词');
-        
-        Ec.askClose();
-        process.exit(0);
-    } catch (error) {
-        Ec.error(`❌ 处理 Agent 时发生错误: ${error.message}`);
-        Ec.askClose();
-        process.exit(1);
-    }
-};