From c1ca9b7e1d314631f7dbc3e53a38f9fe46d07c0b Mon Sep 17 00:00:00 2001
From: "Denis K." <konsistentsi@gmail.com>
Date: Mon, 27 Jan 2025 19:35:06 +0700
Subject: [PATCH] WIP: build

---
 Cargo.lock          | 1 +
 node/Cargo.toml     | 1 +
 node/src/command.rs | 4 +++-
 3 files changed, 5 insertions(+), 1 deletion(-)

diff --git a/Cargo.lock b/Cargo.lock
index c03e12e8..748379e4 100644
--- a/Cargo.lock
+++ b/Cargo.lock
@@ -9729,6 +9729,7 @@ dependencies = [
  "clap",
  "color-print",
  "cumulus-client-cli",
+ "cumulus-client-service",
  "cumulus-primitives-core",
  "dev-runtime",
  "frame-benchmarking-cli",
diff --git a/node/Cargo.toml b/node/Cargo.toml
index b976ef84..85ccb191 100644
--- a/node/Cargo.toml
+++ b/node/Cargo.toml
@@ -39,6 +39,7 @@ polkadot-cli = { workspace = true }
 
 # cumulus
 cumulus-client-cli = { workspace = true }
+cumulus-client-service = { workspace = true }
 cumulus-primitives-core = { workspace = true }
 
 [build-dependencies]
diff --git a/node/src/command.rs b/node/src/command.rs
index cb41ba08..c81e890b 100644
--- a/node/src/command.rs
+++ b/node/src/command.rs
@@ -24,6 +24,7 @@ use crate::{
 use robonomics_primitives::{AccountId, Block, CommunityAccount};
 use robonomics_service as service;
 
+use cumulus_client_service::storage_proof_size::HostFunctions as ReclaimHostFunctions;
 use cumulus_primitives_core::ParaId;
 use frame_benchmarking_cli::{BenchmarkCmd, SUBSTRATE_REFERENCE_HARDWARE};
 use log::{info, warn};
@@ -302,7 +303,8 @@ pub fn run() -> sc_cli::Result<()> {
             match cmd {
                 BenchmarkCmd::Pallet(cmd) => {
                     if cfg!(feature = "runtime-benchmarks") {
-                        runner.sync_run(|config| cmd.run::<Block, ()>(config))
+                        // runner.sync_run(|config| cmd.run::<Block, ()>(config))
+                        runner.sync_run(|config| cmd.run_with_spec::<sp_runtime::traits::HashingFor<Block>, ReclaimHostFunctions>(Some(config.chain_spec)))
                     } else {
                         Err("Benchmarking wasn't enabled when building the node. \
                 You can enable it with `--features runtime-benchmarks`."