@eggjs/agent-runtime 3.73.0-beta.0 → 4.0.2-beta.4

package/dist/src/enhanceAgentController.js

@@ -1,90 +0,0 @@
-"use strict";
-var __importDefault = (this && this.__importDefault) || function (mod) {
-    return (mod && mod.__esModule) ? mod : { "default": mod };
-};
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.enhanceAgentController = enhanceAgentController;
-const node_path_1 = __importDefault(require("node:path"));
-const FileAgentStore_1 = require("./FileAgentStore");
-const agentDefaults_1 = require("./agentDefaults");
-const AGENT_METHOD_NAMES = [
-    'createThread', 'getThread', 'asyncRun', 'streamRun', 'syncRun', 'getRun', 'cancelRun',
-];
-const NOT_IMPLEMENTED = Symbol.for('AGENT_NOT_IMPLEMENTED');
-const AGENT_ENHANCED = Symbol.for('AGENT_CONTROLLER_ENHANCED');
-// Enhance an AgentController class with smart default implementations.
-//
-// Called by the plugin/controller lifecycle hook AFTER the decorator has set
-// HTTP metadata and injected stub methods. Detects which methods are
-// user-defined vs stubs (via Symbol.for('AGENT_NOT_IMPLEMENTED') marker)
-// and replaces stubs with store-backed default implementations.
-// Also wraps init()/destroy() to manage the AgentStore lifecycle.
-//
-// Prerequisites:
-// - The class must be marked with Symbol.for('AGENT_CONTROLLER') (otherwise this is a no-op).
-// - Stub methods must be marked with Symbol.for('AGENT_NOT_IMPLEMENTED').
-function enhanceAgentController(clazz) {
-    // Only enhance classes marked by @AgentController decorator
-    if (!clazz[Symbol.for('AGENT_CONTROLLER')]) {
-        return;
-    }
-    // Guard against repeated enhancement (e.g., multiple lifecycle hook calls)
-    if (clazz[AGENT_ENHANCED]) {
-        return;
-    }
-    // Identify which methods are stubs vs user-defined
-    const stubMethods = new Set();
-    for (const name of AGENT_METHOD_NAMES) {
-        const method = clazz.prototype[name];
-        if (!method || method[NOT_IMPLEMENTED]) {
-            stubMethods.add(name);
-        }
-    }
-    // Wrap init() lifecycle to create store and task tracking
-    const originalInit = clazz.prototype.init;
-    clazz.prototype.init = async function () {
-        // Allow user to provide custom store via createStore()
-        if (typeof this.createStore === 'function') {
-            this.__agentStore = await this.createStore();
-        }
-        else {
-            const dataDir = process.env.TEGG_AGENT_DATA_DIR || node_path_1.default.join(process.cwd(), '.agent-data');
-            this.__agentStore = new FileAgentStore_1.FileAgentStore({ dataDir });
-        }
-        if (this.__agentStore.init) {
-            await this.__agentStore.init();
-        }
-        this.__runningTasks = new Map();
-        if (originalInit) {
-            await originalInit.call(this);
-        }
-    };
-    // Wrap destroy() lifecycle to wait for in-flight tasks and cleanup
-    const originalDestroy = clazz.prototype.destroy;
-    clazz.prototype.destroy = async function () {
-        var _a, _b;
-        // Wait for in-flight background tasks
-        if ((_a = this.__runningTasks) === null || _a === void 0 ? void 0 : _a.size) {
-            const pending = Array.from(this.__runningTasks.values()).map((t) => t.promise);
-            await Promise.allSettled(pending);
-        }
-        // Destroy store
-        if ((_b = this.__agentStore) === null || _b === void 0 ? void 0 : _b.destroy) {
-            await this.__agentStore.destroy();
-        }
-        if (originalDestroy) {
-            await originalDestroy.call(this);
-        }
-    };
-    // Inject smart defaults for stub methods
-    for (const methodName of AGENT_METHOD_NAMES) {
-        if (!stubMethods.has(methodName))
-            continue;
-        const factory = agentDefaults_1.AGENT_DEFAULT_FACTORIES[methodName];
-        if (factory) {
-            clazz.prototype[methodName] = factory();
-        }
-    }
-    clazz[AGENT_ENHANCED] = true;
-}
-//# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoiZW5oYW5jZUFnZW50Q29udHJvbGxlci5qcyIsInNvdXJjZVJvb3QiOiIiLCJzb3VyY2VzIjpbIi4uLy4uL3NyYy9lbmhhbmNlQWdlbnRDb250cm9sbGVyLnRzIl0sIm5hbWVzIjpbXSwibWFwcGluZ3MiOiI7Ozs7O0FBd0JBLHdEQXVFQztBQS9GRCwwREFBNkI7QUFFN0IscURBQWtEO0FBRWxELG1EQUEwRDtBQUUxRCxNQUFNLGtCQUFrQixHQUFHO0lBQ3pCLGNBQWMsRUFBRSxXQUFXLEVBQUUsVUFBVSxFQUFFLFdBQVcsRUFBRSxTQUFTLEVBQUUsUUFBUSxFQUFFLFdBQVc7Q0FDdkYsQ0FBQztBQUVGLE1BQU0sZUFBZSxHQUFHLE1BQU0sQ0FBQyxHQUFHLENBQUMsdUJBQXVCLENBQUMsQ0FBQztBQUM1RCxNQUFNLGNBQWMsR0FBRyxNQUFNLENBQUMsR0FBRyxDQUFDLDJCQUEyQixDQUFDLENBQUM7QUFFL0QsdUVBQXVFO0FBQ3ZFLEVBQUU7QUFDRiw2RUFBNkU7QUFDN0UscUVBQXFFO0FBQ3JFLHlFQUF5RTtBQUN6RSxnRUFBZ0U7QUFDaEUsa0VBQWtFO0FBQ2xFLEVBQUU7QUFDRixpQkFBaUI7QUFDakIsOEZBQThGO0FBQzlGLDBFQUEwRTtBQUMxRSxTQUFnQixzQkFBc0IsQ0FBQyxLQUF3QjtJQUM3RCw0REFBNEQ7SUFDNUQsSUFBSSxDQUFFLEtBQWEsQ0FBQyxNQUFNLENBQUMsR0FBRyxDQUFDLGtCQUFrQixDQUFDLENBQUMsRUFBRSxDQUFDO1FBQ3BELE9BQU87SUFDVCxDQUFDO0lBRUQsMkVBQTJFO0lBQzNFLElBQUssS0FBYSxDQUFDLGNBQWMsQ0FBQyxFQUFFLENBQUM7UUFDbkMsT0FBTztJQUNULENBQUM7SUFFRCxtREFBbUQ7SUFDbkQsTUFBTSxXQUFXLEdBQUcsSUFBSSxHQUFHLEVBQVUsQ0FBQztJQUN0QyxLQUFLLE1BQU0sSUFBSSxJQUFJLGtCQUFrQixFQUFFLENBQUM7UUFDdEMsTUFBTSxNQUFNLEdBQUcsS0FBSyxDQUFDLFNBQVMsQ0FBQyxJQUFJLENBQUMsQ0FBQztRQUNyQyxJQUFJLENBQUMsTUFBTSxJQUFLLE1BQWMsQ0FBQyxlQUFlLENBQUMsRUFBRSxDQUFDO1lBQ2hELFdBQVcsQ0FBQyxHQUFHLENBQUMsSUFBSSxDQUFDLENBQUM7UUFDeEIsQ0FBQztJQUNILENBQUM7SUFFRCwwREFBMEQ7SUFDMUQsTUFBTSxZQUFZLEdBQUcsS0FBSyxDQUFDLFNBQVMsQ0FBQyxJQUFJLENBQUM7SUFDMUMsS0FBSyxDQUFDLFNBQVMsQ0FBQyxJQUFJLEdBQUcsS0FBSztRQUMxQix1REFBdUQ7UUFDdkQsSUFBSSxPQUFPLElBQUksQ0FBQyxXQUFXLEtBQUssVUFBVSxFQUFFLENBQUM7WUFDM0MsSUFBSSxDQUFDLFlBQVksR0FBRyxNQUFNLElBQUksQ0FBQyxXQUFXLEVBQUUsQ0FBQztRQUMvQyxDQUFDO2FBQU0sQ0FBQztZQUNOLE1BQU0sT0FBTyxHQUFHLE9BQU8sQ0FBQyxHQUFHLENBQUMsbUJBQW1CLElBQUksbUJBQUksQ0FBQyxJQUFJLENBQUMsT0FBTyxDQUFDLEdBQUcsRUFBRSxFQUFFLGFBQWEsQ0FBQyxDQUFDO1lBQzNGLElBQUksQ0FBQyxZQUFZLEdBQUcsSUFBSSwrQkFBYyxDQUFDLEVBQUUsT0FBTyxFQUFFLENBQUMsQ0FBQztRQUN0RCxDQUFDO1FBRUQsSUFBSSxJQUFJLENBQUMsWUFBWSxDQUFDLElBQUksRUFBRSxDQUFDO1lBQzNCLE1BQU8sSUFBSSxDQUFDLFlBQTJCLENBQUMsSUFBSyxFQUFFLENBQUM7UUFDbEQsQ0FBQztRQUVELElBQUksQ0FBQyxjQUFjLEdBQUcsSUFBSSxHQUFHLEVBQUUsQ0FBQztRQUVoQyxJQUFJLFlBQVksRUFBRSxDQUFDO1lBQ2pCLE1BQU0sWUFBWSxDQUFDLElBQUksQ0FBQyxJQUFJLENBQUMsQ0FBQztRQUNoQyxDQUFDO0lBQ0gsQ0FBQyxDQUFDO0lBRUYsbUVBQW1FO0lBQ25FLE1BQU0sZUFBZSxHQUFHLEtBQUssQ0FBQyxTQUFTLENBQUMsT0FBTyxDQUFDO0lBQ2hELEtBQUssQ0FBQyxTQUFTLENBQUMsT0FBTyxHQUFHLEtBQUs7O1FBQzdCLHNDQUFzQztRQUN0QyxJQUFJLE1BQUEsSUFBSSxDQUFDLGNBQWMsMENBQUUsSUFBSSxFQUFFLENBQUM7WUFDOUIsTUFBTSxPQUFPLEdBQUcsS0FBSyxDQUFDLElBQUksQ0FBQyxJQUFJLENBQUMsY0FBYyxDQUFDLE1BQU0sRUFBRSxDQUFDLENBQUMsR0FBRyxDQUFDLENBQUMsQ0FBTSxFQUFFLEVBQUUsQ0FBQyxDQUFDLENBQUMsT0FBTyxDQUFDLENBQUM7WUFDcEYsTUFBTSxPQUFPLENBQUMsVUFBVSxDQUFDLE9BQU8sQ0FBQyxDQUFDO1FBQ3BDLENBQUM7UUFFRCxnQkFBZ0I7UUFDaEIsSUFBSSxNQUFBLElBQUksQ0FBQyxZQUFZLDBDQUFFLE9BQU8sRUFBRSxDQUFDO1lBQy9CLE1BQU0sSUFBSSxDQUFDLFlBQVksQ0FBQyxPQUFPLEVBQUUsQ0FBQztRQUNwQyxDQUFDO1FBRUQsSUFBSSxlQUFlLEVBQUUsQ0FBQztZQUNwQixNQUFNLGVBQWUsQ0FBQyxJQUFJLENBQUMsSUFBSSxDQUFDLENBQUM7UUFDbkMsQ0FBQztJQUNILENBQUMsQ0FBQztJQUVGLHlDQUF5QztJQUN6QyxLQUFLLE1BQU0sVUFBVSxJQUFJLGtCQUFrQixFQUFFLENBQUM7UUFDNUMsSUFBSSxDQUFDLFdBQVcsQ0FBQyxHQUFHLENBQUMsVUFBVSxDQUFDO1lBQUUsU0FBUztRQUMzQyxNQUFNLE9BQU8sR0FBRyx1Q0FBdUIsQ0FBQyxVQUFVLENBQUMsQ0FBQztRQUNwRCxJQUFJLE9BQU8sRUFBRSxDQUFDO1lBQ1osS0FBSyxDQUFDLFNBQVMsQ0FBQyxVQUFVLENBQUMsR0FBRyxPQUFPLEVBQUUsQ0FBQztRQUMxQyxDQUFDO0lBQ0gsQ0FBQztJQUVBLEtBQWEsQ0FBQyxjQUFjLENBQUMsR0FBRyxJQUFJLENBQUM7QUFDeEMsQ0FBQyJ9