foliko 2.0.21 → 2.0.23

package/plugins/core/workflow/js-runner.js

@@ -1,318 +1,318 @@
-/**
- * JS Workflow Runner - Total.js Style (Secondary)
- * Executes workflows defined in JavaScript format using $.push(), $.next(), $.send() API
- */
-
-const { runScriptSafely } = require('../../../src/utils/sandbox');
-
-class JsRunner {
-  constructor(framework) {
-    this.framework = framework;
-    this._log = require('../../../src/common/logger').logger.child('JsRunner');
-  }
-
-  /**
-   * Execute a JS workflow
-   * @param {string|Function} workflowCode - Workflow code or exported function
-   * @param {Object} input - Input data
-   * @param {string} sessionId - Session ID for tool execution
-   * @returns {Promise<Object>} execution result
-   */
-  async execute(workflowCode, input = {}, sessionId = null) {
-    this._log.debug('Executing JS workflow');
-
-    const { FlowContext } = require('./context');
-    const context = new FlowContext(input, sessionId);
-    context.setFramework(this.framework);
-
-    // Create stage executor and attach to context
-    const stageExecutor = new JsStageExecutor(this.framework);
-    context._stageExecutor = stageExecutor;
-
-    try {
-      // Get the workflow function
-      let workflowFn;
-      if (typeof workflowCode === 'function') {
-        workflowFn = workflowCode;
-      } else if (typeof workflowCode === 'string') {
-        workflowFn = this._parseWorkflow(workflowCode);
-      } else {
-        throw new Error('Workflow must be a function or string');
-      }
-
-      // Create the $ object for Total.js style API
-      const $ = this._createFlowApi(context, stageExecutor);
-
-      // Execute the workflow function
-      const flowPromise = workflowFn($);
-
-      // If it returns a promise, wait for it
-      if (flowPromise && typeof flowPromise.then === 'function') {
-        await flowPromise;
-      }
-
-      // Execute stages until destroyed or no more pending next
-      await this._runStages(context, stageExecutor);
-
-      // Build output
-      const output = {};
-      for (const [key, value] of context._results) {
-        output[key] = value;
-      }
-
-      return {
-        success: !context._destroyed,
-        output,
-        result: context._sentData || output,
-      };
-    } catch (err) {
-      this._log.error(`JS Workflow execution error: ${err.message}`);
-      return {
-        success: false,
-        error: err.message,
-      };
-    }
-  }
-
-  /**
-   * Parse workflow code string into a function
-   * Supports:
-   *   FLOW(function($) { ... })  - Direct function
-   *   module.exports = function($) { ... }  - CommonJS export
-   *   module.exports.default = function($) { ... }  - Default export
-   */
-  _parseWorkflow(code) {
-    const sandboxContext = {
-      console: { log() {}, error() {}, warn() {}, info() {} },
-      Math,
-      JSON,
-      Promise,
-      Array,
-      Object,
-      Date,
-      RegExp,
-      Error,
-      Symbol,
-      Map,
-      Set,
-      module: { exports: {} },
-      exports: {},
-      FlowContext: require('./context').FlowContext,
-    };
-
-    try {
-      const ctx = require('vm').createContext(sandboxContext);
-
-      // Check which format the code uses
-      const isModuleExport = code.includes('module.exports');
-      let fn;
-
-      if (isModuleExport) {
-        // CommonJS export format: module.exports = function($) { ... }
-        const wrapped = `
-          (function(module, exports, FlowContext) {
-            ${code}
-            return module.exports;
-          })
-        `;
-        const factory = require('vm').runInContext(wrapped, ctx, { timeout: 5000 });
-        fn = factory(ctx.module, ctx.exports, require('./context').FlowContext);
-      } else if (code.includes('FLOW(')) {
-        // FLOW wrapper format: FLOW(function($) { ... })
-        const wrapped = `(function(FlowContext) { return (${code}) })`;
-        fn = require('vm').runInContext(wrapped, ctx, { timeout: 5000 });
-        fn = fn(require('./context').FlowContext);
-      } else {
-        // Assume it's a function directly
-        const wrapped = `(function(FlowContext) { return (${code}) })`;
-        fn = require('vm').runInContext(wrapped, ctx, { timeout: 5000 });
-        fn = fn(require('./context').FlowContext);
-      }
-
-      // Handle module.exports.default
-      if (fn && fn.default) {
-        fn = fn.default;
-      }
-
-      return fn;
-    } catch (err) {
-      throw new Error(`Failed to parse workflow: ${err.message}`);
-    }
-  }
-
-  /**
-   * Create the $ API object for Total.js style
-   */
-  _createFlowApi(context, stageExecutor) {
-    const $ = {
-      // Register a stage
-      push: (name, fn) => {
-        context.push(name, fn);
-        return $;
-      },
-
-      // Go to next stage (sync) - 支持直接传参
-      next: (name, ...args) => {
-        context.next(name, ...args);
-        return $;
-      },
-
-      // Go to next stage (async helper)
-      next2: (name, ...args) => {
-        return () => {
-          context.next(name, ...args);
-        };
-      },
-
-      // Send data to next stage
-      send: (data) => {
-        context.send(data);
-        return $;
-      },
-
-      // Input from previous stage (返回 sentData �?workflow input)
-      get input() {
-        return context._sentData !== undefined ? context._sentData : context.input;
-      },
-
-      // Flow-level persistent data
-      data: context.data,
-
-      // Built-in delay
-      wait: (ms) => context.wait(ms),
-
-      // $.tool - Tool accessor with list(), execute(), register(), get()
-      get tool() { return context.tool; },
-
-      // $.extension - Extension accessor with list(), register(), remove(), execute(), get()
-      get extension() { return context.extension; },
-
-      // $.workflow - Workflow accessor with list(), execute(), get()
-      get workflow() { return context.workflow; },
-
-      // $.agent - Agent accessor with create(), chat(), list()
-      get agent() { return context.agent; },
-
-      // $.prompt - Prompt accessor with send(), chat()
-      get prompt() { return context.prompt; },
-
-      // $.skill - Skill accessor with list(), execute(), get()
-      get skill() { return context.skill; },
-
-      // Execute script
-      script: (code) => context.script(code),
-
-      // End workflow
-      destroy: () => context.destroy(),
-
-      // Current stage info
-      get current() { return context.current; },
-      get prev() { return context.prev; },
-
-      // Stage registry
-      _stages: context._stages,
-    };
-
-    return $;
-  }
-
-  /**
-   * Run stages sequentially based on $.next() calls
-   */
-  async _runStages(context, stageExecutor) {
-    // Create $ API object for this run
-    const $ = this._createFlowApi(context, stageExecutor);
-
-    while (!context._destroyed && context._pendingNext) {
-      const nextName = context._pendingNext;
-
-      // Get args that were passed via $.next(name, ...args)
-      const args = context._sentData;
-      context._sentData = null;
-      context._pendingNext = null;
-
-      const fn = context._stages?.get(nextName);
-      if (!fn) {
-        this._log.warn(`Stage "${nextName}" not found`);
-        break;
-      }
-
-      context.prev = context.current;
-      context.current = nextName;
-
-      this._log.debug(`Executing JS stage: ${nextName}`);
-
-      try {
-        // Call fn with $ as first arg, then unpack args
-        // Supports: fn($) or fn($, a, b, c)
-        const fnArgs = args !== undefined
-          ? (Array.isArray(args) ? [$, ...args] : [$, args])
-          : [$];
-
-        const result = fn(...fnArgs);
-        if (result && typeof result.then === 'function') {
-          await result;
-        }
-      } catch (err) {
-        this._log.error(`Stage "${nextName}" error: ${err.message}`);
-        context.destroy();
-        throw err;
-      }
-    }
-  }
-}
-
-/**
- * JS Stage Executor - handles stage execution in JS workflows
- */
-class JsStageExecutor {
-  constructor(framework) {
-    this.framework = framework;
-    this._log = require('../../../src/common/logger').logger.child('JsStageExecutor');
-  }
-
-  /**
-   * Execute a stage from JSON config within a JS workflow
-   */
-  async executeStage(config, context) {
-    const { ActionStage } = require('./stages/action');
-    const { ChoiceStage } = require('./stages/choice');
-    const { EachStage } = require('./stages/each');
-    const { ParallelStage } = require('./stages/parallel');
-    const type = this._getStageType(config);
-
-    switch (type) {
-      case 'action': {
-        const stage = new ActionStage(config);
-        return await stage.execute(context);
-      }
-      case 'choice': {
-        const stage = new ChoiceStage(config);
-        const nextStage = await stage.execute(context);
-        if (nextStage) {
-          context._pendingNext = nextStage;
-        }
-        return { nextStage };
-      }
-      case 'each': {
-        const stage = new EachStage(config);
-        return await stage.execute(context);
-      }
-      case 'parallel': {
-        const stage = new ParallelStage(config);
-        return await stage.execute(context);
-      }
-      default:
-        throw new Error(`Unknown stage type: ${type}`);
-    }
-  }
-
-  _getStageType(config) {
-    if (config.choice || config.when || config.default) return 'choice';
-    if (config.each || config.over) return 'each';
-    if (config.parallel) return 'parallel';
-    return 'action';
-  }
-}
-
-module.exports = { JsRunner };
+/**
+ * JS Workflow Runner - Total.js Style (Secondary)
+ * Executes workflows defined in JavaScript format using $.push(), $.next(), $.send() API
+ */
+
+const { runScriptSafely } = require('../../../src/utils/sandbox');
+
+class JsRunner {
+  constructor(framework) {
+    this.framework = framework;
+    this._log = require('../../../src/common/logger').logger.child('JsRunner');
+  }
+
+  /**
+   * Execute a JS workflow
+   * @param {string|Function} workflowCode - Workflow code or exported function
+   * @param {Object} input - Input data
+   * @param {string} sessionId - Session ID for tool execution
+   * @returns {Promise<Object>} execution result
+   */
+  async execute(workflowCode, input = {}, sessionId = null) {
+    this._log.debug('Executing JS workflow');
+
+    const { FlowContext } = require('./context');
+    const context = new FlowContext(input, sessionId);
+    context.setFramework(this.framework);
+
+    // Create stage executor and attach to context
+    const stageExecutor = new JsStageExecutor(this.framework);
+    context._stageExecutor = stageExecutor;
+
+    try {
+      // Get the workflow function
+      let workflowFn;
+      if (typeof workflowCode === 'function') {
+        workflowFn = workflowCode;
+      } else if (typeof workflowCode === 'string') {
+        workflowFn = this._parseWorkflow(workflowCode);
+      } else {
+        throw new Error('Workflow must be a function or string');
+      }
+
+      // Create the $ object for Total.js style API
+      const $ = this._createFlowApi(context, stageExecutor);
+
+      // Execute the workflow function
+      const flowPromise = workflowFn($);
+
+      // If it returns a promise, wait for it
+      if (flowPromise && typeof flowPromise.then === 'function') {
+        await flowPromise;
+      }
+
+      // Execute stages until destroyed or no more pending next
+      await this._runStages(context, stageExecutor);
+
+      // Build output
+      const output = {};
+      for (const [key, value] of context._results) {
+        output[key] = value;
+      }
+
+      return {
+        success: !context._destroyed,
+        output,
+        result: context._sentData || output,
+      };
+    } catch (err) {
+      this._log.error(`JS Workflow execution error: ${err.message}`);
+      return {
+        success: false,
+        error: err.message,
+      };
+    }
+  }
+
+  /**
+   * Parse workflow code string into a function
+   * Supports:
+   *   FLOW(function($) { ... })  - Direct function
+   *   module.exports = function($) { ... }  - CommonJS export
+   *   module.exports.default = function($) { ... }  - Default export
+   */
+  _parseWorkflow(code) {
+    const sandboxContext = {
+      console: { log() {}, error() {}, warn() {}, info() {} },
+      Math,
+      JSON,
+      Promise,
+      Array,
+      Object,
+      Date,
+      RegExp,
+      Error,
+      Symbol,
+      Map,
+      Set,
+      module: { exports: {} },
+      exports: {},
+      FlowContext: require('./context').FlowContext,
+    };
+
+    try {
+      const ctx = require('vm').createContext(sandboxContext);
+
+      // Check which format the code uses
+      const isModuleExport = code.includes('module.exports');
+      let fn;
+
+      if (isModuleExport) {
+        // CommonJS export format: module.exports = function($) { ... }
+        const wrapped = `
+          (function(module, exports, FlowContext) {
+            ${code}
+            return module.exports;
+          })
+        `;
+        const factory = require('vm').runInContext(wrapped, ctx, { timeout: 5000 });
+        fn = factory(ctx.module, ctx.exports, require('./context').FlowContext);
+      } else if (code.includes('FLOW(')) {
+        // FLOW wrapper format: FLOW(function($) { ... })
+        const wrapped = `(function(FlowContext) { return (${code}) })`;
+        fn = require('vm').runInContext(wrapped, ctx, { timeout: 5000 });
+        fn = fn(require('./context').FlowContext);
+      } else {
+        // Assume it's a function directly
+        const wrapped = `(function(FlowContext) { return (${code}) })`;
+        fn = require('vm').runInContext(wrapped, ctx, { timeout: 5000 });
+        fn = fn(require('./context').FlowContext);
+      }
+
+      // Handle module.exports.default
+      if (fn && fn.default) {
+        fn = fn.default;
+      }
+
+      return fn;
+    } catch (err) {
+      throw new Error(`Failed to parse workflow: ${err.message}`);
+    }
+  }
+
+  /**
+   * Create the $ API object for Total.js style
+   */
+  _createFlowApi(context, stageExecutor) {
+    const $ = {
+      // Register a stage
+      push: (name, fn) => {
+        context.push(name, fn);
+        return $;
+      },
+
+      // Go to next stage (sync) - 支持直接传参
+      next: (name, ...args) => {
+        context.next(name, ...args);
+        return $;
+      },
+
+      // Go to next stage (async helper)
+      next2: (name, ...args) => {
+        return () => {
+          context.next(name, ...args);
+        };
+      },
+
+      // Send data to next stage
+      send: (data) => {
+        context.send(data);
+        return $;
+      },
+
+      // Input from previous stage (返回 sentData �?workflow input)
+      get input() {
+        return context._sentData !== undefined ? context._sentData : context.input;
+      },
+
+      // Flow-level persistent data
+      data: context.data,
+
+      // Built-in delay
+      wait: (ms) => context.wait(ms),
+
+      // $.tool - Tool accessor with list(), execute(), register(), get()
+      get tool() { return context.tool; },
+
+      // $.extension - Extension accessor with list(), register(), remove(), execute(), get()
+      get extension() { return context.extension; },
+
+      // $.workflow - Workflow accessor with list(), execute(), get()
+      get workflow() { return context.workflow; },
+
+      // $.agent - Agent accessor with create(), chat(), list()
+      get agent() { return context.agent; },
+
+      // $.prompt - Prompt accessor with send(), chat()
+      get prompt() { return context.prompt; },
+
+      // $.skill - Skill accessor with list(), execute(), get()
+      get skill() { return context.skill; },
+
+      // Execute script
+      script: (code) => context.script(code),
+
+      // End workflow
+      destroy: () => context.destroy(),
+
+      // Current stage info
+      get current() { return context.current; },
+      get prev() { return context.prev; },
+
+      // Stage registry
+      _stages: context._stages,
+    };
+
+    return $;
+  }
+
+  /**
+   * Run stages sequentially based on $.next() calls
+   */
+  async _runStages(context, stageExecutor) {
+    // Create $ API object for this run
+    const $ = this._createFlowApi(context, stageExecutor);
+
+    while (!context._destroyed && context._pendingNext) {
+      const nextName = context._pendingNext;
+
+      // Get args that were passed via $.next(name, ...args)
+      const args = context._sentData;
+      context._sentData = null;
+      context._pendingNext = null;
+
+      const fn = context._stages?.get(nextName);
+      if (!fn) {
+        this._log.warn(`Stage "${nextName}" not found`);
+        break;
+      }
+
+      context.prev = context.current;
+      context.current = nextName;
+
+      this._log.debug(`Executing JS stage: ${nextName}`);
+
+      try {
+        // Call fn with $ as first arg, then unpack args
+        // Supports: fn($) or fn($, a, b, c)
+        const fnArgs = args !== undefined
+          ? (Array.isArray(args) ? [$, ...args] : [$, args])
+          : [$];
+
+        const result = fn(...fnArgs);
+        if (result && typeof result.then === 'function') {
+          await result;
+        }
+      } catch (err) {
+        this._log.error(`Stage "${nextName}" error: ${err.message}`);
+        context.destroy();
+        throw err;
+      }
+    }
+  }
+}
+
+/**
+ * JS Stage Executor - handles stage execution in JS workflows
+ */
+class JsStageExecutor {
+  constructor(framework) {
+    this.framework = framework;
+    this._log = require('../../../src/common/logger').logger.child('JsStageExecutor');
+  }
+
+  /**
+   * Execute a stage from JSON config within a JS workflow
+   */
+  async executeStage(config, context) {
+    const { ActionStage } = require('./stages/action');
+    const { ChoiceStage } = require('./stages/choice');
+    const { EachStage } = require('./stages/each');
+    const { ParallelStage } = require('./stages/parallel');
+    const type = this._getStageType(config);
+
+    switch (type) {
+      case 'action': {
+        const stage = new ActionStage(config);
+        return await stage.execute(context);
+      }
+      case 'choice': {
+        const stage = new ChoiceStage(config);
+        const nextStage = await stage.execute(context);
+        if (nextStage) {
+          context._pendingNext = nextStage;
+        }
+        return { nextStage };
+      }
+      case 'each': {
+        const stage = new EachStage(config);
+        return await stage.execute(context);
+      }
+      case 'parallel': {
+        const stage = new ParallelStage(config);
+        return await stage.execute(context);
+      }
+      default:
+        throw new Error(`Unknown stage type: ${type}`);
+    }
+  }
+
+  _getStageType(config) {
+    if (config.choice || config.when || config.default) return 'choice';
+    if (config.each || config.over) return 'each';
+    if (config.parallel) return 'parallel';
+    return 'action';
+  }
+}
+
+module.exports = { JsRunner };