youtube-music/src/loader/main.ts

import { BrowserWindow, ipcMain } from 'electron';

import { deepmerge } from 'deepmerge-ts';
import { allPlugins, mainPlugins } from 'virtual:plugins';

import config from '@/config';
import { startPlugin, stopPlugin } from '@/utils';

import type { PluginConfig, PluginDef } from '@/types/plugins';
import type { BackendContext } from '@/types/contexts';

const loadedPluginMap: Record<string, PluginDef<unknown, unknown, unknown>> = {};

const createContext = (id: string, win: BrowserWindow): BackendContext<PluginConfig> => ({
  getConfig: () =>
    deepmerge(
      allPlugins[id].config ?? { enabled: false },
      config.get(`plugins.${id}`) ?? {},
    ) as PluginConfig,
  setConfig: (newConfig) => {
    config.setPartial(`plugins.${id}`, newConfig, allPlugins[id].config);
  },

  ipc: {
    send: (event: string, ...args: unknown[]) => {
      win.webContents.send(event, ...args);
    },
    handle: (event: string, listener: CallableFunction) => {
      // eslint-disable-next-line @typescript-eslint/no-unsafe-return
      ipcMain.handle(event, (_, ...args: unknown[]) => listener(...args));
    },
    on: (event: string, listener: CallableFunction) => {
      ipcMain.on(event, (_, ...args: unknown[]) => {
        listener(...args);
      });
    },
    removeHandler: (event: string) => {
      ipcMain.removeHandler(event);
    }
  },

  window: win,
});

export const forceUnloadMainPlugin = async (
  id: string,
  win: BrowserWindow,
): Promise<void> => {
  const plugin = loadedPluginMap[id];
  if (!plugin) return;

  return new Promise<void>((resolve, reject) => {
    try {
      const hasStopped = stopPlugin(id, plugin, {
        ctx: 'backend',
        context: createContext(id, win),
      });
      if (
        hasStopped ||
        (
          hasStopped === null &&
          typeof plugin.backend !== 'function' && plugin.backend
        )
      ) {
        delete loadedPluginMap[id];
        console.log('[YTMusic]', `"${id}" plugin is unloaded`);
        resolve();
      } else {
        console.log(
          '[YTMusic]',
          `Cannot unload "${id}" plugin`,
        );
        reject();
        return;
      }
    } catch (err) {
      console.log('[YTMusic]', `Cannot unload "${id}" plugin: ${String(err)}`);
      reject(err);
    }
  });
};

export const forceLoadMainPlugin = async (
  id: string,
  win: BrowserWindow,
): Promise<void> => {
  const plugin = mainPlugins[id];
  if (!plugin) return;

  return new Promise<void>((resolve, reject) => {
    try {
      const hasStarted = startPlugin(id, plugin, {
        ctx: 'backend',
        context: createContext(id, win),
      });
      if (
        hasStarted ||
        (
          hasStarted === null &&
          typeof plugin.backend !== 'function' && plugin.backend
        )
      ) {
        loadedPluginMap[id] = plugin;
        resolve();
      } else {
        console.log('[YTMusic]', `Cannot load "${id}" plugin`);
        reject();
      }
    } catch (err) {
      console.error(
        '[YTMusic]',
        `Cannot initialize "${id}" plugin: `,
      );
      console.trace(err);
      reject(err);
    }
  });
};

export const loadAllMainPlugins = async (win: BrowserWindow) => {
  console.log('[YTMusic]', 'Loading all plugins');
  const pluginConfigs = config.plugins.getPlugins();
  const queue: Promise<void>[] = [];

  for (const [plugin, pluginDef] of Object.entries(mainPlugins)) {
    const config = deepmerge(pluginDef.config, pluginConfigs[plugin] ?? {});
    if (config.enabled) {
      queue.push(forceLoadMainPlugin(plugin, win));
    } else if (loadedPluginMap[plugin]) {
      queue.push(forceUnloadMainPlugin(plugin, win));
    }
  }

  await Promise.allSettled(queue);
};

export const unloadAllMainPlugins = (win: BrowserWindow) => {
  for (const id of Object.keys(loadedPluginMap)) {
    forceUnloadMainPlugin(id, win);
  }
};

export const getLoadedMainPlugin = (id: string): PluginDef<unknown, unknown, unknown> | undefined => {
  return loadedPluginMap[id];
};

export const getAllLoadedMainPlugins = () => {
  return loadedPluginMap;
};