85 lines
3.1 KiB
TypeScript
85 lines
3.1 KiB
TypeScript
// File path: service/extensionService.ts
|
|
|
|
/**
|
|
* ExtensionManage 是一个单例类，用于管理扩展的实例。
|
|
* 提供注册、触发和移除插件扩展的功能。
|
|
*/
|
|
import { ExtensionType } from "types/extensionType";
|
|
import React from "react";
|
|
|
|
class ExtensionManage {
|
|
/** 存储扩展的映射，键为扩展名称，值为插件名称和扩展的集合 */
|
|
private extensions: Map<string, Set<{ pluginName: string; extension: ExtensionType }>> = new Map();
|
|
/** ExtensionManage 的唯一实例 */
|
|
private static instance: ExtensionManage;
|
|
|
|
/** 私有构造函数，防止外部实例化 */
|
|
private constructor() {}
|
|
|
|
/**
|
|
* 获取 ExtensionManage 的唯一实例。
|
|
* @returns {ExtensionManage} 返回 ExtensionManage 的唯一实例。
|
|
*/
|
|
public static getInstance(): ExtensionManage {
|
|
if (!this.instance) {
|
|
this.instance = new ExtensionManage();
|
|
}
|
|
return this.instance;
|
|
}
|
|
|
|
/** 注册扩展 */
|
|
private register(extensionName: string, pluginName: string, extension: ExtensionType) {
|
|
const handlers = this.extensions.get(extensionName) || new Set();
|
|
handlers.add({ pluginName, extension });
|
|
this.extensions.set(extensionName, handlers);
|
|
}
|
|
|
|
/** 执行扩展方法 */
|
|
private executeExtensionMethod<T>(extensionName: string, method: keyof ExtensionType, ...args: any[]): Set<T> {
|
|
const result = new Set<T>();
|
|
const handlers = this.extensions.get(extensionName);
|
|
|
|
if (handlers) {
|
|
handlers.forEach(({ extension }) => {
|
|
const methodFunction = extension[method];
|
|
if (methodFunction) {
|
|
try {
|
|
const value = methodFunction(...args);
|
|
if (value && (typeof value === 'string' || React.isValidElement(value))) {
|
|
result.add(value as T);
|
|
}
|
|
} catch (error) {
|
|
console.error(`Error executing hook ${extensionName}:`, error);
|
|
}
|
|
}
|
|
});
|
|
}
|
|
return result;
|
|
}
|
|
|
|
/** 触发扩展的动作 */
|
|
private triggerAction(extensionName: string, ...args: any[]): void {
|
|
this.executeExtensionMethod<void>(extensionName, 'action', ...args);
|
|
}
|
|
|
|
/** 触发扩展的组件 */
|
|
private triggerComponent(extensionName: string, ...args: any[]): Set<React.FC> {
|
|
return this.executeExtensionMethod<React.FC>(extensionName, 'component', ...args);
|
|
}
|
|
|
|
/** 触发扩展的文本 */
|
|
private triggerText(extensionName: string, ...args: any[]): Set<string> {
|
|
return this.executeExtensionMethod<string>(extensionName, 'text', ...args);
|
|
}
|
|
|
|
/** 移除指定插件的扩展 */
|
|
private removePluginExtensions(pluginName: string) {
|
|
this.extensions.forEach((handlers, extensionName) => {
|
|
const newHandlers = new Set(
|
|
Array.from(handlers).filter(handler => handler.pluginName !== pluginName)
|
|
);
|
|
this.extensions.set(extensionName, newHandlers);
|
|
});
|
|
}
|
|
}
|