From d317b4a0fbe54fed247b626e8d0d0928497d812b Mon Sep 17 00:00:00 2001
From: Matthias Seitz <matthias.seitz@outlook.de>
Date: Mon, 1 Jul 2024 15:07:49 +0200
Subject: [PATCH] chore: add builder with rng secret key fn (#9218)

---
 crates/net/network/src/config.rs | 16 ++++++++++++++++
 1 file changed, 16 insertions(+)

diff --git a/crates/net/network/src/config.rs b/crates/net/network/src/config.rs
index b6c955212..3f7887a9b 100644
--- a/crates/net/network/src/config.rs
+++ b/crates/net/network/src/config.rs
@@ -88,6 +88,11 @@ impl NetworkConfig<()> {
     pub fn builder(secret_key: SecretKey) -> NetworkConfigBuilder {
         NetworkConfigBuilder::new(secret_key)
     }
+
+    /// Convenience method for creating the corresponding builder type with a random secret key.
+    pub fn builder_with_rng_secret_key() -> NetworkConfigBuilder {
+        NetworkConfigBuilder::with_rng_secret_key()
+    }
 }
 
 impl<C> NetworkConfig<C> {
@@ -176,6 +181,12 @@ pub struct NetworkConfigBuilder {
 
 #[allow(missing_docs)]
 impl NetworkConfigBuilder {
+    /// Create a new builder instance with a random secret key.
+    pub fn with_rng_secret_key() -> Self {
+        Self::new(rng_secret_key())
+    }
+
+    /// Create a new builder instance with the given secret key.
     pub fn new(secret_key: SecretKey) -> Self {
         Self {
             secret_key,
@@ -212,6 +223,11 @@ impl NetworkConfigBuilder {
         pk2id(&self.secret_key.public_key(SECP256K1))
     }
 
+    /// Returns the configured [`SecretKey`], from which the node's identity is derived.
+    pub const fn secret_key(&self) -> &SecretKey {
+        &self.secret_key
+    }
+
     /// Sets the chain spec.
     pub fn chain_spec(mut self, chain_spec: Arc<ChainSpec>) -> Self {
         self.chain_spec = chain_spec;