style(thread): improve delegation typing

visual-cortex · Aug 15, 2023 · bab593a · bab593a
1 parent e95e650
commit bab593a
Showing 1 changed file with 11 additions and 3 deletions.
diff --git a/source/platform/browser/worker/imgstry.thread.ts b/source/platform/browser/worker/imgstry.thread.ts
@@ -53,12 +53,20 @@ export class ImgstryThread implements IImgstryThread, IDisposable {
         this._logger = new Logger(!!_options.isDebugEnabled);
         this._worker = new ImgstryWorker();
 
-        const delegate = (event: string, delegate: Function) => {
-            fromEvent<MessageEvent | ErrorEvent>(this._worker, event)
+        type EventMap = {
+            message: MessageEvent
+            error: ErrorEvent
+        };
+
+        const delegate = <
+            TEvent extends keyof EventMap,
+            TMessage = EventMap[TEvent]
+        >(event: TEvent, cb: (message: TMessage) => void) => {
+            fromEvent<TMessage>(this._worker, event)
                 .pipe(
                     takeUntil(this._disposed$),
                 )
-                .subscribe(delegate);
+                .subscribe(cb);
         };
 
         delegate('message', this._handleMessage);