feat: add nushell support

Author: LittleJianCH

src/cli/self_update.rs

@@ -384,6 +384,7 @@ the corresponding `env` file under {cargo_home}.
 This is usually done by running one of the following (note the leading DOT):
     . "{cargo_home}/env"            # For sh/bash/zsh/ash/dash/pdksh
     source "{cargo_home}/env.fish"  # For fish
+    source "{cargo_home}/env.nu"    # For nushell
 "#
     };
 }

src/cli/self_update/env.nu

@@ -0,0 +1,3 @@
+if ("{cargo_bin}" not-in ($env.Path | split row (char esep))) {
+  $env.Path = ($env.Path | prepend "{cargo_bin}")
+}

src/cli/self_update/shell.rs

@@ -77,6 +77,7 @@ fn enumerate_shells() -> Vec<Shell> {
         Box::new(Bash),
         Box::new(Zsh),
         Box::new(Fish),
+        Box::new(Nu),
     ]
 }
 
@@ -92,7 +93,7 @@ pub(crate) trait UnixShell {
     fn does_exist(&self, process: &Process) -> bool;
 
     // Gives all rcfiles of a given shell that Rustup is concerned with.
-    // Used primarily in checking rcfiles for cleanup.
+    // Used primarily in checking rcfiles c
     fn rcfiles(&self, process: &Process) -> Vec<PathBuf>;
 
     // Gives rcs that should be written to.
@@ -255,6 +256,60 @@ impl UnixShell for Fish {
     }
 }
 
+struct Nu;
+
+impl UnixShell for Nu {
+    fn does_exist(&self, process: &Process) -> bool {
+        // nu has to either be the shell or be callable for nu setup.
+        matches!(process.var("SHELL"), Ok(sh) if sh.contains("nu"))
+            || utils::find_cmd(&["nu"], process).is_some()
+    }
+
+    fn rcfiles(&self, process: &Process) -> Vec<PathBuf> {
+        let p0: Vec<_> = process
+            .var("XDG_CONFIG_HOME")
+            .ok()
+            .iter()
+            .flat_map(|p| {
+                let path = PathBuf::from(p).join("nushell/");
+                [path.join("env.nu"), path.join("config.nu")]
+            })
+            .collect();
+
+        let p1: Vec<_> = process
+            .home_dir()
+            .iter()
+            .flat_map(|p| {
+                let path = PathBuf::from(p).join(".config/nushell/");
+                [path.join("env.nu"), path.join("config.nu")]
+            })
+            .collect();
+
+        [p0, p1].concat()
+    }
+
+    fn update_rcs(&self, process: &Process) -> Vec<PathBuf> {
+        let rcs = self.rcfiles(process);
+        if rcs.len() == 4 {
+            // The first two rcfile takes precedence(XDG_CONFIG_HOME).
+            rcs[0..2].to_vec()
+        } else {
+            rcs
+        }
+    }
+
+    fn env_script(&self) -> ShellScript {
+        ShellScript {
+            name: "env.nu",
+            content: include_str!("env.nu"),
+        }
+    }
+
+    fn source_string(&self, process: &Process) -> Result<String> {
+        Ok(format!(r#"source "{}/env.nu""#, cargo_home_str(process)?))
+    }
+}
+
 pub(crate) fn legacy_paths(process: &Process) -> impl Iterator<Item = PathBuf> + '_ {
     let zprofiles = Zsh::zdotdir(process)
         .into_iter()