snapshot before rebranding

2025-12-14 07:37:21 +03:00
parent 5520d491a5
commit 09735eb97a
1752 changed files with 58116 additions and 15986 deletions
@@ -0,0 +1,290 @@
+// Copyright (C) Parity Technologies (UK) Ltd.
+// This file is dual-licensed as Apache-2.0 or GPL-3.0.
+// see LICENSE for license details.
+
+#![allow(dead_code)]
+#![allow(unused_variables)]
+
+use clap::Parser;
+use codec::Compact;
+use std::{fs, fs::File, io::BufReader, time::Duration};
+use bizinikiwi_txtesttool::{
+	block_monitor::BlockMonitor,
+	cli::{Cli, CliCommand},
+	execution_log::{journal::Journal, make_stats, STAT_TARGET},
+	init_logger,
+	runner::DefaultTxTask,
+	scenario::{AccountsDescription, ChainType, ScenarioBuilder, ScenarioType},
+	subxt_transaction::{
+		self, generate_ecdsa_keypair, generate_sr25519_keypair, EthRuntimeConfig, EthTransaction,
+		EthTransactionsSink, SubstrateTransaction, SubstrateTransactionsSink, SENDER_SEED,
+	},
+};
+use subxt::{ext::frame_metadata::RuntimeMetadataPrefixed, PolkadotConfig};
+use tracing::info;
+
+macro_rules! populate_scenario_builder {
+	($scenario_builder:expr, $scenario_type:expr) => {{
+		match $scenario_type {
+			ScenarioType::OneShot { account, nonce } =>
+				$scenario_builder.with_account_id(account.clone()).with_nonce_from(*nonce),
+			ScenarioType::FromSingleAccount { account, from, count } => $scenario_builder
+				.with_account_id(account.clone())
+				.with_nonce_from(*from)
+				.with_txs_count(*count),
+			ScenarioType::FromManyAccounts { start_id, last_id, from, count } => $scenario_builder
+				.with_start_id(*start_id)
+				.with_last_id(*last_id)
+				.with_nonce_from(*from)
+				.with_txs_count(*count),
+		}
+	}};
+}
+
+#[tokio::main]
+async fn main() -> Result<(), Box<dyn std::error::Error>> {
+	init_logger();
+
+	let cli = Cli::parse();
+
+	match &cli.command {
+		CliCommand::Tx {
+			chain,
+			ws,
+			unwatched,
+			block_monitor,
+			mortal,
+			log_file,
+			scenario,
+			send_threshold,
+			remark,
+			tip,
+			use_legacy_backend,
+		} => match chain {
+			ChainType::Fake => {
+				unimplemented!()
+			},
+			ChainType::Eth => {
+				let mut scenario_builder = ScenarioBuilder::new()
+					.with_rpc_uri(ws.to_string())
+					.with_chain_type(chain.clone())
+					.with_send_threshold(*send_threshold as usize)
+					.with_block_monitoring(*block_monitor)
+					.with_watched_txs(!unwatched)
+					.with_installed_ctrlc_stop_hook(true)
+					.with_legacy_backend(*use_legacy_backend)
+					.with_tip(*tip);
+
+				scenario_builder = populate_scenario_builder!(scenario_builder, scenario);
+
+				if let Some(mortality) = mortal {
+					scenario_builder = scenario_builder.with_mortality(*mortality);
+				}
+
+				if let Some(inner) = remark {
+					scenario_builder = scenario_builder.with_remark_recipe(*inner);
+				}
+
+				let scenario_executor = scenario_builder.build().await;
+				let _ = scenario_executor.execute().await;
+			},
+			ChainType::Sub => {
+				let mut scenario_builder = ScenarioBuilder::new()
+					.with_rpc_uri(ws.to_string())
+					.with_chain_type(chain.clone())
+					.with_send_threshold(*send_threshold as usize)
+					.with_block_monitoring(*block_monitor)
+					.with_watched_txs(!unwatched)
+					.with_installed_ctrlc_stop_hook(true)
+					.with_legacy_backend(*use_legacy_backend)
+					.with_tip(*tip);
+
+				scenario_builder = populate_scenario_builder!(scenario_builder, scenario);
+
+				if let Some(mortality) = mortal {
+					scenario_builder = scenario_builder.with_mortality(*mortality);
+				}
+
+				if let Some(inner) = remark {
+					scenario_builder = scenario_builder.with_remark_recipe(*inner);
+				}
+
+				let scenario_executor = scenario_builder.build().await;
+				let _ = scenario_executor.execute().await;
+			},
+		},
+		CliCommand::CheckNonce { chain, ws, account } => {
+			match chain {
+				ChainType::Fake => {
+					panic!("check nonce not supported for fake chain");
+				},
+				ChainType::Eth => {
+					let desc = if let Ok(id) = account.parse::<u32>() {
+						AccountsDescription::Derived(id..id + 1)
+					} else {
+						AccountsDescription::Keyring(account.clone())
+					};
+					let sink = EthTransactionsSink::new_with_uri_with_accounts_description(
+						ws,
+						desc,
+						generate_ecdsa_keypair,
+						None,
+						false,
+					)
+					.await;
+					let account =
+						sink.get_from_account_id(account).ok_or("account shall be correct")?;
+					let nonce = sink.check_account_nonce(account).await?;
+					info!(target:STAT_TARGET, "{nonce:?}");
+				},
+				ChainType::Sub => {
+					let desc = if let Ok(id) = account.parse::<u32>() {
+						AccountsDescription::Derived(id..id + 1)
+					} else {
+						AccountsDescription::Keyring(account.clone())
+					};
+					let sink = SubstrateTransactionsSink::new_with_uri_with_accounts_description(
+						ws,
+						desc,
+						generate_sr25519_keypair,
+						None,
+						false,
+					)
+					.await;
+					let account =
+						sink.get_from_account_id(account).ok_or("account shall be correct")?;
+					let nonce = sink.check_account_nonce(account).await?;
+					info!(target:STAT_TARGET, "{nonce:?}");
+				},
+			};
+		},
+		CliCommand::Metadata { ws } => {
+			// Handle metadata command
+			use codec::Decode;
+			let api = subxt::OnlineClient::<EthRuntimeConfig>::from_insecure_url(ws).await?;
+			let runtime_apis = api.runtime_api().at_latest().await?;
+			let raw_bytes: Vec<u8> = runtime_apis.call_raw("Metadata_metadata", None).await?;
+			let (_, meta): (Compact<u32>, RuntimeMetadataPrefixed) =
+				Decode::decode(&mut &raw_bytes[..]).map_err(|e| format!("Decode error: {e}"))?;
+			println!("{meta:#?}");
+		},
+		CliCommand::BlockMonitor { chain, ws, display } => {
+			match chain {
+				ChainType::Sub => {
+					let block_monitor =
+						BlockMonitor::<PolkadotConfig>::new_with_options(ws, *display).await;
+					async {
+						loop {
+							tokio::time::sleep(Duration::from_secs(10)).await
+						}
+					}
+					.await;
+				},
+				ChainType::Eth => {
+					let block_monitor =
+						BlockMonitor::<EthRuntimeConfig>::new_with_options(ws, *display).await;
+					async {
+						loop {
+							tokio::time::sleep(Duration::from_secs(10)).await
+						}
+					}
+					.await;
+				},
+				ChainType::Fake => {
+					unimplemented!()
+				},
+			};
+		},
+		CliCommand::LoadLog { chain, log_file, show_graphs, out_csv_filename, .. } => match chain {
+			ChainType::Sub => {
+				let logs = Journal::<DefaultTxTask<SubstrateTransaction>>::load_logs(
+					log_file,
+					out_csv_filename,
+				);
+				make_stats(logs.values().cloned(), *show_graphs);
+			},
+			ChainType::Eth => {
+				let logs =
+					Journal::<DefaultTxTask<EthTransaction>>::load_logs(log_file, out_csv_filename);
+				make_stats(logs.values().cloned(), *show_graphs);
+			},
+			ChainType::Fake => {
+				unimplemented!()
+			},
+		},
+		CliCommand::GenerateEndowedAccounts {
+			chain,
+			start_id,
+			last_id,
+			balance,
+			out_file_name,
+			chain_spec,
+		} => {
+			let accounts_description = AccountsDescription::Derived(*start_id..last_id + 1);
+			let funded_accounts = match chain {
+				ChainType::Sub => {
+					let accounts = subxt_transaction::derive_accounts::<PolkadotConfig, _, _>(
+						accounts_description.clone(),
+						SENDER_SEED,
+						generate_sr25519_keypair,
+					);
+					accounts
+						.values()
+						.map(|keypair| {
+							serde_json::json!((
+								<PolkadotConfig as subxt::Config>::AccountId::from(
+									keypair.0.clone().public_key()
+								),
+								balance,
+							))
+						})
+						.collect::<Vec<_>>()
+				},
+				ChainType::Eth => {
+					let accounts = subxt_transaction::derive_accounts::<EthRuntimeConfig, _, _>(
+						accounts_description.clone(),
+						SENDER_SEED,
+						generate_ecdsa_keypair,
+					);
+					accounts
+						.values()
+						.map(|keypair| {
+							serde_json::json!((
+								"0x".to_string() + &hex::encode(keypair.0.clone().public_key()),
+								balance,
+							))
+						})
+						.collect::<Vec<_>>()
+				},
+				ChainType::Fake => Default::default(),
+			};
+
+			if let Some(chain_spec) = chain_spec {
+				let file = File::open(chain_spec)?;
+				let reader = BufReader::new(file);
+				let mut chain_spec: serde_json::Value = serde_json::from_reader(reader)?;
+
+				if let Some(balances) = chain_spec["genesis"]["runtimeGenesis"]["patch"]["balances"]
+					["balances"]
+					.as_array_mut()
+				{
+					balances.extend(funded_accounts);
+				} else {
+					return Err("Balances array not found in provided chain-spec".into());
+				}
+
+				fs::write(
+					out_file_name,
+					serde_json::to_string_pretty(&chain_spec)
+						.map_err(|e| format!("to pretty failed: {e}"))?,
+				)
+				.map_err(|err| err.to_string())?;
+			} else {
+				let json_object = serde_json::json!({"balances":{"balances":funded_accounts}});
+				fs::write(out_file_name, serde_json::to_string_pretty(&json_object)?.as_bytes())?;
+			}
+		},
+	};
+
+	Ok(())
+}