momo-ai 1.0.20 → 1.0.22

package/src/executor/executeAsk.js

@@ -0,0 +1,274 @@
+const fs = require('fs');
+const path = require('path');
+const Ec = require('../epic');
+const { selectSingle } = require('../utils/momo-menu');
+const { parseFile, exists } = require('../utils/momo-file-utils');
+require('colors');
+
+/**
+ * 扫描模板目录，加载所有提示词文件
+ * @param {string} templateDir 模板目录路径
+ * @returns {Array} 提示词文件列表
+ */
+const _scanTemplates = (templateDir) => {
+    if (!exists(templateDir)) {
+        return [];
+    }
+    
+    const files = fs.readdirSync(templateDir);
+    const templates = [];
+    
+    for (const file of files) {
+        if (!file.endsWith('.md')) {
+            continue;
+        }
+        
+        const filePath = path.join(templateDir, file);
+        try {
+            if (fs.statSync(filePath).isFile()) {
+                const parsed = parseFile(filePath);
+                if (parsed && parsed.attributes) {
+                    // 从 attributes 中提取 title 和 version
+                    // 注意：parseYaml 返回的值可能包含前后空格
+                    const title = String(parsed.attributes.title || '').trim() || file.replace('.md', '');
+                    const version = String(parsed.attributes.version || '').trim() || '1.0.0';
+                    
+                    // 读取完整文件内容用于提取 BEGIN/END 之间的内容
+                    const fullContent = fs.readFileSync(filePath, 'utf8');
+                    
+                    templates.push({
+                        file: file,
+                        filePath: filePath,
+                        title: title,
+                        version: version,
+                        content: fullContent  // 使用完整内容，而不是 parsed.body
+                    });
+                } else {
+                    // 如果解析失败，尝试直接读取文件内容
+                    const fullContent = fs.readFileSync(filePath, 'utf8');
+                    // 尝试手动解析 front-matter
+                    const frontMatterMatch = fullContent.match(/^---\s*\n([\s\S]+?)\n---/);
+                    if (frontMatterMatch) {
+                        const yamlContent = frontMatterMatch[1];
+                        const lines = yamlContent.split('\n');
+                        let title = file.replace('.md', '');
+                        let version = '1.0.0';
+                        
+                        for (const line of lines) {
+                            const titleMatch = line.match(/^title:\s*(.+)$/);
+                            if (titleMatch) {
+                                title = titleMatch[1].trim();
+                            }
+                            const versionMatch = line.match(/^version:\s*(.+)$/);
+                            if (versionMatch) {
+                                version = versionMatch[1].trim();
+                            }
+                        }
+                        
+                        templates.push({
+                            file: file,
+                            filePath: filePath,
+                            title: title,
+                            version: version,
+                            content: fullContent
+                        });
+                    } else {
+                        Ec.warn(`⚠ 无法解析模板文件: ${file} (缺少 front-matter)`);
+                    }
+                }
+            }
+        } catch (error) {
+            Ec.warn(`⚠ 读取模板文件失败: ${file} - ${error.message}`);
+        }
+    }
+    
+    return templates;
+};
+
+/**
+ * 提取 --- BEGIN 和 --- END 之间的内容
+ * @param {string} content 文件内容
+ * @returns {string|null} 提取的内容，如果未找到返回 null
+ */
+const _extractPromptContent = (content) => {
+    const beginMatch = content.indexOf('--- BEGIN');
+    const endMatch = content.indexOf('--- END');
+    
+    if (beginMatch === -1 || endMatch === -1 || endMatch <= beginMatch) {
+        return null;
+    }
+    
+    // 提取 BEGIN 之后到 END 之前的内容
+    const startPos = content.indexOf('\n', beginMatch) + 1;
+    const endPos = endMatch;
+    
+    if (startPos <= 0 || endPos <= startPos) {
+        return null;
+    }
+    
+    return content.substring(startPos, endPos).trim();
+};
+
+/**
+ * 显示模板列表（表格形式）
+ * @param {Array} templates 模板列表
+ */
+const _displayTemplatesTable = (templates) => {
+    if (templates.length === 0) {
+        Ec.warn('未找到任何提示词模板');
+        return;
+    }
+    
+    // 计算列宽
+    const titleWidth = Math.max(
+        '标题'.length,
+        ...templates.map(t => t.title.length)
+    );
+    const versionWidth = Math.max(
+        '版本'.length,
+        ...templates.map(t => t.version.length)
+    );
+    
+    // 表头
+    console.log('');
+    console.log('─'.repeat(60));
+    console.log(' 提示词模板列表'.green.bold);
+    console.log('─'.repeat(60));
+    
+    // 表头行
+    const header = `  ${'标题'.padEnd(titleWidth)}  ${'版本'.padEnd(versionWidth)}`;
+    console.log(header.cyan);
+    console.log('─'.repeat(60));
+    
+    // 数据行
+    templates.forEach((template, index) => {
+        const title = template.title.padEnd(titleWidth);
+        const version = template.version.padEnd(versionWidth);
+        console.log(`  ${title}  ${version}`);
+    });
+    
+    console.log('─'.repeat(60));
+    console.log('');
+};
+
+/**
+ * 创建菜单项列表
+ * @param {Array} templates 模板列表
+ * @returns {Array} 菜单项列表
+ */
+const _createMenuItems = (templates) => {
+    return templates.map((template, index) => ({
+        name: `${template.title} (${template.version})`,
+        description: template.file,
+        index: index
+    }));
+};
+
+module.exports = async (options) => {
+    try {
+        // 获取主项目根目录（momo 命令所在的目录）
+        const mainProjectRoot = path.resolve(__dirname, '../..');
+        const templateDir = path.join(mainProjectRoot, 'src', '_template', 'R2MO');
+        
+        // 1. 检查模板目录是否存在
+        if (!exists(templateDir)) {
+            Ec.error(`❌ 模板目录不存在: ${templateDir}`);
+            console.log('');
+            Ec.warn('请确保主项目中存在 src/_template/R2MO 目录');
+            console.log('');
+            process.exit(1);
+        }
+        
+        // 2. 扫描模板文件
+        Ec.waiting('正在扫描提示词模板...');
+        const templates = _scanTemplates(templateDir);
+        
+        if (templates.length === 0) {
+            Ec.warn('未找到任何提示词模板文件');
+            console.log('');
+            Ec.info(`请确保在 ${templateDir} 目录中存在 .md 格式的模板文件`);
+            console.log('');
+            process.exit(1);
+        }
+        
+        Ec.info(`✓ 找到 ${templates.length} 个提示词模板`);
+        
+        // 3. 显示表格
+        _displayTemplatesTable(templates);
+        
+        // 4. 创建菜单并让用户选择
+        const menuItems = _createMenuItems(templates);
+        const selectedItem = await selectSingle(menuItems, '请选择提示词模板');
+        
+        if (selectedItem === null || selectedItem === undefined) {
+            Ec.warn('已取消选择');
+            process.exit(0);
+        }
+        
+        // 从选中的菜单项中获取索引
+        const selectedIndex = selectedItem.index;
+        if (selectedIndex === undefined || selectedIndex === null || selectedIndex < 0 || selectedIndex >= templates.length) {
+            Ec.error('❌ 无效的选择');
+            process.exit(1);
+        }
+        
+        const selectedTemplate = templates[selectedIndex];
+        
+        if (!selectedTemplate || !selectedTemplate.content) {
+            Ec.error('❌ 模板数据无效');
+            process.exit(1);
+        }
+        
+        // 5. 提取提示词内容
+        const promptContent = _extractPromptContent(selectedTemplate.content);
+        
+        if (!promptContent) {
+            Ec.error(`❌ 无法从模板中提取提示词内容`);
+            console.log('');
+            Ec.warn('请确保模板文件包含 --- BEGIN 和 --- END 标记');
+            console.log('');
+            process.exit(1);
+        }
+        
+        // 6. 复制到剪切板
+        try {
+            await Ec.outCopy(promptContent);
+            Ec.info('✓ 提示词已复制到剪切板');
+        } catch (copyError) {
+            Ec.warn(`⚠ 复制到剪切板失败: ${copyError.message}`);
+            console.log('');
+            Ec.info('提示词内容：');
+            console.log('─'.repeat(60));
+            console.log(promptContent);
+            console.log('─'.repeat(60));
+            console.log('');
+            process.exit(1);
+        }
+        
+        // 7. 显示详细信息
+        console.log('');
+        console.log('─'.repeat(60));
+        console.log(' 模板详细信息'.green.bold);
+        console.log('─'.repeat(60));
+        console.log(` 文件: ${selectedTemplate.file}`);
+        console.log(` 标题: ${selectedTemplate.title}`);
+        console.log(` 版本: ${selectedTemplate.version}`);
+        console.log(` 路径: ${selectedTemplate.filePath}`);
+        console.log('─'.repeat(60));
+        console.log('');
+        console.log(' 提示词内容：'.cyan);
+        console.log('─'.repeat(60));
+        console.log(promptContent);
+        console.log('─'.repeat(60));
+        console.log('');
+        
+        Ec.info('🎉 操作完成！');
+        console.log('');
+        
+        process.exit(0);
+    } catch (error) {
+        Ec.error(`❌ 执行失败: ${error.message}`);
+        console.log('');
+        process.exit(1);
+    }
+};