back

packages/devtools-shared/src/receive.ts

@@ -0,0 +1,22 @@
+// receive ws message by adding a listener to the ws object
+
+import { DevtoolsEvent, DevtoolsEventPayloads } from "./event-types";
+
+export function receive<T extends DevtoolsEvent>(
+    ws: WebSocket,
+    event: T,
+    callback: (payload: DevtoolsEventPayloads[T]) => void,
+) {
+    const listener = (e: MessageEvent) => {
+        const { event: receivedEvent, payload } = JSON.parse(e.data);
+        if (event === receivedEvent) {
+            callback(payload);
+        }
+    };
+
+    ws.addEventListener("message", listener);
+
+    return () => {
+        ws.removeEventListener("message", listener);
+    };
+}