From 02cce39e44f290320c364c7a6fcd10fa2e23e317 Mon Sep 17 00:00:00 2001
From: CanadaHonk <oj@oojmed.com>
Date: Thu, 26 Jan 2023 23:08:09 +0000
Subject: [PATCH] lib/cdp: rewrite onMessage for internals

---
 src/lib/cdp.js | 12 ++++++------
 1 file changed, 6 insertions(+), 6 deletions(-)

diff --git a/src/lib/cdp.js b/src/lib/cdp.js
index 1c3dc59..068c8dc 100644
--- a/src/lib/cdp.js
+++ b/src/lib/cdp.js
@@ -129,13 +129,13 @@ export default async ({ pipe: { pipeWrite, pipeRead } = {}, port }) => {
   }
 
   return {
-    onMessage: (_callback, once = false) => {
-      const callback = once ? msg => {
-        _callback(msg);
-        messageCallbacks.splice(messageCallbacks.indexOf(callback), 1); // remove self
-      } : _callback;
-
+    onMessage: (callback) => {
       messageCallbacks.push(callback);
+
+      // return function to unhook
+      return () => {
+        messageCallbacks.splice(messageCallbacks.indexOf(callback), 1);
+      };
     },
 
     sendMessage,