stellar_scaffold_cli/commands/build/

clients.rs

#![allow(clippy::struct_excessive_bools)]
use super::env_toml::Network;
use crate::arg_parsing;
use crate::arg_parsing::ArgParser;
use crate::commands::build::clients::Error::UpgradeArgsError;
use crate::commands::build::env_toml;
use indexmap::IndexMap;
use regex::Regex;
use serde_json;
use shlex::split;
use std::hash::Hash;
use std::path::Path;
use std::process::Command;
use std::{fmt::Debug, path::PathBuf};
use stellar_cli::{
    commands as cli,
    commands::contract::info::shared::{
        self as contract_spec, fetch, Args as FetchArgs, Error as FetchError,
    },
    commands::NetworkRunnable,
    print::Print,
    utils::contract_hash,
    utils::contract_spec::Spec,
    CommandParser,
};
use stellar_strkey::{self, Contract};
use stellar_xdr::curr::ScSpecEntry::FunctionV0;
use stellar_xdr::curr::{Error as xdrError, ScSpecEntry, ScSpecTypeBytesN, ScSpecTypeDef};

#[derive(Clone, Copy, Debug, Eq, Hash, PartialEq, clap::ValueEnum)]
pub enum ScaffoldEnv {
    Development,
    Testing,
    Staging,
    Production,
}

impl std::fmt::Display for ScaffoldEnv {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        write!(f, "{}", format!("{self:?}").to_lowercase())
    }
}

#[derive(clap::Args, Clone, Debug)]
pub struct Args {
    #[arg(env = "STELLAR_SCAFFOLD_ENV", value_enum)]
    pub env: Option<ScaffoldEnv>,
    #[arg(skip)]
    pub workspace_root: Option<std::path::PathBuf>,
    /// Directory where wasm files are located
    #[arg(skip)]
    pub out_dir: Option<std::path::PathBuf>,
    #[arg(skip)]
    pub global_args: Option<stellar_cli::commands::global::Args>,
}

#[derive(thiserror::Error, Debug)]
pub enum Error {
    #[error(transparent)]
    EnvironmentsToml(#[from] env_toml::Error),
    #[error("⛔ ️invalid network: must either specify a network name or both network_passphrase and rpc_url"
    )]
    MalformedNetwork,
    #[error(transparent)]
    ParsingNetwork(#[from] cli::network::Error),
    #[error(transparent)]
    GeneratingKey(#[from] cli::keys::generate::Error),
    #[error("⛔ ️can only have one default account; marked as default: {0:?}")]
    OnlyOneDefaultAccount(Vec<String>),
    #[error(transparent)]
    InvalidPublicKey(#[from] cli::keys::public_key::Error),
    #[error(transparent)]
    AddressParsing(#[from] stellar_cli::config::address::Error),
    #[error("⛔ ️you need to provide at least one account, to use as the source account for contract deployment and other operations"
    )]
    NeedAtLeastOneAccount,
    #[error("⛔ ️No contract named {0:?}")]
    BadContractName(String),
    #[error("⛔ ️Invalid contract ID: {0:?}")]
    InvalidContractID(String),
    #[error("⛔ ️An ID must be set for a contract in production or staging. E.g. <name>.id = C...")]
    MissingContractID(String),
    #[error("⛔ ️Unable to parse script: {0:?}")]
    ScriptParseFailure(String),
    #[error(transparent)]
    RpcClient(#[from] soroban_rpc::Error),
    #[error("⛔ ️Failed to execute subcommand: {0:?}\n{1:?}")]
    SubCommandExecutionFailure(String, String),
    #[error(transparent)]
    ContractInstall(#[from] cli::contract::upload::Error),
    #[error(transparent)]
    ContractDeploy(#[from] cli::contract::deploy::wasm::Error),
    #[error(transparent)]
    ContractBindings(#[from] cli::contract::bindings::typescript::Error),
    #[error(transparent)]
    ContractFetch(#[from] cli::contract::fetch::Error),
    #[error(transparent)]
    ConfigLocator(#[from] stellar_cli::config::locator::Error),
    #[error(transparent)]
    ConfigNetwork(#[from] stellar_cli::config::network::Error),
    #[error(transparent)]
    ContractInvoke(#[from] cli::contract::invoke::Error),
    #[error(transparent)]
    ContractInfo(#[from] cli::contract::info::interface::Error),
    #[error(transparent)]
    Clap(#[from] clap::Error),
    #[error(transparent)]
    WasmHash(#[from] xdrError),
    #[error(transparent)]
    Io(#[from] std::io::Error),
    #[error(transparent)]
    Json(#[from] serde_json::Error),
    #[error("⛔ ️Failed to run npm command in {0:?}: {1:?}")]
    NpmCommandFailure(std::path::PathBuf, String),
    #[error(transparent)]
    AccountFund(#[from] cli::keys::fund::Error),
    #[error("Failed to get upgrade operator: {0:?}")]
    UpgradeArgsError(arg_parsing::Error),
    #[error(transparent)]
    FetchError(#[from] FetchError),
    #[error(transparent)]
    SpecError(#[from] stellar_cli::get_spec::contract_spec::Error),
}

impl Args {
    fn printer(&self) -> Print {
        Print::new(self.global_args.as_ref().is_some_and(|args| args.quiet))
    }

    pub async fn run(&self, package_names: Vec<String>) -> Result<(), Error> {
        let workspace_root = self
            .workspace_root
            .as_ref()
            .expect("workspace_root must be set before running");

        let Some(current_env) = env_toml::Environment::get(
            workspace_root,
            &self.stellar_scaffold_env(ScaffoldEnv::Development),
        )?
        else {
            return Ok(());
        };

        self.add_network_to_env(&current_env.network)?;
        // Create the '.config' directory if it doesn't exist
        std::fs::create_dir_all(workspace_root.join(".config/stellar"))
            .map_err(stellar_cli::config::locator::Error::Io)?;
        self.clone()
            .handle_accounts(current_env.accounts.as_deref(), &current_env.network)
            .await?;
        self.clone()
            .handle_contracts(
                current_env.contracts.as_ref(),
                package_names,
                &current_env.network,
            )
            .await?;

        Ok(())
    }

    fn stellar_scaffold_env(&self, default: ScaffoldEnv) -> String {
        self.env.unwrap_or(default).to_string().to_lowercase()
    }

    /// Parse the network settings from the environments.toml file and set `STELLAR_RPC_URL` and
    /// `STELLAR_NETWORK_PASSPHRASE`.
    ///
    /// We could set `STELLAR_NETWORK` instead, but when importing contracts, we want to hard-code
    /// the network passphrase. So if given a network name, we use soroban-cli to fetch the RPC url
    /// & passphrase for that named network, and still set the environment variables.
    fn add_network_to_env(&self, network: &env_toml::Network) -> Result<(), Error> {
        let printer = self.printer();
        match &network {
            Network {
                name: Some(name), ..
            } => {
                let stellar_cli::config::network::Network {
                    rpc_url,
                    network_passphrase,
                    ..
                } = (stellar_cli::config::network::Args {
                    network: Some(name.clone()),
                    rpc_url: None,
                    network_passphrase: None,
                    rpc_headers: Vec::new(),
                })
                .get(&stellar_cli::config::locator::Args {
                    global: false,
                    config_dir: None,
                })?;
                printer.infoln(format!("Using {name} network"));
                std::env::set_var("STELLAR_RPC_URL", rpc_url);
                std::env::set_var("STELLAR_NETWORK_PASSPHRASE", network_passphrase);
            }
            Network {
                rpc_url: Some(rpc_url),
                network_passphrase: Some(passphrase),
                ..
            } => {
                std::env::set_var("STELLAR_RPC_URL", rpc_url);
                std::env::set_var("STELLAR_NETWORK_PASSPHRASE", passphrase);
                printer.infoln(format!("Using network at {rpc_url}"));
            }
            _ => return Err(Error::MalformedNetwork),
        }

        Ok(())
    }

    fn get_network_args(network: &Network) -> stellar_cli::config::network::Args {
        stellar_cli::config::network::Args {
            rpc_url: network.rpc_url.clone(),
            network_passphrase: network.network_passphrase.clone(),
            network: network.name.clone(),
            rpc_headers: network.rpc_headers.clone().unwrap_or_default(),
        }
    }

    fn get_config_dir(&self) -> PathBuf {
        self.workspace_root
            .as_ref()
            .expect("workspace_root not set")
            .join(".config")
            .join("stellar")
    }

    fn get_config_locator(&self) -> stellar_cli::config::locator::Args {
        let config_dir = Some(self.get_config_dir());
        stellar_cli::config::locator::Args {
            global: false,
            config_dir,
        }
    }

    fn get_contract_alias(
        &self,
        name: &str,
    ) -> Result<Option<Contract>, stellar_cli::config::locator::Error> {
        let config_dir = self.get_config_locator();
        let network_passphrase = std::env::var("STELLAR_NETWORK_PASSPHRASE")
            .expect("No STELLAR_NETWORK_PASSPHRASE environment variable set");
        config_dir.get_contract_id(name, &network_passphrase)
    }

    async fn get_contract_hash(
        &self,
        contract_id: &Contract,
        network: &Network,
    ) -> Result<Option<String>, Error> {
        let result = cli::contract::fetch::Cmd {
            contract_id: stellar_cli::config::UnresolvedContract::Resolved(*contract_id),
            out_file: None,
            locator: self.get_config_locator(),
            network: Self::get_network_args(network),
        }
        .run_against_rpc_server(self.global_args.as_ref(), None)
        .await;

        match result {
            Ok(result) => {
                let ctrct_hash = contract_hash(&result)?;
                Ok(Some(hex::encode(ctrct_hash)))
            }
            Err(e) => {
                if e.to_string().contains("Contract not found") {
                    Ok(None)
                } else {
                    Err(Error::ContractFetch(e))
                }
            }
        }
    }

    fn save_contract_alias(
        &self,
        name: &str,
        contract_id: &Contract,
        network: &Network,
    ) -> Result<(), stellar_cli::config::locator::Error> {
        let config_dir = self.get_config_locator();
        let passphrase = network
            .network_passphrase
            .clone()
            .expect("You must set a network passphrase.");
        config_dir.save_contract_id(&passphrase, contract_id, name)
    }

    fn create_contract_template(&self, name: &str, contract_id: &str) -> Result<(), Error> {
        let allow_http = if ["development", "test"]
            .contains(&self.stellar_scaffold_env(ScaffoldEnv::Production).as_str())
        {
            "\n  allowHttp: true,"
        } else {
            ""
        };
        let network = std::env::var("STELLAR_NETWORK_PASSPHRASE")
            .expect("No STELLAR_NETWORK_PASSPHRASE environment variable set");
        let template = format!(
            r"import * as Client from '{name}';
import {{ rpcUrl }} from './util';

export default new Client.Client({{
  networkPassphrase: '{network}',
  contractId: '{contract_id}',
  rpcUrl,{allow_http}
  publicKey: undefined,
}});
"
        );
        let workspace_root = self
            .workspace_root
            .as_ref()
            .expect("workspace_root not set");
        let path = workspace_root.join(format!("src/contracts/{name}.ts"));
        std::fs::write(path, template)?;
        Ok(())
    }

    async fn generate_contract_bindings(&self, name: &str, contract_id: &str) -> Result<(), Error> {
        let printer = self.printer();
        printer.infoln(format!("Binding {name:?} contract"));
        let workspace_root = self
            .workspace_root
            .as_ref()
            .expect("workspace_root not set");
        let final_output_dir = workspace_root.join(format!("packages/{name}"));

        // Create a temporary directory for building the new client
        let temp_dir = workspace_root.join(format!("target/packages/{name}"));
        let temp_dir_display = temp_dir.display();
        cli::contract::bindings::typescript::Cmd::parse_arg_vec(&[
            "--contract-id",
            contract_id,
            "--output-dir",
            temp_dir.to_str().expect("we do not support non-utf8 paths"),
            "--config-dir",
            self.get_config_dir()
                .to_str()
                .expect("we do not support non-utf8 paths"),
            "--overwrite",
        ])?
        .run_against_rpc_server(self.global_args.as_ref(), None)
        .await?;

        // Run `npm i` in the temp directory
        printer.infoln(format!("Running 'npm install' in {temp_dir_display:?}"));
        let output = std::process::Command::new("npm")
            .current_dir(&temp_dir)
            .arg("install")
            .arg("--loglevel=error") // Reduce noise from warnings
            .arg("--no-workspaces") // fix issue where stellar sometimes isnt installed locally causing tsc to fail
            .output()?;

        if !output.status.success() {
            // Clean up temp directory on failure
            let _ = std::fs::remove_dir_all(&temp_dir);
            return Err(Error::NpmCommandFailure(
                temp_dir.clone(),
                format!(
                    "npm install failed with status: {:?}\nError: {}",
                    output.status.code(),
                    String::from_utf8_lossy(&output.stderr)
                ),
            ));
        }
        printer.checkln(format!("'npm install' succeeded in {temp_dir_display}"));

        printer.infoln(format!("Running 'npm run build' in {temp_dir_display}"));
        let output = std::process::Command::new("npm")
            .current_dir(&temp_dir)
            .arg("run")
            .arg("build")
            .arg("--loglevel=error") // Reduce noise from warnings
            .output()?;

        if !output.status.success() {
            // Clean up temp directory on failure
            let _ = std::fs::remove_dir_all(&temp_dir);
            return Err(Error::NpmCommandFailure(
                temp_dir.clone(),
                format!(
                    "npm run build failed with status: {:?}\nError: {}",
                    output.status.code(),
                    String::from_utf8_lossy(&output.stderr)
                ),
            ));
        }
        printer.checkln(format!("'npm run build' succeeded in {temp_dir_display}"));

        // Now atomically replace the old directory with the new one
        if final_output_dir.exists() {
            for p in ["dist/index.d.ts", "dist/index.js", "src/index.ts"]
                .iter()
                .map(Path::new)
            {
                std::fs::copy(temp_dir.join(p), final_output_dir.join(p))?;
            }
            printer.checkln(format!("Client {name:?} updated successfully"));
        } else {
            std::fs::create_dir_all(&final_output_dir)?;
            // No existing directory, just move temp to final location
            std::fs::rename(&temp_dir, &final_output_dir)?;
            printer.checkln(format!("Client {name:?} created successfully"));
            // Run npm install in the final output directory to ensure proper linking
            let output = std::process::Command::new("npm")
                .current_dir(&final_output_dir)
                .arg("install")
                .arg("--loglevel=error")
                .output()?;

            if !output.status.success() {
                return Err(Error::NpmCommandFailure(
                    final_output_dir.clone(),
                    format!(
                        "npm install in final directory failed with status: {:?}\nError: {}",
                        output.status.code(),
                        String::from_utf8_lossy(&output.stderr)
                    ),
                ));
            }
        }

        self.create_contract_template(name, contract_id)?;
        Ok(())
    }

    async fn handle_accounts(
        &self,
        accounts: Option<&[env_toml::Account]>,
        network: &Network,
    ) -> Result<(), Error> {
        let printer = self.printer();
        let Some(accounts) = accounts else {
            return Err(Error::NeedAtLeastOneAccount);
        };

        let default_account_candidates = accounts
            .iter()
            .filter(|&account| account.default)
            .map(|account| account.name.clone())
            .collect::<Vec<_>>();

        let default_account = match (default_account_candidates.as_slice(), accounts) {
            ([], []) => return Err(Error::NeedAtLeastOneAccount),
            ([], [env_toml::Account { name, .. }, ..]) => name.clone(),
            ([candidate], _) => candidate.to_string(),
            _ => return Err(Error::OnlyOneDefaultAccount(default_account_candidates)),
        };
        let config = self.get_config_locator();

        for account in accounts {
            printer.infoln(format!("Creating keys for {:?}", account.name));
            // Use provided global args or create default
            let args =
                self.global_args
                    .clone()
                    .unwrap_or_else(|| stellar_cli::commands::global::Args {
                        locator: config.clone(),
                        ..Default::default()
                    });

            let generate_cmd = cli::keys::generate::Cmd {
                name: account.name.clone().parse()?,
                fund: true,
                config_locator: config.clone(),
                network: Self::get_network_args(network),
                seed: None,
                hd_path: None,
                as_secret: false,
                secure_store: false,
                overwrite: false,
            };

            match generate_cmd.run(&args).await {
                Err(e) if e.to_string().contains("already exists") => {
                    printer.blankln(e);
                    // Check if account exists on chain
                    let rpc_client = soroban_rpc::Client::new(
                        network
                            .rpc_url
                            .as_ref()
                            .expect("network contains the RPC url"),
                    )?;

                    let public_key_cmd = cli::keys::public_key::Cmd {
                        name: account.name.parse()?,
                        locator: config.clone(),
                        hd_path: None,
                    };
                    let address = public_key_cmd.public_key().await?;

                    if (rpc_client.get_account(&address.to_string()).await).is_err() {
                        printer.infoln("Account not found on chain, funding...");
                        let fund_cmd = cli::keys::fund::Cmd {
                            network: Self::get_network_args(network),
                            address: public_key_cmd,
                        };
                        fund_cmd.run(&args).await?;
                    }
                }
                other_result => other_result?,
            }
        }

        std::env::set_var("STELLAR_ACCOUNT", &default_account);
        Ok(())
    }

    fn maintain_user_ordering(
        package_names: &[String],
        contracts: Option<&IndexMap<Box<str>, env_toml::Contract>>,
    ) -> Vec<String> {
        contracts.map_or_else(
            || package_names.to_vec(),
            |contracts| {
                let mut reordered: Vec<String> = contracts
                    .keys()
                    .filter_map(|contract_name| {
                        package_names
                            .iter()
                            .find(|&name| name == contract_name.as_ref())
                            .cloned()
                    })
                    .collect();

                reordered.extend(
                    package_names
                        .iter()
                        .filter(|name| !contracts.contains_key(name.as_str()))
                        .cloned(),
                );

                reordered
            },
        )
    }

    async fn handle_production_contracts(
        &self,
        contracts: &IndexMap<Box<str>, env_toml::Contract>,
    ) -> Result<(), Error> {
        for (name, contract) in contracts.iter().filter(|(_, settings)| settings.client) {
            if let Some(id) = &contract.id {
                if stellar_strkey::Contract::from_string(id).is_err() {
                    return Err(Error::InvalidContractID(id.to_string()));
                }
                self.clone()
                    .generate_contract_bindings(name, &id.to_string())
                    .await?;
            } else {
                return Err(Error::MissingContractID(name.to_string()));
            }
        }
        Ok(())
    }

    async fn handle_contracts(
        &self,
        contracts: Option<&IndexMap<Box<str>, env_toml::Contract>>,
        package_names: Vec<String>,
        network: &Network,
    ) -> Result<(), Error> {
        let printer = self.printer();
        if package_names.is_empty() {
            return Ok(());
        }

        let env = self.stellar_scaffold_env(ScaffoldEnv::Production);
        if env == "production" || env == "staging" {
            if let Some(contracts) = contracts {
                self.handle_production_contracts(contracts).await?;
            }
            return Ok(());
        }

        self.validate_contract_names(contracts)?;

        let names = Self::maintain_user_ordering(&package_names, contracts);

        let mut results: Vec<(String, Result<(), String>)> = Vec::new();

        for name in names {
            let settings = contracts
                .and_then(|contracts| contracts.get(name.as_str()))
                .cloned()
                .unwrap_or_default();

            // Skip if client generation is disabled
            if !settings.client {
                continue;
            }

            match self
                .process_single_contract(&name, settings, network, &env)
                .await
            {
                Ok(()) => {
                    printer.checkln(format!("Successfully generated client for: {name}"));
                    results.push((name, Ok(())));
                }
                Err(e) => {
                    printer.errorln(format!("Failed to generate client for: {name}"));
                    results.push((name, Err(e.to_string())));
                }
            }
        }

        // Partition results into successes and failures
        let (successes, failures): (Vec<_>, Vec<_>) =
            results.into_iter().partition(|(_, result)| result.is_ok());

        // Print summary
        printer.infoln("Client Generation Summary:");
        printer.blankln(format!("Successfully processed: {}", successes.len()));
        printer.blankln(format!("Failed: {}", failures.len()));

        if !failures.is_empty() {
            printer.infoln("Failures:");
            for (name, result) in &failures {
                if let Err(e) = result {
                    printer.blankln(format!("{name}: {e}"));
                }
            }
        }

        Ok(())
    }

    fn get_wasm_path(&self, contract_name: &str) -> std::path::PathBuf {
        // Check if out_dir was specified and use it, otherwise fall back to target directory
        if let Some(out_dir) = &self.out_dir {
            out_dir.join(format!("{contract_name}.wasm"))
        } else {
            let workspace_root = self
                .workspace_root
                .as_ref()
                .expect("workspace_root not set");
            let target_dir = workspace_root.join("target");
            stellar_build::stellar_wasm_out_file(&target_dir, contract_name)
        }
    }

    fn validate_contract_names(
        &self,
        contracts: Option<&IndexMap<Box<str>, env_toml::Contract>>,
    ) -> Result<(), Error> {
        if let Some(contracts) = contracts {
            for (name, _) in contracts.iter().filter(|(_, settings)| settings.client) {
                let wasm_path = self.get_wasm_path(name);
                if !wasm_path.exists() {
                    return Err(Error::BadContractName(name.to_string()));
                }
            }
        }
        Ok(())
    }

    fn get_package_dir(&self, name: &str) -> Result<std::path::PathBuf, Error> {
        let workspace_root = self
            .workspace_root
            .as_ref()
            .expect("workspace_root must be set before running");
        let package_dir = workspace_root.join(format!("packages/{name}"));
        if !package_dir.exists() {
            return Err(Error::BadContractName(name.to_string()));
        }
        Ok(package_dir)
    }

    async fn process_single_contract(
        &self,
        name: &str,
        settings: env_toml::Contract,
        network: &Network,
        env: &str,
    ) -> Result<(), Error> {
        let printer = self.printer();
        // First check if we have an ID in settings
        let contract_id = if let Some(id) = &settings.id {
            Contract::from_string(id).map_err(|_| Error::InvalidContractID(id.clone()))?
        } else {
            let wasm_path = self.get_wasm_path(name);
            if !wasm_path.exists() {
                return Err(Error::BadContractName(name.to_string()));
            }
            let new_hash = self.upload_contract_wasm(name, &wasm_path).await?;
            let mut upgraded_contract = None;

            // Check existing alias - if it exists and matches hash, we can return early
            if let Some(existing_contract_id) = self.get_contract_alias(name)? {
                let hash = self
                    .get_contract_hash(&existing_contract_id, network)
                    .await?;
                if let Some(current_hash) = hash {
                    if current_hash == new_hash {
                        printer.checkln(format!("Contract {name:?} is up to date"));
                        // If there is not a package at packages/<name>, generate bindings
                        if self.get_package_dir(name).is_err() {
                            self.generate_contract_bindings(
                                name,
                                &existing_contract_id.to_string(),
                            )
                            .await?;
                        }
                        return Ok(());
                    }
                    upgraded_contract = self
                        .try_upgrade_contract(
                            name,
                            existing_contract_id,
                            &current_hash,
                            &new_hash,
                            network,
                        )
                        .await?;
                }
                printer.infoln(format!("Updating contract {name:?}"));
            }

            // Deploy new contract if we got here (don't deploy if we already run an upgrade)
            let contract_id = if let Some(upgraded) = upgraded_contract {
                upgraded
            } else {
                self.deploy_contract(name, &new_hash, &settings).await?
            };
            // Run after_deploy script if in development or test environment
            if let Some(after_deploy) = settings.after_deploy.as_deref() {
                if env == "development" || env == "testing" {
                    printer.infoln(format!("Running after_deploy script for {name:?}"));
                    self.run_after_deploy_script(name, &contract_id, after_deploy)
                        .await?;
                }
            }
            self.save_contract_alias(name, &contract_id, network)?;
            contract_id
        };

        self.generate_contract_bindings(name, &contract_id.to_string())
            .await?;

        Ok(())
    }

    async fn upload_contract_wasm(
        &self,
        name: &str,
        wasm_path: &std::path::Path,
    ) -> Result<String, Error> {
        let printer = self.printer();
        printer.infoln(format!("Installing {name:?} wasm bytecode on-chain..."));
        let hash = cli::contract::upload::Cmd::parse_arg_vec(&[
            "--wasm",
            wasm_path
                .to_str()
                .expect("we do not support non-utf8 paths"),
            "--config-dir",
            self.get_config_dir()
                .to_str()
                .expect("we do not support non-utf8 paths"),
        ])?
        .run_against_rpc_server(self.global_args.as_ref(), None)
        .await?
        .into_result()
        .expect("no hash returned by 'contract upload'")
        .to_string();
        printer.infoln(format!("    ↳ hash: {hash}"));
        Ok(hash)
    }

    fn parse_script_line(line: &str) -> Result<(Option<String>, Vec<String>), Error> {
        let re = Regex::new(r"\$\((.*?)\)").expect("Invalid regex pattern");
        let (shell, flag) = if cfg!(windows) {
            ("cmd", "/C")
        } else {
            ("sh", "-c")
        };

        let resolved_line = Self::resolve_line(&re, line, shell, flag)?;
        let parts = split(&resolved_line)
            .ok_or_else(|| Error::ScriptParseFailure(resolved_line.to_string()))?;

        let (source_account, command_parts): (Vec<_>, Vec<_>) = parts
            .iter()
            .partition(|&part| part.starts_with("STELLAR_ACCOUNT="));

        let source = source_account.first().map(|account| {
            account
                .strip_prefix("STELLAR_ACCOUNT=")
                .unwrap()
                .to_string()
        });

        Ok((
            source,
            command_parts.iter().map(|s| (*s).to_string()).collect(),
        ))
    }

    async fn deploy_contract(
        &self,
        name: &str,
        hash: &str,
        settings: &env_toml::Contract,
    ) -> Result<Contract, Error> {
        let printer = self.printer();
        let mut deploy_args = vec![
            "--alias".to_string(),
            name.to_string(),
            "--wasm-hash".to_string(),
            hash.to_string(),
            "--config-dir".to_string(),
            self.get_config_dir()
                .to_str()
                .expect("we do not support non-utf8 paths")
                .to_string(),
        ];

        if let Some(constructor_script) = &settings.constructor_args {
            let (source_account, mut args) = Self::parse_script_line(constructor_script)?;

            if let Some(account) = source_account {
                deploy_args.extend_from_slice(&["--source-account".to_string(), account]);
            }

            deploy_args.push("--".to_string());
            deploy_args.append(&mut args);
        }

        printer.infoln(format!("Instantiating {name:?} smart contract"));
        let deploy_arg_refs: Vec<&str> = deploy_args
            .iter()
            .map(std::string::String::as_str)
            .collect();
        let contract_id = cli::contract::deploy::wasm::Cmd::parse_arg_vec(&deploy_arg_refs)?
            .run_against_rpc_server(self.global_args.as_ref(), None)
            .await?
            .into_result()
            .expect("no contract id returned by 'contract deploy'");
        printer.infoln(format!("    ↳ contract_id: {contract_id}"));

        Ok(contract_id)
    }

    async fn try_upgrade_contract(
        &self,
        name: &str,
        existing_contract_id: Contract,
        existing_hash: &str,
        hash: &str,
        network: &Network,
    ) -> Result<Option<Contract>, Error> {
        let printer = self.printer();
        let existing_spec = fetch_contract_spec(existing_hash, network).await?;
        let spec_to_upgrade = fetch_contract_spec(hash, network).await?;
        let Some(legacy_upgradeable) = Self::is_legacy_upgradeable(existing_spec) else {
            return Ok(None);
        };

        if Self::is_legacy_upgradeable(spec_to_upgrade).is_none() {
            printer.warnln("New WASM is not upgradable. Contract will be redeployed instead of being upgraded.");
            return Ok(None);
        }

        printer
            .infoln("Upgradable contract found, will use 'upgrade' function instead of redeploy");

        let existing_contract_id_str = existing_contract_id.to_string();
        let mut redeploy_args = vec![
            "--id",
            existing_contract_id_str.as_str(),
            "--",
            "upgrade",
            "--new_wasm_hash",
            hash,
        ];

        let invoke_cmd = if legacy_upgradeable {
            let upgrade_operator = ArgParser::get_upgrade_args(name).map_err(UpgradeArgsError)?;
            redeploy_args.push("--operator");
            redeploy_args.push(&upgrade_operator);
            cli::contract::invoke::Cmd::parse_arg_vec(&redeploy_args)
        } else {
            cli::contract::invoke::Cmd::parse_arg_vec(&redeploy_args)
        }?;
        printer.infoln(format!("Upgrading {name:?} smart contract"));
        invoke_cmd
            .run_against_rpc_server(self.global_args.as_ref(), None)
            .await?
            .into_result()
            .expect("no result returned by 'contract invoke'");
        printer.infoln(format!("Contract upgraded: {existing_contract_id}"));

        Ok(Some(existing_contract_id))
    }

    /// Returns: none if not upgradable, Some(true) if legacy upgradeable, Some(false) if new upgradeable
    fn is_legacy_upgradeable(spec: Vec<ScSpecEntry>) -> Option<bool> {
        spec.iter()
            .filter_map(|x| if let FunctionV0(e) = x { Some(e) } else { None })
            .filter(|x| x.name.to_string() == "upgrade")
            .find(|x| {
                x.inputs
                    .iter()
                    .any(|y| matches!(y.type_, ScSpecTypeDef::BytesN(ScSpecTypeBytesN { n: 32 })))
            })
            .map(|x| x.inputs.iter().any(|y| y.type_ == ScSpecTypeDef::Address))
    }

    fn resolve_line(re: &Regex, line: &str, shell: &str, flag: &str) -> Result<String, Error> {
        let mut result = String::new();
        let mut last_match = 0;
        for cap in re.captures_iter(line) {
            let whole_match = cap.get(0).unwrap();
            result.push_str(&line[last_match..whole_match.start()]);
            let cmd = &cap[1];
            let output = Self::execute_subcommand(shell, flag, cmd)?;
            result.push_str(&output);
            last_match = whole_match.end();
        }
        result.push_str(&line[last_match..]);
        Ok(result)
    }

    fn execute_subcommand(shell: &str, flag: &str, cmd: &str) -> Result<String, Error> {
        match Command::new(shell).arg(flag).arg(cmd).output() {
            Ok(output) => {
                let stdout = String::from_utf8_lossy(&output.stdout).trim().to_string();
                let stderr = String::from_utf8_lossy(&output.stderr).trim().to_string();

                if output.status.success() {
                    Ok(stdout)
                } else {
                    Err(Error::SubCommandExecutionFailure(cmd.to_string(), stderr))
                }
            }
            Err(e) => Err(Error::SubCommandExecutionFailure(
                cmd.to_string(),
                e.to_string(),
            )),
        }
    }

    async fn run_after_deploy_script(
        &self,
        name: &str,
        contract_id: &Contract,
        after_deploy_script: &str,
    ) -> Result<(), Error> {
        let printer = self.printer();
        let config_dir_path = self.get_config_dir();
        let config_dir = config_dir_path.to_str().unwrap();
        for line in after_deploy_script.lines() {
            let line = line.trim();
            if line.is_empty() {
                continue;
            }

            let (source_account, command_parts) = Self::parse_script_line(line)?;

            let contract_id_arg = contract_id.to_string();
            let mut args = vec!["--id", &contract_id_arg, "--config-dir", config_dir];
            if let Some(account) = source_account.as_ref() {
                args.extend_from_slice(&["--source-account", account]);
            }
            args.extend_from_slice(&["--"]);
            args.extend(command_parts.iter().map(std::string::String::as_str));

            printer.infoln(format!(
                "  ↳ Executing: stellar contract invoke {}",
                args.join(" ")
            ));
            let result = cli::contract::invoke::Cmd::parse_arg_vec(&args)?
                .run_against_rpc_server(self.global_args.as_ref(), None)
                .await?;
            printer.infoln(format!("  ↳ Result: {result:?}"));
        }
        printer.checkln(format!(
            "After deploy script for {name:?} completed successfully"
        ));
        Ok(())
    }
}

async fn fetch_contract_spec(
    wasm_hash: &str,
    network: &Network,
) -> Result<Vec<ScSpecEntry>, Error> {
    let fetched = fetch(
        &FetchArgs {
            wasm_hash: Some(wasm_hash.to_string()),
            network: network.into(),
            ..Default::default()
        },
        // Quiets the output of the fetch command
        &Print::new(true),
    )
    .await?;

    match fetched.contract {
        contract_spec::Contract::Wasm { wasm_bytes } => Ok(Spec::new(&wasm_bytes)?.spec),
        contract_spec::Contract::StellarAssetContract => unreachable!(),
    }
}

#[cfg(test)]
mod tests {
    use super::*;
    use tempfile::TempDir;

    #[test]
    fn test_get_package_dir() {
        let temp_dir = TempDir::new().unwrap();
        let package_path = temp_dir.path().join("packages/existing_package");
        std::fs::create_dir_all(&package_path).unwrap();
        let args = Args {
            env: Some(ScaffoldEnv::Development),
            workspace_root: Some(temp_dir.path().to_path_buf()),
            out_dir: None,
            global_args: None,
        };
        let result = args.get_package_dir("existing_package");
        assert!(result.is_ok());
        let path = result.unwrap();
        assert_eq!(path.file_name().unwrap(), "existing_package");
    }

    #[test]
    fn test_get_package_dir_nonexistent() {
        let args = Args {
            env: Some(ScaffoldEnv::Development),
            workspace_root: Some(std::path::PathBuf::from("tests/nonexistent_workspace")),
            out_dir: None,
            global_args: None,
        };
        let result = args.get_package_dir("nonexistent_package");
        assert!(result.is_err());
        if let Err(Error::BadContractName(name)) = result {
            assert_eq!(name, "nonexistent_package");
        } else {
            panic!("Expected BadContractName error");
        }
    }
}