From 795a1a52aa0c384c29f914ccc35c16e4dfe73efd Mon Sep 17 00:00:00 2001
From: Pasha Podolsky <ppodolsky@me.com>
Date: Sat, 2 Mar 2024 21:35:09 +0300
Subject: [PATCH] [feat] Expose GetSyncPeers

---
 iroh/src/client.rs          | 16 ++++++++++++----
 iroh/src/node.rs            |  6 ++++++
 iroh/src/rpc_protocol.rs    | 22 +++++++++++++++++++++-
 iroh/src/sync_engine/rpc.rs |  9 +++++++++
 4 files changed, 48 insertions(+), 5 deletions(-)

diff --git a/iroh/src/client.rs b/iroh/src/client.rs
index 5acc31eefb4..07adc32f6ff 100644
--- a/iroh/src/client.rs
+++ b/iroh/src/client.rs
@@ -25,7 +25,7 @@ use iroh_net::ticket::BlobTicket;
 use iroh_net::{key::PublicKey, magic_endpoint::ConnectionInfo, NodeAddr};
 use iroh_sync::actor::OpenState;
 use iroh_sync::store::DownloadPolicy;
-use iroh_sync::{store::Query, AuthorId, CapabilityKind, NamespaceId};
+use iroh_sync::{store::Query, AuthorId, CapabilityKind, NamespaceId, PeerIdBytes};
 use iroh_sync::{ContentStatus, RecordIdentifier};
 use quic_rpc::message::RpcMsg;
 use quic_rpc::{RpcClient, ServiceConnection};
@@ -42,9 +42,9 @@ use crate::rpc_protocol::{
     BlobReadAtRequest, BlobReadAtResponse, BlobValidateRequest, CounterStats,
     CreateCollectionRequest, CreateCollectionResponse, DeleteTagRequest, DocCloseRequest,
     DocCreateRequest, DocDelRequest, DocDelResponse, DocDropRequest, DocExportFileRequest,
-    DocGetDownloadPolicyRequest, DocGetExactRequest, DocGetManyRequest, DocImportFileRequest,
-    DocImportProgress, DocImportRequest, DocLeaveRequest, DocListRequest, DocOpenRequest,
-    DocSetDownloadPolicyRequest, DocSetHashRequest, DocSetRequest, DocShareRequest,
+    DocGetDownloadPolicyRequest, DocGetExactRequest, DocGetManyRequest, DocGetSyncPeersRequest,
+    DocImportFileRequest, DocImportProgress, DocImportRequest, DocLeaveRequest, DocListRequest,
+    DocOpenRequest, DocSetDownloadPolicyRequest, DocSetHashRequest, DocSetRequest, DocShareRequest,
     DocStartSyncRequest, DocStatusRequest, DocSubscribeRequest, DocTicket, DownloadProgress,
     ListTagsRequest, ListTagsResponse, NodeConnectionInfoRequest, NodeConnectionInfoResponse,
     NodeConnectionsRequest, NodeShutdownRequest, NodeStatsRequest, NodeStatusRequest,
@@ -1036,6 +1036,14 @@ where
             .await??;
         Ok(res.policy)
     }
+
+    /// Get sync peers for this document
+    pub async fn get_sync_peers(&self) -> Result<Option<Vec<PeerIdBytes>>> {
+        let res = self
+            .rpc(DocGetSyncPeersRequest { doc_id: self.id() })
+            .await??;
+        Ok(res.peers)
+    }
 }
 
 impl<'a, C: ServiceConnection<ProviderService>> From<&'a Doc<C>>
diff --git a/iroh/src/node.rs b/iroh/src/node.rs
index fd599e73554..1f6a2000a8d 100644
--- a/iroh/src/node.rs
+++ b/iroh/src/node.rs
@@ -1771,6 +1771,12 @@ fn handle_rpc_request<D: BaoStore, E: ServiceEndpoint<ProviderService>>(
                 })
                 .await
             }
+            DocGetSyncPeers(msg) => {
+                chan.rpc(msg, handler, |handler, req| async move {
+                    handler.inner.sync.doc_get_sync_peers(req).await
+                })
+                .await
+            }
         }
     });
 }
diff --git a/iroh/src/rpc_protocol.rs b/iroh/src/rpc_protocol.rs
index 68915d80be3..381bdb1a766 100644
--- a/iroh/src/rpc_protocol.rs
+++ b/iroh/src/rpc_protocol.rs
@@ -21,7 +21,7 @@ use iroh_net::{
 use iroh_sync::{
     actor::OpenState,
     store::{DownloadPolicy, Query},
-    {AuthorId, CapabilityKind, Entry, NamespaceId, SignedEntry},
+    PeerIdBytes, {AuthorId, CapabilityKind, Entry, NamespaceId, SignedEntry},
 };
 use quic_rpc::{
     message::{BidiStreaming, BidiStreamingMsg, Msg, RpcMsg, ServerStreaming, ServerStreamingMsg},
@@ -930,6 +930,24 @@ pub struct DocGetDownloadPolicyResponse {
     pub policy: DownloadPolicy,
 }
 
+/// Get peers for document
+#[derive(Serialize, Deserialize, Debug)]
+pub struct DocGetSyncPeersRequest {
+    /// The document id
+    pub doc_id: NamespaceId,
+}
+
+impl RpcMsg<ProviderService> for DocGetSyncPeersRequest {
+    type Response = RpcResult<DocGetSyncPeersResponse>;
+}
+
+/// Response to [`DocGetSyncPeersRequest`]
+#[derive(Serialize, Deserialize, Debug)]
+pub struct DocGetSyncPeersResponse {
+    /// The download policy
+    pub peers: Option<Vec<PeerIdBytes>>,
+}
+
 /// Get the bytes for a hash
 #[derive(Serialize, Deserialize, Debug)]
 pub struct BlobReadAtRequest {
@@ -1070,6 +1088,7 @@ pub enum ProviderRequest {
     DocSubscribe(DocSubscribeRequest),
     DocGetDownloadPolicy(DocGetDownloadPolicyRequest),
     DocSetDownloadPolicy(DocSetDownloadPolicyRequest),
+    DocGetSyncPeers(DocGetSyncPeersRequest),
 
     AuthorList(AuthorListRequest),
     AuthorCreate(AuthorCreateRequest),
@@ -1121,6 +1140,7 @@ pub enum ProviderResponse {
     DocSubscribe(RpcResult<DocSubscribeResponse>),
     DocGetDownloadPolicy(RpcResult<DocGetDownloadPolicyResponse>),
     DocSetDownloadPolicy(RpcResult<DocSetDownloadPolicyResponse>),
+    DocGetSyncPeers(RpcResult<DocGetSyncPeersResponse>),
 
     AuthorList(RpcResult<AuthorListResponse>),
     AuthorCreate(RpcResult<AuthorCreateResponse>),
diff --git a/iroh/src/sync_engine/rpc.rs b/iroh/src/sync_engine/rpc.rs
index f11907504ca..4b3afffa3ce 100644
--- a/iroh/src/sync_engine/rpc.rs
+++ b/iroh/src/sync_engine/rpc.rs
@@ -6,6 +6,7 @@ use iroh_bytes::{store::Store as BaoStore, BlobFormat};
 use iroh_sync::{Author, NamespaceSecret};
 use tokio_stream::StreamExt;
 
+use crate::rpc_protocol::{DocGetSyncPeersRequest, DocGetSyncPeersResponse};
 use crate::{
     rpc_protocol::{
         AuthorCreateRequest, AuthorCreateResponse, AuthorListRequest, AuthorListResponse,
@@ -258,4 +259,12 @@ impl SyncEngine {
         let policy = self.sync.get_download_policy(req.doc_id).await?;
         Ok(DocGetDownloadPolicyResponse { policy })
     }
+
+    pub async fn doc_get_sync_peers(
+        &self,
+        req: DocGetSyncPeersRequest,
+    ) -> RpcResult<DocGetSyncPeersResponse> {
+        let peers = self.sync.get_sync_peers(req.doc_id).await?;
+        Ok(DocGetSyncPeersResponse { peers })
+    }
 }