refactor: Move callsites to liteque to be behind a plugin

1 files changed, 16 insertions, 12 deletions

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<T extends PluginType> {
@@ -21,37 +24,38 @@ export interface PluginProvider<T> {
   getClient(): Promise<T | null>;
 }
 
+// Preserve the key-dependent value type: for K, store TPlugin<K>[]
+type ProviderMap = { [K in PluginType]: TPlugin<K>[] };
+
 export class PluginManager {
-  private static providers = new Map<PluginType, TPlugin<PluginType>[]>();
+  private static providers: ProviderMap = {
+    [PluginType.Search]: [],
+    [PluginType.Queue]: [],
+  };
 
   static register<T extends PluginType>(plugin: TPlugin<T>): 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<T extends PluginType>(
     type: T,
   ): Promise<PluginTypeMap[T] | null> {
-    const provider = PluginManager.providers.get(type);
-    if (!provider) {
+    const providers: TPlugin<T>[] = 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<T extends PluginType>(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;