feat: add ability to run a script once network is initialized

crates/pop-cli/src/commands/up/parachain.rs

@@ -6,6 +6,7 @@ use cliclack::{
 	clear_screen, confirm, intro, log, multi_progress, outro, outro_cancel, set_theme, ProgressBar,
 };
 use console::{Emoji, Style};
+use duct::cmd;
 use pop_parachains::{NetworkNode, Status, Zombienet};
 use std::time::Duration;
 use tokio::time::sleep;
@@ -26,6 +27,12 @@ pub(crate) struct ZombienetCommand {
 	/// The url of the git repository of a parachain to be used, with branch/release tag specified as #fragment (e.g. 'https://github.com/org/repository#tag'). A specific binary name can also be optionally specified via query string parameter (e.g. 'https://github.com/org/repository?binaryname#tag'), defaulting to the name of the repository when not specified.
 	#[arg(short, long)]
 	parachain: Option<Vec<String>>,
+	/// The path to a script to be run after the network has been set up.
+	#[clap(name = "init", short = 'i', long)]
+	setup_script: Option<String>,
+	/// The script arguments, encoded as strings.
+	#[clap(name = "args", short = 'a', long, num_args = 0..)]
+	args: Vec<String>,
 	/// Whether the output should be verbose.
 	#[arg(short, long, action)]
 	verbose: bool,
@@ -138,6 +145,16 @@ impl ZombienetCommand {
 					}
 				}
 
+				if self.setup_script.is_some() {
+					spinner.set_message("Initializing your setup script...");
+					// sleep for 15 seconds to allow the network to start up
+					sleep(Duration::from_secs(15)).await;
+					cmd(self.setup_script.clone().unwrap(), self.args.clone())
+						.run()
+						.map_err(|e| anyhow::Error::new(e).context("Error running the script"))?;
+					clear_screen()?;
+				}
+
 				spinner.stop(result);
 				tokio::signal::ctrl_c().await?;
 				outro("Done")?;

tests/zombienet.toml

@@ -3,6 +3,7 @@ chain = "rococo-local"
 
 [[relaychain.nodes]]
 name = "alice"
+rpc_port = 8833
 validator = true
 
 [[relaychain.nodes]]
@@ -22,4 +23,5 @@ default_command = "pop-node"
 
 [[parachains.collators]]
 name = "pop"
+rpc_port = 9944
 args = ["-lruntime::contracts=debug"]