feat(cli/self-update): add support for PowerShell on Unix systems

Author: paxbun

src/cli/self_update.rs

@@ -401,6 +401,7 @@ This is usually done by running one of the following (note the leading DOT):
     source "{cargo_home}/env.fish"  # For fish
     source $"{cargo_home_nushell}/env.nu"  # For nushell
     source "{cargo_home}/env.tcsh"  # For tcsh
+    . "{cargo_home}/env.ps1"        # For pwsh
 "#
     };
 }

src/cli/self_update/env.ps1

@@ -0,0 +1,4 @@
+# rustup shell setup
+if (-not ":${env:PATH}:".Contains(":{cargo_bin}:")) {
+    ${env:PATH} = "{cargo_bin}:${env:PATH}";
+}

src/cli/self_update/shell.rs

@@ -58,7 +58,7 @@ fn cargo_home_str_with_home(home: &str, process: &Process) -> Result<Cow<'static
 }
 
 // TODO: Tcsh (BSD)
-// TODO?: Make a decision on Ion Shell, Power Shell, Nushell
+// TODO?: Make a decision on Ion Shell
 // Cross-platform non-POSIX shells have not been assessed for integration yet
 fn enumerate_shells() -> Vec<Shell> {
     vec![
@@ -68,6 +68,7 @@ fn enumerate_shells() -> Vec<Shell> {
         Box::new(Fish),
         Box::new(Nu),
         Box::new(Tcsh),
+        Box::new(Pwsh),
     ]
 }
 
@@ -368,6 +369,82 @@ impl UnixShell for Tcsh {
     }
 }
 
+struct Pwsh;
+
+impl UnixShell for Pwsh {
+    fn does_exist(&self, process: &Process) -> bool {
+        matches!(process.var("SHELL"), Ok(sh) if sh.contains("pwsh"))
+            || utils::find_cmd(&["pwsh"], process).is_some()
+    }
+
+    fn rcfiles(&self, process: &Process) -> Vec<PathBuf> {
+        let mut paths = vec![];
+
+        let Some(mut config_dir) = process.home_dir() else {
+            return paths;
+        };
+        config_dir.extend([".config", "powershell"]);
+
+        // PowerShell provides many kinds of user-specific and host-specific
+        // profile files. When the system has multiple profiles, PowerShell
+        // executes them in a defined order.
+        //
+        // For more details, please refer to:
+        // https://learn.microsoft.com/en-us/powershell/module/microsoft.powershell.core/about/about_profiles
+        //
+        // `~/.config/powershell/profile.ps1` is the "Current User, All Hosts"
+        // profile file. It affects all PowerShell hosts of the current user.
+        paths.push(config_dir.join("profile.ps1"));
+
+        let Ok(config_dir) = config_dir.read_dir() else {
+            return paths;
+        };
+
+        // Some editors like Visual Studio Code or PowerShell ISE use their
+        // own dedicated profile files, whose file names are
+        // `<Host Profile ID>_profile.ps1`. Such customization may use
+        // PowerShell Editor Services for IDE integration.
+        // https://github.com/PowerShell/PowerShellEditorServices
+        for host_profile in config_dir {
+            let Ok(host_profile) = host_profile else {
+                continue;
+            };
+            let host_profile_path = host_profile.path();
+            if !host_profile_path.is_file() {
+                continue;
+            }
+            if host_profile_path.ends_with("_profile.ps1") {
+                paths.push(host_profile_path);
+            }
+        }
+
+        paths
+    }
+
+    fn update_rcs(&self, process: &Process) -> Vec<PathBuf> {
+        let mut paths = vec![];
+        // Always modify the "Current User, All Hosts" profile.
+        let Some(mut profile) = process.home_dir() else {
+            return paths;
+        };
+
+        profile.extend([".config", "powershell", "profile.ps1"]);
+        paths.push(profile);
+        paths
+    }
+
+    fn env_script(&self) -> ShellScript {
+        ShellScript {
+            name: "env.ps1",
+            content: include_str!("env.ps1"),
+        }
+    }
+
+    fn source_string(&self, process: &Process) -> Result<String> {
+        Ok(format!(r#". "{}/env.ps1""#, self.cargo_home_str(process)?))
+    }
+}
+
 pub(crate) fn legacy_paths(process: &Process) -> impl Iterator<Item = PathBuf> + '_ {
     let zprofiles = Zsh::zdotdir(process)
         .into_iter()