From f2e17c6064a4ffa240466eefc2c5b47ad60bd851 Mon Sep 17 00:00:00 2001
From: Sander <hey@sandydoo.me>
Date: Thu, 26 Sep 2024 17:50:52 +0400
Subject: [PATCH] devenv: escape shell vars in task exports

Fixes sourcing exported variables with spaces in them. Hello, macOS.
---
 devenv/src/tasks.rs | 28 +++++++++++++++++++++++++++-
 1 file changed, 27 insertions(+), 1 deletion(-)

diff --git a/devenv/src/tasks.rs b/devenv/src/tasks.rs
index 26e65875d..c91ef96a0 100644
--- a/devenv/src/tasks.rs
+++ b/devenv/src/tasks.rs
@@ -178,7 +178,11 @@ impl TaskState {
                     for (env_key, env_value) in env_obj {
                         if let Some(env_str) = env_value.as_str() {
                             command.env(env_key, env_str);
-                            devenv_env.push_str(&format!("export {}={}\n", env_key, env_str));
+                            devenv_env.push_str(&format!(
+                                "export {}={}\n",
+                                env_key,
+                                shell_escape(env_str)
+                            ));
                         }
                     }
                 }
@@ -836,6 +840,21 @@ impl TasksUi {
     }
 }
 
+/// Escape a shell variable by wrapping it in single quotes.
+/// Any single quotes within the variable are escaped.
+fn shell_escape(s: &str) -> String {
+    let mut escaped = String::with_capacity(s.len() + 2);
+    escaped.push('\'');
+    for c in s.chars() {
+        match c {
+            '\'' => escaped.push_str("'\\''"),
+            _ => escaped.push(c),
+        }
+    }
+    escaped.push('\'');
+    escaped
+}
+
 #[cfg(test)]
 mod test {
     use super::*;
@@ -846,6 +865,13 @@ mod test {
     use std::io::Write;
     use std::os::unix::fs::PermissionsExt;
 
+    #[test]
+    fn test_shell_escape() {
+        let escaped = shell_escape("foo'bar");
+        eprintln!("{escaped}");
+        assert_eq!(escaped, "'foo'\\''bar'");
+    }
+
     #[tokio::test]
     async fn test_task_name() -> Result<(), Error> {
         let invalid_names = vec![