From cecdf611e948257a79e7c262501bdadbd174e44d Mon Sep 17 00:00:00 2001
From: greged93 <82421016+greged93@users.noreply.github.com>
Date: Sat, 26 Oct 2024 08:11:27 +0200
Subject: [PATCH] feat: `map_chainspec` for `NodeConfig` (#12068)

---
 crates/exex/exex/src/dyn_context.rs | 22 +++-------------------
 crates/node/core/src/node_config.rs | 23 +++++++++++++++++++++++
 2 files changed, 26 insertions(+), 19 deletions(-)

diff --git a/crates/exex/exex/src/dyn_context.rs b/crates/exex/exex/src/dyn_context.rs
index 226f3a3feb9a..b48a6ebc951f 100644
--- a/crates/exex/exex/src/dyn_context.rs
+++ b/crates/exex/exex/src/dyn_context.rs
@@ -1,7 +1,7 @@
 //! Mirrored version of [`ExExContext`](`crate::ExExContext`)
 //! without generic abstraction over [Node](`reth_node_api::FullNodeComponents`)
 
-use std::{fmt::Debug, sync::Arc};
+use std::fmt::Debug;
 
 use reth_chainspec::{EthChainSpec, Head};
 use reth_node_api::FullNodeComponents;
@@ -55,24 +55,8 @@ where
     Node::Executor: Debug,
 {
     fn from(ctx: ExExContext<Node>) -> Self {
-        // convert `NodeConfig` with generic over chainspec into `NodeConfig<Box<dyn EthChainSpec>`
-        let chain: Arc<Box<dyn EthChainSpec + 'static>> =
-            Arc::new(Box::new(ctx.config.chain) as Box<dyn EthChainSpec>);
-        let config = NodeConfig {
-            chain,
-            datadir: ctx.config.datadir,
-            config: ctx.config.config,
-            metrics: ctx.config.metrics,
-            instance: ctx.config.instance,
-            network: ctx.config.network,
-            rpc: ctx.config.rpc,
-            txpool: ctx.config.txpool,
-            builder: ctx.config.builder,
-            debug: ctx.config.debug,
-            db: ctx.config.db,
-            dev: ctx.config.dev,
-            pruning: ctx.config.pruning,
-        };
+        let config =
+            ctx.config.map_chainspec(|chainspec| Box::new(chainspec) as Box<dyn EthChainSpec>);
         let notifications = Box::new(ctx.notifications) as Box<dyn ExExNotificationsStream>;
 
         Self {
diff --git a/crates/node/core/src/node_config.rs b/crates/node/core/src/node_config.rs
index a8799d80df1c..80fb5152e7bb 100644
--- a/crates/node/core/src/node_config.rs
+++ b/crates/node/core/src/node_config.rs
@@ -422,6 +422,29 @@ impl<ChainSpec> NodeConfig<ChainSpec> {
             Err(e) => Err(eyre!("Failed to load configuration: {e}")),
         }
     }
+
+    /// Modifies the [`ChainSpec`] generic of the config using the provided closure.
+    pub fn map_chainspec<F, C>(self, f: F) -> NodeConfig<C>
+    where
+        F: FnOnce(Arc<ChainSpec>) -> C,
+    {
+        let chain = Arc::new(f(self.chain));
+        NodeConfig {
+            chain,
+            datadir: self.datadir,
+            config: self.config,
+            metrics: self.metrics,
+            instance: self.instance,
+            network: self.network,
+            rpc: self.rpc,
+            txpool: self.txpool,
+            builder: self.builder,
+            debug: self.debug,
+            db: self.db,
+            dev: self.dev,
+            pruning: self.pruning,
+        }
+    }
 }
 
 impl Default for NodeConfig<ChainSpec> {