fix(core): Fix client hook edge cases around multiple callbacks

Author: Lms24

packages/core/src/client.ts

@@ -137,7 +137,7 @@ export abstract class Client<O extends ClientOptions = ClientOptions> {
   private _outcomes: { [key: string]: number };
 
   // eslint-disable-next-line @typescript-eslint/ban-types
-  private _hooks: Record<string, Function[]>;
+  private _hooks: Record<string, Set<Function>>;
 
   /**
    * Initializes this client instance.
@@ -685,21 +685,23 @@ export abstract class Client<O extends ClientOptions = ClientOptions> {
    * Register a hook on this client.
    */
   public on(hook: string, callback: unknown): () => void {
-    const hooks = (this._hooks[hook] = this._hooks[hook] || []);
+    const hookCallbacks = (this._hooks[hook] = this._hooks[hook] || new Set());
 
-    // @ts-expect-error We assume the types are correct
-    hooks.push(callback);
+    // Wrap the callback in a function so that registering the same callback instance multiple
+    // times results in the callback being called multiple times.
+    // @ts-expect-error - The `callback` type is correct and must be a function due to the
+    // individual, specific overloads of this function.
+    // eslint-disable-next-line @typescript-eslint/ban-types
+    const uniqueCallback: Function = (...args: unknown[]) => callback(...args);
+
+    hookCallbacks.add(uniqueCallback);
 
     // This function returns a callback execution handler that, when invoked,
     // deregisters a callback. This is crucial for managing instances where callbacks
     // need to be unregistered to prevent self-referencing in callback closures,
     // ensuring proper garbage collection.
     return () => {
-      // @ts-expect-error We assume the types are correct
-      const cbIndex = hooks.indexOf(callback);
-      if (cbIndex > -1) {
-        hooks.splice(cbIndex, 1);
-      }
+      hookCallbacks.delete(uniqueCallback);
     };
   }
 

packages/core/test/lib/client.test.ts

@@ -2397,10 +2397,8 @@ describe('Client', () => {
 
       client.emit('beforeEnvelope', mockEnvelope);
     });
-  });
 
-  describe('hook removal with `on`', () => {
-    it('should return a cleanup function that, when executed, unregisters a hook', async () => {
+    it('returns a cleanup function that, when executed, unregisters a hook', async () => {
       vi.useFakeTimers();
       expect.assertions(8);
 
@@ -2420,7 +2418,7 @@ describe('Client', () => {
       const callback = vi.fn();
       const removeAfterSendEventListenerFn = client.on('afterSendEvent', callback);
 
-      expect(client['_hooks']['afterSendEvent']).toEqual([callback]);
+      expect(client['_hooks']['afterSendEvent']!.size).toBe(1);
 
       client.sendEvent(errorEvent);
       vi.runAllTimers();
@@ -2435,7 +2433,7 @@ describe('Client', () => {
 
       // Should unregister `afterSendEvent` callback.
       removeAfterSendEventListenerFn();
-      expect(client['_hooks']['afterSendEvent']).toEqual([]);
+      expect(client['_hooks']['afterSendEvent']!.size).toBe(0);
 
       client.sendEvent(errorEvent);
       vi.runAllTimers();
@@ -2450,6 +2448,67 @@ describe('Client', () => {
       expect(callback).toBeCalledTimes(1);
       expect(callback).toBeCalledWith(errorEvent, { statusCode: 200 });
     });
+
+    it('allows synchronously unregistering multiple callbacks from within the callback', () => {
+      const client = new TestClient(
+        getDefaultTestClientOptions({
+          dsn: PUBLIC_DSN,
+          enableSend: true,
+        }),
+      );
+
+      const callback1 = vi.fn();
+      const callback2 = vi.fn();
+
+      const removeCallback1 = client.on('close', () => {
+        callback1();
+        removeCallback1();
+      });
+      const removeCallback2 = client.on('close', () => {
+        callback2();
+        removeCallback2();
+      });
+
+      client.emit('close');
+
+      expect(callback1).toHaveBeenCalledTimes(1);
+      expect(callback2).toHaveBeenCalledTimes(1);
+
+      callback1.mockReset();
+      callback2.mockReset();
+
+      client.emit('close');
+
+      expect(callback1).toHaveBeenCalledTimes(0);
+      expect(callback2).toHaveBeenCalledTimes(0);
+    });
+
+    it('allows registering and unregistering the same callback multiple times', () => {
+      const client = new TestClient(options);
+      const callback = vi.fn();
+
+      const unregister1 = client.on('close', callback);
+      const unregister2 = client.on('close', callback);
+
+      client.emit('close');
+
+      expect(callback).toHaveBeenCalledTimes(2);
+
+      unregister1();
+
+      callback.mockReset();
+
+      client.emit('close');
+
+      expect(callback).toHaveBeenCalledTimes(1);
+
+      unregister2();
+
+      callback.mockReset();
+      client.emit('close');
+
+      expect(callback).toHaveBeenCalledTimes(0);
+    });
   });
 
   describe('withMonitor', () => {