Support PowerShell core (fix #189)

src/steps/powershell.rs

@@ -0,0 +1,77 @@
+use crate::error::Error;
+#[cfg(windows)]
+use crate::error::ErrorKind;
+use crate::executor::{CommandExt, RunType};
+use crate::terminal::{is_dumb, print_separator};
+use crate::utils::{require_option, which, PathExt};
+use std::path::PathBuf;
+use std::process::Command;
+
+pub struct Powershell {
+    path: Option<PathBuf>,
+    profile: Option<PathBuf>,
+}
+
+impl Powershell {
+    /// Returns a powershell instance.
+    ///
+    /// If the powershell binary is not found, or the current terminal is dumb
+    /// then the instance of this struct will skip all the powershell steps.
+    pub fn new() -> Self {
+        let path = which("pwsh").or_else(|| which("powershell")).filter(|_| !is_dumb());
+
+        let profile = path.as_ref().and_then(|path| {
+            Command::new(path)
+                .args(&["-Command", "echo $profile"])
+                .check_output()
+                .map(|output| PathBuf::from(output.trim()))
+                .and_then(|p| p.require())
+                .ok()
+        });
+
+        Powershell { path, profile }
+    }
+
+    #[cfg(windows)]
+    pub fn has_module(powershell: &PathBuf, command: &str) -> bool {
+        || -> Result<(), Error> {
+            Command::new(&powershell)
+                .args(&["-Command", &format!("Get-Module -ListAvailable {}", command)])
+                .check_output()?;
+            Ok(())
+        }()
+        .is_ok()
+    }
+
+    pub fn profile(&self) -> Option<&PathBuf> {
+        self.profile.as_ref()
+    }
+
+    pub fn update_modules(&self, run_type: RunType) -> Result<(), Error> {
+        let powershell = require_option(self.path.as_ref())?;
+
+        print_separator("Powershell Modules Update");
+        run_type
+            .execute(&powershell)
+            .args(&["-Command", "Update-Module", "-v"])
+            .check_run()
+    }
+
+    #[cfg(windows)]
+    pub fn windows_update(&self, run_type: RunType) -> Result<(), Error> {
+        let powershell = require_option(self.path.as_ref())?;
+
+        if !Self::has_module(&powershell, "PSWindowsUpdate") {
+            Err(ErrorKind::SkipStep)?;
+        }
+        print_separator("Windows Update");
+
+        run_type
+            .execute(&powershell)
+            .args(&[
+                "-Command",
+                "Import-Module PSWindowsUpdate; Install-WindowsUpdate -MicrosoftUpdate -AcceptAll -Verbose",
+            ])
+            .check_run()
+    }
+}