feat: implement support for --cwd flag

packages/zpm-switch/src/commands/mod.rs

@@ -29,7 +29,7 @@ pub async fn run_default() -> ExitCode {
         cwd,
         args,
         version,
-    } = extract_bin_meta();
+    } = extract_bin_meta().expect("Failed to extract binary metadata");
 
     if let Some(cwd) = cwd {
         set_fake_cwd(cwd);

packages/zpm-switch/src/commands/proxy.rs

@@ -3,7 +3,7 @@ use std::process::ExitStatus;
 use clipanion::cli;
 use zpm_utils::ToFileString;
 
-use crate::{cwd::{get_fake_cwd, get_final_cwd}, errors::Error, manifest::{find_closest_package_manager, validate_package_manager}, yarn::get_default_yarn_version, yarn_enums::ReleaseLine};
+use crate::{cwd::{get_final_cwd, restore_args}, errors::Error, manifest::{find_closest_package_manager, validate_package_manager}, yarn::get_default_yarn_version, yarn_enums::ReleaseLine};
 
 use super::switch::explicit::ExplicitCommand;
 
@@ -39,9 +39,7 @@ impl ProxyCommand {
             = self.args.clone();
 
         // Don't forget to add back the cwd parameter that was removed earlier on!
-        if let Some(cwd) = get_fake_cwd() {
-            args.insert(0, cwd.to_file_string());
-        }
+        restore_args(&mut args);
 
         ExplicitCommand::run(&reference, &args).await
     }

packages/zpm-switch/src/commands/switch/explicit.rs

@@ -3,7 +3,7 @@ use std::process::{Command, ExitStatus, Stdio};
 use clipanion::cli;
 use zpm_utils::ToFileString;
 
-use crate::{cwd::{get_fake_cwd, get_final_cwd}, errors::Error, install::install_package_manager, manifest::{find_closest_package_manager, PackageManagerReference, VersionPackageManagerReference}, yarn::resolve_selector, yarn_enums::Selector};
+use crate::{cwd::{get_final_cwd, restore_args}, errors::Error, install::install_package_manager, manifest::{find_closest_package_manager, PackageManagerReference, VersionPackageManagerReference}, yarn::resolve_selector, yarn_enums::Selector};
 
 #[cli::command(proxy)]
 #[cli::path("switch")]
@@ -44,9 +44,7 @@ impl ExplicitCommand {
             = self.args.clone();
 
         // Don't forget to add back the cwd parameter that was removed earlier on!
-        if let Some(cwd) = get_fake_cwd() {
-            args.insert(0, cwd.to_file_string());
-        }
+        restore_args(&mut args);
 
         let version
             = resolve_selector(&self.selector).await?;

packages/zpm-switch/src/cwd.rs

@@ -6,7 +6,7 @@
 
 use std::sync::Mutex;
 
-use zpm_utils::{Path, PathError};
+use zpm_utils::{Path, PathError, ToFileString};
 
 static FAKE_CWD: Mutex<Option<Path>> = Mutex::new(None);
 
@@ -25,3 +25,10 @@ pub fn get_final_cwd() -> Result<Path, PathError> {
         Path::current_dir()
     }
 }
+
+pub fn restore_args(args: &mut Vec<String>) {
+    if let Some(cwd) = get_fake_cwd() {
+        // We add an explicit `--cwd` so that both implicit and explicit cwd arguments are correctly forwarded.
+        args.insert(0, format!("--cwd={}", cwd.to_file_string()));
+    }
+}

packages/zpm-switch/src/yarn.rs

@@ -2,7 +2,7 @@ use serde::Deserialize;
 use serde_with::serde_as;
 use std::{collections::BTreeMap, str::FromStr};
 use zpm_semver::{Range, Version};
-use zpm_utils::{ExplicitPath, FromFileString, Path, ToFileString};
+use zpm_utils::{ExplicitPath, FromFileString, Path, PathError, RawPath, ToFileString};
 
 use crate::{errors::Error, http::fetch, manifest::{PackageManagerReference, VersionPackageManagerReference}, yarn_enums::{ChannelSelector, Selector}};
 
@@ -105,29 +105,53 @@ pub fn get_bin_version() -> String {
         .to_string()
 }
 
-pub fn extract_bin_meta() -> BinMeta {
+const CWD_FLAG: &str = "--cwd";
+const CWD_FLAG_EQUALS: &str = "--cwd=";
+
+pub fn extract_bin_meta() -> Result<BinMeta, PathError> {
     let mut cwd = None;
 
     let mut args = std::env::args()
         .skip(1)
         .collect::<Vec<_>>();
 
+    // TODO: Use clipanion to error on incorrect placement of `--cwd` argument.
+    if args.len() >= 2 && args[0] == CWD_FLAG {
+        let raw_path
+            = RawPath::from_str(&args[1])?;
+
+        cwd = Some(raw_path.path);
+        args.drain(..2);
+    } else if args.len() >= 1 && args[0].starts_with(CWD_FLAG_EQUALS) {
+        let raw_path
+            = RawPath::from_str(&args[0][CWD_FLAG_EQUALS.len()..])?;
+
+        cwd = Some(raw_path.path);
+        args.remove(0);
+    }
+
     if let Some(first_arg) = args.first() {
         let explicit_path
             = ExplicitPath::from_str(first_arg);
 
-        if let Ok(explicit_path) = explicit_path {
-            cwd = Some(explicit_path.raw_path.path);
-            args.remove(0);
+        match explicit_path {
+            Ok(explicit_path) => {
+                cwd = Some(explicit_path.raw_path.path);
+                args.remove(0);
+            },
+            Err(PathError::InvalidExplicitPathParameter(_)) => {
+                // This is not a valid explicit path, so we don't modify `cwd`.
+            },
+            Err(err) => return Err(err),
         }
     }
 
     let version
         = get_bin_version();
 
-    BinMeta {
+    Ok(BinMeta {
         cwd,
         args,
         version,
-    }
+    })
 }

packages/zpm/src/commands/mod.rs

@@ -62,7 +62,7 @@ pub fn run_default() -> ExitCode {
         cwd,
         args,
         version,
-    } = extract_bin_meta();
+    } = extract_bin_meta().expect("Failed to extract binary metadata");
 
     if let Some(cwd) = cwd {
         cwd.sys_set_current_dir()