23d9a5bf63
Despite the fact that Topgrade explicitly tries to run the Windows Powershell, gsudo has an odd behaviour which causes it to detect whether it's being run from Powershell for Cmd. If run from Powershell, It will launch the current powershell with the given command. If the current shell is Powershell, it will try to run the command with it. However, Get-WmiObject doesn't exist on Powershell core.
124 lines
3.8 KiB
Rust
124 lines
3.8 KiB
Rust
use crate::executor::{CommandExt, RunType};
|
|
use crate::powershell;
|
|
use crate::terminal::print_separator;
|
|
use crate::utils::require;
|
|
use crate::{error::SkipStep, steps::git::Repositories};
|
|
use crate::{execution_context::ExecutionContext, utils::require_option};
|
|
use anyhow::Result;
|
|
use log::debug;
|
|
use std::convert::TryFrom;
|
|
use std::path::Path;
|
|
use std::{ffi::OsStr, process::Command};
|
|
|
|
pub fn run_chocolatey(ctx: &ExecutionContext) -> Result<()> {
|
|
let choco = require("choco")?;
|
|
let yes = ctx.config().yes();
|
|
|
|
print_separator("Chocolatey");
|
|
|
|
let mut cmd = &choco;
|
|
let mut args = vec!["upgrade", "all"];
|
|
|
|
if let Some(sudo) = ctx.sudo() {
|
|
cmd = sudo;
|
|
args.insert(0, "choco");
|
|
}
|
|
|
|
let mut command = ctx.run_type().execute(&cmd);
|
|
|
|
command.args(&args);
|
|
|
|
if yes {
|
|
command.arg("--yes");
|
|
}
|
|
|
|
command.check_run()
|
|
}
|
|
|
|
pub fn run_scoop(cleanup: bool, run_type: RunType) -> Result<()> {
|
|
let scoop = require("scoop")?;
|
|
|
|
print_separator("Scoop");
|
|
|
|
run_type.execute(&scoop).args(&["update"]).check_run()?;
|
|
run_type.execute(&scoop).args(&["update", "*"]).check_run()?;
|
|
|
|
if cleanup {
|
|
run_type.execute(&scoop).args(&["cleanup", "*"]).check_run()?;
|
|
}
|
|
|
|
Ok(())
|
|
}
|
|
|
|
pub fn run_wsl_topgrade(ctx: &ExecutionContext) -> Result<()> {
|
|
let wsl = require("wsl")?;
|
|
let topgrade = Command::new(&wsl)
|
|
.args(&["bash", "-lc", "which topgrade"])
|
|
.check_output()
|
|
.map_err(|_| SkipStep(String::from("Could not find Topgrade installed in WSL")))?;
|
|
|
|
let mut command = ctx.run_type().execute(&wsl);
|
|
command
|
|
.args(&["bash", "-c"])
|
|
.arg(format!("TOPGRADE_PREFIX=WSL exec {}", topgrade));
|
|
|
|
if ctx.config().yes() {
|
|
command.arg("-y");
|
|
}
|
|
|
|
command.check_run()
|
|
}
|
|
|
|
pub fn windows_update(ctx: &ExecutionContext) -> Result<()> {
|
|
let powershell = powershell::Powershell::windows_powershell();
|
|
|
|
if powershell.supports_windows_update() {
|
|
print_separator("Windows Update");
|
|
return powershell.windows_update(ctx);
|
|
}
|
|
|
|
let usoclient = require("UsoClient")?;
|
|
|
|
print_separator("Windows Update");
|
|
println!("Running Windows Update. Check the control panel for progress.");
|
|
ctx.run_type().execute(&usoclient).arg("ScanInstallWait").check_run()?;
|
|
ctx.run_type().execute(&usoclient).arg("StartInstall").check_run()
|
|
}
|
|
|
|
pub fn upgrade_store_apps(ctx: &ExecutionContext) -> Result<()> {
|
|
let powershell = powershell::Powershell::windows_powershell();
|
|
|
|
let path = powershell.path().as_ref().unwrap();
|
|
let sudo = require_option(ctx.sudo().as_ref(), String::from("Sudo is required"))?;
|
|
print_separator("Microsoft Store");
|
|
println!("Updating Microsoft Store applications in the background");
|
|
|
|
ctx.run_type().execute("cmd").arg("/c").args(&[sudo, path]).args(&["-NoProfile", "-Command", "(Get-WmiObject -Namespace 'root\\cimv2\\mdm\\dmmap' -Class 'MDM_EnterpriseModernAppManagement_AppManagement01').UpdateScanMethod() > $null"]).check_run()
|
|
}
|
|
|
|
pub fn reboot() {
|
|
Command::new("shutdown").args(&["/R", "/T", "0"]).spawn().ok();
|
|
}
|
|
|
|
pub fn insert_startup_scripts(ctx: &ExecutionContext, git_repos: &mut Repositories) -> Result<()> {
|
|
let startup_dir = ctx
|
|
.base_dirs()
|
|
.data_dir()
|
|
.join("Microsoft\\Windows\\Start Menu\\Programs\\Startup");
|
|
for entry in std::fs::read_dir(&startup_dir)? {
|
|
if let Ok(entry) = entry {
|
|
let path = entry.path();
|
|
if path.extension().and_then(OsStr::to_str) == Some("lnk") {
|
|
if let Ok(lnk) = parselnk::Lnk::try_from(Path::new(&path)) {
|
|
debug!("Startup link: {:?}", lnk);
|
|
if let Some(path) = lnk.relative_path() {
|
|
git_repos.insert_if_repo(&startup_dir.join(path));
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
Ok(())
|
|
}
|