OverlayScrollbars/packages/overlayscrollbars/src/plugins/plugins.ts

import { each, isArray, keys } from 'support';
import { OverlayScrollbars, OverlayScrollbarsStatic } from 'overlayscrollbars';

export type PluginInstance =
  | Record<string, unknown>
  | ((staticObj: OverlayScrollbarsStatic, instanceObj: OverlayScrollbars) => void);
export type Plugin<T extends PluginInstance = PluginInstance> = {
  [pluginName: string]: T;
};

const pluginRegistry: Record<string, PluginInstance> = {};

export const getPlugins = () => pluginRegistry;

export const addPlugin = (addedPlugin: Plugin | Plugin[]): void => {
  each((isArray(addedPlugin) ? addedPlugin : [addedPlugin]) as Plugin[], (plugin) => {
    const pluginName = keys(plugin)[0];
    pluginRegistry[pluginName] = plugin[pluginName];
  });
};