import { plugin, server, command, event, MiaoScriptConsole } from '@ms/api' import { injectable, inject, postConstruct, Container, DefaultContainer as container } from '@ms/container' import * as fs from '@ms/common/dist/fs' import { getPluginMetadatas, getPluginCommandMetadata, getPluginListenerMetadata, getPlugin, getPluginTabCompleterMetadata } from './utils' import { interfaces } from './interfaces'; @injectable() export class PluginManagerImpl implements plugin.PluginManager { @inject(plugin.PluginInstance) private pluginInstance: any; @inject(server.ServerType) private serverType: string; @inject(server.Console) private Console: MiaoScriptConsole; @inject(command.Command) private CommandManager: command.Command; @inject(event.Event) private EventManager: event.Event; private pluginMap: Map; @postConstruct() init() { if (this.pluginInstance !== null) { // 如果plugin不等于null 则代表是正式环境 console.info(`Initialization MiaoScript Plugin System: ${this.pluginInstance} ...`); this.pluginMap = new Map(); console.info(`${this.EventManager.mapEventName().toFixed(0)} ${this.serverType} Event Mapping Complate...`); } } scan(folder: string): void { var plugin = fs.file(root, folder); var files = [] // load common plugin .concat(this.scanFloder(plugin)) // load space plugin .concat(this.scanFloder(fs.file(plugin, this.serverType))) this.loadPlugins(files); } build(container: Container): void { this.buildPlugins(container); } load(...args: any[]): void { this.checkAndGet(args[0]).forEach(pl => this.runCatch(pl, 'load')); } enable(...args: any[]): void { this.checkAndGet(args[0]).forEach(pl => this.runCatch(pl, 'enable')); } disable(...args: any[]): void { this.checkAndGet(args[0]).forEach(pl => { this.runCatch(pl, 'disable'); this.EventManager.disable(pl); }); } reload(...args: any[]): void { this.checkAndGet(arguments[0]).forEach((pl: interfaces.Plugin) => { this.disable(pl); this.EventManager.disable(pl); this.loadPlugin(pl.description.source); pl = this.buildPlugin(getPlugin(pl.description.name)); this.load(pl); this.enable(pl); }) } private runCatch(pl: any, func: string) { try { pl[func].call(pl); } catch (ex) { console.console(`§6插件 §b${pl.description.name} §6执行 §d${func} §6方法时发生错误 §4${ex}`); console.ex(ex); } } private checkAndGet(name: string | interfaces.Plugin | undefined): Map | interfaces.Plugin[] { if (name == undefined) { return this.pluginMap; } if (!(name instanceof interfaces.Plugin) && !this.pluginMap.has(name)) { throw new Error(`Plugin ${name} not exist!`); } // 如果是插件 则直接返回 return [name as interfaces.Plugin]; } private scanFloder(plugin: any): string[] { var files = []; console.info(`Scanning Plugins in ${plugin} ...`); this.checkUpdateFolder(plugin); fs.list(plugin).forEach((file: any) => files.push(file.toFile())); return files; } /** * 更新插件 * @param path */ private checkUpdateFolder(path) { var update = fs.file(path, "update"); if (!update.exists()) { update.mkdirs(); } } private loadPlugins(files: any[]): void { this.loadJsPlugins(files); } /** * JS类型插件预加载 */ private loadJsPlugins(files: any[]) { files.filter(file => file.name.endsWith(".js")).forEach(file => this.loadPlugin(file)) } private loadPlugin(file: any) { try { this.updatePlugin(file); this.createPlugin(file); } catch (ex) { console.console(`§6插件 §b${file.name} §6初始化时发生错误 §4${ex.message}`); console.ex(ex); } } private updatePlugin(file: any) { var update = fs.file(fs.file(file.parentFile, 'update'), file.name); if (update.exists()) { console.info(`Auto Update Plugin ${file.name} ...`); fs.move(update, file, true); } } private createPlugin(file) { //@ts-ignore require(file, { cache: false }); } private beforeLoadHook(origin) { var result = origin; // // 注入 console 对象 // 给插件注入单独的 console // result += '\nvar console = new Console(); module.exports.console = console;'; // // 插件注入 self 对象 // result += '\nvar self = {}; module.exports.self = self;'; return result; } private buildPlugins(container: Container) { let pluginMetadatas = getPluginMetadatas(); pluginMetadatas.forEach(metadata => { this.buildPlugin(metadata); }); } private buildPlugin(metadata: interfaces.PluginMetadata) { try { let pluginInstance = container.getNamed(plugin.Plugin, metadata.name) if (pluginInstance.description.source + '' !== metadata.source + '') { console.warn(`find duplicate plugin ${pluginInstance.description.source} and ${metadata.source}. the first plugin will be ignore!`) } container.rebind(plugin.Plugin).to(metadata.target).inSingletonScope().whenTargetNamed(metadata.name); } catch{ container.bind(plugin.Plugin).to(metadata.target).inSingletonScope().whenTargetNamed(metadata.name); } let pluginInstance = container.getNamed(plugin.Plugin, metadata.name) this.pluginMap.set(metadata.name, pluginInstance); pluginInstance.description = metadata; // @ts-ignore pluginInstance.logger = new this.Console(metadata.name); this.registryCommand(pluginInstance); this.registryListener(pluginInstance); return pluginInstance; } private registryCommand(pluginInstance: interfaces.Plugin) { let cmds = getPluginCommandMetadata(pluginInstance); let tabs = getPluginTabCompleterMetadata(pluginInstance); cmds.forEach(cmd => { let tab = tabs.get(cmd.name); this.CommandManager.on(pluginInstance, cmd.name, { cmd: pluginInstance[cmd.executor].bind(pluginInstance), tab: tab ? pluginInstance[tab.executor].bind(pluginInstance) : undefined }); }) } private registryListener(pluginInstance: interfaces.Plugin) { let events = getPluginListenerMetadata(pluginInstance) for (const event of events) { // here must bind this to pluginInstance this.EventManager.listen(pluginInstance, event.name, pluginInstance[event.executor].bind(pluginInstance)); } } }