feat(task): support NO_COLOR

https://no-color.org/

Author: fengmk2

crates/vite_task/src/execute.rs

@@ -179,6 +179,7 @@ const DEFAULT_PASSTHROUGH_ENVS: &[&str] = &[
     "TERM_PROGRAM",
     "DISPLAY",
     "FORCE_COLOR",
+    "NO_COLOR",
     // Temporary directories
     "TMP",
     "TEMP",
@@ -275,7 +276,7 @@ impl TaskEnvs {
         // Automatically add FORCE_COLOR environment variable if not already set
         // This enables color output in subprocesses when color is supported
         // TODO: will remove this temporarily until we have a better solution
-        if !all_envs.contains_key("FORCE_COLOR") {
+        if !all_envs.contains_key("FORCE_COLOR") && !all_envs.contains_key("NO_COLOR") {
             if let Some(support) = on(Stream::Stdout) {
                 let force_color_value = if support.has_16m {
                     "3" // True color (16 million colors)
@@ -528,6 +529,23 @@ mod tests {
         unsafe {
             std::env::remove_var("FORCE_COLOR");
         }
+
+        // Test when NO_COLOR is already set - should not be overridden
+        unsafe {
+            std::env::set_var("NO_COLOR", "1");
+        }
+
+        let result3 = TaskEnvs::resolve(base_dir, &resolved_task_config).unwrap();
+        assert!(result3.all_envs.contains_key("NO_COLOR"));
+        let no_color_value = result3.all_envs.get("NO_COLOR").unwrap();
+        assert_eq!(no_color_value.to_str().unwrap(), "1");
+        // FORCE_COLOR should not automatically added since NO_COLOR is set
+        assert!(!result3.all_envs.contains_key("FORCE_COLOR"));
+
+        // Clean up
+        unsafe {
+            std::env::remove_var("NO_COLOR");
+        }
     }
 
     #[test]

packages/cli/snap-tests/pass-no-color-env/check.js

@@ -0,0 +1 @@
+console.log('NO_COLOR=%s', process.env.NO_COLOR);

packages/cli/snap-tests/pass-no-color-env/package.json

@@ -0,0 +1,5 @@
+{
+  "scripts": {
+    "check": "node check.js"
+  }
+}

packages/cli/snap-tests/pass-no-color-env/snap.txt

@@ -0,0 +1,7 @@
+> NO_COLOR=1 vite-plus run check -- --foo # get NO_COLOR=1 from env
+Cache not found
+NO_COLOR=1
+
+> NO_COLOR=2 vite-plus run check -- --bar # get NO_COLOR=2 from env
+Cache not found
+NO_COLOR=2

packages/cli/snap-tests/pass-no-color-env/steps.json

@@ -0,0 +1,9 @@
+{
+  "env": {
+    "VITE_DISABLE_AUTO_INSTALL": "1"
+  },
+  "commands": [
+    "NO_COLOR=1 vite-plus run check -- --foo # get NO_COLOR=1 from env",
+    "NO_COLOR=2 vite-plus run check -- --bar # get NO_COLOR=2 from env"
+  ]
+}