feat(iroh): allow to override provenance for StaticProvider

iroh/src/discovery/static_provider.rs

@@ -64,10 +64,19 @@ use super::{Discovery, DiscoveryError, DiscoveryItem, NodeData, NodeInfo};
 /// ```
 ///
 /// [`NodeTicket`]: https://docs.rs/iroh-base/latest/iroh_base/ticket/struct.NodeTicket
-#[derive(Debug, Default, Clone)]
-#[repr(transparent)]
+#[derive(Debug, Clone)]
 pub struct StaticProvider {
     nodes: Arc<RwLock<BTreeMap<NodeId, StoredNodeInfo>>>,
+    provenance: &'static str,
+}
+
+impl Default for StaticProvider {
+    fn default() -> Self {
+        Self {
+            nodes: Default::default(),
+            provenance: Self::PROVENANCE,
+        }
+    }
 }
 
 #[derive(Debug)]
@@ -90,6 +99,20 @@ impl StaticProvider {
         Self::default()
     }
 
+    /// Creates a new static discovery instance with the provided `provenance`.
+    ///
+    /// The provenance is part of [`DiscoveryItem`]s returned from [`Self::resolve`].
+    /// It is mostly used for debugging information and allows understanding the origin of
+    /// addressing information used by an iroh [`Endpoint`].
+    ///
+    /// [`Endpoint`]: crate::Endpoint
+    pub fn with_provenance(provenance: &'static str) -> Self {
+        Self {
+            nodes: Default::default(),
+            provenance,
+        }
+    }
+
     /// Creates a static discovery instance from node addresses.
     ///
     /// # Examples
@@ -196,7 +219,7 @@ impl Discovery for StaticProvider {
                     .as_micros() as u64;
                 let item = DiscoveryItem::new(
                     NodeInfo::from_parts(node_id, node_info.data.clone()),
-                    Self::PROVENANCE,
+                    self.provenance,
                     Some(last_updated),
                 );
                 Some(stream::iter(Some(Ok(item))).boxed())
@@ -248,4 +271,22 @@ mod tests {
 
         Ok(())
     }
+
+    #[tokio::test]
+    async fn test_provenance() -> Result {
+        let discovery = StaticProvider::with_provenance("foo");
+        let key = SecretKey::from_bytes(&[0u8; 32]);
+        let addr = NodeAddr {
+            node_id: key.public(),
+            relay_url: Some("https://example.com".parse()?),
+            direct_addresses: Default::default(),
+        };
+        discovery.add_node_info(addr);
+        let mut stream = discovery.resolve(key.public()).unwrap();
+        let item = stream.next().await.unwrap()?;
+        assert_eq!(item.provenance(), "foo");
+        assert_eq!(item.relay_url(), Some(&("https://example.com".parse()?)));
+
+        Ok(())
+    }
 }