adding in listenerCount

jaredwray · Oct 15, 2024 · 6b17b31 · 6b17b31
1 parent 6a222dc
commit 6b17b31
Show file tree

Hide file tree

Showing 2 changed files with 27 additions and 1 deletion.
diff --git a/src/eventified.ts b/src/eventified.ts
@@ -22,7 +22,11 @@ export class Eventified implements IEventEmitter {
 		return this;
 	}
 
-	listenerCount(eventName: string | symbol): number {
+	listenerCount(eventName?: string | symbol): number {
+		if (!eventName) {
+			return this.getAllListeners().length;
+		}
+
 		const listeners = this._eventListeners.get(eventName as string);
 		return listeners ? listeners.length : 0;
 	}
@@ -130,4 +134,13 @@ export class Eventified implements IEventEmitter {
 			}
 		}
 	}
+
+	public getAllListeners(): EventListener[] {
+		let result = new Array<EventListener>();
+		for (const listeners of this._eventListeners.values()) {
+			result = result.concat(listeners);
+		}
+
+		return result;
+	}
 }
diff --git a/test/eventified.test.ts b/test/eventified.test.ts
@@ -148,4 +148,17 @@ describe('Eventified', () => {
 
 		t.expect(dataReceived).toBe(1);
 	});
+
+	test('get listener count', t => {
+		const emitter = new Eventified();
+		const listener = () => {};
+
+		emitter.on('test-event', listener);
+		emitter.on('test-event', listener);
+		emitter.on('test-event1', listener);
+		emitter.on('test-event2', listener);
+
+		t.expect(emitter.listenerCount()).toBe(4);
+		t.expect(emitter.listenerCount('test-event')).toBe(2);
+	});
 });