Send block as generic payload to be subscribed Ref: #31

pinax-network · Nov 16, 2023 · bebc08b · bebc08b
1 parent 2f89e6d
commit bebc08b
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 3 deletions.
diff --git a/src/websocket/parseMessage.ts b/src/websocket/parseMessage.ts
@@ -11,8 +11,6 @@ export function parseMessage(message: string|Buffer): { method: string, id: stri
 
     if ( method === 'subscribe' ) {
         if ( params === null || params === undefined) throw new Error('Missing required \'params\' in JSON request.');
-        const { moduleHash } = params;
-        if ( moduleHash === null || moduleHash === undefined) throw new Error('Missing required \'moduleHash\' in JSON request.');
     }
 
     return json;

diff --git a/src/websocket/subscribe.ts b/src/websocket/subscribe.ts
@@ -6,7 +6,13 @@ import * as sqlite from "../sqlite.js";
 import * as prometheus from "../prometheus.js";
 
 export default function (ws: ServerWebSocket<ServerWebSocketData>, params: {[key: string]: any}, id?: string) {
-    const { moduleHash, chain } = params;
+    const { chain, block } = params;
+    let { moduleHash } = params;
+
+    if ( block === true ) {
+        moduleHash = sqlite.selectAll(db, "moduleHash")[0].key;
+    }
+
     const topic = chain ? `${chain}:${moduleHash}` : moduleHash;
     if ( ws.isSubscribed(topic) ) throw new Error(`Already subscribed to [${topic}] topic.`);