Memory MCP Server

/** * 触发条件匹配引擎 * 使用 Node.js 内置 vm 模块提供沙箱化的 JS 代码执行环境 */ import vm from 'vm'; import { matchKeys, matchKeysAll } from '../nlp/keywords.js'; // 默认超时时间（毫秒） const DEFAULT_TIMEOUT = 1000; /** * 创建安全的沙箱上下文 * @param {Object} context - 用户提供的上下文 * @returns {Object} 安全的沙箱上下文 */ function createSafeContext(context = {}) { // 创建一个干净的上下文对象，只包含安全的内置对象 const safeContext = { ...context, // 提供安全的内置对象（只读） Date: Date, Math: Math, RegExp: RegExp, JSON: JSON, Array: Array, Object: Object, String: String, Number: Number, Boolean: Boolean, parseInt: parseInt, parseFloat: parseFloat, isNaN: isNaN, isFinite: isFinite, // 明确禁止的对象 require: undefined, process: undefined, global: undefined, globalThis: undefined, Buffer: undefined, setTimeout: undefined, setInterval: undefined, setImmediate: undefined, clearTimeout: undefined, clearInterval: undefined, clearImmediate: undefined }; return safeContext; } /** * 在沙箱中执行 JS 代码 * @param {string} code - 要执行的 JS 代码 * @param {Object} context - 执行上下文 * @param {number} [timeout=DEFAULT_TIMEOUT] - 超时时间 * @returns {Promise<{result: any, error: Error|null}>} */ export async function executeSandboxed(code, context = {}, timeout = DEFAULT_TIMEOUT) { try { // 创建安全的上下文 const safeContext = createSafeContext(context); // 创建 VM 上下文 const vmContext = vm.createContext(safeContext); // 执行代码（带超时） const result = vm.runInContext(code, vmContext, { timeout: timeout, displayErrors: true, breakOnSigint: true }); // 返回结果 return { result: result, error: null }; } catch (error) { // 捕获并返回错误 return { result: null, error: error instanceof Error ? error : new Error(String(error)) }; } } /** * 评估触发条件 * @param {string} triggerCode - 触发条件的 JS 代码 * @param {Object} executionContext - 执行上下文 * @param {Array} executionContext.messages - 消息数组 * @param {string} executionContext.conversation_id - 对话 ID * @param {Object} executionContext.participants - 参与者信息 * @returns {Promise<boolean>} 是否触发 */ export async function evaluateTrigger(triggerCode, executionContext) { if (!triggerCode || typeof triggerCode !== 'string') { console.warn('Invalid trigger code'); return false; } try { // 准备沙箱上下文 const sandboxContext = { context: executionContext, // 注入关键词匹配函数 match_keys: function(messages, keys, scope = 'any', depth = 4) { return matchKeys(messages, keys, scope, depth); }, match_keys_all: function(messages, keys, scope = 'any', depth = 4) { return matchKeysAll(messages, keys, scope, depth); }, // 注入安全的内置对象 Date: Date, Math: Math, RegExp: RegExp, JSON: JSON, Array: Array, Object: Object, String: String, Number: Number, Boolean: Boolean }; // 执行触发条件代码 const { result, error } = await executeSandboxed(triggerCode, sandboxContext); if (error) { console.error('Error evaluating trigger:', error); return false; } // 将结果转换为布尔值 return Boolean(result); } catch (error) { console.error('Unexpected error in evaluateTrigger:', error); return false; } } /** * 测试触发条件（用于验证触发条件语法） * @param {string} triggerCode - 触发条件的 JS 代码 * @returns {Promise<{valid: boolean, error: Error|null}>} */ export async function testTrigger(triggerCode) { try { // 使用模拟上下文测试 const mockContext = { messages: [ { role: 'user', content: 'test message' } ], conversation_id: 'test', participants: {} }; const { error } = await executeSandboxed(triggerCode, { context: mockContext, match_keys: function() { return 0; }, match_keys_all: function() { return false; }, Date: Date, Math: Math, RegExp: RegExp }); return { valid: error === null, error: error }; } catch (error) { return { valid: false, error: error instanceof Error ? error : new Error(String(error)) }; } } export default { executeSandboxed, evaluateTrigger, testTrigger };