sloth-d2c-mcp 1.0.4-beta100

package/dist/build/plugin/manager.js

@@ -0,0 +1,129 @@
+import { loadAllPlugins, loadWorkspacePlugins } from './loader.js';
+/**
+ * 插件管理器 - 负责管理生命周期 hooks 并按顺序调用插件
+ */
+class PluginManager {
+    plugins = [];
+    initialized = false;
+    workspaceRoot = '';
+    /**
+     * 设置工作区根目录
+     * @param root 工作区根目录
+     */
+    setWorkspaceRoot(root) {
+        this.workspaceRoot = root;
+        // 重置初始化状态，以便重新加载工作区插件
+        if (this.initialized) {
+            this.reset();
+        }
+    }
+    /**
+     * 获取工作区根目录
+     */
+    getWorkspaceRoot() {
+        return this.workspaceRoot;
+    }
+    /**
+     * 初始化插件管理器
+     * 如果设置了工作区根目录，则加载工作区插件
+     * 否则加载全局插件（兼容旧逻辑）
+     */
+    async initialize() {
+        if (this.initialized) {
+            return;
+        }
+        try {
+            if (this.workspaceRoot) {
+                // 加载工作区声明的插件
+                console.log(`[sloth] 从工作区加载插件: ${this.workspaceRoot}`);
+                this.plugins = await loadWorkspacePlugins(this.workspaceRoot);
+            }
+            else {
+                // 兼容旧逻辑：加载全局插件
+                console.log(`[sloth] 未设置工作区，加载全局插件`);
+                this.plugins = await loadAllPlugins();
+            }
+            this.initialized = true;
+            console.log(`[sloth] 插件管理器初始化完成，已加载 ${this.plugins.length} 个插件`);
+        }
+        catch (error) {
+            console.error('[sloth] 插件管理器初始化失败:', error);
+            // 如果是工作区模式，抛出错误让调用方知道
+            if (this.workspaceRoot) {
+                throw error;
+            }
+            this.plugins = [];
+            this.initialized = true;
+        }
+    }
+    /**
+     * 获取已加载的插件列表
+     */
+    getPlugins() {
+        return [...this.plugins];
+    }
+    /**
+     * 注册一个插件
+     */
+    registerPlugin(plugin) {
+        this.plugins.push(plugin);
+        console.log(`[sloth] 插件 ${plugin.name} 已注册`);
+    }
+    /**
+     * 执行指定的 hook
+     * @param hookName hook 名称
+     * @param params hook 参数
+     * 按顺序调用所有插件，每个插件都会执行，返回最后一个成功的结果
+     * 每个插件执行时会收到上一个插件的信息 { lastPlugin, returnVal }
+     */
+    async executeHook(hookName, params) {
+        if (!this.initialized) {
+            await this.initialize();
+        }
+        const pluginsWithHook = this.plugins.filter((p) => typeof p[hookName] === 'function');
+        if (pluginsWithHook.length === 0) {
+            console.log(`[sloth] 没有插件注册 ${hookName} hook`);
+            return params;
+        }
+        let lastPlugin = '';
+        let lastReturnVal = null;
+        for (const plugin of pluginsWithHook) {
+            try {
+                console.log(`[sloth] 执行插件 ${plugin.name} 的 ${hookName} hook`);
+                const hookFn = plugin[hookName];
+                const context = {
+                    lastPlugin,
+                    returnVal: lastReturnVal,
+                    config: plugin._config,
+                };
+                const result = await hookFn({
+                    ...params,
+                    ...(lastReturnVal || {}),
+                    _config: context.config,
+                    workspaceRoot: this.workspaceRoot,
+                }, context);
+                // 更新上一个插件信息
+                lastPlugin = plugin.name;
+                lastReturnVal = {
+                    ...params,
+                    ...(lastReturnVal || {}),
+                    ...(result || {}),
+                    workspaceRoot: this.workspaceRoot,
+                };
+            }
+            catch (error) {
+                console.error(`[sloth] 插件 ${plugin.name} 执行 ${hookName} 出错:`, error);
+            }
+        }
+        return lastReturnVal || params;
+    }
+    /**
+     * 重置插件管理器
+     */
+    reset() {
+        this.plugins = [];
+        this.initialized = false;
+    }
+}
+// 导出单例实例
+export const pluginManager = new PluginManager();

package/dist/build/plugin/types.js

@@ -0,0 +1,6 @@
+export const PLUGIN_HOOKS = [
+    'afterSubmitConfigData',
+    'onFetchImage',
+    'onSamplingComplete',
+    'beforeFinalPrompt',
+];