From 5e19b2d70fca942813048b0b079cbbce11046726 Mon Sep 17 00:00:00 2001
From: DarkWiiPlayer <darkwiiplayer@hotmail.com>
Date: Thu, 17 Mar 2022 09:49:25 +0100
Subject: [PATCH] Add return values to listen and forget methods

---
 listener.js | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/listener.js b/listener.js
index 5aafd0f..84aa7f3 100644
--- a/listener.js
+++ b/listener.js
@@ -9,7 +9,7 @@ Example:
 */
 
 const registry = new Map()
-export const listener = (target={}) => {
+const listener = (target={}) => {
 	const callbacks = new Map()
 	const methods = Object.create(null)
 	methods.listen = function(name, fn, {once=false}={}) {
@@ -19,13 +19,14 @@ export const listener = (target={}) => {
 		let set = callbacks.get(name) ?? new Set()
 		callbacks.set(name, set)
 		set.add(callback)
+		return this
 	}
 	methods.forget = function(name, callback) {
 		if (callback) {
 			const set = callbacks.get(name)
-			if (set) set.delete(callback)
+			if (set) return set.delete(callback)
 		} else {
-			callbacks.delete(name)
+			return callbacks.delete(name)
 		}
 	}
 	let proxy = new Proxy(target, {