refactor(plugin): apply new plugin loader at all type of plugin

This commit is contained in:
Su-Yong
2023-11-12 00:50:09 +09:00
parent dfcc4107b7
commit 9c59f56aac
5 changed files with 208 additions and 70 deletions

90
src/loader/preload.ts Normal file
View File

@ -0,0 +1,90 @@
import { deepmerge } from 'deepmerge-ts';
import {
PluginBaseConfig,
PluginBuilder,
PreloadPlugin,
PluginContext,
PreloadPluginFactory
} from '../plugins/utils/builder';
import config from '../config';
const allPluginFactoryList: Record<string, PreloadPluginFactory<PluginBaseConfig>> = {};
const allPluginBuilders: Record<string, PluginBuilder<string, PluginBaseConfig>> = {};
const unregisterStyleMap: Record<string, (() => void)[]> = {};
const loadedPluginMap: Record<string, PreloadPlugin<PluginBaseConfig>> = {};
const createContext = <
Key extends keyof PluginBuilderList,
Config extends PluginBaseConfig = PluginBuilderList[Key]['config'],
>(id: Key): PluginContext<Config> => ({
getConfig: () => deepmerge(allPluginBuilders[id].config, config.get(`plugins.${id}`) ?? {}) as Config,
setConfig: (newConfig) => {
config.setPartial(`plugins.${id}`, newConfig);
},
});
const forceUnloadPreloadPlugin = (id: keyof PluginBuilderList) => {
unregisterStyleMap[id]?.forEach((unregister) => unregister());
delete unregisterStyleMap[id];
loadedPluginMap[id]?.onUnload?.();
delete loadedPluginMap[id];
console.log('[YTMusic]', `"${id}" plugin is unloaded`);
};
export const forceLoadPreloadPlugin = async (id: keyof PluginBuilderList) => {
try {
const factory = allPluginFactoryList[id];
if (!factory) return;
const context = createContext(id);
const plugin = await factory(context);
loadedPluginMap[id] = plugin;
plugin.onLoad?.();
console.log('[YTMusic]', `"${id}" plugin is loaded`);
} catch (err) {
console.log('[YTMusic]', `Cannot initialize "${id}" plugin: ${String(err)}`);
}
};
export const loadAllPreloadPlugins = async () => {
const pluginConfigs = config.plugins.getPlugins();
for (const [pluginId, builder] of Object.entries(allPluginBuilders)) {
const typedBuilder = builder as PluginBuilderList[keyof PluginBuilderList];
const config = deepmerge(typedBuilder.config, pluginConfigs[pluginId as keyof PluginBuilderList] ?? {});
if (config.enabled) {
await forceLoadPreloadPlugin(pluginId as keyof PluginBuilderList);
} else {
if (loadedPluginMap[pluginId as keyof PluginBuilderList]) {
forceUnloadPreloadPlugin(pluginId as keyof PluginBuilderList);
}
}
}
};
export const unloadAllPreloadPlugins = () => {
for (const id of Object.keys(loadedPluginMap)) {
forceUnloadPreloadPlugin(id as keyof PluginBuilderList);
}
};
export const getLoadedPreloadPlugin = <Key extends keyof PluginBuilderList>(id: Key): PreloadPlugin<PluginBuilderList[Key]['config']> | undefined => {
return loadedPluginMap[id];
};
export const getAllLoadedPreloadPlugins = () => {
return loadedPluginMap;
};
export const registerPreloadPlugin = (
id: string,
builder: PluginBuilder<string, PluginBaseConfig>,
factory?: PreloadPluginFactory<PluginBaseConfig>,
) => {
if (factory) allPluginFactoryList[id] = factory;
allPluginBuilders[id] = builder;
};