adding waku_get_connected_peers

waku-org · Oct 28, 2024 · b28101e · b28101e
1 parent 4cf7f82
commit b28101e
Show file tree

Hide file tree

Showing 3 changed files with 24 additions and 0 deletions.
diff --git a/library/libwaku.h b/library/libwaku.h
@@ -147,6 +147,10 @@ int waku_listen_addresses(void* ctx,
                           WakuCallBack callback,
                           void* userData);
 
+int waku_get_connected_peers(void* ctx,
+                          WakuCallBack callback,
+                          void* userData);
+
 // Returns a list of multiaddress given a url to a DNS discoverable ENR tree
 // Parameters
 //     char* entTreeUrl: URL containing a discoverable ENR tree

diff --git a/library/libwaku.nim b/library/libwaku.nim
@@ -553,6 +553,19 @@ proc waku_get_peerids_from_peerstore(
   )
   .handleRes(callback, userData)
 
+proc waku_get_connected_peers(
+    ctx: ptr WakuContext, callback: WakuCallBack, userData: pointer
+): cint {.dynlib, exportc.} =
+  checkLibwakuParams(ctx, callback, userData)
+
+  waku_thread
+  .sendRequestToWakuThread(
+    ctx,
+    RequestType.PEER_MANAGER,
+    PeerManagementRequest.createShared(PeerManagementMsgType.GET_CONNECTED_PEERS),
+  )
+  .handleRes(callback, userData)
+
 proc waku_get_peerids_by_protocol(
     ctx: ptr WakuContext, protocol: cstring, callback: WakuCallBack, userData: pointer
 ): cint {.dynlib, exportc.} =

diff --git a/library/waku_thread/inter_thread_communication/requests/peer_manager_request.nim b/library/waku_thread/inter_thread_communication/requests/peer_manager_request.nim
@@ -13,6 +13,7 @@ type PeerManagementMsgType* {.pure.} = enum
   DISCONNECT_PEER_BY_ID
   DIAL_PEER
   DIAL_PEER_BY_ID
+  GET_CONNECTED_PEERS
 
 type PeerManagementRequest* = object
   operation: PeerManagementMsgType
@@ -114,5 +115,11 @@ proc process*(
       let msg = "failed dialing peer"
       error "DIAL_PEER_BY_ID failed", error = msg
       return err(msg)
+  of GET_CONNECTED_PEERS:
+    ## returns a comma-separated string of peerIDs
+    let
+      (inPeerIds, outPeerIds) = waku.node.peerManager.connectedPeers()
+      connectedPeerids = concat(inPeerIds, outPeerIds)
+    return ok(connectedPeerids.mapIt($it).join(","))
 
   return ok("")