fix: dump_decalared_classes fix goerli (#1295)

fix: dump_declared_classes fix goerli

Author: Yael-Starkware

crates/dump_declared_classes/src/main.rs

@@ -2,23 +2,24 @@ use clap::{Arg, Command};
 use papyrus_storage::utils::dump_declared_classes_table_by_block_range;
 
 /// This executable dumps the declared_classes table from the storage to a file.
-
 fn main() {
     let cli_params = get_cli_params();
     match dump_declared_classes_table_by_block_range(
         cli_params.start_block,
         cli_params.end_block,
         &cli_params.file_path,
+        &cli_params.chain_id,
     ) {
         Ok(_) => println!("Dumped declared_classes table to file: {} .", cli_params.file_path),
-        Err(e) => println!("Failed dumping declared_classes table with error: {} .", e),
+        Err(e) => println!("Failed dumping declared_classes table with error: {}", e),
     }
 }
 
 struct CliParams {
     start_block: u64,
     end_block: u64,
     file_path: String,
+    chain_id: String,
 }
 
 /// The start_block and end_block arguments are mandatory and define the block range to dump,
@@ -47,10 +48,17 @@ fn get_cli_params() -> CliParams {
                 .required(true)
                 .help("The block number to end dumping at."),
         )
+        .arg(
+            Arg::new("chain_id")
+                .short('c')
+                .long("chain_id")
+                .required(true)
+                .help("The chain id SN_MAIN/SN_GOERLI, default value is SN_MAIN."),
+        )
         .get_matches();
 
     let file_path =
-        matches.get_one::<String>("file_path").expect("Failed parsing file_path").as_str();
+        matches.get_one::<String>("file_path").expect("Failed parsing file_path").to_string();
     let start_block = matches
         .get_one::<String>("start_block")
         .expect("Failed parsing start_block")
@@ -64,5 +72,7 @@ fn get_cli_params() -> CliParams {
     if start_block >= end_block {
         panic!("start_block must be smaller than end_block");
     }
-    CliParams { start_block, end_block, file_path: file_path.to_string() }
+    let chain_id =
+        matches.get_one::<String>("chain_id").expect("Failed parsing chain_id").to_string();
+    CliParams { start_block, end_block, file_path, chain_id }
 }

crates/papyrus_storage/src/utils.rs

@@ -9,7 +9,7 @@ use std::io::{BufWriter, Write};
 
 use serde::Serialize;
 use starknet_api::block::BlockNumber;
-use starknet_api::core::{ClassHash, CompiledClassHash};
+use starknet_api::core::{ChainId, ClassHash, CompiledClassHash};
 use starknet_api::hash::StarkFelt;
 use starknet_api::state::{EntryPoint, EntryPointType};
 
@@ -61,8 +61,10 @@ pub fn dump_declared_classes_table_by_block_range(
     start_block: u64,
     end_block: u64,
     file_path: &str,
+    chain_id: &str,
 ) -> StorageResult<()> {
-    let storage_config = StorageConfig::default();
+    let mut storage_config = StorageConfig::default();
+    storage_config.db_config.chain_id = ChainId(chain_id.to_string());
     let (storage_reader, _) = open_storage(storage_config)?;
     let txn = storage_reader.begin_ro_txn()?;
     let compiled_class_marker = txn.get_compiled_class_marker()?;