#![allow(unknown_lints)]
#![allow(clippy::literal_string_with_formatting_args)]
use crate::config::Settings;
use std::borrow::Cow;
use std::fmt::Display;
use indoc::formatdoc;
use crate::shell::{self, ActivateOptions, Shell};
#[derive(Default)]
pub struct Pwsh {}
impl Pwsh {}
impl Shell for Pwsh {
fn activate(&self, opts: ActivateOptions) -> String {
let exe = opts.exe;
let flags = opts.flags;
let exe = exe.to_string_lossy();
let mut out = String::new();
out.push_str(&shell::build_deactivation_script(self));
out.push_str(&self.format_activate_prelude(&opts.prelude));
out.push_str(&formatdoc! {r#"
$env:MISE_SHELL = 'pwsh'
if (-not (Test-Path -Path Env:/__MISE_ORIG_PATH)) {{
$env:__MISE_ORIG_PATH = $env:PATH
}}
function mise {{
[CmdletBinding()]
param(
[Parameter(ValueFromRemainingArguments=$true)] # Allow any number of arguments, including none
[string[]] $arguments
)
$previous_out_encoding = $OutputEncoding
$previous_console_out_encoding = [Console]::OutputEncoding
$OutputEncoding = [Console]::OutputEncoding = [Text.UTF8Encoding]::UTF8
function _reset_output_encoding {{
$OutputEncoding = $previous_out_encoding
[Console]::OutputEncoding = $previous_console_out_encoding
}}
if ($arguments.count -eq 0) {{
& "{exe}"
_reset_output_encoding
return
}} elseif ($arguments -contains '-h' -or $arguments -contains '--help') {{
& "{exe}" @arguments
_reset_output_encoding
return
}}
$command = $arguments[0]
if ($arguments.Length -gt 1) {{
$remainingArgs = $arguments[1..($arguments.Length - 1)]
}} else {{
$remainingArgs = @()
}}
switch ($command) {{
{{ $_ -in 'deactivate', 'shell', 'sh' }} {{
& "{exe}" $command @remainingArgs | Out-String | Invoke-Expression -ErrorAction SilentlyContinue
_reset_output_encoding
}}
default {{
& "{exe}" $command @remainingArgs
$status = $LASTEXITCODE
if ($(Test-Path -Path Function:\_mise_hook)){{
_mise_hook
}}
_reset_output_encoding
# Pass down exit code from mise after _mise_hook
if ($PSVersionTable.PSVersion.Major -ge 7) {{
pwsh -NoProfile -Command exit $status
}} else {{
powershell -NoProfile -Command exit $status
}}
}}
}}
}}
"#});
if !opts.no_hook_env {
out.push_str(&formatdoc! {r#"
function Global:_mise_hook {{
if ($env:MISE_SHELL -eq "pwsh"){{
$output = & "{exe}" hook-env{flags} $args -s pwsh | Out-String
if ($output -and $output.Trim()) {{
$output | Invoke-Expression
}}
}}
}}
function __enable_mise_chpwd{{
if ($PSVersionTable.PSVersion.Major -lt 7) {{
if ($env:MISE_PWSH_CHPWD_WARNING -ne '0') {{
Write-Warning "mise: chpwd functionality requires PowerShell version 7 or higher. Your current version is $($PSVersionTable.PSVersion). You can add `$env:MISE_PWSH_CHPWD_WARNING=0` to your environment to disable this warning."
}}
return
}}
if (-not $__mise_pwsh_chpwd){{
$Global:__mise_pwsh_chpwd= $true
$_mise_chpwd_hook = [EventHandler[System.Management.Automation.LocationChangedEventArgs]] {{
param([object] $source, [System.Management.Automation.LocationChangedEventArgs] $eventArgs)
end {{
_mise_hook
}}
}};
$__mise_pwsh_previous_chpwd_function=$ExecutionContext.SessionState.InvokeCommand.LocationChangedAction;
if ($__mise_original_pwsh_chpwd_function) {{
$ExecutionContext.SessionState.InvokeCommand.LocationChangedAction = [Delegate]::Combine($__mise_pwsh_previous_chpwd_function, $_mise_chpwd_hook)
}}
else {{
$ExecutionContext.SessionState.InvokeCommand.LocationChangedAction = $_mise_chpwd_hook
}}
}}
}}
__enable_mise_chpwd
Remove-Item -ErrorAction SilentlyContinue -Path Function:/__enable_mise_chpwd
function __enable_mise_prompt {{
if (-not $__mise_pwsh_previous_prompt_function){{
$Global:__mise_pwsh_previous_prompt_function=$function:prompt
function global:prompt {{
if (Test-Path -Path Function:\_mise_hook){{
_mise_hook
}}
& $__mise_pwsh_previous_prompt_function
}}
}}
}}
__enable_mise_prompt
Remove-Item -ErrorAction SilentlyContinue -Path Function:/__enable_mise_prompt
_mise_hook
"#});
}
if Settings::get().not_found_auto_install {
out.push_str(&formatdoc! {r#"
if (-not $__mise_pwsh_command_not_found){{
$Global:__mise_pwsh_command_not_found= $true
function __enable_mise_command_not_found {{
$_mise_pwsh_cmd_not_found_hook = [EventHandler[System.Management.Automation.CommandLookupEventArgs]] {{
param([object] $Name, [System.Management.Automation.CommandLookupEventArgs] $eventArgs)
end {{
if ([Microsoft.PowerShell.PSConsoleReadLine]::GetHistoryItems()[-1].CommandLine -match ([regex]::Escape($Name))) {{
if (& "{exe}" hook-not-found -s pwsh -- $Name){{
_mise_hook
if (Get-Command $Name -ErrorAction SilentlyContinue){{
$EventArgs.Command = Get-Command $Name
$EventArgs.StopSearch = $true
}}
}}
}}
}}
}}
$current_command_not_found_function = $ExecutionContext.SessionState.InvokeCommand.CommandNotFoundAction
if ($current_command_not_found_function) {{
$ExecutionContext.SessionState.InvokeCommand.CommandNotFoundAction = [Delegate]::Combine($current_command_not_found_function, $_mise_pwsh_cmd_not_found_hook)
}}
else {{
$ExecutionContext.SessionState.InvokeCommand.CommandNotFoundAction = $_mise_pwsh_cmd_not_found_hook
}}
}}
__enable_mise_command_not_found
Remove-Item -ErrorAction SilentlyContinue -Path Function:/__enable_mise_command_not_found
}}
"#});
}
out
}
fn deactivate(&self) -> String {
formatdoc! {r#"
Remove-Item -ErrorAction SilentlyContinue function:mise
Remove-Item -ErrorAction SilentlyContinue -Path Env:/MISE_SHELL
Remove-Item -ErrorAction SilentlyContinue -Path Env:/__MISE_DIFF
Remove-Item -ErrorAction SilentlyContinue -Path Env:/__MISE_SESSION
"#}
}
fn set_env(&self, k: &str, v: &str) -> String {
let k = powershell_escape(k.into());
let v = powershell_escape(v.into());
format!("$Env:{k}='{v}'\n")
}
fn prepend_env(&self, k: &str, v: &str) -> String {
let k = powershell_escape(k.into());
let v = powershell_escape(v.into());
format!("$Env:{k}='{v}'+[IO.Path]::PathSeparator+$env:{k}\n")
}
fn unset_env(&self, k: &str) -> String {
let k = powershell_escape(k.into());
format!("Remove-Item -ErrorAction SilentlyContinue -Path Env:/{k}\n")
}
}
impl Display for Pwsh {
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
write!(f, "pwsh")
}
}
fn powershell_escape(s: Cow<str>) -> Cow<str> {
let needs_escape = s.is_empty();
if !needs_escape {
return s;
}
let mut es = String::with_capacity(s.len());
let mut chars = s.chars().peekable();
loop {
match chars.next() {
Some('\t') => {
es.push_str("`t");
}
Some('\n') => {
es.push_str("`n");
}
Some('\r') => {
es.push_str("`r");
}
Some('\'') => {
es.push_str("`'");
}
Some('`') => {
es.push_str("``");
}
Some(c) => {
es.push(c);
}
None => {
break;
}
}
}
es.into()
}
#[cfg(test)]
mod tests {
use insta::assert_snapshot;
use std::path::Path;
use test_log::test;
use crate::test::replace_path;
use super::*;
#[test]
fn test_activate() {
unsafe {
std::env::remove_var("__MISE_ORIG_PATH");
std::env::remove_var("__MISE_DIFF");
}
let pwsh = Pwsh::default();
let exe = Path::new("/some/dir/mise");
let opts = ActivateOptions {
exe: exe.to_path_buf(),
flags: " --status".into(),
no_hook_env: false,
prelude: vec![],
};
assert_snapshot!(pwsh.activate(opts));
}
#[test]
fn test_set_env() {
assert_snapshot!(Pwsh::default().set_env("FOO", "1"));
}
#[test]
fn test_prepend_env() {
let pwsh = Pwsh::default();
assert_snapshot!(replace_path(&pwsh.prepend_env("PATH", "/some/dir:/2/dir")));
}
#[test]
fn test_unset_env() {
assert_snapshot!(Pwsh::default().unset_env("FOO"));
}
#[test]
fn test_deactivate() {
let deactivate = Pwsh::default().deactivate();
assert_snapshot!(replace_path(&deactivate));
}
}