feat: wallet integration call contract

Cargo.toml

@@ -49,17 +49,14 @@ subxt = "0.38.0"
 ink_env = "5.0.0"
 sp-core = "32.0.0"
 sp-weights = "31.0.0"
-scale-info = { version = "2.11.4", default-features = false, features = ["derive"] }
-scale-value = { version = "0.17.0", default-features = false, features = ["from-string", "parser-ss58"] }
 # TODO: git deps
 #contract-build = "5.0.0-alpha"
 contract-build = { git = "https://github.com/use-ink/cargo-contract", branch = "peter/chore-make-types-pub" }
 #contract-extrinsics = "5.0.0-alpha"
 contract-extrinsics = { git = "https://github.com/use-ink/cargo-contract", branch = "peter/chore-make-types-pub" }
-#contract-transcode = "5.0.0"
+#contract-transcode = "5.0.0-alpha"
 contract-transcode = { git = "https://github.com/use-ink/cargo-contract", branch = "peter/chore-make-types-pub" }
 heck = "0.5.0"
-hex = { version = "0.4.3", default-features = false }
 
 # parachains
 askama = "0.12"
@@ -68,6 +65,8 @@ walkdir = "2.5"
 indexmap = "2.2"
 toml_edit = { version = "0.22", features = ["serde"] }
 symlink = "0.1"
+scale-info = { version = "2.11.4", default-features = false, features = ["derive"] }
+scale-value = { version = "0.17.0", default-features = false, features = ["from-string", "parser-ss58"] }
 serde_json = { version = "1.0", features = ["preserve_order"] }
 serde = { version = "1.0", features = ["derive"] }
 zombienet-sdk = "0.2.18"

crates/pop-cli/Cargo.toml

@@ -28,12 +28,12 @@ url.workspace = true
 clap.workspace = true
 cliclack.workspace = true
 console.workspace = true
+sp-core.workspace = true
 strum.workspace = true
 strum_macros.workspace = true
 
 # contracts
 pop-contracts = { path = "../pop-contracts", version = "0.5.0", optional = true }
-sp-core = { workspace = true, optional = true }
 sp-weights = { workspace = true, optional = true }
 
 # parachains
@@ -49,7 +49,7 @@ pop-common = { path = "../pop-common", version = "0.5.0" }
 
 # wallet-integration
 axum.workspace = true
-tower-http = { workspace = true, features = ["fs"] }
+tower-http = { workspace = true, features = ["fs", "cors"] }
 
 [dev-dependencies]
 assert_cmd.workspace = true
@@ -59,6 +59,6 @@ subxt-signer.workspace = true
 
 [features]
 default = ["contract", "parachain", "telemetry"]
-contract = ["dep:pop-contracts", "dep:sp-core", "dep:sp-weights", "dep:dirs"]
+contract = ["dep:pop-contracts", "dep:sp-weights", "dep:dirs"]
 parachain = ["dep:pop-parachains", "dep:dirs"]
 telemetry = ["dep:pop-telemetry"]

crates/pop-cli/src/cli.rs

@@ -150,7 +150,6 @@ impl traits::Confirm for Confirm {
 
 /// A input prompt using cliclack.
 struct Input(cliclack::Input);
-
 impl traits::Input for Input {
 	/// Sets the default value for the input.
 	fn default_input(mut self, value: &str) -> Self {

crates/pop-cli/src/commands/call/chain.rs

@@ -2,14 +2,17 @@
 
 use std::path::Path;
 
-use crate::cli::{self, traits::*};
+use crate::{
+	cli::{self, traits::*},
+	common::wallet::wait_for_signature,
+};
 use anyhow::{anyhow, Result};
 use clap::Args;
 use pop_parachains::{
 	construct_extrinsic, construct_sudo_extrinsic, decode_call_data, encode_call_data,
 	find_dispatchable_by_name, find_pallet_by_name, parse_chain_metadata, set_up_client,
-	sign_and_submit_extrinsic, supported_actions, Action, CallData, DynamicPayload, Function,
-	OnlineClient, Pallet, Param, SubstrateConfig,
+	sign_and_submit_extrinsic, submit_signed_extrinsic, supported_actions, Action, CallData,
+	DynamicPayload, Function, OnlineClient, Pallet, Param, Payload, SubstrateConfig,
 };
 use url::Url;
 
@@ -40,6 +43,9 @@
 	/// - with a password "//Alice///SECRET_PASSWORD"
 	#[arg(short, long)]
 	suri: Option<String>,
+	/// Use your browser wallet to sign the extrinsic.
+	#[arg(name = "use-wallet", long, default_value = "false", conflicts_with = "suri")]
+	use_wallet: bool,
 	/// SCALE encoded bytes representing the call data of the extrinsic.
 	#[arg(name = "call", short, long, conflicts_with_all = ["pallet", "function", "args"])]
 	call_data: Option<String>,
@@ -95,7 +101,15 @@
 			};
 
 			// Sign and submit the extrinsic.
-			if let Err(e) = call.submit_extrinsic(&chain.client, &chain.url, xt, &mut cli).await {
+			let result = if self.use_wallet {
+				let call_data = xt.encode_call_data(&chain.client.metadata())?;
+				submit_extrinsic_with_secure_signing(&chain.client, &chain.url, call_data, &mut cli)
+					.await
+			} else {
+				call.submit_extrinsic(&chain.client, &chain.url, xt, &mut cli).await
+			};
+
+			if let Err(e) = result {
 				display_message(&e.to_string(), false, &mut cli)?;
 				break;
 			}
@@ -197,11 +211,25 @@
 			// sudo.
 			self.configure_sudo(chain, cli)?;
 
-			// Resolve who is signing the extrinsic.
+			// Resolve who is signing the extrinsic. If a `suri` was provided via the command line,
+			// skip the prompt.
 			let suri = match self.suri.as_ref() {
 				Some(suri) => suri.clone(),
-				None =>
-					cli.input("Signer of the extrinsic:").default_input(DEFAULT_URI).interact()?,
+				None => {
+					if !self.use_wallet {
+						if cli.confirm("Do you want to use your browser wallet to sign the transaction? (Selecting 'No' will prompt you to manually enter the secret key URI for signing, e.g., '//Alice')")
+						.initial_value(true)
+						.interact()? {
+							self.use_wallet = true;
+							DEFAULT_URI.to_string() // Default value because the user is using the browser wallet.
+						}
+						else {
+							cli.input("Signer of the extrinsic:").default_input(DEFAULT_URI).interact()?
+						}
+					} else {
+						DEFAULT_URI.to_string() // Default value because the user is using the browser wallet.
+					}
+				},
 			};
 
 			return Ok(Call {
@@ -210,6 +238,7 @@
 				suri,
 				skip_confirm: self.skip_confirm,
 				sudo: self.sudo,
+				use_wallet: self.use_wallet,
 			});
 		}
 	}
@@ -222,11 +251,36 @@
 		call_data: &str,
 		cli: &mut impl Cli,
 	) -> Result<()> {
-		// Resolve who is signing the extrinsic.
+		// Resolve who is signing the extrinsic. If a `suri` was provided via the command line,
+		// skip the prompt.
+		let mut use_wallet = self.use_wallet;
 		let suri = match self.suri.as_ref() {
-			Some(suri) => suri,
-			None => &cli.input("Signer of the extrinsic:").default_input(DEFAULT_URI).interact()?,
+			Some(suri) => suri.clone(),
+			None =>
+				if !self.use_wallet {
+					if cli.confirm("Do you want to use your browser wallet to sign the transaction? (Selecting 'No' will prompt you to manually enter the secret key URI for signing, e.g., '//Alice')")
+						.initial_value(true)
+						.interact()? {
+							use_wallet = true;
+							DEFAULT_URI.to_string()
+						}
+						else {
+							cli.input("Signer of the extrinsic:").default_input(DEFAULT_URI).interact()?
+						}
+				} else {
+					DEFAULT_URI.to_string()
+				},
 		};
+		// Perform signing steps with wallet integration and return early.
+		if use_wallet {
+			let call_data_bytes =
+				decode_call_data(call_data).map_err(|err| anyhow!("{}", format!("{err:?}")))?;
+			submit_extrinsic_with_secure_signing(client, &url, call_data_bytes, cli)
+				.await
+				.map_err(|err| anyhow!("{}", format!("{err:?}")))?;
+			display_message("Call complete.", true, cli)?;
+			return Ok(());
+		}
 		cli.info(format!("Encoded call data: {}", call_data))?;
 		if !self.skip_confirm &&
 			!cli.confirm("Do you want to submit the extrinsic?")
@@ -244,7 +298,7 @@
 		spinner.start("Signing and submitting the extrinsic and then waiting for finalization, please be patient...");
 		let call_data_bytes =
 			decode_call_data(call_data).map_err(|err| anyhow!("{}", format!("{err:?}")))?;
-		let result = sign_and_submit_extrinsic(client, url, CallData::new(call_data_bytes), suri)
+		let result = sign_and_submit_extrinsic(client, url, CallData::new(call_data_bytes), &suri)
 			.await
 			.map_err(|err| anyhow!("{}", format!("{err:?}")))?;
 
@@ -283,6 +337,7 @@
 		self.function = None;
 		self.args.clear();
 		self.sudo = false;
+		self.use_wallet = false;
 	}
 
 	// Function to check if all required fields are specified.
@@ -334,6 +389,8 @@
 	/// - for a dev account "//Alice"
 	/// - with a password "//Alice///SECRET_PASSWORD"
 	suri: String,
+	/// Whether to use your browser wallet to sign the extrinsic.
+	use_wallet: bool,
 	/// Whether to automatically sign and submit the extrinsic without prompting for confirmation.
 	skip_confirm: bool,
 	/// Whether to dispatch the function call with `Root` origin.
@@ -411,14 +468,45 @@
 				.collect();
 			full_message.push_str(&format!(" --args {}", args.join(" ")));
 		}
-		full_message.push_str(&format!(" --url {} --suri {}", chain.url, self.suri));
+		full_message.push_str(&format!(" --url {}", chain.url));
+		if self.use_wallet {
+			full_message.push_str(" --use-wallet");
+		} else {
+			full_message.push_str(&format!(" --suri {}", self.suri));
+		}
 		if self.sudo {
 			full_message.push_str(" --sudo");
 		}
 		full_message
 	}
 }
 
+// Sign and submit an extrinsic using wallet integration.
+async fn submit_extrinsic_with_secure_signing(
+	client: &OnlineClient<SubstrateConfig>,
+	url: &Url,
+	call_data: Vec<u8>,
+	cli: &mut impl Cli,
+) -> Result<()> {
+	let maybe_payload = wait_for_signature(call_data, url.to_string()).await?;
+	if let Some(payload) = maybe_payload {
+		cli.success("Signed payload received.")?;
+		let spinner = cliclack::spinner();
+		spinner.start(
+			"Submitting the extrinsic and then waiting for finalization, please be patient...",
+		);
+
+		let result = submit_signed_extrinsic(client.clone(), payload)
+			.await
+			.map_err(|err| anyhow!("{}", format!("{err:?}")))?;
+
+		spinner.stop(format!("Extrinsic submitted with hash: {:?}", result));
+	} else {
+		display_message("No signed payload received.", false, cli)?;
+	}
+	Ok(())
+}
+
 // Displays a message to the user, with formatting based on the success status.
 fn display_message(message: &str, success: bool, cli: &mut impl Cli) -> Result<()> {
 	if success {
@@ -642,7 +730,7 @@
 		)
 		.expect_input("The value for `remark` might be too large to enter. You may enter the path to a file instead.", "0x11".into())
 		.expect_confirm("Would you like to dispatch this function call with `Root` origin?", true)
-		.expect_input("Signer of the extrinsic:", "//Bob".into());
+		.expect_confirm("Do you want to use your browser wallet to sign the transaction? (Selecting 'No' will prompt you to manually enter the secret key URI for signing, e.g., '//Alice')", true);
 
 		let chain = call_config.configure_chain(&mut cli).await?;
 		assert_eq!(chain.url, Url::parse(POP_NETWORK_TESTNET_URL)?);
@@ -651,9 +739,10 @@
 		assert_eq!(call_chain.function.pallet, "System");
 		assert_eq!(call_chain.function.name, "remark");
 		assert_eq!(call_chain.args, ["0x11".to_string()].to_vec());
-		assert_eq!(call_chain.suri, "//Bob");
+		assert_eq!(call_chain.suri, "//Alice"); // Default value
+		assert!(call_chain.use_wallet);
 		assert!(call_chain.sudo);
-		assert_eq!(call_chain.display(&chain), "pop call chain --pallet System --function remark --args \"0x11\" --url wss://rpc1.paseo.popnetwork.xyz/ --suri //Bob --sudo");
+		assert_eq!(call_chain.display(&chain), "pop call chain --pallet System --function remark --args \"0x11\" --url wss://rpc1.paseo.popnetwork.xyz/ --use-wallet --sudo");
 		cli.verify()
 	}
 
@@ -714,6 +803,7 @@
 			},
 			args: vec!["0x11".to_string()].to_vec(),
 			suri: DEFAULT_URI.to_string(),
+			use_wallet: false,
 			skip_confirm: false,
 			sudo: false,
 		};
@@ -753,6 +843,7 @@
 			function: find_dispatchable_by_name(&pallets, "System", "remark")?.clone(),
 			args: vec!["0x11".to_string()].to_vec(),
 			suri: DEFAULT_URI.to_string(),
+			use_wallet: false,
 			skip_confirm: false,
 			sudo: false,
 		};
@@ -776,11 +867,13 @@
 			args: vec![].to_vec(),
 			url: Some(Url::parse(POP_NETWORK_TESTNET_URL)?),
 			suri: None,
+			use_wallet: false,
 			skip_confirm: false,
 			call_data: Some("0x00000411".to_string()),
 			sudo: false,
 		};
 		let mut cli = MockCli::new()
+			.expect_confirm("Do you want to use your browser wallet to sign the transaction? (Selecting 'No' will prompt you to manually enter the secret key URI for signing, e.g., '//Alice')", false)
 			.expect_input("Signer of the extrinsic:", "//Bob".into())
 			.expect_confirm("Do you want to submit the extrinsic?", false)
 			.expect_outro_cancel("Extrinsic with call data 0x00000411 was not submitted.");
@@ -803,8 +896,9 @@
 			pallet: None,
 			function: None,
 			args: vec![].to_vec(),
-			url: Some(Url::parse("wss://polkadot-rpc.publicnode.com")?),
+			url: Some(Url::parse(POLKADOT_NETWORK_URL)?),
 			suri: Some("//Alice".to_string()),
+			use_wallet: false,
 			skip_confirm: false,
 			call_data: Some("0x00000411".to_string()),
 			sudo: true,
@@ -836,6 +930,7 @@
 			function: Some("remark".to_string()),
 			args: vec!["0x11".to_string()].to_vec(),
 			url: Some(Url::parse(POP_NETWORK_TESTNET_URL)?),
+			use_wallet: true,
 			suri: Some(DEFAULT_URI.to_string()),
 			skip_confirm: false,
 			call_data: None,
@@ -846,6 +941,7 @@
 		assert_eq!(call_config.function, None);
 		assert_eq!(call_config.args.len(), 0);
 		assert!(!call_config.sudo);
+		assert!(!call_config.use_wallet);
 		Ok(())
 	}
 
@@ -857,6 +953,7 @@
 			args: vec!["0x11".to_string()].to_vec(),
 			url: Some(Url::parse(POP_NETWORK_TESTNET_URL)?),
 			suri: Some(DEFAULT_URI.to_string()),
+			use_wallet: false,
 			skip_confirm: false,
 			call_data: None,
 			sudo: false,
@@ -875,6 +972,7 @@
 			args: vec!["2000".to_string(), "0x1".to_string(), "0x12".to_string()].to_vec(),
 			url: Some(Url::parse(POP_NETWORK_TESTNET_URL)?),
 			suri: Some(DEFAULT_URI.to_string()),
+			use_wallet: false,
 			call_data: None,
 			skip_confirm: false,
 			sudo: false,