object subscription w/ optional params

topology-foundation · Jul 7, 2024 · 726010b · 726010b
1 parent 2f433bd
commit 726010b
Showing 1 changed file with 15 additions and 6 deletions.
diff --git a/packages/node/src/index.ts b/packages/node/src/index.ts
@@ -86,19 +86,28 @@ export class TopologyNode {
 
   /// Subscribe to the object's PubSub group
   /// and fetch it from a peer
-  async subscribeObject(objectId: string) {
+  async subscribeObject(objectId: string, fetch = false, peerId = "") {
     this._networkNode.subscribe(objectId);
+    if (!fetch) return;
     const message = `{
       "type": "object_fetch",
       "sender": "${this._networkNode.peerId}",
       "data": [${uint8ArrayFromString(objectId)}]
     }`;
 
-    await this._networkNode.sendGroupMessageRandomPeer(
-      objectId,
-      ["/topology/message/0.0.1"],
-      message,
-    );
+    if (peerId === "") {
+      await this._networkNode.sendGroupMessageRandomPeer(
+        objectId,
+        ["/topology/message/0.0.1"],
+        message,
+      );
+    } else {
+      await this._networkNode.sendMessage(
+        peerId,
+        ["/topology/message/0.0.1"],
+        message,
+      );
+    }
   }
 
   getPeers() {