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

src/loader/menu.ts

@@ -0,0 +1,72 @@
+import { deepmerge } from 'deepmerge-ts';
+
+import { MenuPluginContext, MenuPluginFactory, PluginBaseConfig, PluginBuilder } from '../plugins/utils/builder';
+import config from '../config';
+import { setApplicationMenu } from '../menu';
+
+import type { BrowserWindow, MenuItemConstructorOptions } from 'electron';
+
+const allPluginFactoryList: Record<string, MenuPluginFactory<PluginBaseConfig>> = {};
+const allPluginBuilders: Record<string, PluginBuilder<string, PluginBaseConfig>> = {};
+const menuTemplateMap: Record<string, MenuItemConstructorOptions[]> = {};
+
+const createContext = <
+  Key extends keyof PluginBuilderList,
+  Config extends PluginBaseConfig = PluginBuilderList[Key]['config'],
+>(id: Key, win: BrowserWindow): MenuPluginContext<Config> => ({
+  getConfig: () => deepmerge(allPluginBuilders[id].config, config.get(`plugins.${id}`) ?? {}) as Config,
+  setConfig: (newConfig) => {
+    config.setPartial(`plugins.${id}`, newConfig);
+  },
+  window: win,
+  refresh: async () => {
+    await setApplicationMenu(win);
+
+    if (config.plugins.isEnabled('in-app-menu')) {
+      win.webContents.send('refresh-in-app-menu');
+    }
+  },
+});
+
+export const forceLoadMenuPlugin = async (id: keyof PluginBuilderList, win: BrowserWindow) => {
+  try {
+    const factory = allPluginFactoryList[id];
+    if (!factory) return;
+
+    const context = createContext(id, win);
+    menuTemplateMap[id] = await factory(context);
+
+    console.log('[YTMusic]', `"${id}" plugin is loaded`);
+  } catch (err) {
+    console.log('[YTMusic]', `Cannot initialize "${id}" plugin: ${String(err)}`);
+  }
+};
+
+export const loadAllMenuPlugins = async (win: BrowserWindow) => {
+  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 forceLoadMenuPlugin(pluginId as keyof PluginBuilderList, win);
+    }
+  }
+};
+
+export const getMenuTemplate = <Key extends keyof PluginBuilderList>(id: Key): MenuItemConstructorOptions[] | undefined => {
+  return menuTemplateMap[id];
+};
+export const getAllMenuTemplate = () => {
+  return menuTemplateMap;
+};
+export const registerMenuPlugin = (
+  id: string,
+  builder: PluginBuilder<string, PluginBaseConfig>,
+  factory?: MenuPluginFactory<PluginBaseConfig>,
+) => {
+  if (factory) allPluginFactoryList[id] = factory;
+  allPluginBuilders[id] = builder;
+};

src/loader/preload.ts

@@ -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;
+};