add mmr rpc to the node

Author: vedhavyas

Cargo.lock

@@ -24,6 +24,7 @@ futures = "0.3.29"
 hex = "0.4.3"
 jsonrpsee = { version = "0.16.3", features = ["server"] }
 mmr-gadget = { version = "4.0.0-dev", git = "https://github.com/subspace/polkadot-sdk", rev = "d6b500960579d73c43fc4ef550b703acfa61c4c8" }
+mmr-rpc = { version = "4.0.0-dev", git = "https://github.com/subspace/polkadot-sdk", rev = "d6b500960579d73c43fc4ef550b703acfa61c4c8" }
 pallet-transaction-payment-rpc = { version = "4.0.0-dev", git = "https://github.com/subspace/polkadot-sdk", rev = "d6b500960579d73c43fc4ef550b703acfa61c4c8" }
 parity-scale-codec = "3.6.9"
 parking_lot = "0.12.1"

crates/subspace-service/Cargo.toml

@@ -1055,6 +1055,7 @@ where
             let archived_segment_notification_stream = archived_segment_notification_stream.clone();
             let transaction_pool = transaction_pool.clone();
             let chain_spec = config.base.chain_spec.cloned_box();
+            let backend = backend.clone();
 
             Box::new(move |deny_unsafe, subscription_executor| {
                 let deps = rpc::FullDeps {
@@ -1071,6 +1072,7 @@ where
                     segment_headers_store: segment_headers_store.clone(),
                     sync_oracle: sync_oracle.clone(),
                     kzg: subspace_link.kzg().clone(),
+                    backend: backend.clone(),
                 };
 
                 rpc::create_full(deps).map_err(Into::into)

crates/subspace-service/src/lib.rs

@@ -22,6 +22,7 @@
 #![warn(missing_docs)]
 
 use jsonrpsee::RpcModule;
+use mmr_rpc::{Mmr, MmrApiServer};
 use pallet_transaction_payment_rpc::{TransactionPayment, TransactionPaymentApiServer};
 use sc_client_api::{AuxStore, BlockBackend};
 use sc_consensus_subspace::archiver::{ArchivedSegmentNotification, SegmentHeadersStore};
@@ -42,13 +43,14 @@ use sp_consensus_subspace::{FarmerPublicKey, SubspaceApi};
 use sp_objects::ObjectsApi;
 use std::sync::Arc;
 use subspace_core_primitives::crypto::kzg::Kzg;
+use subspace_core_primitives::BlockNumber;
 use subspace_networking::libp2p::Multiaddr;
 use subspace_runtime_primitives::opaque::Block;
 use subspace_runtime_primitives::{AccountId, Balance, Nonce};
 use substrate_frame_rpc_system::{System, SystemApiServer};
 
 /// Full client dependencies.
-pub struct FullDeps<C, P, SO, AS>
+pub struct FullDeps<C, P, SO, AS, B>
 where
     SO: SyncOracle + Send + Sync + Clone,
 {
@@ -78,11 +80,13 @@ where
     pub sync_oracle: SubspaceSyncOracle<SO>,
     /// Kzg instance.
     pub kzg: Kzg,
+    /// Backend used by the node.
+    pub backend: Arc<B>,
 }
 
 /// Instantiate all full RPC extensions.
-pub fn create_full<C, P, SO, AS>(
-    deps: FullDeps<C, P, SO, AS>,
+pub fn create_full<C, P, SO, AS, B>(
+    deps: FullDeps<C, P, SO, AS, B>,
 ) -> Result<RpcModule<()>, Box<dyn std::error::Error + Send + Sync>>
 where
     C: ProvideRuntimeApi<Block>
@@ -96,10 +100,13 @@ where
         + pallet_transaction_payment_rpc::TransactionPaymentRuntimeApi<Block, Balance>
         + BlockBuilder<Block>
         + SubspaceApi<Block, FarmerPublicKey>
+        + mmr_rpc::MmrRuntimeApi<Block, <Block as sp_runtime::traits::Block>::Hash, BlockNumber>
         + ObjectsApi<Block>,
     P: TransactionPool + 'static,
     SO: SyncOracle + Send + Sync + Clone + 'static,
     AS: AuxStore + Send + Sync + 'static,
+    B: sc_client_api::Backend<Block> + Send + Sync + 'static,
+    B::State: sc_client_api::StateBackend<sp_runtime::traits::HashingFor<Block>>,
 {
     let mut module = RpcModule::new(());
     let FullDeps {
@@ -115,6 +122,7 @@ where
         segment_headers_store,
         sync_oracle,
         kzg,
+        backend,
     } = deps;
 
     let chain_name = chain_spec.name().to_string();
@@ -127,7 +135,7 @@ where
 
     module.merge(
         SubspaceRpc::new(SubspaceRpcConfig {
-            client,
+            client: client.clone(),
             subscription_executor,
             new_slot_notification_stream,
             reward_signing_notification_stream,
@@ -140,6 +148,15 @@ where
         })?
         .into_rpc(),
     )?;
+    module.merge(
+        Mmr::new(
+            client,
+            backend
+                .offchain_storage()
+                .ok_or("Backend doesn't provide the required offchain storage")?,
+        )
+        .into_rpc(),
+    )?;
 
     Ok(module)
 }