lib/cdp: rewrite onMessage for internals

2023-01-26 23:08:09 +00:00 · 2023-01-26 23:08:09 +00:00 · 02cce39e44
commit 02cce39e44
parent b537650f03
1 changed files with 6 additions and 6 deletions
--- 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,