From 8d32055485858210252096483bb20533dc8bdf60 Mon Sep 17 00:00:00 2001 From: Mohamed Bassem Date: Sun, 14 Sep 2025 18:16:40 +0000 Subject: refactor: Move callsites to liteque to be behind a plugin --- packages/shared/plugins.ts | 28 ++++++++++++++++------------ 1 file changed, 16 insertions(+), 12 deletions(-) (limited to 'packages/shared/plugins.ts') diff --git a/packages/shared/plugins.ts b/packages/shared/plugins.ts index 2ce5826a..2aa7df4a 100644 --- a/packages/shared/plugins.ts +++ b/packages/shared/plugins.ts @@ -1,14 +1,17 @@ // Implementation inspired from Outline +import type { QueueClient } from "./queueing"; import logger from "./logger"; import { SearchIndexClient } from "./search"; export enum PluginType { Search = "search", + Queue = "queue", } interface PluginTypeMap { [PluginType.Search]: SearchIndexClient; + [PluginType.Queue]: QueueClient; } export interface TPlugin { @@ -21,37 +24,38 @@ export interface PluginProvider { getClient(): Promise; } +// Preserve the key-dependent value type: for K, store TPlugin[] +type ProviderMap = { [K in PluginType]: TPlugin[] }; + export class PluginManager { - private static providers = new Map[]>(); + private static providers: ProviderMap = { + [PluginType.Search]: [], + [PluginType.Queue]: [], + }; static register(plugin: TPlugin): void { - const p = PluginManager.providers.get(plugin.type); - if (!p) { - PluginManager.providers.set(plugin.type, [plugin]); - return; - } - p.push(plugin); + PluginManager.providers[plugin.type].push(plugin); } static async getClient( type: T, ): Promise { - const provider = PluginManager.providers.get(type); - if (!provider) { + const providers: TPlugin[] = PluginManager.providers[type]; + if (providers.length === 0) { return null; } - return await provider[provider.length - 1].provider.getClient(); + return await providers[providers.length - 1]!.provider.getClient(); } static isRegistered(type: T): boolean { - return !!PluginManager.providers.get(type); + return PluginManager.providers[type].length > 0; } static logAllPlugins() { logger.info("Plugins (Last one wins):"); for (const type of Object.values(PluginType)) { logger.info(` ${type}:`); - const plugins = PluginManager.providers.get(type); + const plugins = PluginManager.providers[type]; if (!plugins) { logger.info(" - None"); continue; -- cgit v1.2.3-70-g09d2