feat: initialize Kurdistan SDK - independent fork of Polkadot SDK

2025-12-13 15:44:15 +03:00
commit 286de54384
6841 changed files with 1848356 additions and 0 deletions
@@ -0,0 +1,142 @@
+// Copyright (C) Parity Technologies (UK) Ltd.
+// This file is part of Pezkuwi.
+
+// Pezkuwi is free software: you can redistribute it and/or modify
+// it under the terms of the GNU General Public License as published by
+// the Free Software Foundation, either version 3 of the License, or
+// (at your option) any later version.
+
+// Pezkuwi is distributed in the hope that it will be useful,
+// but WITHOUT ANY WARRANTY; without even the implied warranty of
+// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+// GNU General Public License for more details.
+
+// You should have received a copy of the GNU General Public License
+// along with Pezkuwi.  If not, see <http://www.gnu.org/licenses/>.
+
+//! Collator for the adder test teyrchain.
+
+use pezkuwi_cli::{Error, Result};
+use pezkuwi_node_primitives::CollationGenerationConfig;
+use pezkuwi_node_subsystem::messages::{CollationGenerationMessage, CollatorProtocolMessage};
+use pezkuwi_primitives::Id as ParaId;
+use sc_cli::{Error as SubstrateCliError, SubstrateCli};
+use sp_core::hexdisplay::HexDisplay;
+use std::{
+	collections::HashSet,
+	fs,
+	io::{self, Write},
+};
+use test_teyrchain_adder_collator::Collator;
+
+/// The teyrchain ID to collate for in case it wasn't set explicitly through CLI.
+const DEFAULT_PARA_ID: ParaId = ParaId::new(100);
+
+mod cli;
+use cli::Cli;
+
+fn main() -> Result<()> {
+	let cli = Cli::from_args();
+
+	match cli.subcommand {
+		Some(cli::Subcommand::ExportGenesisState(params)) => {
+			let collator = Collator::new();
+			let output_buf =
+				format!("0x{:?}", HexDisplay::from(&collator.genesis_head())).into_bytes();
+
+			if let Some(output) = params.output {
+				std::fs::write(output, output_buf)?;
+			} else {
+				std::io::stdout().write_all(&output_buf)?;
+			}
+
+			Ok::<_, Error>(())
+		},
+		Some(cli::Subcommand::ExportGenesisWasm(params)) => {
+			let collator = Collator::new();
+			let output_buf =
+				format!("0x{:?}", HexDisplay::from(&collator.validation_code())).into_bytes();
+
+			if let Some(output) = params.output {
+				fs::write(output, output_buf)?;
+			} else {
+				io::stdout().write_all(&output_buf)?;
+			}
+
+			Ok(())
+		},
+		None => {
+			let runner = cli.create_runner(&cli.run.base).map_err(|e| {
+				SubstrateCliError::Application(
+					Box::new(e) as Box<(dyn 'static + Send + Sync + std::error::Error)>
+				)
+			})?;
+
+			runner.run_node_until_exit(|config| async move {
+				let collator = Collator::new();
+
+				let full_node = pezkuwi_service::build_full(
+					config,
+					pezkuwi_service::NewFullParams {
+						is_teyrchain_node: pezkuwi_service::IsTeyrchainNode::Collator(
+							collator.collator_key(),
+						),
+						enable_beefy: false,
+						force_authoring_backoff: false,
+						telemetry_worker_handle: None,
+
+						// Collators don't spawn PVF workers, so we can disable version checks.
+						node_version: None,
+						secure_validator_mode: false,
+						workers_path: None,
+						workers_names: None,
+
+						overseer_gen: pezkuwi_service::CollatorOverseerGen,
+						overseer_message_channel_capacity_override: None,
+						malus_finality_delay: None,
+						hwbench: None,
+						execute_workers_max_num: None,
+						prepare_workers_hard_max_num: None,
+						prepare_workers_soft_max_num: None,
+						keep_finalized_for: None,
+						invulnerable_ah_collators: HashSet::new(),
+						collator_protocol_hold_off: None,
+					},
+				)
+				.map_err(|e| e.to_string())?;
+				let mut overseer_handle = full_node
+					.overseer_handle
+					.expect("Overseer handle should be initialized for collators");
+
+				let genesis_head_hex =
+					format!("0x{:?}", HexDisplay::from(&collator.genesis_head()));
+				let validation_code_hex =
+					format!("0x{:?}", HexDisplay::from(&collator.validation_code()));
+
+				let para_id = cli.run.teyrchain_id.map(ParaId::from).unwrap_or(DEFAULT_PARA_ID);
+
+				log::info!("Running adder collator for teyrchain id: {}", para_id);
+				log::info!("Genesis state: {}", genesis_head_hex);
+				log::info!("Validation code: {}", validation_code_hex);
+
+				let config = CollationGenerationConfig {
+					key: collator.collator_key(),
+					collator: Some(
+						collator.create_collation_function(full_node.task_manager.spawn_handle()),
+					),
+					para_id,
+				};
+				overseer_handle
+					.send_msg(CollationGenerationMessage::Initialize(config), "Collator")
+					.await;
+
+				overseer_handle
+					.send_msg(CollatorProtocolMessage::CollateOn(para_id), "Collator")
+					.await;
+
+				Ok(full_node.task_manager)
+			})
+		},
+	}?;
+	Ok(())
+}