pop_chains/build/

mod.rs

// SPDX-License-Identifier: GPL-3.0

use crate::errors::{Error, handle_command_error};
use anyhow::{Result, anyhow};
use duct::cmd;
use pop_common::{Profile, account_id::convert_to_evm_accounts, manifest::from_path};
use sc_chain_spec::{GenericChainSpec, NoExtension};
use serde_json::{Value, json};
use sp_core::bytes::to_hex;
use std::{
	fs,
	path::{Path, PathBuf},
	str::FromStr,
};

/// Build the deterministic runtime.
pub mod runtime;

/// A builder for generating chain specifications.
///
/// This enum represents two different ways to build a chain specification:
/// - Using an existing node.
/// - Using a runtime.
pub enum ChainSpecBuilder {
	/// A node-based chain specification builder.
	Node {
		/// Path to the node directory.
		node_path: PathBuf,
		/// Whether to include a default bootnode in the specification.
		default_bootnode: bool,
		/// The build profile to use (debug, release, production, etc).
		profile: Profile,
	},
	/// A runtime-based chain specification builder.
	Runtime {
		/// Path to the runtime directory.
		runtime_path: PathBuf,
		/// The build profile to use (debug, release, production, etc).
		profile: Profile,
	},
}

impl ChainSpecBuilder {
	/// Builds the chain specification using the provided profile and features.
	///
	/// # Arguments
	/// * `features` - A list of cargo features to enable during the build
	///
	/// # Returns
	/// The path to the built artifact
	pub fn build(&self, features: &[String]) -> Result<PathBuf> {
		build_project(&self.path(), None, &self.profile(), features, None)?;
		// Check the artifact is found after being built
		self.artifact_path()
	}

	/// Gets the path associated with this chain specification builder.
	///
	/// # Returns
	/// The path to either the node or runtime directory.
	pub fn path(&self) -> PathBuf {
		match self {
			ChainSpecBuilder::Node { node_path, .. } => node_path,
			ChainSpecBuilder::Runtime { runtime_path, .. } => runtime_path,
		}
		.clone()
	}

	/// Gets the build profile associated with this chain specification builder.
	///
	/// # Returns
	/// The build profile (debug, release, production, etc.) to use when building the chain.
	pub fn profile(&self) -> Profile {
		*match self {
			ChainSpecBuilder::Node { profile, .. } => profile,
			ChainSpecBuilder::Runtime { profile, .. } => profile,
		}
	}

	/// Gets the path to the built artifact.
	///
	/// # Returns
	/// The path to the built artifact (node binary or runtime WASM).
	pub fn artifact_path(&self) -> Result<PathBuf> {
		let manifest = from_path(&self.path())?;
		let package = manifest.package().name();
		let root_folder = rustilities::manifest::find_workspace_manifest(self.path())
			.ok_or(anyhow::anyhow!("Not inside a workspace"))?
			.parent()
			.expect("Path to Cargo.toml workspace root folder must exist")
			.to_path_buf();
		let path = match self {
			ChainSpecBuilder::Node { profile, .. } =>
				profile.target_directory(&root_folder).join(package),
			ChainSpecBuilder::Runtime { profile, .. } => {
				let base = profile.target_directory(&root_folder).join("wbuild").join(package);
				let wasm_file = package.replace("-", "_");
				let compact_compressed = base.join(format!("{wasm_file}.compact.compressed.wasm"));
				let raw = base.join(format!("{wasm_file}.wasm"));
				if compact_compressed.is_file() {
					compact_compressed
				} else if raw.is_file() {
					raw
				} else {
					return Err(anyhow::anyhow!("No runtime found"));
				}
			},
		};
		Ok(path.canonicalize()?)
	}

	/// Generates a plain (human readable) chain specification file.
	///
	/// # Arguments
	/// * `chain_or_preset` - The chain (when using a node) or preset (when using a runtime) name.
	/// * `output_file` - The path where the chain spec should be written.
	/// * `name` - The name to be used on the chain spec if specified.
	/// * `id` - The ID to be used on the chain spec if specified.
	pub fn generate_plain_chain_spec(
		&self,
		chain_or_preset: &str,
		output_file: &Path,
		name: Option<&str>,
		id: Option<&str>,
	) -> Result<(), Error> {
		match self {
			ChainSpecBuilder::Node { default_bootnode, .. } => generate_plain_chain_spec_with_node(
				&self.artifact_path()?,
				output_file,
				*default_bootnode,
				chain_or_preset,
			),
			ChainSpecBuilder::Runtime { .. } => generate_plain_chain_spec_with_runtime(
				fs::read(self.artifact_path()?)?,
				output_file,
				chain_or_preset,
				name,
				id,
			),
		}
	}

	/// Generates a raw (encoded) chain specification file from a plain one.
	///
	/// # Arguments
	/// * `plain_chain_spec` - The path to the plain chain spec file.
	/// * `raw_chain_spec_name` - The name for the generated raw chain spec file.
	///
	/// # Returns
	/// The path to the generated raw chain spec file.
	pub fn generate_raw_chain_spec(
		&self,
		plain_chain_spec: &Path,
		raw_chain_spec_name: &str,
	) -> Result<PathBuf, Error> {
		match self {
			ChainSpecBuilder::Node { .. } => generate_raw_chain_spec_with_node(
				&self.artifact_path()?,
				plain_chain_spec,
				raw_chain_spec_name,
			),
			ChainSpecBuilder::Runtime { .. } =>
				generate_raw_chain_spec_with_runtime(plain_chain_spec, raw_chain_spec_name),
		}
	}

	/// Extracts and exports the WebAssembly runtime code from a raw chain specification.
	///
	/// # Arguments
	/// * `raw_chain_spec` - Path to the raw chain specification file to extract the runtime from.
	/// * `wasm_file_name` - Name for the file where the extracted runtime will be saved.
	///
	/// # Returns
	/// The path to the generated WASM runtime file.
	///
	/// # Errors
	/// Returns an error if:
	/// - The chain specification file cannot be read or parsed.
	/// - The runtime cannot be extracted from the chain spec.
	/// - The runtime cannot be written to the output file.
	pub fn export_wasm_file(
		&self,
		raw_chain_spec: &Path,
		wasm_file_name: &str,
	) -> Result<PathBuf, Error> {
		match self {
			ChainSpecBuilder::Node { .. } =>
				export_wasm_file_with_node(&self.artifact_path()?, raw_chain_spec, wasm_file_name),
			ChainSpecBuilder::Runtime { .. } =>
				export_wasm_file_with_runtime(raw_chain_spec, wasm_file_name),
		}
	}
}

/// Build the chain and returns the path to the binary.
///
/// # Arguments
/// * `path` - The path to the chain manifest.
/// * `package` - The optional package to be built.
/// * `profile` - Whether the chain should be built without any debugging functionality.
/// * `node_path` - An optional path to the node directory. Defaults to the `node` subdirectory of
///   the project path if not provided.
/// * `features` - A set of features the project is built with.
pub fn build_chain(
	path: &Path,
	package: Option<String>,
	profile: &Profile,
	node_path: Option<&Path>,
	features: &[String],
) -> Result<PathBuf, Error> {
	build_project(path, package, profile, features, None)?;
	binary_path(&profile.target_directory(path), node_path.unwrap_or(&path.join("node")))
}

/// Build the Rust project.
///
/// # Arguments
/// * `path` - The optional path to the project manifest, defaulting to the current directory if not
///   specified.
/// * `package` - The optional package to be built.
/// * `profile` - Whether the project should be built without any debugging functionality.
/// * `features` - A set of features the project is built with.
/// * `target` - The optional target to be specified.
pub fn build_project(
	path: &Path,
	package: Option<String>,
	profile: &Profile,
	features: &[String],
	target: Option<&str>,
) -> Result<(), Error> {
	let mut args = vec!["build"];
	if let Some(package) = package.as_deref() {
		args.push("--package");
		args.push(package)
	}
	if profile == &Profile::Release {
		args.push("--release");
	} else if profile == &Profile::Production {
		args.push("--profile=production");
	}

	let feature_args = features.join(",");
	if !features.is_empty() {
		args.push("--features");
		args.push(&feature_args);
	}

	if let Some(target) = target {
		args.push("--target");
		args.push(target);
	}

	cmd("cargo", args).dir(path).run()?;
	Ok(())
}

/// Determines whether the manifest at the supplied path is a supported chain project.
///
/// # Arguments
/// * `path` - The optional path to the manifest, defaulting to the current directory if not
///   specified.
pub fn is_supported(path: &Path) -> bool {
	let manifest = match from_path(path) {
		Ok(m) => m,
		Err(_) => return false,
	};
	// Simply check for a chain dependency
	const DEPENDENCIES: [&str; 4] =
		["cumulus-client-collator", "cumulus-primitives-core", "parachains-common", "polkadot-sdk"];
	DEPENDENCIES.into_iter().any(|d| {
		manifest.dependencies.contains_key(d) ||
			manifest.workspace.as_ref().is_some_and(|w| w.dependencies.contains_key(d))
	})
}

/// Constructs the node binary path based on the target path and the node directory path.
///
/// # Arguments
/// * `target_path` - The path where the binaries are expected to be found.
/// * `node_path` - The path to the node from which the node name will be parsed.
pub fn binary_path(target_path: &Path, node_path: &Path) -> Result<PathBuf, Error> {
	build_binary_path(node_path, |node_name| target_path.join(node_name))
}

/// Constructs the runtime binary path based on the target path and the directory path.
///
/// # Arguments
/// * `target_path` - The path where the binaries are expected to be found.
/// * `runtime_path` - The path to the runtime from which the runtime name will be parsed.
pub fn runtime_binary_path(target_path: &Path, runtime_path: &Path) -> Result<PathBuf, Error> {
	build_binary_path(runtime_path, |runtime_name| {
		target_path.join(format!("{runtime_name}/{}.wasm", runtime_name.replace("-", "_")))
	})
}

fn build_binary_path<F>(project_path: &Path, path_builder: F) -> Result<PathBuf, Error>
where
	F: Fn(&str) -> PathBuf,
{
	let manifest = from_path(project_path)?;
	let project_name = manifest.package().name();
	let release = path_builder(project_name);
	if !release.exists() {
		return Err(Error::MissingBinary(project_name.to_string()));
	}
	Ok(release)
}

/// Generates a raw chain specification file from a plain chain specification for a runtime.
///
/// # Arguments
/// * `plain_chain_spec` - Location of the plain chain specification file.
/// * `raw_chain_spec_name` - The name of the raw chain specification file to be generated.
///
/// # Returns
/// The path to the generated raw chain specification file.
pub fn generate_raw_chain_spec_with_runtime(
	plain_chain_spec: &Path,
	raw_chain_spec_name: &str,
) -> Result<PathBuf, Error> {
	let chain_spec = GenericChainSpec::<Option<()>>::from_json_file(plain_chain_spec.to_path_buf())
		.map_err(|e| anyhow::anyhow!(e))?;
	let raw_chain_spec = chain_spec.as_json(true).map_err(|e| anyhow::anyhow!(e))?;
	let raw_chain_spec_file = plain_chain_spec.with_file_name(raw_chain_spec_name);
	fs::write(&raw_chain_spec_file, raw_chain_spec)?;
	Ok(raw_chain_spec_file)
}

/// Generates a plain chain specification file for a runtime.
///
/// # Arguments
/// * `wasm` - The WebAssembly runtime bytes.
/// * `plain_chain_spec` - The path where the plain chain specification should be written.
/// * `preset` - Preset name for genesis configuration.
/// * `name` - The name to be used on the chain spec if specified.
/// * `id` - The ID to be used on the chain spec if specified.
pub fn generate_plain_chain_spec_with_runtime(
	wasm: Vec<u8>,
	plain_chain_spec: &Path,
	preset: &str,
	name: Option<&str>,
	id: Option<&str>,
) -> Result<(), Error> {
	let mut chain_spec = GenericChainSpec::<NoExtension>::builder(&wasm[..], None)
		.with_genesis_config_preset_name(preset.trim());

	if let Some(name) = name {
		chain_spec = chain_spec.with_name(name);
	}

	if let Some(id) = id {
		chain_spec = chain_spec.with_id(id);
	}

	let chain_spec = chain_spec.build().as_json(false).map_err(|e| anyhow::anyhow!(e))?;
	fs::write(plain_chain_spec, chain_spec)?;

	Ok(())
}

/// Extracts and exports the WebAssembly runtime from a raw chain specification.
///
/// # Arguments
/// * `raw_chain_spec` - The path to the raw chain specification file to extract the runtime from.
/// * `wasm_file_name` - The name of the file where the extracted runtime will be saved.
///
/// # Returns
/// The path to the generated WASM runtime file wrapped in a Result.
///
/// # Errors
/// Returns an error if:
/// - The chain specification file cannot be read or parsed.
/// - The runtime cannot be extracted from the chain spec.
/// - The runtime cannot be written to the output file.
pub fn export_wasm_file_with_runtime(
	raw_chain_spec: &Path,
	wasm_file_name: &str,
) -> Result<PathBuf, Error> {
	let chain_spec = GenericChainSpec::<Option<()>>::from_json_file(raw_chain_spec.to_path_buf())
		.map_err(|e| anyhow::anyhow!(e))?;
	let raw_wasm_blob =
		cumulus_client_cli::extract_genesis_wasm(&chain_spec).map_err(|e| anyhow::anyhow!(e))?;
	let wasm_file = raw_chain_spec.parent().unwrap_or(Path::new("./")).join(wasm_file_name);
	fs::write(&wasm_file, raw_wasm_blob)?;
	Ok(wasm_file)
}

/// Generates the plain text chain specification for a chain with its own node.
///
/// # Arguments
/// * `binary_path` - The path to the node binary executable that contains the `build-spec` command.
/// * `plain_chain_spec` - Location of the plain_chain_spec file to be generated.
/// * `default_bootnode` - Whether to include localhost as a bootnode.
/// * `chain` - The chain specification. It can be one of the predefined ones (e.g. dev, local or a
///   custom one) or the path to an existing chain spec.
pub fn generate_plain_chain_spec_with_node(
	binary_path: &Path,
	plain_chain_spec: &Path,
	default_bootnode: bool,
	chain: &str,
) -> Result<(), Error> {
	check_command_exists(binary_path, "build-spec")?;
	let mut args = vec!["build-spec", "--chain", chain];
	if !default_bootnode {
		args.push("--disable-default-bootnode");
	}
	// Create a temporary file.
	let temp_file = tempfile::NamedTempFile::new_in(std::env::temp_dir())?;
	// Run the command and redirect output to the temporary file.
	let output = cmd(binary_path, args)
		.stdout_path(temp_file.path())
		.stderr_capture()
		.unchecked()
		.run()?;
	// Check if the command failed.
	handle_command_error(&output, Error::BuildSpecError)?;
	// Atomically replace the chain spec file with the temporary file.
	temp_file.persist(plain_chain_spec).map_err(|e| {
		Error::AnyhowError(anyhow!(
			"Failed to replace the chain spec file with the temporary file: {e}"
		))
	})?;
	Ok(())
}

/// Generates a raw chain specification file for a chain.
///
/// # Arguments
/// * `binary_path` - The path to the node binary executable that contains the `build-spec` command.
/// * `plain_chain_spec` - Location of the plain chain specification file.
/// * `chain_spec_file_name` - The name of the chain specification file to be generated.
pub fn generate_raw_chain_spec_with_node(
	binary_path: &Path,
	plain_chain_spec: &Path,
	chain_spec_file_name: &str,
) -> Result<PathBuf, Error> {
	if !plain_chain_spec.exists() {
		return Err(Error::MissingChainSpec(plain_chain_spec.display().to_string()));
	}
	check_command_exists(binary_path, "build-spec")?;
	let raw_chain_spec = plain_chain_spec.with_file_name(chain_spec_file_name);
	let output = cmd(
		binary_path,
		vec![
			"build-spec",
			"--chain",
			&plain_chain_spec.display().to_string(),
			"--disable-default-bootnode",
			"--raw",
		],
	)
	.stdout_path(&raw_chain_spec)
	.stderr_capture()
	.unchecked()
	.run()?;
	handle_command_error(&output, Error::BuildSpecError)?;
	Ok(raw_chain_spec)
}

/// Export the WebAssembly runtime for the chain.
///
/// # Arguments
/// * `binary_path` - The path to the node binary executable that contains the `export-genesis-wasm`
///   command.
/// * `raw_chain_spec` - Location of the raw chain specification file.
/// * `wasm_file_name` - The name of the wasm runtime file to be generated.
pub fn export_wasm_file_with_node(
	binary_path: &Path,
	raw_chain_spec: &Path,
	wasm_file_name: &str,
) -> Result<PathBuf, Error> {
	if !raw_chain_spec.exists() {
		return Err(Error::MissingChainSpec(raw_chain_spec.display().to_string()));
	}
	check_command_exists(binary_path, "export-genesis-wasm")?;
	let wasm_file = raw_chain_spec.parent().unwrap_or(Path::new("./")).join(wasm_file_name);
	let output = cmd(
		binary_path,
		vec![
			"export-genesis-wasm",
			"--chain",
			&raw_chain_spec.display().to_string(),
			&wasm_file.display().to_string(),
		],
	)
	.stdout_null()
	.stderr_capture()
	.unchecked()
	.run()?;
	handle_command_error(&output, Error::BuildSpecError)?;
	Ok(wasm_file)
}

/// Generate the chain genesis state.
///
/// # Arguments
/// * `binary_path` - The path to the node binary executable that contains the
///   `export-genesis-state` command.
/// * `raw_chain_spec` - Location of the raw chain specification file.
/// * `genesis_file_name` - The name of the genesis state file to be generated.
pub fn generate_genesis_state_file_with_node(
	binary_path: &Path,
	raw_chain_spec: &Path,
	genesis_file_name: &str,
) -> Result<PathBuf, Error> {
	if !raw_chain_spec.exists() {
		return Err(Error::MissingChainSpec(raw_chain_spec.display().to_string()));
	}
	check_command_exists(binary_path, "export-genesis-state")?;
	let genesis_file = raw_chain_spec.parent().unwrap_or(Path::new("./")).join(genesis_file_name);
	let output = cmd(
		binary_path,
		vec![
			"export-genesis-state",
			"--chain",
			&raw_chain_spec.display().to_string(),
			&genesis_file.display().to_string(),
		],
	)
	.stdout_null()
	.stderr_capture()
	.unchecked()
	.run()?;
	handle_command_error(&output, Error::BuildSpecError)?;
	Ok(genesis_file)
}

/// Checks if a given command exists and can be executed by running it with the "--help" argument.
fn check_command_exists(binary_path: &Path, command: &str) -> Result<(), Error> {
	cmd(binary_path, vec![command, "--help"]).stdout_null().run().map_err(|_err| {
		Error::MissingCommand {
			command: command.to_string(),
			binary: binary_path.display().to_string(),
		}
	})?;
	Ok(())
}

/// A chain specification.
pub struct ChainSpec(Value);
impl ChainSpec {
	/// Parses a chain specification from a path.
	///
	/// # Arguments
	/// * `path` - The path to a chain specification file.
	pub fn from(path: &Path) -> Result<ChainSpec> {
		Ok(ChainSpec(Value::from_str(&fs::read_to_string(path)?)?))
	}

	/// Get the chain type from the chain specification.
	pub fn get_chain_type(&self) -> Option<&str> {
		self.0.get("chainType").and_then(|v| v.as_str())
	}

	/// Get the name from the chain specification.
	pub fn get_name(&self) -> Option<&str> {
		self.0.get("name").and_then(|v| v.as_str())
	}

	/// Get the chain ID from the chain specification.
	pub fn get_chain_id(&self) -> Option<u64> {
		self.0.get("para_id").and_then(|v| v.as_u64())
	}

	/// Get the property `basedOn` from the chain specification.
	pub fn get_property_based_on(&self) -> Option<&str> {
		self.0.get("properties").and_then(|v| v.get("basedOn")).and_then(|v| v.as_str())
	}

	/// Get the protocol ID from the chain specification.
	pub fn get_protocol_id(&self) -> Option<&str> {
		self.0.get("protocolId").and_then(|v| v.as_str())
	}

	/// Get the relay chain from the chain specification.
	pub fn get_relay_chain(&self) -> Option<&str> {
		self.0.get("relay_chain").and_then(|v| v.as_str())
	}

	/// Get the sudo key from the chain specification.
	pub fn get_sudo_key(&self) -> Option<&str> {
		self.0
			.get("genesis")
			.and_then(|genesis| genesis.get("runtimeGenesis"))
			.and_then(|runtime_genesis| runtime_genesis.get("patch"))
			.and_then(|patch| patch.get("sudo"))
			.and_then(|sudo| sudo.get("key"))
			.and_then(|key| key.as_str())
	}

	/// Replaces the chain id with the provided `para_id`.
	///
	/// # Arguments
	/// * `para_id` - The new value for the para_id.
	pub fn replace_para_id(&mut self, para_id: u32) -> Result<(), Error> {
		// Replace para_id
		let root = self
			.0
			.as_object_mut()
			.ok_or_else(|| Error::Config("expected root object".into()))?;
		root.insert("para_id".to_string(), json!(para_id));

		// Replace genesis.runtimeGenesis.patch.parachainInfo.parachainId
		let replace = self.0.pointer_mut("/genesis/runtimeGenesis/patch/parachainInfo/parachainId");
		// If this fails, it means it is a raw chainspec
		if let Some(replace) = replace {
			*replace = json!(para_id);
		}
		Ok(())
	}

	/// Replaces the relay chain name with the given one.
	///
	/// # Arguments
	/// * `relay_name` - The new value for the relay chain field in the specification.
	pub fn replace_relay_chain(&mut self, relay_name: &str) -> Result<(), Error> {
		// Replace relay_chain
		let root = self
			.0
			.as_object_mut()
			.ok_or_else(|| Error::Config("expected root object".into()))?;
		root.insert("relay_chain".to_string(), json!(relay_name));
		Ok(())
	}

	/// Replaces the chain type with the given one.
	///
	/// # Arguments
	/// * `chain_type` - The new value for the chain type.
	pub fn replace_chain_type(&mut self, chain_type: &str) -> Result<(), Error> {
		// Replace chainType
		let replace = self
			.0
			.get_mut("chainType")
			.ok_or_else(|| Error::Config("expected `chainType`".into()))?;
		*replace = json!(chain_type);
		Ok(())
	}

	/// Replaces the protocol ID with the given one.
	///
	/// # Arguments
	/// * `protocol_id` - The new value for the protocolId of the given specification.
	pub fn replace_protocol_id(&mut self, protocol_id: &str) -> Result<(), Error> {
		// Replace protocolId
		let replace = self
			.0
			.get_mut("protocolId")
			.ok_or_else(|| Error::Config("expected `protocolId`".into()))?;
		*replace = json!(protocol_id);
		Ok(())
	}

	/// Replaces the properties with the given ones.
	///
	/// # Arguments
	/// * `raw_properties` - Comma-separated, key-value pairs. Example: "KEY1=VALUE1,KEY2=VALUE2".
	pub fn replace_properties(&mut self, raw_properties: &str) -> Result<(), Error> {
		// Replace properties
		let replace = self
			.0
			.get_mut("properties")
			.ok_or_else(|| Error::Config("expected `properties`".into()))?;
		let mut properties = serde_json::Map::new();
		let mut iter = raw_properties
			.split(',')
			.flat_map(|s| s.split('=').map(|p| p.trim()).collect::<Vec<_>>())
			.collect::<Vec<_>>()
			.into_iter();
		while let Some(key) = iter.next() {
			let value = iter.next().expect("Property value expected but not found");
			properties.insert(key.to_string(), Value::String(value.to_string()));
		}
		*replace = Value::Object(properties);
		Ok(())
	}

	/// Replaces the invulnerables session keys in the chain specification with the provided
	/// `collator_keys`.
	///
	/// # Arguments
	/// * `collator_keys` - A list of new collator keys.
	pub fn replace_collator_keys(&mut self, collator_keys: Vec<String>) -> Result<(), Error> {
		let uses_evm_keys = self
			.0
			.get("properties")
			.and_then(|p| p.get("isEthereum"))
			.and_then(|v| v.as_bool())
			.unwrap_or(false);

		let keys = if uses_evm_keys {
			convert_to_evm_accounts(collator_keys.clone())?
		} else {
			collator_keys.clone()
		};

		let invulnerables = self
			.0
			.get_mut("genesis")
			.ok_or_else(|| Error::Config("expected `genesis`".into()))?
			.get_mut("runtimeGenesis")
			.ok_or_else(|| Error::Config("expected `runtimeGenesis`".into()))?
			.get_mut("patch")
			.ok_or_else(|| Error::Config("expected `patch`".into()))?
			.get_mut("collatorSelection")
			.ok_or_else(|| Error::Config("expected `collatorSelection`".into()))?
			.get_mut("invulnerables")
			.ok_or_else(|| Error::Config("expected `invulnerables`".into()))?;

		*invulnerables = json!(keys);

		let session_keys = keys
			.iter()
			.zip(collator_keys.iter())
			.map(|(address, original_address)| {
				json!([
					address,
					address,
					{ "aura": original_address } // Always the original address
				])
			})
			.collect::<Vec<_>>();

		let session_keys_field = self
			.0
			.get_mut("genesis")
			.ok_or_else(|| Error::Config("expected `genesis`".into()))?
			.get_mut("runtimeGenesis")
			.ok_or_else(|| Error::Config("expected `runtimeGenesis`".into()))?
			.get_mut("patch")
			.ok_or_else(|| Error::Config("expected `patch`".into()))?
			.get_mut("session")
			.ok_or_else(|| Error::Config("expected `session`".into()))?
			.get_mut("keys")
			.ok_or_else(|| Error::Config("expected `session.keys`".into()))?;

		*session_keys_field = json!(session_keys);

		Ok(())
	}

	/// Converts the chain specification to a string.
	pub fn to_string(&self) -> Result<String> {
		Ok(serde_json::to_string_pretty(&self.0)?)
	}

	/// Writes the chain specification to a file.
	///
	/// # Arguments
	/// * `path` - The path to the chain specification file.
	pub fn to_file(&self, path: &Path) -> Result<()> {
		fs::write(path, self.to_string()?)?;
		Ok(())
	}

	/// Updates the runtime code in the chain specification.
	///
	/// # Arguments
	/// * `bytes` - The new runtime code.
	pub fn update_runtime_code(&mut self, bytes: &[u8]) -> Result<(), Error> {
		// Replace `genesis.runtimeGenesis.code`
		let code = self
			.0
			.get_mut("genesis")
			.ok_or_else(|| Error::Config("expected `genesis`".into()))?
			.get_mut("runtimeGenesis")
			.ok_or_else(|| Error::Config("expected `runtimeGenesis`".into()))?
			.get_mut("code")
			.ok_or_else(|| Error::Config("expected `runtimeGenesis.code`".into()))?;
		let hex = to_hex(bytes, true);
		*code = json!(hex);
		Ok(())
	}
}

#[cfg(test)]
mod tests {
	use super::*;
	use crate::{
		Config, Error, new_chain::instantiate_standard_template, templates::ChainTemplate,
		up::Zombienet,
	};
	use anyhow::Result;
	use pop_common::{
		manifest::{Dependency, add_feature},
		set_executable_permission,
	};
	use sp_core::bytes::from_hex;
	use std::{
		fs::{self, write},
		io::Write,
		path::Path,
	};
	use strum::VariantArray;
	use tempfile::{Builder, TempDir, tempdir};

	static MOCK_WASM: &[u8] = include_bytes!("../../../../tests/runtimes/base_parachain.wasm");

	fn setup_template_and_instantiate() -> Result<TempDir> {
		let temp_dir = tempdir().expect("Failed to create temp dir");
		let config = Config {
			symbol: "DOT".to_string(),
			decimals: 18,
			initial_endowment: "1000000".to_string(),
		};
		instantiate_standard_template(&ChainTemplate::Standard, temp_dir.path(), config, None)?;
		Ok(temp_dir)
	}

	// Function that mocks the build process generating the target dir and release.
	fn mock_build_process(temp_dir: &Path) -> Result<(), Error> {
		// Create a target directory
		let target_dir = temp_dir.join("target");
		fs::create_dir(&target_dir)?;
		fs::create_dir(target_dir.join("release"))?;
		// Create a release file
		fs::File::create(target_dir.join("release/parachain-template-node"))?;
		Ok(())
	}

	// Function create a mocked node directory with Cargo.toml
	fn mock_node(temp_dir: &Path) -> Result<(), Error> {
		let node_dir = temp_dir.join("node");
		fs::create_dir(&node_dir)?;
		fs::write(
			node_dir.join("Cargo.toml"),
			r#"[package]
name = "parachain-template-node"
version = "0.1.0"
edition = "2021"
"#,
		)?;
		Ok(())
	}

	// Function that mocks the build process of WASM runtime generating the target dir and release.
	fn mock_build_runtime_process(temp_dir: &Path) -> Result<(), Error> {
		let runtime = "parachain-template-runtime";
		// Create a target directory
		let target_dir = temp_dir.join("target");
		fs::create_dir(&target_dir)?;
		fs::create_dir(target_dir.join("release"))?;
		fs::create_dir(target_dir.join("release/wbuild"))?;
		fs::create_dir(target_dir.join(format!("release/wbuild/{runtime}")))?;
		// Create a WASM binary file
		fs::File::create(
			target_dir.join(format!("release/wbuild/{runtime}/{}.wasm", runtime.replace("-", "_"))),
		)?;
		Ok(())
	}

	// Function that generates a Cargo.toml inside node directory for testing.
	fn generate_mock_node(temp_dir: &Path, name: Option<&str>) -> Result<PathBuf, Error> {
		// Create a node directory
		let target_dir = temp_dir.join(name.unwrap_or("node"));
		fs::create_dir(&target_dir)?;
		// Create a Cargo.toml file
		let mut toml_file = fs::File::create(target_dir.join("Cargo.toml"))?;
		writeln!(
			toml_file,
			r#"
			[package]
			name = "parachain_template_node"
			version = "0.1.0"

			[dependencies]

			"#
		)?;
		Ok(target_dir)
	}

	// Function that fetch a binary from pop network
	async fn fetch_binary(cache: &Path) -> Result<String, Error> {
		let config = Builder::new().suffix(".toml").tempfile()?;
		writeln!(
			config.as_file(),
			r#"
            [relaychain]
            chain = "paseo-local"

			[[parachains]]
			id = 4385
			default_command = "pop-node"
			"#
		)?;
		let mut zombienet = Zombienet::new(
			cache,
			config.path().try_into()?,
			None,
			None,
			None,
			None,
			Some(&vec!["https://github.com/r0gue-io/pop-node#node-v0.3.0".to_string()]),
		)
		.await?;
		let mut binary_name: String = "".to_string();
		for binary in zombienet.binaries().filter(|b| !b.exists() && b.name() == "pop-node") {
			binary_name = format!("{}-{}", binary.name(), binary.version().unwrap());
			binary.source(true, &(), true).await?;
		}
		Ok(binary_name)
	}

	// Replace the binary fetched with the mocked binary
	fn replace_mock_with_binary(temp_dir: &Path, binary_name: String) -> Result<PathBuf, Error> {
		let binary_path = temp_dir.join(binary_name);
		let content = fs::read(&binary_path)?;
		write(temp_dir.join("target/release/parachain-template-node"), content)?;
		// Make executable
		set_executable_permission(temp_dir.join("target/release/parachain-template-node"))?;
		Ok(binary_path)
	}

	fn add_production_profile(project: &Path) -> Result<()> {
		let root_toml_path = project.join("Cargo.toml");
		let mut root_toml_content = fs::read_to_string(&root_toml_path)?;
		root_toml_content.push_str(
			r#"
			[profile.production]
			codegen-units = 1
			inherits = "release"
			lto = true
			"#,
		);
		// Write the updated content back to the file
		write(&root_toml_path, root_toml_content)?;
		Ok(())
	}

	#[test]
	fn build_chain_works() -> Result<()> {
		let name = "parachain_template_node";
		let temp_dir = tempdir()?;
		cmd("cargo", ["new", name, "--bin"]).dir(temp_dir.path()).run()?;
		let project = temp_dir.path().join(name);
		add_production_profile(&project)?;
		add_feature(&project, ("dummy-feature".to_string(), vec![]))?;
		for node in [None, Some("custom_node")] {
			let node_path = generate_mock_node(&project, node)?;
			for package in [None, Some(String::from("parachain_template_node"))] {
				for profile in Profile::VARIANTS {
					let node_path = node.map(|_| node_path.as_path());
					let binary = build_chain(
						&project,
						package.clone(),
						profile,
						node_path,
						&["dummy-feature".to_string()],
					)?;
					let target_directory = profile.target_directory(&project);
					assert!(target_directory.exists());
					assert!(target_directory.join("parachain_template_node").exists());
					assert_eq!(
						binary.display().to_string(),
						target_directory.join("parachain_template_node").display().to_string()
					);
				}
			}
		}
		Ok(())
	}

	#[test]
	fn build_project_works() -> Result<()> {
		let name = "example_project";
		let temp_dir = tempdir()?;
		cmd("cargo", ["new", name, "--bin"]).dir(temp_dir.path()).run()?;
		let project = temp_dir.path().join(name);
		add_production_profile(&project)?;
		add_feature(&project, ("dummy-feature".to_string(), vec![]))?;
		for package in [None, Some(String::from(name))] {
			for profile in Profile::VARIANTS {
				build_project(
					&project,
					package.clone(),
					profile,
					&["dummy-feature".to_string()],
					None,
				)?;
				let target_directory = profile.target_directory(&project);
				let binary = build_binary_path(&project, |runtime_name| {
					target_directory.join(runtime_name)
				})?;
				assert!(target_directory.exists());
				assert!(target_directory.join(name).exists());
				assert_eq!(
					binary.display().to_string(),
					target_directory.join(name).display().to_string()
				);
			}
		}
		Ok(())
	}

	#[test]
	fn binary_path_of_node_works() -> Result<()> {
		let temp_dir =
			setup_template_and_instantiate().expect("Failed to setup template and instantiate");
		mock_build_process(temp_dir.path())?;
		mock_node(temp_dir.path())?;
		let release_path =
			binary_path(&temp_dir.path().join("target/release"), &temp_dir.path().join("node"))?;
		assert_eq!(
			release_path.display().to_string(),
			format!("{}/target/release/parachain-template-node", temp_dir.path().display())
		);
		Ok(())
	}

	#[test]
	fn binary_path_of_runtime_works() -> Result<()> {
		let temp_dir =
			setup_template_and_instantiate().expect("Failed to setup template and instantiate");
		// Ensure binary path works for the runtime.
		let runtime = "parachain-template-runtime";
		mock_build_runtime_process(temp_dir.path())?;
		let release_path = runtime_binary_path(
			&temp_dir.path().join("target/release/wbuild"),
			&temp_dir.path().join("runtime"),
		)?;
		assert_eq!(
			release_path.display().to_string(),
			format!(
				"{}/target/release/wbuild/{runtime}/{}.wasm",
				temp_dir.path().display(),
				runtime.replace("-", "_")
			)
		);

		Ok(())
	}

	#[test]
	fn binary_path_fails_missing_binary() -> Result<()> {
		let temp_dir =
			setup_template_and_instantiate().expect("Failed to setup template and instantiate");
		mock_node(temp_dir.path())?;
		assert!(matches!(
			binary_path(&temp_dir.path().join("target/release"), &temp_dir.path().join("node")),
			Err(Error::MissingBinary(error)) if error == "parachain-template-node"
		));
		Ok(())
	}

	#[tokio::test]
	async fn generate_files_works() -> Result<()> {
		let temp_dir =
			setup_template_and_instantiate().expect("Failed to setup template and instantiate");
		mock_build_process(temp_dir.path())?;
		let binary_name = fetch_binary(temp_dir.path()).await?;
		let binary_path = replace_mock_with_binary(temp_dir.path(), binary_name)?;
		// Test generate chain spec
		let plain_chain_spec = &temp_dir.path().join("plain-parachain-chainspec.json");
		generate_plain_chain_spec_with_node(
			&binary_path,
			&temp_dir.path().join("plain-parachain-chainspec.json"),
			false,
			"local",
		)?;
		assert!(plain_chain_spec.exists());
		{
			let mut chain_spec = ChainSpec::from(plain_chain_spec)?;
			chain_spec.replace_para_id(2001)?;
			chain_spec.to_file(plain_chain_spec)?;
		}
		let raw_chain_spec = generate_raw_chain_spec_with_node(
			&binary_path,
			plain_chain_spec,
			"raw-parachain-chainspec.json",
		)?;
		assert!(raw_chain_spec.exists());
		let content = fs::read_to_string(raw_chain_spec.clone()).expect("Could not read file");
		assert!(content.contains("\"para_id\": 2001"));
		assert!(content.contains("\"bootNodes\": []"));
		// Test export wasm file
		let wasm_file =
			export_wasm_file_with_node(&binary_path, &raw_chain_spec, "para-2001-wasm")?;
		assert!(wasm_file.exists());
		// Test generate chain state file
		let genesis_file = generate_genesis_state_file_with_node(
			&binary_path,
			&raw_chain_spec,
			"para-2001-genesis-state",
		)?;
		assert!(genesis_file.exists());
		Ok(())
	}

	#[tokio::test]
	async fn generate_plain_chain_spec_with_runtime_works_with_name_and_id_override() -> Result<()>
	{
		let temp_dir =
			setup_template_and_instantiate().expect("Failed to setup template and instantiate");
		// Test generate chain spec
		let plain_chain_spec = &temp_dir.path().join("plain-parachain-chainspec.json");
		generate_plain_chain_spec_with_runtime(
			Vec::from(MOCK_WASM),
			plain_chain_spec,
			"local_testnet",
			Some("POP Chain Spec"),
			Some("pop-chain-spec"),
		)?;
		assert!(plain_chain_spec.exists());
		let raw_chain_spec =
			generate_raw_chain_spec_with_runtime(plain_chain_spec, "raw-parachain-chainspec.json")?;
		assert!(raw_chain_spec.exists());
		let content = fs::read_to_string(raw_chain_spec.clone()).expect("Could not read file");
		assert!(content.contains("\"name\": \"POP Chain Spec\""));
		assert!(content.contains("\"id\": \"pop-chain-spec\""));
		assert!(content.contains("\"bootNodes\": []"));
		Ok(())
	}

	#[tokio::test]
	async fn generate_plain_chain_spec_with_runtime_works_with_name_override() -> Result<()> {
		let temp_dir =
			setup_template_and_instantiate().expect("Failed to setup template and instantiate");
		// Test generate chain spec
		let plain_chain_spec = &temp_dir.path().join("plain-parachain-chainspec.json");
		generate_plain_chain_spec_with_runtime(
			Vec::from(MOCK_WASM),
			plain_chain_spec,
			"local_testnet",
			Some("POP Chain Spec"),
			None,
		)?;
		assert!(plain_chain_spec.exists());
		let raw_chain_spec =
			generate_raw_chain_spec_with_runtime(plain_chain_spec, "raw-parachain-chainspec.json")?;
		assert!(raw_chain_spec.exists());
		let content = fs::read_to_string(raw_chain_spec.clone()).expect("Could not read file");
		assert!(content.contains("\"name\": \"POP Chain Spec\""));
		assert!(content.contains("\"id\": \"dev\""));
		assert!(content.contains("\"bootNodes\": []"));
		Ok(())
	}

	#[tokio::test]
	async fn generate_plain_chain_spec_with_runtime_works_with_id_override() -> Result<()> {
		let temp_dir =
			setup_template_and_instantiate().expect("Failed to setup template and instantiate");
		// Test generate chain spec
		let plain_chain_spec = &temp_dir.path().join("plain-parachain-chainspec.json");
		generate_plain_chain_spec_with_runtime(
			Vec::from(MOCK_WASM),
			plain_chain_spec,
			"local_testnet",
			None,
			Some("pop-chain-spec"),
		)?;
		assert!(plain_chain_spec.exists());
		let raw_chain_spec =
			generate_raw_chain_spec_with_runtime(plain_chain_spec, "raw-parachain-chainspec.json")?;
		assert!(raw_chain_spec.exists());
		let content = fs::read_to_string(raw_chain_spec.clone()).expect("Could not read file");
		assert!(content.contains("\"name\": \"Development\""));
		assert!(content.contains("\"id\": \"pop-chain-spec\""));
		assert!(content.contains("\"bootNodes\": []"));
		Ok(())
	}

	#[tokio::test]
	async fn generate_plain_chain_spec_with_runtime_works_without_name_and_id_override()
	-> Result<()> {
		let temp_dir =
			setup_template_and_instantiate().expect("Failed to setup template and instantiate");
		// Test generate chain spec
		let plain_chain_spec = &temp_dir.path().join("plain-parachain-chainspec.json");
		generate_plain_chain_spec_with_runtime(
			Vec::from(MOCK_WASM),
			plain_chain_spec,
			"local_testnet",
			None,
			None,
		)?;
		assert!(plain_chain_spec.exists());
		let raw_chain_spec =
			generate_raw_chain_spec_with_runtime(plain_chain_spec, "raw-parachain-chainspec.json")?;
		assert!(raw_chain_spec.exists());
		let content = fs::read_to_string(raw_chain_spec.clone()).expect("Could not read file");
		assert!(content.contains("\"name\": \"Development\""));
		assert!(content.contains("\"id\": \"dev\""));
		assert!(content.contains("\"bootNodes\": []"));
		Ok(())
	}

	#[tokio::test]
	async fn fails_to_generate_plain_chain_spec_when_file_missing() -> Result<()> {
		let temp_dir =
			setup_template_and_instantiate().expect("Failed to setup template and instantiate");
		mock_build_process(temp_dir.path())?;
		let binary_name = fetch_binary(temp_dir.path()).await?;
		let binary_path = replace_mock_with_binary(temp_dir.path(), binary_name)?;
		assert!(matches!(
			generate_plain_chain_spec_with_node(
				&binary_path,
				&temp_dir.path().join("plain-parachain-chainspec.json"),
				false,
				&temp_dir.path().join("plain-parachain-chainspec.json").display().to_string(),
			),
			Err(Error::BuildSpecError(message)) if message.contains("No such file or directory")
		));
		assert!(!temp_dir.path().join("plain-parachain-chainspec.json").exists());
		Ok(())
	}

	#[test]
	fn raw_chain_spec_fails_wrong_chain_spec() -> Result<()> {
		assert!(matches!(
			generate_raw_chain_spec_with_node(
				Path::new("./binary"),
				Path::new("./plain-parachain-chainspec.json"),
				"plain-parachain-chainspec.json"
			),
			Err(Error::MissingChainSpec(..))
		));
		Ok(())
	}

	#[test]
	fn export_wasm_file_fails_wrong_chain_spec() -> Result<()> {
		assert!(matches!(
			export_wasm_file_with_node(
				Path::new("./binary"),
				Path::new("./raw-parachain-chainspec"),
				"para-2001-wasm"
			),
			Err(Error::MissingChainSpec(..))
		));
		Ok(())
	}

	#[test]
	fn generate_genesis_state_file_wrong_chain_spec() -> Result<()> {
		assert!(matches!(
			generate_genesis_state_file_with_node(
				Path::new("./binary"),
				Path::new("./raw-parachain-chainspec"),
				"para-2001-genesis-state",
			),
			Err(Error::MissingChainSpec(..))
		));
		Ok(())
	}

	#[test]
	fn get_chain_type_works() -> Result<()> {
		let chain_spec = ChainSpec(json!({
			"chainType": "test",
		}));
		assert_eq!(chain_spec.get_chain_type(), Some("test"));
		Ok(())
	}

	#[test]
	fn get_chain_name_works() -> Result<()> {
		assert_eq!(ChainSpec(json!({})).get_name(), None);
		let chain_spec = ChainSpec(json!({
			"name": "test",
		}));
		assert_eq!(chain_spec.get_name(), Some("test"));
		Ok(())
	}

	#[test]
	fn get_chain_id_works() -> Result<()> {
		let chain_spec = ChainSpec(json!({
			"para_id": 2002,
		}));
		assert_eq!(chain_spec.get_chain_id(), Some(2002));
		Ok(())
	}

	#[test]
	fn get_property_based_on_works() -> Result<()> {
		assert_eq!(ChainSpec(json!({})).get_property_based_on(), None);
		let chain_spec = ChainSpec(json!({
			"properties": {
				"basedOn": "test",
			}
		}));
		assert_eq!(chain_spec.get_property_based_on(), Some("test"));
		Ok(())
	}

	#[test]
	fn get_protocol_id_works() -> Result<()> {
		let chain_spec = ChainSpec(json!({
			"protocolId": "test",
		}));
		assert_eq!(chain_spec.get_protocol_id(), Some("test"));
		Ok(())
	}

	#[test]
	fn get_relay_chain_works() -> Result<()> {
		let chain_spec = ChainSpec(json!({
			"relay_chain": "test",
		}));
		assert_eq!(chain_spec.get_relay_chain(), Some("test"));
		Ok(())
	}

	#[test]
	fn get_sudo_key_works() -> Result<()> {
		assert_eq!(ChainSpec(json!({})).get_sudo_key(), None);
		let chain_spec = ChainSpec(json!({
			"para_id": 1000,
			"genesis": {
				"runtimeGenesis": {
					"patch": {
						"sudo": {
							"key": "sudo-key"
						}
					}
				}
			},
		}));
		assert_eq!(chain_spec.get_sudo_key(), Some("sudo-key"));
		Ok(())
	}

	#[test]
	fn replace_para_id_works() -> Result<()> {
		let mut chain_spec = ChainSpec(json!({
			"para_id": 1000,
			"genesis": {
				"runtimeGenesis": {
					"patch": {
						"parachainInfo": {
							"parachainId": 1000
						}
					}
				}
			},
		}));
		chain_spec.replace_para_id(2001)?;
		assert_eq!(
			chain_spec.0,
			json!({
				"para_id": 2001,
				"genesis": {
					"runtimeGenesis": {
						"patch": {
							"parachainInfo": {
								"parachainId": 2001
							}
						}
					}
				},
			})
		);
		Ok(())
	}

	#[test]
	fn replace_para_id_fails() -> Result<()> {
		let mut chain_spec = ChainSpec(json!({
			"para_id": 2001,
			"": {
				"runtimeGenesis": {
					"patch": {
						"parachainInfo": {
							"parachainId": 1000
						}
					}
				}
			},
		}));
		assert!(chain_spec.replace_para_id(2001).is_ok());
		chain_spec = ChainSpec(json!({
			"para_id": 2001,
			"genesis": {
				"": {
					"patch": {
						"parachainInfo": {
							"parachainId": 1000
						}
					}
				}
			},
		}));
		assert!(chain_spec.replace_para_id(2001).is_ok());
		chain_spec = ChainSpec(json!({
			"para_id": 2001,
			"genesis": {
				"runtimeGenesis": {
					"": {
						"parachainInfo": {
							"parachainId": 1000
						}
					}
				}
			},
		}));
		assert!(chain_spec.replace_para_id(2001).is_ok());
		chain_spec = ChainSpec(json!({
			"para_id": 2001,
			"genesis": {
				"runtimeGenesis": {
					"patch": {
						"": {
							"parachainId": 1000
						}
					}
				}
			},
		}));
		assert!(chain_spec.replace_para_id(2001).is_ok());
		chain_spec = ChainSpec(json!({
			"para_id": 2001,
			"genesis": {
				"runtimeGenesis": {
					"patch": {
						"parachainInfo": {
						}
					}
				}
			},
		}));
		assert!(chain_spec.replace_para_id(2001).is_ok());
		Ok(())
	}

	#[test]
	fn replace_relay_chain_works() -> Result<()> {
		let mut chain_spec = ChainSpec(json!({"relay_chain": "old-relay"}));
		chain_spec.replace_relay_chain("new-relay")?;
		assert_eq!(chain_spec.0, json!({"relay_chain": "new-relay"}));
		Ok(())
	}

	#[test]
	fn replace_chain_type_works() -> Result<()> {
		let mut chain_spec = ChainSpec(json!({"chainType": "old-chainType"}));
		chain_spec.replace_chain_type("new-chainType")?;
		assert_eq!(chain_spec.0, json!({"chainType": "new-chainType"}));
		Ok(())
	}

	#[test]
	fn replace_chain_type_fails() -> Result<()> {
		let mut chain_spec = ChainSpec(json!({"": "old-chainType"}));
		assert!(
			matches!(chain_spec.replace_chain_type("new-chainType"), Err(Error::Config(error)) if error == "expected `chainType`")
		);
		Ok(())
	}

	#[test]
	fn replace_protocol_id_works() -> Result<()> {
		let mut chain_spec = ChainSpec(json!({"protocolId": "old-protocolId"}));
		chain_spec.replace_protocol_id("new-protocolId")?;
		assert_eq!(chain_spec.0, json!({"protocolId": "new-protocolId"}));
		Ok(())
	}

	#[test]
	fn replace_protocol_id_fails() -> Result<()> {
		let mut chain_spec = ChainSpec(json!({"": "old-protocolId"}));
		assert!(
			matches!(chain_spec.replace_protocol_id("new-protocolId"), Err(Error::Config(error)) if error == "expected `protocolId`")
		);
		Ok(())
	}

	#[test]
	fn replace_collator_keys_works() -> Result<()> {
		let mut chain_spec = ChainSpec(json!({
			"para_id": 1000,
			"genesis": {
				"runtimeGenesis": {
					"patch": {
						"collatorSelection": {
							"invulnerables": [
							  "5GrwvaEF5zXb26Fz9rcQpDWS57CtERHpNehXCPcNoHGKutQY",
							  "5FHneW46xGXgs5mUiveU4sbTyGBzmstUspZC92UhjJM694ty"
							]
						  },
						  "session": {
							"keys": [
							  [
								"5GrwvaEF5zXb26Fz9rcQpDWS57CtERHpNehXCPcNoHGKutQY",
								"5GrwvaEF5zXb26Fz9rcQpDWS57CtERHpNehXCPcNoHGKutQY",
								{
								  "aura": "5GrwvaEF5zXb26Fz9rcQpDWS57CtERHpNehXCPcNoHGKutQY"
								}
							  ],
							  [
								"5FHneW46xGXgs5mUiveU4sbTyGBzmstUspZC92UhjJM694ty",
								"5FHneW46xGXgs5mUiveU4sbTyGBzmstUspZC92UhjJM694ty",
								{
								  "aura": "5FHneW46xGXgs5mUiveU4sbTyGBzmstUspZC92UhjJM694ty"
								}
							  ]
							]
						  },
					}
				}
			},
		}));
		chain_spec.replace_collator_keys(vec![
			"5Gw3s7q4QLkSWwknsi8jj5P1K79e5N4b6pfsNUzS97H1DXYF".to_string(),
		])?;
		assert_eq!(
			chain_spec.0,
			json!({
				"para_id": 1000,
				"genesis": {
				"runtimeGenesis": {
					"patch": {
						"collatorSelection": {
							"invulnerables": [
							  "5Gw3s7q4QLkSWwknsi8jj5P1K79e5N4b6pfsNUzS97H1DXYF",
							]
						  },
						  "session": {
							"keys": [
							  [
								"5Gw3s7q4QLkSWwknsi8jj5P1K79e5N4b6pfsNUzS97H1DXYF",
								"5Gw3s7q4QLkSWwknsi8jj5P1K79e5N4b6pfsNUzS97H1DXYF",
								{
								  "aura": "5Gw3s7q4QLkSWwknsi8jj5P1K79e5N4b6pfsNUzS97H1DXYF"
								}
							  ],
							]
						  },
					}
				}
			},
			})
		);
		Ok(())
	}

	#[test]
	fn replace_use_evm_collator_keys_works() -> Result<()> {
		let mut chain_spec = ChainSpec(json!({
			"para_id": 1000,
			"properties": {
				"isEthereum": true
			},
			"genesis": {
				"runtimeGenesis": {
					"patch": {
						"collatorSelection": {
							"invulnerables": [
							  "5FHneW46xGXgs5mUiveU4sbTyGBzmstUspZC92UhjJM694ty"
							]
						  },
						  "session": {
							"keys": [
							  [
								"5FHneW46xGXgs5mUiveU4sbTyGBzmstUspZC92UhjJM694ty",
								"5FHneW46xGXgs5mUiveU4sbTyGBzmstUspZC92UhjJM694ty",
								{
								  "aura": "5FHneW46xGXgs5mUiveU4sbTyGBzmstUspZC92UhjJM694ty"
								}
							  ]
							]
						  },
					}
				}
			},
		}));
		chain_spec.replace_collator_keys(vec![
			"5GrwvaEF5zXb26Fz9rcQpDWS57CtERHpNehXCPcNoHGKutQY".to_string(),
		])?;
		assert_eq!(
			chain_spec.0,
			json!({
				"para_id": 1000,
				"properties": {
					"isEthereum": true
				},
				"genesis": {
				"runtimeGenesis": {
					"patch": {
						"collatorSelection": {
							"invulnerables": [
							  "0x9621dde636de098b43efb0fa9b61facfe328f99d",
							]
						  },
						  "session": {
							"keys": [
							  [
								"0x9621dde636de098b43efb0fa9b61facfe328f99d",
								"0x9621dde636de098b43efb0fa9b61facfe328f99d",
								{
								  "aura": "5GrwvaEF5zXb26Fz9rcQpDWS57CtERHpNehXCPcNoHGKutQY"
								}
							  ],
							]
						  },
					}
				}
			},
			})
		);
		Ok(())
	}

	#[test]
	fn update_runtime_code_works() -> Result<()> {
		let mut chain_spec =
			ChainSpec(json!({"genesis": {"runtimeGenesis" : {  "code": "0x00" }}}));

		chain_spec.update_runtime_code(&from_hex("0x1234")?)?;
		assert_eq!(chain_spec.0, json!({"genesis": {"runtimeGenesis" : {  "code": "0x1234" }}}));
		Ok(())
	}

	#[test]
	fn update_runtime_code_fails() -> Result<()> {
		let mut chain_spec =
			ChainSpec(json!({"invalidKey": {"runtimeGenesis" : {  "code": "0x00" }}}));
		assert!(
			matches!(chain_spec.update_runtime_code(&from_hex("0x1234")?), Err(Error::Config(error)) if error == "expected `genesis`")
		);

		chain_spec = ChainSpec(json!({"genesis": {"invalidKey" : {  "code": "0x00" }}}));
		assert!(
			matches!(chain_spec.update_runtime_code(&from_hex("0x1234")?), Err(Error::Config(error)) if error == "expected `runtimeGenesis`")
		);

		chain_spec = ChainSpec(json!({"genesis": {"runtimeGenesis" : {  "invalidKey": "0x00" }}}));
		assert!(
			matches!(chain_spec.update_runtime_code(&from_hex("0x1234")?), Err(Error::Config(error)) if error == "expected `runtimeGenesis.code`")
		);
		Ok(())
	}

	#[test]
	fn check_command_exists_fails() -> Result<()> {
		let binary_path = PathBuf::from("/bin");
		let cmd = "nonexistent_command";
		assert!(matches!(
			check_command_exists(&binary_path, cmd),
			Err(Error::MissingCommand {command, binary })
			if command == cmd && binary == binary_path.display().to_string()
		));
		Ok(())
	}

	#[test]
	fn is_supported_works() -> Result<()> {
		let temp_dir = tempdir()?;
		let path = temp_dir.path();

		// Standard rust project
		let name = "hello_world";
		cmd("cargo", ["new", name]).dir(path).run()?;
		assert!(!is_supported(&path.join(name)));

		// Chain
		let mut manifest = from_path(&path.join(name))?;
		manifest
			.dependencies
			.insert("cumulus-client-collator".into(), Dependency::Simple("^0.14.0".into()));
		let manifest = toml_edit::ser::to_string_pretty(&manifest)?;
		write(path.join(name).join("Cargo.toml"), manifest)?;
		assert!(is_supported(&path.join(name)));
		Ok(())
	}

	#[test]
	fn chain_spec_builder_node_path_works() -> Result<()> {
		let node_path = PathBuf::from("/test/node");
		let builder = ChainSpecBuilder::Node {
			node_path: node_path.clone(),
			default_bootnode: true,
			profile: Profile::Release,
		};
		assert_eq!(builder.path(), node_path);
		Ok(())
	}

	#[test]
	fn chain_spec_builder_runtime_path_works() -> Result<()> {
		let runtime_path = PathBuf::from("/test/runtime");
		let builder = ChainSpecBuilder::Runtime {
			runtime_path: runtime_path.clone(),
			profile: Profile::Release,
		};
		assert_eq!(builder.path(), runtime_path);
		Ok(())
	}

	#[test]
	fn chain_spec_builder_node_profile_works() -> Result<()> {
		for profile in Profile::VARIANTS {
			let builder = ChainSpecBuilder::Node {
				node_path: PathBuf::from("/test/node"),
				default_bootnode: true,
				profile: *profile,
			};
			assert_eq!(builder.profile(), *profile);
		}
		Ok(())
	}

	#[test]
	fn chain_spec_builder_runtime_profile_works() -> Result<()> {
		for profile in Profile::VARIANTS {
			let builder = ChainSpecBuilder::Runtime {
				runtime_path: PathBuf::from("/test/runtime"),
				profile: *profile,
			};
			assert_eq!(builder.profile(), *profile);
		}
		Ok(())
	}

	#[test]
	fn chain_spec_builder_node_artifact_path_works() -> Result<()> {
		let temp_dir =
			setup_template_and_instantiate().expect("Failed to setup template and instantiate");
		mock_build_process(temp_dir.path())?;
		mock_node(temp_dir.path())?;
		let builder = ChainSpecBuilder::Node {
			node_path: temp_dir.path().join("node"),
			default_bootnode: true,
			profile: Profile::Release,
		};
		let artifact_path = builder.artifact_path()?;
		assert!(artifact_path.exists());
		assert!(artifact_path.ends_with("parachain-template-node"));
		Ok(())
	}

	#[test]
	fn chain_spec_builder_runtime_artifact_path_works() -> Result<()> {
		let temp_dir =
			setup_template_and_instantiate().expect("Failed to setup template and instantiate");
		mock_build_runtime_process(temp_dir.path())?;

		let builder = ChainSpecBuilder::Runtime {
			runtime_path: temp_dir.path().join("runtime"),
			profile: Profile::Release,
		};
		let artifact_path = builder.artifact_path()?;
		assert!(artifact_path.is_file());
		assert!(artifact_path.ends_with("parachain_template_runtime.wasm"));
		Ok(())
	}

	#[test]
	fn chain_spec_builder_node_artifact_path_fails() -> Result<()> {
		let temp_dir =
			setup_template_and_instantiate().expect("Failed to setup template and instantiate");

		let builder = ChainSpecBuilder::Node {
			node_path: temp_dir.path().join("node"),
			default_bootnode: true,
			profile: Profile::Release,
		};
		assert!(builder.artifact_path().is_err());
		Ok(())
	}

	#[test]
	fn chain_spec_builder_runtime_artifact_path_fails() -> Result<()> {
		let temp_dir =
			setup_template_and_instantiate().expect("Failed to setup template and instantiate");

		let builder = ChainSpecBuilder::Runtime {
			runtime_path: temp_dir.path().join("runtime"),
			profile: Profile::Release,
		};
		let result = builder.artifact_path();
		assert!(result.is_err());
		assert!(matches!(result, Err(e) if e.to_string().contains("No runtime found")));
		Ok(())
	}

	#[test]
	fn chain_spec_builder_generate_raw_chain_spec_works() -> Result<()> {
		let temp_dir = tempdir()?;
		let builder = ChainSpecBuilder::Runtime {
			runtime_path: temp_dir.path().join("runtime"),
			profile: Profile::Release,
		};
		let original_chain_spec_path =
			PathBuf::from("artifacts/passet-hub-spec.json").canonicalize()?;
		assert!(original_chain_spec_path.exists());
		let chain_spec_path = temp_dir.path().join(original_chain_spec_path.file_name().unwrap());
		fs::copy(&original_chain_spec_path, &chain_spec_path)?;
		let raw_chain_spec_path = temp_dir.path().join("raw.json");
		let final_raw_path = builder.generate_raw_chain_spec(
			&chain_spec_path,
			raw_chain_spec_path.file_name().unwrap().to_str().unwrap(),
		)?;
		assert!(final_raw_path.is_file());
		assert_eq!(final_raw_path, raw_chain_spec_path);

		// Check raw chain spec contains expected fields
		let raw_content = fs::read_to_string(&raw_chain_spec_path)?;
		let raw_json: Value = serde_json::from_str(&raw_content)?;
		assert!(raw_json.get("genesis").is_some());
		assert!(raw_json.get("genesis").unwrap().get("raw").is_some());
		assert!(raw_json.get("genesis").unwrap().get("raw").unwrap().get("top").is_some());
		Ok(())
	}

	#[test]
	fn chain_spec_builder_export_wasm_works() -> Result<()> {
		let temp_dir = tempdir()?;
		let builder = ChainSpecBuilder::Runtime {
			runtime_path: temp_dir.path().join("runtime"),
			profile: Profile::Release,
		};
		let original_chain_spec_path =
			PathBuf::from("artifacts/passet-hub-spec.json").canonicalize()?;
		let chain_spec_path = temp_dir.path().join(original_chain_spec_path.file_name().unwrap());
		fs::copy(&original_chain_spec_path, &chain_spec_path)?;
		let final_wasm_path = temp_dir.path().join("runtime.wasm");
		let final_raw_path = builder.generate_raw_chain_spec(&chain_spec_path, "raw.json")?;
		let wasm_path = builder.export_wasm_file(
			&final_raw_path,
			final_wasm_path.file_name().unwrap().to_str().unwrap(),
		)?;
		assert!(wasm_path.is_file());
		assert_eq!(final_wasm_path, wasm_path);
		Ok(())
	}
}