341 lines
9.9 KiB
JavaScript
341 lines
9.9 KiB
JavaScript
#!/usr/bin/env node
|
||
|
||
const fs = require('fs');
|
||
const path = require('path');
|
||
|
||
// 简单的命令行参数处理
|
||
const args = process.argv.slice(2);
|
||
const command = args[0] || 'protocols'; // 默认执行protocols命令，而不是help
|
||
const param = args[1]; // role命令时的角色文件路径
|
||
|
||
// 获取脚本所在目录和PromptX根目录
|
||
const scriptDir = __dirname;
|
||
const promptxDir = scriptDir; // 脚本现在就在PromptX目录内
|
||
|
||
/**
|
||
* 打印所有协议内容
|
||
*/
|
||
function printProtocols() {
|
||
// 定义目录优先级顺序
|
||
const directories = [
|
||
{ path: path.join(promptxDir, 'protocol'), name: 'protocol' },
|
||
{ path: path.join(promptxDir, 'core'), name: 'core' },
|
||
{ path: path.join(promptxDir, 'resource'), name: 'resource' }
|
||
];
|
||
|
||
let allFiles = [];
|
||
|
||
// 递归查找文件函数
|
||
function collectMarkdownFiles(dir) {
|
||
if (!fs.existsSync(dir)) {
|
||
console.warn(`警告: 目录不存在 ${dir}`);
|
||
return [];
|
||
}
|
||
|
||
let files = [];
|
||
const entries = fs.readdirSync(dir, { withFileTypes: true });
|
||
|
||
for (const entry of entries) {
|
||
const fullPath = path.join(dir, entry.name);
|
||
|
||
if (entry.isDirectory()) {
|
||
files = files.concat(collectMarkdownFiles(fullPath));
|
||
} else if (entry.isFile() && entry.name.endsWith('.md')) {
|
||
files.push(fullPath);
|
||
}
|
||
}
|
||
|
||
return files;
|
||
}
|
||
|
||
// 按目录优先级收集文件
|
||
for (const dir of directories) {
|
||
const dirFiles = collectMarkdownFiles(dir.path);
|
||
|
||
// 每个目录内的文件按字母顺序排序
|
||
dirFiles.sort();
|
||
|
||
// 合并到总文件列表
|
||
allFiles = allFiles.concat(dirFiles);
|
||
|
||
console.log(`从 ${dir.name} 目录收集了 ${dirFiles.length} 个文件`);
|
||
}
|
||
|
||
// 没有文件时的提示
|
||
if (allFiles.length === 0) {
|
||
console.log("未找到任何协议文件。请确认PromptX目录结构是否正确。");
|
||
return;
|
||
}
|
||
|
||
// 打印每个文件
|
||
for (const file of allFiles) {
|
||
const relativePath = path.relative(promptxDir, file);
|
||
const separator = "=".repeat(80);
|
||
console.log(`\n${separator}\n### 文件: ${relativePath}\n${separator}\n`);
|
||
|
||
try {
|
||
const content = fs.readFileSync(file, 'utf8');
|
||
console.log(content);
|
||
} catch (err) {
|
||
console.error(`读取文件错误: ${file}`, err);
|
||
}
|
||
}
|
||
|
||
console.log(`\n总计读取了 ${allFiles.length} 个协议文件。`);
|
||
}
|
||
|
||
/**
|
||
* 打印指定角色内容
|
||
*/
|
||
function printRole(rolePath) {
|
||
// 如果传入的是相对路径，则基于PromptX目录解析
|
||
let fullPath;
|
||
if (path.isAbsolute(rolePath)) {
|
||
fullPath = rolePath;
|
||
} else {
|
||
fullPath = path.join(promptxDir, rolePath);
|
||
}
|
||
|
||
if (!fs.existsSync(fullPath)) {
|
||
console.error(`错误: 角色文件不存在: ${fullPath}`);
|
||
return;
|
||
}
|
||
|
||
try {
|
||
const content = fs.readFileSync(fullPath, 'utf8');
|
||
const separator = "=".repeat(80);
|
||
console.log(`\n${separator}\n### 角色文件: ${path.relative(promptxDir, fullPath)}\n${separator}\n`);
|
||
console.log(content);
|
||
} catch (err) {
|
||
console.error(`读取角色文件错误: ${fullPath}`, err);
|
||
}
|
||
}
|
||
|
||
/**
|
||
* 打印指定路径的文件内容
|
||
*/
|
||
function printFile(filePath) {
|
||
// 如果传入的是相对路径，则基于PromptX目录解析
|
||
let fullPath;
|
||
if (path.isAbsolute(filePath)) {
|
||
fullPath = filePath;
|
||
} else {
|
||
fullPath = path.join(promptxDir, filePath);
|
||
}
|
||
|
||
if (!fs.existsSync(fullPath)) {
|
||
console.error(`错误: 文件不存在: ${fullPath}`);
|
||
return;
|
||
}
|
||
|
||
try {
|
||
const content = fs.readFileSync(fullPath, 'utf8');
|
||
const separator = "=".repeat(80);
|
||
console.log(`\n${separator}\n### 文件: ${path.relative(promptxDir, fullPath)}\n${separator}\n`);
|
||
console.log(content);
|
||
} catch (err) {
|
||
console.error(`读取文件错误: ${fullPath}`, err);
|
||
}
|
||
}
|
||
|
||
/**
|
||
* 添加记忆条目
|
||
* @param {string} content - 记忆内容
|
||
* @param {object} options - 配置选项
|
||
*/
|
||
function addMemory(content, options = {}) {
|
||
const defaultOptions = {
|
||
tags: ['其他'],
|
||
score: 5,
|
||
duration: '短期',
|
||
timestamp: new Date().toLocaleString('zh-CN', {
|
||
year: 'numeric',
|
||
month: '2-digit',
|
||
day: '2-digit',
|
||
hour: '2-digit',
|
||
minute: '2-digit'
|
||
})
|
||
};
|
||
|
||
// 确保选项中的数组和对象被正确合并
|
||
const finalOptions = {
|
||
...defaultOptions,
|
||
...options,
|
||
tags: options.tags && options.tags.length > 0 ? options.tags : defaultOptions.tags
|
||
};
|
||
|
||
console.log('最终选项:', finalOptions); // 添加调试输出
|
||
|
||
// 构建记忆条目,确保格式统一
|
||
const memoryEntry = `\n- ${content.trim()} ${finalOptions.tags.map(tag => `#${tag}`).join(' ')} #评分:${finalOptions.score} #有效期:${finalOptions.duration} #时间:${finalOptions.timestamp}\n`;
|
||
|
||
// 确保.memory目录存在
|
||
const memoryDir = path.join(process.cwd(), '.memory');
|
||
if (!fs.existsSync(memoryDir)) {
|
||
fs.mkdirSync(memoryDir, { recursive: true });
|
||
}
|
||
|
||
// 追加到记忆文件
|
||
const memoryFile = path.join(memoryDir, 'declarative.md');
|
||
try {
|
||
// 如果文件不存在,创建文件并添加标题
|
||
if (!fs.existsSync(memoryFile)) {
|
||
fs.writeFileSync(memoryFile, '# 陈述性记忆\n\n## 高价值记忆（评分 ≥ 7）\n');
|
||
}
|
||
|
||
fs.appendFileSync(memoryFile, memoryEntry);
|
||
console.log('✅ 记忆已成功保存');
|
||
|
||
// 如果评分大于等于7,输出高价值提醒
|
||
if (finalOptions.score >= 7) {
|
||
console.log('🌟 这是一条高价值记忆');
|
||
}
|
||
} catch (err) {
|
||
console.error('❌ 记忆保存失败:', err);
|
||
}
|
||
}
|
||
|
||
/**
|
||
* 解析记忆命令参数
|
||
* @param {string} content - 记忆内容
|
||
* @param {string[]} args - 其他参数
|
||
*/
|
||
function parseMemoryArgs(content, args) {
|
||
const options = {
|
||
tags: [],
|
||
score: 5,
|
||
duration: '短期'
|
||
};
|
||
|
||
console.log('原始参数:', args); // 调试输出
|
||
|
||
// 解析标签和其他选项
|
||
for (let arg of args) {
|
||
// 移除参数前后的引号（如果有）
|
||
arg = arg.replace(/^['"]|['"]$/g, '').trim();
|
||
|
||
console.log('处理参数:', arg); // 调试输出
|
||
|
||
if (arg.startsWith('#')) {
|
||
// 检查是否是特殊标记
|
||
if (arg.includes('评分:')) {
|
||
const score = parseInt(arg.split('评分:')[1]);
|
||
console.log('解析评分:', score); // 调试输出
|
||
if (!isNaN(score) && score >= 1 && score <= 10) {
|
||
options.score = score;
|
||
}
|
||
} else if (arg.includes('有效期:')) {
|
||
const duration = arg.split('有效期:')[1];
|
||
console.log('解析有效期:', duration); // 调试输出
|
||
if (['短期', '中期', '长期'].includes(duration)) {
|
||
options.duration = duration;
|
||
}
|
||
} else {
|
||
// 普通标签
|
||
const tag = arg.slice(1);
|
||
console.log('解析标签:', tag); // 调试输出
|
||
if (tag) {
|
||
options.tags.push(tag);
|
||
}
|
||
}
|
||
}
|
||
}
|
||
|
||
// 如果没有标签,使用默认标签
|
||
if (options.tags.length === 0) {
|
||
options.tags = ['其他'];
|
||
}
|
||
|
||
console.log('解析结果:', options); // 调试输出
|
||
return options;
|
||
}
|
||
|
||
/**
|
||
* 打印帮助信息
|
||
*/
|
||
function printHelp() {
|
||
console.log(`
|
||
PromptX 工具 - 协议和角色内容查看器
|
||
|
||
使用方法:
|
||
node promptx.js - 打印所有协议内容 (按protocol、core、resource顺序)
|
||
node promptx.js protocols - 同上，打印所有协议内容
|
||
node promptx.js role <路径> - 打印指定角色文件内容
|
||
node promptx.js file <路径> - 打印指定文件内容
|
||
node promptx.js remember <内容> [选项] - 添加记忆条目
|
||
node promptx.js help - 显示此帮助信息
|
||
|
||
记忆命令选项:
|
||
#标签名 - 添加标签 (可多个)
|
||
score:数字 - 设置重要性评分 (1-10)
|
||
duration:时长 - 设置有效期 (短期/长期)
|
||
|
||
示例:
|
||
node promptx.js remember "用户提出了重要建议" #用户反馈 #改进建议 score:7 duration:长期
|
||
node promptx.js remember "临时配置信息" #配置 score:3
|
||
`);
|
||
}
|
||
|
||
// 根据命令执行相应功能
|
||
switch (command) {
|
||
case 'protocols':
|
||
printProtocols();
|
||
break;
|
||
case 'role':
|
||
if (!param) {
|
||
console.error('错误: 缺少角色文件路径');
|
||
printHelp();
|
||
} else {
|
||
printRole(param);
|
||
}
|
||
break;
|
||
case 'file':
|
||
if (!param) {
|
||
console.error('错误: 缺少文件路径');
|
||
printHelp();
|
||
} else {
|
||
printFile(param);
|
||
}
|
||
break;
|
||
case 'remember':
|
||
if (!param) {
|
||
console.error('错误: 缺少记忆内容');
|
||
console.log('使用方法: node promptx.js remember 记忆内容 [#标签1 #标签2] [#评分:7] [#有效期:长期]');
|
||
} else {
|
||
try {
|
||
// 获取所有参数
|
||
const allArgs = process.argv.slice(2); // 从 remember 开始的所有参数
|
||
console.log('所有参数:', allArgs);
|
||
|
||
// 找到第一个标签（以#开头的参数）的位置
|
||
const tagStartIndex = allArgs.findIndex(arg => arg.startsWith('#'));
|
||
console.log('标签起始位置:', tagStartIndex);
|
||
|
||
// 如果没有找到标签，使用所有剩余参数作为内容
|
||
const contentEndIndex = tagStartIndex === -1 ? allArgs.length : tagStartIndex;
|
||
|
||
// 组合记忆内容（去掉 remember 命令）
|
||
const memoryContent = allArgs.slice(1, contentEndIndex).join(' ');
|
||
console.log('记忆内容:', memoryContent);
|
||
|
||
// 获取所有标签和选项
|
||
const memoryArgs = allArgs.slice(contentEndIndex);
|
||
console.log('记忆参数:', memoryArgs);
|
||
|
||
const options = parseMemoryArgs(memoryContent, memoryArgs);
|
||
console.log('解析的选项:', options);
|
||
addMemory(memoryContent, options);
|
||
} catch (err) {
|
||
console.error('错误:', err.message);
|
||
console.log('使用方法: node promptx.js remember 记忆内容 [#标签1 #标签2] [#评分:7] [#有效期:长期]');
|
||
}
|
||
}
|
||
break;
|
||
case 'help':
|
||
printHelp();
|
||
break;
|
||
default:
|
||
console.error(`错误: 未知命令 "${command}"`);
|
||
printHelp();
|
||
break;
|
||
}
|