Add API to get the content of a single tag

Author: rklaehn

src/rpc.rs

@@ -317,6 +317,16 @@ impl<D: crate::store::Store> Handler<D> {
             #[allow(clippy::manual_flatten)]
             for item in tags {
                 if let Ok((name, HashAndFormat { hash, format })) = item {
+                    if let Some(from) = msg.from.as_ref() {
+                        if &name < from {
+                            continue;
+                        }
+                    }
+                    if let Some(to) = msg.to.as_ref() {
+                        if &name >= to {
+                            break;
+                        }
+                    }
                     if (format.is_raw() && msg.raw) || (format.is_hash_seq() && msg.hash_seq) {
                         co.yield_(TagInfo { name, hash, format }).await;
                     }

src/rpc/client/tags.rs

@@ -42,6 +42,12 @@ where
         Self { rpc }
     }
 
+    /// Get the value of a single tag
+    pub async fn get(&self, name: Tag) -> Result<Option<TagInfo>> {
+        let mut stream = self.rpc.server_streaming(ListRequest::single(name)).await?;
+        Ok(stream.next().await.transpose()?)
+    }
+
     /// Lists all tags.
     pub async fn list(&self) -> Result<impl Stream<Item = Result<TagInfo>>> {
         let stream = self.rpc.server_streaming(ListRequest::all()).await?;
@@ -59,6 +65,15 @@ where
         self.rpc.rpc(DeleteRequest { name }).await??;
         Ok(())
     }
+
+    pub async fn get(&self, name: Tag) -> Result<TagInfo> {
+        let mut stream = self.rpc.server_streaming(ListRequest::single(name)).await?;
+        let res = stream
+            .next()
+            .await
+            .ok_or_else(|| anyhow::anyhow!("no response"))??;
+        Ok(res)
+    }
 }
 
 /// Information about a tag.

src/rpc/proto/tags.rs

@@ -1,10 +1,13 @@
 //! Tags RPC protocol
+use bytes::Bytes;
 use nested_enum_utils::enum_conversions;
 use quic_rpc_derive::rpc_requests;
 use serde::{Deserialize, Serialize};
 
 use super::{RpcResult, RpcService};
-use crate::{net_protocol::BatchId, rpc::client::tags::TagInfo, HashAndFormat, Tag};
+use crate::{
+    net_protocol::BatchId, rpc::client::tags::TagInfo, util::increment_vec, HashAndFormat, Tag,
+};
 
 #[allow(missing_docs)]
 #[derive(strum::Display, Debug, Serialize, Deserialize)]
@@ -73,14 +76,33 @@ pub struct ListRequest {
     pub raw: bool,
     /// List hash seq tags
     pub hash_seq: bool,
+    /// From tag
+    pub from: Option<Tag>,
+    /// To tag (exclusive)
+    pub to: Option<Tag>,
 }
 
 impl ListRequest {
+    /// List a single tag
+    pub fn single(name: Tag) -> Self {
+        let mut next = name.0.to_vec();
+        increment_vec(&mut next);
+        let next = Bytes::from(next).into();
+        Self {
+            raw: true,
+            hash_seq: true,
+            from: Some(name),
+            to: Some(next),
+        }
+    }
+
     /// List all tags
     pub fn all() -> Self {
         Self {
             raw: true,
             hash_seq: true,
+            from: None,
+            to: None,
         }
     }
 
@@ -89,6 +111,8 @@ impl ListRequest {
         Self {
             raw: true,
             hash_seq: false,
+            from: None,
+            to: None,
         }
     }
 
@@ -97,6 +121,8 @@ impl ListRequest {
         Self {
             raw: false,
             hash_seq: true,
+            from: None,
+            to: None,
         }
     }
 }

src/util.rs

@@ -302,6 +302,19 @@ pub(crate) fn raw_outboard_size(size: u64) -> u64 {
     BaoTree::new(size, IROH_BLOCK_SIZE).outboard_size()
 }
 
+/// Increment a byte vector, lexographically.
+pub(crate) fn increment_vec(bytes: &mut Vec<u8>) {
+    for byte in bytes.iter_mut().rev() {
+        if *byte < 255 {
+            *byte += 1;
+            return;
+        }
+        *byte = 0;
+    }
+
+    bytes.push(0);
+}
+
 /// Synchronously compute the outboard of a file, and return hash and outboard.
 ///
 /// It is assumed that the file is not modified while this is running.