feat(bin/node): support proofs exex cli flags

Author: refcell

bin/node/src/cli.rs

@@ -1,6 +1,8 @@
 //! Contains the CLI arguments
 
-use base_reth_runner::{BaseNodeConfig, FlashblocksConfig, TracingConfig};
+use std::path::PathBuf;
+
+use base_reth_runner::{BaseNodeConfig, FlashblocksConfig, ProofsConfig, TracingConfig};
 use reth_optimism_node::args::RollupArgs;
 
 /// CLI Arguments
@@ -37,6 +39,19 @@ pub struct Args {
     /// Enable metering RPC for transaction bundle simulation
     #[arg(long = "enable-metering", value_name = "ENABLE_METERING")]
     pub enable_metering: bool,
+
+    /// If true, initialize external-proofs exex to save and serve trie nodes to provide proofs
+    /// faster.
+    #[arg(long = "proofs-history", value_name = "PROOFS_HISTORY", default_value = "false")]
+    pub proofs_history: bool,
+
+    /// The path to the storage DB for proofs history.
+    #[arg(
+        long = "proofs-history.storage-path",
+        value_name = "PROOFS_HISTORY_STORAGE_PATH",
+        required_if_eq("proofs_history", "true")
+    )]
+    pub proofs_history_storage_path: Option<PathBuf>,
 }
 
 impl Args {
@@ -45,22 +60,39 @@ impl Args {
     pub const fn flashblocks_enabled(&self) -> bool {
         self.websocket_url.is_some()
     }
+
+    /// Returns the [`FlashblocksConfig`] if flashblocks is enabled.
+    pub fn flashblocks_config(&self) -> Option<FlashblocksConfig> {
+        self.websocket_url.as_ref().map(|websocket_url| FlashblocksConfig {
+            websocket_url: websocket_url.clone(),
+            max_pending_blocks_depth: self.max_pending_blocks_depth,
+        })
+    }
+
+    /// Returns the [`ProofsConfig`].
+    pub fn proofs_config(&self) -> ProofsConfig {
+        ProofsConfig {
+            enabled: self.proofs_history,
+            storage_path: self.proofs_history_storage_path.clone(),
+        }
+    }
+
+    /// Returns the [`TracingConfig`].
+    pub const fn tracing_config(&self) -> TracingConfig {
+        TracingConfig {
+            enabled: self.enable_transaction_tracing,
+            logs_enabled: self.enable_transaction_tracing_logs,
+        }
+    }
 }
 
 impl From<Args> for BaseNodeConfig {
     fn from(args: Args) -> Self {
-        let flashblocks = args.websocket_url.map(|websocket_url| FlashblocksConfig {
-            websocket_url,
-            max_pending_blocks_depth: args.max_pending_blocks_depth,
-        });
-
         Self {
+            flashblocks: args.flashblocks_config(),
+            proofs: args.proofs_config(),
+            tracing: args.tracing_config(),
             rollup_args: args.rollup_args,
-            flashblocks,
-            tracing: TracingConfig {
-                enabled: args.enable_transaction_tracing,
-                logs_enabled: args.enable_transaction_tracing_logs,
-            },
             metering_enabled: args.enable_metering,
         }
     }

crates/runner/src/config.rs

@@ -11,6 +11,8 @@ pub struct BaseNodeConfig {
     pub flashblocks: Option<FlashblocksConfig>,
     /// Execution extension tracing toggles.
     pub tracing: TracingConfig,
+    /// Proofs extension configuration.
+    pub proofs: ProofsConfig,
     /// Indicates whether the metering RPC surface should be installed.
     pub metering_enabled: bool,
 }
@@ -31,6 +33,16 @@ pub struct FlashblocksConfig {
     pub max_pending_blocks_depth: u64,
 }
 
+/// Proofs Extension Configuration.
+#[derive(Debug, Clone)]
+pub struct ProofsConfig {
+    /// If true, initializes external-proofs ExEx to save and serve trie nodes to provide proofs
+    /// faster.
+    pub enabled: bool,
+    /// The path to the storage DB for proofs history.
+    pub storage_path: Option<std::path::PathBuf>,
+}
+
 /// Transaction tracing toggles.
 #[derive(Debug, Clone, Copy)]
 pub struct TracingConfig {

crates/runner/src/lib.rs

@@ -8,7 +8,7 @@ mod runner;
 pub use runner::BaseNodeRunner;
 
 mod config;
-pub use config::{BaseNodeConfig, FlashblocksConfig, TracingConfig};
+pub use config::{BaseNodeConfig, FlashblocksConfig, ProofsConfig, TracingConfig};
 
 mod extensions;
 pub use extensions::{BaseRpcExtension, FlashblocksCanonExtension, TransactionTracingExtension};