feat: ✨ add maintenance mode to node

client/blockchain-service/src/handler.rs

@@ -126,6 +126,8 @@ where
     ///
     /// Only required if the node is running as a provider.
     pub(crate) capacity_manager: Option<CapacityRequestQueue>,
+    /// Whether the node is running in maintenance mode.
+    pub(crate) maintenance_mode: bool,
 }
 
 /// Event loop for the BlockchainService actor.
@@ -216,6 +218,12 @@ where
         message: Self::Message,
     ) -> impl std::future::Future<Output = ()> + Send {
         async {
+            // If the node is running in maintenance mode, we don't process any messages.
+            if self.maintenance_mode {
+                info!(target: LOG_TARGET, "🔒 Maintenance mode is enabled. Skipping message processing.");
+                return;
+            }
+
             match message {
                 BlockchainServiceCommand::SendExtrinsic {
                     call,
@@ -1115,6 +1123,7 @@ where
         rocksdb_root_path: impl Into<PathBuf>,
         notify_period: Option<u32>,
         capacity_request_queue: Option<CapacityRequestQueue>,
+        maintenance_mode: bool,
     ) -> Self {
         Self {
             event_bus_provider: BlockchainServiceEventBusProvider::new(),
@@ -1130,6 +1139,7 @@ where
             persistent_state: BlockchainServiceStateStore::new(rocksdb_root_path.into()),
             notify_period,
             capacity_manager: capacity_request_queue,
+            maintenance_mode,
         }
     }
 
@@ -1285,6 +1295,12 @@ where
     ) where
         Block: cumulus_primitives_core::BlockT<Hash = H256>,
     {
+        // If the node is running in maintenance mode, we don't process block imports.
+        if self.maintenance_mode {
+            info!(target: LOG_TARGET, "🔒 Maintenance mode is enabled. Skipping processing ofblock import #{}: {}", block_number, block_hash);
+            return;
+        }
+
         trace!(target: LOG_TARGET, "📠 Processing block import #{}: {}", block_number, block_hash);
 
         // Provider-specific code to run on every block import.
@@ -1377,6 +1393,12 @@ where
         let block_hash: H256 = notification.hash;
         let block_number: BlockNumber = (*notification.header.number()).saturated_into();
 
+        // If the node is running in maintenance mode, we don't process finality notifications.
+        if self.maintenance_mode {
+            info!(target: LOG_TARGET, "🔒 Maintenance mode is enabled. Skipping finality notification #{}: {}", block_number, block_hash);
+            return;
+        }
+
         info!(target: LOG_TARGET, "📨 Finality notification #{}: {}", block_number, block_hash);
 
         // Get events from storage.

client/blockchain-service/src/lib.rs

@@ -30,6 +30,7 @@ pub async fn spawn_blockchain_service<FSH>(
     rocksdb_root_path: impl Into<PathBuf>,
     notify_period: Option<u32>,
     capacity_config: Option<CapacityConfig>,
+    maintenance_mode: bool,
 ) -> ActorHandle<BlockchainService<FSH>>
 where
     FSH: shc_forest_manager::traits::ForestStorageHandler + Clone + Send + Sync + 'static,
@@ -46,6 +47,7 @@ where
         rocksdb_root_path,
         notify_period,
         capacity_config.map(CapacityRequestQueue::new),
+        maintenance_mode,
     );
 
     task_spawner.spawn_actor(blockchain_service)

node/src/cli.rs

@@ -106,6 +106,12 @@ pub struct ProviderConfigurations {
     #[arg(long)]
     pub provider: bool,
 
+    /// Run the node in maintenance mode.
+    /// In this mode, the node will not import blocks or participate in consensus,
+    /// but will allow specific RPC calls for file and storage management.
+    #[arg(long, default_value = "false")]
+    pub maintenance_mode: bool,
+
     /// Type of StorageHub provider.
     #[clap(
         long,
@@ -173,6 +179,7 @@ impl ProviderConfigurations {
             jump_capacity: self.jump_capacity,
             extrinsic_retry_timeout: self.extrinsic_retry_timeout,
             msp_charging_period: self.msp_charging_period,
+            maintenance_mode: self.maintenance_mode,
         }
     }
 }

node/src/command.rs

@@ -35,6 +35,8 @@ pub struct ProviderOptions {
     pub extrinsic_retry_timeout: u64,
     /// MSP charging fees frequency.
     pub msp_charging_period: Option<u32>,
+    /// Whether the node is running in maintenance mode.
+    pub maintenance_mode: bool,
 }
 
 fn load_spec(id: &str) -> std::result::Result<Box<dyn ChainSpec>, String> {