init helpers: is_contract & is_parachain

Author: weezy20

Cargo.lock

@@ -43,7 +43,7 @@ reqwest = { version = "0.11", optional = true }
 serde_json = { version = "1.0", optional = true }
 serde = { version = "1.0", features = ["derive"], optional = true }
 symlink = { version = "0.1", optional = true }
-toml_edit = { version = "0.22", optional = true }
+toml_edit = { version = "0.22.9" }
 tracing-subscriber = { version = "0.3", optional = true }
 zombienet-sdk = { git = "https://github.com/r0gue-io/zombienet-sdk", branch = "pop", optional = true }
 zombienet-support = { git = "https://github.com/r0gue-io/zombienet-sdk", branch = "pop", optional = true }
@@ -72,7 +72,6 @@ parachain = [
     "dep:reqwest",
     "dep:serde_json",
     "dep:symlink",
-    "dep:toml_edit",
     "dep:tracing-subscriber",
     "dep:url",
     "dep:zombienet-sdk",

Cargo.toml

@@ -1,4 +1,4 @@
-use anyhow::Result;
+use anyhow::{Context, Result};
 use cliclack::{log, outro_cancel};
 use git2::{IndexAddOption, Repository, ResetType};
 use regex::Regex;
@@ -119,7 +119,36 @@ pub(crate) fn resolve_pallet_path(path: Option<String>) -> PathBuf {
 		}
 	}
 }
-
+/// Checks if `path` is a ink contract project directory by searching its dependencies
+pub(crate) fn is_contract(path: &Path) -> Result<bool> {
+	let manifest_path = path.join("Cargo.toml");
+	Ok(if manifest_path.exists() {
+		let manifest =
+			fs::read_to_string(manifest_path).context("is_contract: Failed to read Cargo.toml")?;
+		let manifest: toml_edit::DocumentMut =
+			manifest.parse().context("is_contract: Cargo.toml is not well formed")?;
+		let dependencies =
+			manifest["dependencies"].as_table().expect("dependencies is not a table");
+		dependencies.contains_key("ink") && dependencies.contains_key("scale")
+	} else {
+		false
+	})
+}
+/// Checks if `path` is a substrate parachain project directory by searching its dependencies
+pub(crate) fn is_parachain(path: &Path) -> Result<bool> {
+	let workspace_manifest = path.join("Cargo.toml");
+	if workspace_manifest.exists() {
+		let workspace_manifest = fs::read_to_string(workspace_manifest)
+			.context("is_parachain: Failed to read Cargo.toml")?;
+		let workspace_manifest: toml_edit::DocumentMut = workspace_manifest
+			.parse()
+			.context("is_parachain: Cargo.toml is not well formed")?;
+		todo!("Check if workspace keys are present");
+		Ok(false)
+	} else {
+		Ok(false)
+	}
+}
 #[cfg(test)]
 mod tests {
 	use super::*;

src/helpers.rs

@@ -15,7 +15,7 @@ use std::{
 };
 use symlink::{remove_symlink_file, symlink_file};
 use tempfile::{Builder, NamedTempFile};
-use toml_edit::{value, Document, Formatted, Item, Table, Value};
+use toml_edit::{value, DocumentMut, Formatted, Item, Table, Value};
 use url::Url;
 use zombienet_sdk::{Network, NetworkConfig, NetworkConfigExt};
 use zombienet_support::fs::local::LocalFileSystem;
@@ -26,7 +26,7 @@ pub struct Zombienet {
 	/// The cache location, used for caching binaries.
 	cache: PathBuf,
 	/// The config to be used to launch a network.
-	network_config: (PathBuf, Document),
+	network_config: (PathBuf, DocumentMut),
 	/// The binary required to launch the relay chain.
 	relay_chain: Binary,
 	/// The binaries required to launch parachains.
@@ -43,7 +43,7 @@ impl Zombienet {
 	) -> Result<Self> {
 		// Parse network config
 		let network_config_path = PathBuf::from(network_config);
-		let config = std::fs::read_to_string(&network_config_path)?.parse::<Document>()?;
+		let config = std::fs::read_to_string(&network_config_path)?.parse::<DocumentMut>()?;
 		// Determine binaries
 		let relay_chain_binary = Self::relay_chain(relay_chain_version, &config, &cache).await?;
 		let mut parachain_binaries = IndexMap::new();
@@ -257,7 +257,7 @@ impl Zombienet {
 
 	async fn relay_chain(
 		version: Option<&String>,
-		network_config: &Document,
+		network_config: &DocumentMut,
 		cache: &PathBuf,
 	) -> Result<Binary> {
 		const BINARY: &str = "polkadot";
@@ -602,7 +602,7 @@ mod tests {
 		let cache = PathBuf::from(temp_dir.path());
 
 		let network_config_path = PathBuf::from(CONFIG_FILE_PATH);
-		let config = std::fs::read_to_string(&network_config_path)?.parse::<Document>()?;
+		let config = std::fs::read_to_string(&network_config_path)?.parse::<DocumentMut>()?;
 
 		let binary_relay_chain =
 			Zombienet::relay_chain(Some(&TESTING_POLKADOT_VERSION.to_string()), &config, &cache)
@@ -627,7 +627,7 @@ mod tests {
 		let cache = PathBuf::from(temp_dir.path());
 
 		let network_config_path = PathBuf::from(CONFIG_FILE_PATH);
-		let config = std::fs::read_to_string(&network_config_path)?.parse::<Document>()?;
+		let config = std::fs::read_to_string(&network_config_path)?.parse::<DocumentMut>()?;
 
 		// Ideally here we will Mock GitHub struct and its get_latest_release function response
 		let binary_relay_chain = Zombienet::relay_chain(None, &config, &cache).await?;
@@ -651,7 +651,7 @@ mod tests {
 		let network_config_path = generate_wrong_config_no_relay(&temp_dir)
 			.expect("Error generating the testing toml file");
 
-		let config = std::fs::read_to_string(&network_config_path)?.parse::<Document>()?;
+		let config = std::fs::read_to_string(&network_config_path)?.parse::<DocumentMut>()?;
 
 		let result_error =
 			Zombienet::relay_chain(Some(&TESTING_POLKADOT_VERSION.to_string()), &config, &cache)