refactor(plugin): add renderer-plugin-loader

This commit is contained in:
Su-Yong
2023-11-12 00:21:34 +09:00
parent bc916f3a6e
commit ef71abfff1
4 changed files with 119 additions and 78 deletions

View File

@ -37,7 +37,7 @@ const createContext = <
},
});
const forceUnloadPlugin = (id: keyof PluginBuilderList, win: BrowserWindow) => {
const forceUnloadMainPlugin = (id: keyof PluginBuilderList, win: BrowserWindow) => {
unregisterStyleMap[id]?.forEach((unregister) => unregister());
delete unregisterStyleMap[id];
@ -47,7 +47,7 @@ const forceUnloadPlugin = (id: keyof PluginBuilderList, win: BrowserWindow) => {
console.log('[YTMusic]', `"${id}" plugin is unloaded`);
};
export const forceLoadPlugin = async (id: keyof PluginBuilderList, win: BrowserWindow) => {
export const forceLoadMainPlugin = async (id: keyof PluginBuilderList, win: BrowserWindow) => {
const builder = allPluginBuilders[id];
Promise.allSettled(
@ -88,7 +88,7 @@ export const forceLoadPlugin = async (id: keyof PluginBuilderList, win: BrowserW
}
};
export const loadAllPlugins = async (win: BrowserWindow) => {
export const loadAllMainPlugins = async (win: BrowserWindow) => {
const pluginConfigs = config.plugins.getPlugins();
for (const [pluginId, builder] of Object.entries(allPluginBuilders)) {
@ -97,25 +97,25 @@ export const loadAllPlugins = async (win: BrowserWindow) => {
const config = deepmerge(typedBuilder.config, pluginConfigs[pluginId as keyof PluginBuilderList] ?? {});
if (config.enabled) {
await forceLoadPlugin(pluginId as keyof PluginBuilderList, win);
await forceLoadMainPlugin(pluginId as keyof PluginBuilderList, win);
} else {
if (loadedPluginMap[pluginId as keyof PluginBuilderList]) {
forceUnloadPlugin(pluginId as keyof PluginBuilderList, win);
forceUnloadMainPlugin(pluginId as keyof PluginBuilderList, win);
}
}
}
};
export const unloadAllPlugins = (win: BrowserWindow) => {
export const unloadAllMainPlugins = (win: BrowserWindow) => {
for (const id of Object.keys(loadedPluginMap)) {
forceUnloadPlugin(id as keyof PluginBuilderList, win);
forceUnloadMainPlugin(id as keyof PluginBuilderList, win);
}
};
export const getLoadedPlugin = <Key extends keyof PluginBuilderList>(id: Key): MainPlugin<PluginBuilderList[Key]['config']> | undefined => {
export const getLoadedMainPlugin = <Key extends keyof PluginBuilderList>(id: Key): MainPlugin<PluginBuilderList[Key]['config']> | undefined => {
return loadedPluginMap[id];
};
export const getLoadedAllPlugins = () => {
export const getAllLoadedMainPlugins = () => {
return loadedPluginMap;
};
export const registerMainPlugin = (

97
src/loader/renderer.ts Normal file
View File

@ -0,0 +1,97 @@
import { deepmerge } from 'deepmerge-ts';
import {
PluginBaseConfig, PluginBuilder,
RendererPlugin,
RendererPluginContext,
RendererPluginFactory
} from '../plugins/utils/builder';
const allPluginFactoryList: Record<string, RendererPluginFactory<PluginBaseConfig>> = {};
const allPluginBuilders: Record<string, PluginBuilder<string, PluginBaseConfig>> = {};
const unregisterStyleMap: Record<string, (() => void)[]> = {};
const loadedPluginMap: Record<string, RendererPlugin<PluginBaseConfig>> = {};
const createContext = <
Key extends keyof PluginBuilderList,
Config extends PluginBaseConfig = PluginBuilderList[Key]['config'],
>(id: Key): RendererPluginContext<Config> => ({
getConfig: async () => {
return await window.ipcRenderer.invoke('get-config', id) as Config;
},
setConfig: async (newConfig) => {
await window.ipcRenderer.invoke('set-config', id, newConfig);
},
invoke: async <Return>(event: string, ...args: unknown[]): Promise<Return> => {
return await window.ipcRenderer.invoke(event, ...args) as Return;
},
on: (event: string, listener) => {
window.ipcRenderer.on(event, async (_, ...args) => listener(...args as never));
},
});
const forceUnloadRendererPlugin = (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 forceLoadRendererPlugin = 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 loadAllRendererPlugins = async () => {
const pluginConfigs = window.mainConfig.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 forceLoadRendererPlugin(pluginId as keyof PluginBuilderList);
} else {
if (loadedPluginMap[pluginId as keyof PluginBuilderList]) {
forceUnloadRendererPlugin(pluginId as keyof PluginBuilderList);
}
}
}
};
export const unloadAllRendererPlugins = () => {
for (const id of Object.keys(loadedPluginMap)) {
forceUnloadRendererPlugin(id as keyof PluginBuilderList);
}
};
export const getLoadedRendererPlugin = <Key extends keyof PluginBuilderList>(id: Key): RendererPlugin<PluginBuilderList[Key]['config']> | undefined => {
return loadedPluginMap[id];
};
export const getAllLoadedRendererPlugins = () => {
return loadedPluginMap;
};
export const registerRendererPlugin = (
id: string,
builder: PluginBuilder<string, PluginBaseConfig>,
factory?: RendererPluginFactory<PluginBaseConfig>,
) => {
if (factory) allPluginFactoryList[id] = factory;
allPluginBuilders[id] = builder;
};