pass callbacks to client

dist/index.js

@@ -4,6 +4,12 @@ export class TetherClient {
     subscribedQueries = new Map();
     connect = (url) => {
         this.websocketHandler.startConnection(url);
+        this.websocketHandler.onQuery = (location, data) => {
+            if (location) {
+                const callback = this.subscribedQueries.get(location);
+                callback?.(data);
+            }
+        };
     };
     disconnect = () => {
         this.websocketHandler.close();

dist/utils/websocket.d.ts

@@ -1,9 +1,9 @@
 export declare class WebSocketHandler {
     private ws;
     private url;
-    private subscribedQueries;
-    private onOpen;
-    private onClose;
+    onOpen: () => void;
+    onQuery: (location: string | undefined, data: any) => void;
+    onClose: () => void;
     private reconnectAttempts;
     private maxReconnectAttempts;
     private reconnectInterval;

dist/utils/websocket.js

@@ -1,8 +1,8 @@
 export class WebSocketHandler {
     ws = null;
     url = '';
-    subscribedQueries = new Map();
     onOpen = () => { };
+    onQuery = () => { };
     onClose = () => { };
     reconnectAttempts = 0;
     maxReconnectAttempts = 5;
@@ -24,11 +24,7 @@ export class WebSocketHandler {
         ws.onmessage = (event) => {
             const data = JSON.parse(String(event.data));
             if (data.type === 'query') {
-                this.subscribedQueries.forEach((callback, query) => {
-                    if (data.query === query) {
-                        callback(data.data);
-                    }
-                });
+                this.onQuery(data.location, data.data);
             }
             else if (data.type === 'error') {
                 console.error(data.error);