fix: remove todo about cygwin path and add remaining windows env vars (…

…#802)

Cargo.toml

@@ -115,6 +115,7 @@ tui-input = { version = "0.8.0", optional = true }
 reflink-copy = "0.1.15"
 rayon = "1.10.0"
 patch = "0.7.0"
+regex = "1.10.4"
 
 [dev-dependencies]
 insta = { version = "1.38.0", features = ["yaml"] }

src/windows/env.rs

@@ -1,6 +1,43 @@
-use std::{collections::HashMap, path::Path};
+use std::{
+    collections::HashMap,
+    path::{Component::*, Path, Prefix::Disk},
+};
 
 use rattler_conda_types::Platform;
+use regex::Regex;
+
+fn get_drive_letter(path: &Path) -> Option<char> {
+    path.components().find_map(|component| match component {
+        Prefix(prefix_component) => match prefix_component.kind() {
+            Disk(letter) => Some(letter as char),
+            _ => None,
+        },
+        _ => None,
+    })
+}
+
+fn to_cygdrive(path: &Path) -> String {
+    if let Some(drive_letter) = get_drive_letter(path) {
+        // skip first component, which is the drive letter and the `\` after it
+        let rest = path.iter().skip(2);
+        format!(
+            "/cygdrive/{}/{}",
+            drive_letter.to_lowercase(),
+            rest.map(|c| c.to_string_lossy())
+                .collect::<Vec<_>>()
+                .join("/")
+        )
+    } else {
+        // fallback to `c` if no drive letter is found
+        format!(
+            "/cygdrive/c/{}",
+            path.iter()
+                .map(|c| c.to_string_lossy())
+                .collect::<Vec<_>>()
+                .join("/")
+        )
+    }
+}
 
 pub fn default_env_vars(prefix: &Path, target_platform: &Platform) -> HashMap<String, String> {
     let win_arch = match target_platform {
@@ -39,8 +76,6 @@ pub fn default_env_vars(prefix: &Path, target_platform: &Platform) -> HashMap<St
         library_prefix.join("lib").to_string_lossy().to_string(),
     );
 
-    // vars.insert("CYGWIN_PREFIX", "".join(("/cygdrive/", drive.lower(), tail.replace("\\", "/"))));
-
     let default_vars = vec![
         "ALLUSERSPROFILE",
         "APPDATA",
@@ -87,16 +122,27 @@ pub fn default_env_vars(prefix: &Path, target_platform: &Platform) -> HashMap<St
         std::env::var("BUILD").unwrap_or_else(|_| format!("{}-pc-windows-{}", win_arch, win_msvc)),
     );
 
-    // TODO
-    // get_default(
-    //     "CYGWIN_PREFIX", "".join(("/cygdrive/", drive.lower(), tail.replace("\\", "/")))
-    // )
+    vars.insert("CYGWIN_PREFIX".to_string(), to_cygdrive(prefix));
 
-    // for k in os.environ.keys():
-    //     if re.match("VS[0-9]{2,3}COMNTOOLS", k):
-    //         get_default(k)
-    //     elif re.match("VS[0-9]{4}INSTALLDIR", k):
-    //         get_default(k)
+    let re_vs_comntools = Regex::new(r"^VS[0-9]{2,3}COMNTOOLS$").unwrap();
+    let re_vs_installdir = Regex::new(r"^VS[0-9]{4}INSTALLDIR$").unwrap();
+
+    for (key, val) in std::env::vars() {
+        if re_vs_comntools.is_match(&key) || re_vs_installdir.is_match(&key) {
+            vars.insert(key, val);
+        }
+    }
 
     vars
 }
+
+#[cfg(test)]
+mod test {
+    #[cfg(target_os = "windows")]
+    #[test]
+    fn test_cygdrive() {
+        let path = std::path::Path::new("C:\\Users\\user\\Documents");
+        let cygdrive = super::to_cygdrive(path);
+        assert_eq!(cygdrive, "/cygdrive/c/Users/user/Documents");
+    }
+}