fix: apply event worker context to custom events

Author: twlite

packages/commandkit/src/events/CommandKitEventsChannel.ts

@@ -1,6 +1,7 @@
 import EventEmitter from 'node:events';
 import type { CommandKit } from '../CommandKit';
 import type { AsyncFunction, GenericFunction } from '../context/async-context';
+import { runInEventWorkerContext } from '../app/events/EventWorkerContext';
 
 export type ListenerFunction = GenericFunction | AsyncFunction;
 
@@ -20,15 +21,24 @@ export class CommandKitEventsChannel {
   }
 
   public on(namespace: string, event: string, listener: ListenerFunction) {
-    this.emitter.on(`${namespace}:${event}`, listener);
+    this.emitter.on(
+      `${namespace}:${event}`,
+      this.prepareListener(namespace, event, listener),
+    );
   }
 
   public off(namespace: string, event: string, listener: ListenerFunction) {
-    this.emitter.off(`${namespace}:${event}`, listener);
+    this.emitter.off(
+      `${namespace}:${event}`,
+      this.prepareListener(namespace, event, listener),
+    );
   }
 
   public once(namespace: string, event: string, listener: ListenerFunction) {
-    this.emitter.once(`${namespace}:${event}`, listener);
+    this.emitter.once(
+      `${namespace}:${event}`,
+      this.prepareListener(namespace, event, listener),
+    );
   }
 
   public emit(namespace: string, event: string, ...args: any[]) {
@@ -44,4 +54,28 @@ export class CommandKitEventsChannel {
       this.emitter.removeAllListeners(namespace);
     }
   }
+
+  private prepareListener<L extends ListenerFunction>(
+    namespace: string,
+    event: string,
+    listener: L,
+  ): L {
+    return ((...args) => {
+      return runInEventWorkerContext(
+        {
+          arguments: args,
+          commandkit: this.commandkit,
+          data: {
+            event,
+            namespace,
+            listeners: [],
+            path: `virtual:${namespace}/${event}`,
+          },
+          event,
+          namespace,
+        },
+        listener,
+      );
+    }) as L;
+  }
 }