PSCompletions.psm1
|
Set-Item -Path Function:$($PSCompletions.config.function_name) -Option Constant -Value { $arg = $args Set-Alias Write-Host Microsoft.PowerShell.Utility\Write-Host -ErrorAction SilentlyContinue Set-Alias Write-Output Microsoft.PowerShell.Utility\Write-Output -ErrorAction SilentlyContinue function _replace { param ($data, $separator = '') $data = $data -join $separator if ($data -notlike '*{{*') { return $data } $matches = [regex]::Matches($data, $PSCompletions.replace_pattern) foreach ($match in $matches) { $data = $data.Replace($match.Value, (Invoke-Expression $match.Groups[1].Value) -join $separator ) } if ($data -match $PSCompletions.replace_pattern) { (_replace $data) }else { return $data } } function download_list { $PSCompletions.ensure_dir($PSCompletions.path.temp) if (!(Test-Path $PSCompletions.path.completions_json)) { @{ list = @('psc') } | ConvertTo-Json -Compress | Out-File $PSCompletions.path.completions_json -Encoding utf8 -Force } $current_list = ($PSCompletions.get_raw_content($PSCompletions.path.completions_json) | ConvertFrom-Json).list $isErr = $true $params = @{ ErrorAction = 'Stop' } if ($PSEdition -eq 'Core') { $params['OperationTimeoutSeconds'] = 30 } else { $params['TimeoutSec'] = 30 } $errMsg = @() foreach ($url in $PSCompletions.urls) { $params['Uri'] = "$url/completions.json" try { $response = Invoke-RestMethod @params } catch { $errMsg += $_.Exception.Message continue } $remote_list = $response.list $diff = Compare-Object $remote_list $current_list -PassThru if ($diff) { try { $diff | Out-File $PSCompletions.path.change -Force -Encoding utf8 -ErrorAction Stop $response | ConvertTo-Json -Compress | Out-File $PSCompletions.path.completions_json -Encoding utf8 -Force -ErrorAction Stop $PSCompletions.list = $remote_list } catch { Write-Host $_.Exception.Message -ForegroundColor Red return $false } } else { Clear-Content $PSCompletions.path.change -Force -ErrorAction SilentlyContinue $PSCompletions.list = $current_list } $isErr = $false return $remote_list } if ($isErr) { $errMsg | ForEach-Object { Write-Host $_ -ForegroundColor Red } return $false } } function Show-ParamError { param($flag, $cmd, $err_info = $PSCompletions.info.$cmd.err.$flag, $example = $PSCompletions.info.$cmd.example) $err = if ($flag -eq 'min') { $PSCompletions.info.param_min } elseif ($flag -eq 'max') { $PSCompletions.info.param_max } else { $PSCompletions.info.param_err } $PSCompletions.write_with_color((_replace $err)) if ($err_info) { $PSCompletions.write_with_color((_replace $err_info)) } if ($example) { $PSCompletions.write_with_color($PSCompletions.info.example_color + (_replace $example)) } } function Show-List { $max_len = ($PSCompletions.data.list | Measure-Object -Maximum Length).Maximum $max_len = [Math]::Max($max_len, 10) foreach ($_ in $PSCompletions.data.list) { $alias = $PSCompletions.data.alias.$_ -join ' ' $data.Add([PSCustomObject]@{ Completion = $_ Alias = $alias }) } $data } function Out-Data { if ($PSCompletions._need_update_data) { $PSCompletions.data | ConvertTo-Json -Depth 5 -Compress | Out-File $PSCompletions.path.data -Force -Encoding utf8 $PSCompletions._need_update_data = $null } } function _list { if ($arg.Length -gt 2) { Show-ParamError 'max' 'list' return } $data = [System.Collections.Generic.List[System.Object]]@() if ($arg.Length -eq 2) { if ($arg[1] -eq '--remote') { if (!(download_list)) { return } $max_len = ($PSCompletions.list | Measure-Object -Maximum Length).Maximum foreach ($_ in $PSCompletions.list) { $status = if ($PSCompletions.data.alias[$_]) { $PSCompletions.info.list.added }else { $PSCompletions.info.list.add } $data.Add([PSCustomObject]@{ Completion = $_ Status = $status }) } $data } else { Show-ParamError 'err' 'list' } } else { Show-List } } function _add { if ($arg.Length -lt 2) { Show-ParamError 'min' 'add' return } if (!(download_list)) { return } if ($arg[1] -eq '*') { foreach ($_ in $PSCompletions.list) { $PSCompletions.add_completion($_) $PSCompletions._need_update_data = $true } # 如果没有使用模块提供的补全菜单，需要重启 PowerShell 并重新导入模块 if (!$PSCompletions.use_module_menu) { $PSCompletions.write_with_color((_replace $PSCompletions.info.module.restart)) } return } $completions_list = $arg[1..($arg.Length - 1)] foreach ($completion in $completions_list) { if ($completion -in $PSCompletions.list) { $PSCompletions.add_completion($completion) $PSCompletions._need_update_data = $true } else { $PSCompletions.write_with_color((_replace $PSCompletions.info.add.err.no)) } } if (!$PSCompletions.use_module_menu) { $PSCompletions.write_with_color((_replace $PSCompletions.info.module.restart)) } } function _rm { if ($arg.Length -lt 2) { Show-ParamError 'min' 'rm' return } Clear-Content $PSCompletions.path.update -Force -ErrorAction SilentlyContinue $PSCompletions.update = @() $data = [ordered]@{ list = @() alias = [ordered]@{} aliasMap = [ordered]@{} config = $PSCompletions.data.config } if ($arg[1] -eq '*') { foreach ($completion in $PSCompletions.data.list) { $dir = Join-Path $PSCompletions.path.completions $completion Remove-Item $dir -Recurse -Force -ErrorAction SilentlyContinue if (!(Test-Path $dir)) { $PSCompletions.write_with_color((_replace $PSCompletions.info.rm.done)) } } $data.config.comp_config = @{} } else { $remove_list = @() foreach ($completion in $arg[1..($arg.Length - 1)]) { if ($completion -in $PSCompletions.data.list) { $remove_list += $completion $dir = Join-Path $PSCompletions.path.completions $completion if (!(Test-Path $dir)) { $PSCompletions.write_with_color((_replace $PSCompletions.info.no_completion)) continue } Remove-Item $dir -Recurse -Force -ErrorAction SilentlyContinue if (!(Test-Path $dir)) { $PSCompletions.write_with_color((_replace $PSCompletions.info.rm.done)) } } else { $PSCompletions.write_with_color((_replace $PSCompletions.info.no_completion)) } } foreach ($completion in $PSCompletions.data.list.Where({ $_ -notin $remove_list })) { $data.list += $completion } foreach ($_ in $data.list) { $data.alias.$_ = @() if ($PSCompletions.data.alias[$_]) { foreach ($a in $PSCompletions.data.alias.$_) { $data.alias.$_ += $a $data.aliasMap.$a = $_ } } else { $data.alias.$_ += $_ $data.aliasMap.$_ = $_ } } } $remove = @() foreach ($_ in $data.config.comp_config.keys) { if ($_ -notin $data.list) { $remove += $_ } } foreach ($_ in $remove) { $data.config.comp_config.Remove($_) } $data | ConvertTo-Json -Depth 5 -Compress | Out-File $PScompletions.path.data -Force -Encoding utf8 $PSCompletions.data = $data } function _update { $completion_list = $PSCompletions.data.list.Where({ $_ -in $PSCompletions.list }) $params = @{ ErrorAction = 'Stop' } if ($PSEdition -eq 'Core') { $params['OperationTimeoutSeconds'] = 30 } else { $params['TimeoutSec'] = 30 } if ($arg.Length -lt 2) { # 如果只是使用 psc update 则检查更新 $need_update_list = [System.Collections.Generic.List[string]]@() foreach ($completion in $completion_list) { $params['Uri'] = "$($PSCompletions.url)/completions/$completion/guid.json" try { $response = Invoke-RestMethod @params $old_guid = $PSCompletions.get_raw_content("$($PSCompletions.path.completions)/$completion/guid.json") | ConvertFrom-Json | Select-Object -ExpandProperty guid if ($response.guid -ne $old_guid) { $need_update_list.Add($completion) } } catch { } } $PSCompletions.update = $need_update_list } else { $updated_list = [System.Collections.Generic.List[string]]@() if ($arg[1] -eq '*') { if ($arg[2] -eq '--force') { foreach ($_ in $completion_list) { $PSCompletions.add_completion($_) $updated_list.Add($_) } } else { foreach ($_ in $PSCompletions.update) { $PSCompletions.add_completion($_) $PSCompletions._need_update_data = $true $updated_list.Add($_) } } } else { foreach ($completion in $arg[1..($arg.Length - 1)]) { if ($completion -in $completion_list) { $PSCompletions.add_completion($completion) $PSCompletions._need_update_data = $true $updated_list.Add($completion) } else { $PSCompletions.write_with_color((_replace $PSCompletions.info.no_completion)) } } } $updated_list = $updated_list.Where({ Test-Path "$($PSCompletions.path.completions)/$_/config.json" }) if ($updated_list) { $PSCompletions.update = $PSCompletions.update.Where({ $_ -notin $updated_list }) } else { $PSCompletions._need_update_data = $false } } if ($PSCompletions.update) { $PSCompletions.write_with_color((_replace $PSCompletions.info.update_has)) } else { $PSCompletions.write_with_color((_replace $PSCompletions.info.update_no)) } $PSCompletions.update | Out-File $PSCompletions.path.update -Force -Encoding utf8 } function _search { if ($arg.Length -lt 2) { Show-ParamError 'min' 'search' return } if ($arg.Length -gt 2) { Show-ParamError 'max' 'search' return } if (!(download_list)) { return } $result = $PSCompletions.list.Where({ $_ -like $arg[1] }) if ($result) { $result } else { $PSCompletions.write_with_color((_replace $PSCompletions.info.search.err.no)) } } function _which { if ($arg.Length -lt 2) { Show-ParamError 'min' 'which' return } foreach ($completion in $arg[1..($arg.Length - 1)]) { if ($completion -in $PSCompletions.data.list) { Write-Output (Join-Path $PSCompletions.path.completions $completion) } else { $PSCompletions.write_with_color((_replace $PSCompletions.info.no_completion)) } } } function _alias { $cmd_list = @('list', 'add', 'rm') if ($arg.Length -lt 2) { Show-ParamError 'min' '' $PSCompletions.info.sub_cmd return } if ($arg[1] -notin $cmd_list) { $sub_cmd = $arg[1] Show-ParamError 'err' '' $PSCompletions.info.sub_cmd return } if ($arg[1] -eq 'list') { if ($arg[2]) { Show-ParamError 'max' '' '' $PSCompletions.info.alias.list.example return } $data = [System.Collections.Generic.List[System.Object]]@() Show-List return } $data_alias = [ordered]@{} $data_aliasMap = [ordered]@{} foreach ($_ in $PSCompletions.data.list) { $data_alias[$_] = [System.Collections.Generic.List[string]]@() if ($PSCompletions.data.alias[$_]) { foreach ($a in $PSCompletions.data.alias[$_]) { $data_alias[$_].Add($a) $data_aliasMap[$a] = $_ } } else { $data_alias[$_].Add($_) $data_aliasMap[$_] = $_ } } switch ($arg[1]) { 'add' { $completion = $arg[2] if ($null -eq $arg[2]) { $cmd_list = $PSCompletions.data.list Show-ParamError 'min' '' $PSCompletions.info.sub_cmd $PSCompletions.info.alias.add.example return } else { if ($arg[2] -notin $PSCompletions.data.list) { Show-ParamError 'err' '' $PSCompletions.info.no_completion return } } if ($null -eq $arg[3]) { Show-ParamError 'min' '' $PSCompletions.info.alias.add.err.min_v $PSCompletions.info.alias.add.example return } $add_list = @() foreach ($alias in $arg[3..($arg.Length - 1)]) { $alias = ($alias -split ' ')[0] if ($alias -in $data_alias[$completion]) { Show-ParamError 'err' '' $PSCompletions.info.alias.add.err.exist Write-Host continue } if ([System.Management.Automation.WildcardPattern]::ContainsWildcardCharacters($alias)) { Show-ParamError 'err' '' $PSCompletions.info.err.has_wildcard Write-Host continue } if ($alias -eq 'PSCompletions') { $cmds = Get-Command $has_command = foreach ($c in $cmds) { if ($c.Name -eq $alias) { $c; break } } } else { $has_command = Get-Command $alias -ErrorAction SilentlyContinue } if (($alias -notmatch ".*\.\w+$") -and $has_command.CommandType -eq 'Alias') { Show-ParamError 'err' '' $PSCompletions.info.alias.add.err.cmd_exist continue } $data_alias[$completion].Add($alias) $data_aliasMap[$alias] = $completion $add_list += $alias } if ($add_list.Count) { $PSCompletions.write_with_color((_replace $PSCompletions.info.alias.done)) $need_restart = $true } } 'rm' { $completion = $arg[2] if ($null -eq $arg[2]) { $cmd_list = $PSCompletions.data.list Show-ParamError 'min' '' $PSCompletions.info.sub_cmd $PSCompletions.info.alias.rm.example return } else { if ($arg[2] -notin $PSCompletions.data.list) { Show-ParamError 'err' '' $PSCompletions.info.no_completion return } } if ($null -eq $arg[3]) { Show-ParamError 'min' '' $PSCompletions.info.alias.rm.err.min_v $PSCompletions.info.alias.rm.example return } $rm_list = @() foreach ($alias in $arg[3..($arg.Length - 1)]) { if ($alias -in $PSCompletions.data.aliasMap.Keys) { if ($data_alias[$completion].Count -gt 1) { $null = $data_alias[$completion].Remove($alias) $null = $data_aliasMap.Remove($alias) $rm_list += $alias } else { $PSCompletions.write_with_color((_replace ($PSCompletions.info.alias.rm.err.unique))) } } else { $PSCompletions.write_with_color((_replace ($PSCompletions.info.alias.rm.err.no_alias))) } } if ($rm_list.Count) { $PSCompletions.write_with_color((_replace $PSCompletions.info.alias.done)) $need_restart = $true } } } $PSCompletions.data.alias = $data_alias $PSCompletions.data.aliasMap = $data_aliasMap $PSCompletions.data | ConvertTo-Json -Depth 5 -Compress | Out-File $PScompletions.path.data -Force -Encoding utf8 if ($need_restart) { $PSCompletions.write_with_color((_replace $PSCompletions.info.module.restart)) $need_restart = $null } } function _config { $cmd_list = $PSCompletions.config_item $config_item = $arg[1] if ($arg.Length -lt 2) { Show-ParamError 'min' '' $PSCompletions.info.sub_cmd $PSCompletions.info.config.example return } if ($arg[1] -notin $cmd_list) { $sub_cmd = $arg[1] Show-ParamError 'err' '' $PSCompletions.info.sub_cmd return } if ($arg[1] -in $cmd_list) { if ($arg.Length -gt 3) { Show-ParamError 'max' '' $PSCompletions.info.config.$($arg[1]).err.max $PSCompletions.info.config.$($arg[1]).example return } if ($arg.Length -eq 2) { Write-Output $PSCompletions.wrap_whitespace($PSCompletions.config.$($arg[1])) return } } function handle_done { param([bool]$is_can, $err_info, $example = $PSCompletions.info.config.$($arg[1]).example) if ($arg.Length -eq 3) { if ($is_can) { $old_value = $PSCompletions.config.$config_item $new_value = $arg[2] $PSCompletions.config.$config_item = $new_value $PSCompletions._need_update_data = $true $PSCompletions.write_with_color((_replace $PSCompletions.info.config.done)) } else { Show-ParamError 'err' '' $err_info $example } } } switch ($arg[1]) { 'language' { handle_done ($arg[2] -is [string] -and $arg[2] -ne '') $PSCompletions.info.config.language.err } 'enable_cache' { handle_done ($arg[2] -is [int] -and $arg[2] -in @(1, 0)) $PSCompletions.info.config.err.one_or_zero } 'enable_auto_alias_setup' { handle_done ($arg[2] -is [int] -and $arg[2] -in @(1, 0)) $PSCompletions.info.config.err.one_or_zero } 'enable_completions_update' { handle_done ($arg[2] -is [int] -and $arg[2] -in @(1, 0)) $PSCompletions.info.config.err.one_or_zero } 'enable_module_update' { handle_done ($arg[2] -is [int] -and $arg[2] -in @(1, 0)) $PSCompletions.info.config.err.one_or_zero } 'url' { $arg[2] = $arg[2].TrimEnd('/') handle_done ($arg[2] -match 'http[s]?://' -or $arg[2] -eq '') $PSCompletions.info.config.url.err } 'function_name' { if ([System.Management.Automation.WildcardPattern]::ContainsWildcardCharacters($arg[2])) { Show-ParamError 'err' '' $PSCompletions.info.err.has_wildcard return } if ($arg[2] -eq 'PSCompletions') { $cmds = Get-Command $has_command = foreach ($c in $cmds) { if ($c.Name -eq $arg[2]) { $c; break } } } else { $has_command = Get-Command $arg[2] -ErrorAction SilentlyContinue } handle_done ($arg[2] -ne '' -and !$has_command) $PSCompletions.info.config.function_name.err } } } function _completion { $cmd_list = $PSCompletions.data.list if ($arg.Length -lt 2) { Show-ParamError 'min' '' $PSCompletions.info.sub_cmd $PSCompletions.info.completion.example return } if ($arg[1] -notin $PSCompletions.data.list) { $cmd_list = $PSCompletions.data.list $sub_cmd = $arg[1] Show-ParamError 'err' '' $PSCompletions.info.sub_cmd return } if ($arg.Length -lt 3) { Show-ParamError 'min' 'completion' return } $config_list = $PSCompletions.default_completion_item if ($null -eq $PSCompletions.config.comp_config.$($arg[1])) { $PSCompletions.config.comp_config.$($arg[1]) = @{} } if ($arg[2] -notin $config_list -and $null -eq $PSCompletions.config.comp_config.$($arg[1]).$($arg[2])) { $cmd_list = $config_list + ($PSCompletions.config.comp_config.$($arg[1]).keys.Where({ $_ -notin $config_list })) $sub_cmd = $arg[2] Show-ParamError 'err' '' $PSCompletions.info.sub_cmd return } if ($arg.Length -gt 4) { Show-ParamError 'max' 'completion' return } if ($arg.Length -eq 3) { Write-Output $PSCompletions.wrap_whitespace($PSCompletions.config.comp_config.$($arg[1]).$($arg[2])) return } $completion = $arg[1] $config_item = $arg[2] $old_value = $PSCompletions.config.comp_config[$completion].$config_item $new_value = $arg[3] if ($new_value -match '^-?\d+$') { $new_value = [int]$new_value } if ($config_item -match "(enable_*)|(disable_*)") { if ($new_value -notin @(1, 0)) { $cmd_list = $null $sub_cmd = $value $cmd_info = $PSCompletions.info.menu.config.err.v_1_or_0 Show-ParamError 'err' '' $PSCompletions.info.sub_cmd $PSCompletions.info.completion.example return } } $PSCompletions.config.comp_config[$completion].$config_item = $new_value $PSCompletions._need_update_data = $true foreach ($_ in $PSCompletions.data.list) { if (!$PSCompletions.config.comp_config[$_]) { $PSCompletions.config.comp_config.$_ = @{} } $path = "$($PSCompletions.path.completions)/$_/config.json" $json = $PSCompletions.get_raw_content($path) | ConvertFrom-Json $path = "$($PSCompletions.path.completions)/$_/language/$($json.language[0]).json" $json = $PSCompletions.ConvertFrom_JsonAsHashtable($PSCompletions.get_raw_content($path)) $config_list = $PSCompletions.default_completion_item if ($json.config) { foreach ($item in $json.config) { $config_list += $item.name if ($PSCompletions.config.comp_config[$_].$($item.name) -in @('', $null)) { $PSCompletions.config.comp_config.$_.$($item.name) = $item.value $PSCompletions._need_update_data = $true } } } if ($PSCompletions.config.comp_config[$_]) { $_keys = @() foreach ($k in $PSCompletions.config.comp_config.$_.keys) { if ($k -notin $config_list -and $k -ne 'enable_hooks') { $_keys += $k } } foreach ($k in $_keys) { $null = $PSCompletions.config.comp_config.$_.Remove($c) } } } $PSCompletions.write_with_color((_replace $PSCompletions.info.completion.done)) } function _menu { $cmd_list = @('symbol', 'line_theme', 'color_theme', 'custom', 'config') if ($arg.Length -lt 2) { Show-ParamError 'min' 'menu' $PSCompletions.info.sub_cmd return } if ($arg[1] -notin $cmd_list) { Show-ParamError 'err' 'menu' $PSCompletions.info.sub_cmd } switch ($arg[1]) { 'symbol' { $cmd_list = @('SpaceTab', 'WriteSpaceTab', 'OptionTab') if ($arg.Length -eq 2) { Show-ParamError 'min' '' $PSCompletions.info.sub_cmd $PSCompletions.info.menu.$($arg[1]).example return } if ($arg[2] -notin $cmd_list) { $sub_cmd = $arg[2] Show-ParamError 'err' '' $PSCompletions.info.sub_cmd $PSCompletions.info.menu.$($arg[1]).example } if ($arg.Length -gt 4) { Show-ParamError 'max' '' $PSCompletions.info.menu.$($arg[1]).err.max $PSCompletions.info.menu.$($arg[1]).example return } $config_item = $arg[2] if ($arg.Length -eq 3) { Write-Output $PSCompletions.wrap_whitespace($PSCompletions.config.$config_item) return } if ($arg.Length -eq 4) { $old_value = $PSCompletions.config.$config_item $new_value = $arg[3] $PSCompletions.config.$config_item = $arg[3] $PSCompletions._need_update_data = $true $PSCompletions.write_with_color((_replace $PSCompletions.info.menu.done)) } } 'line_theme' { if ($PSEdition -eq 'Desktop') { $PSCompletions.write_with_color((_replace $PSCompletions.info.err.deny_change_menu_line_style)) return } $cmd_list = @('double_line_rect_border', 'bold_line_rect_border', 'single_line_rect_border', 'single_line_round_border', 'boldTB_slimLR_border', 'slimTB_boldLR_border', 'doubleTB_singleLR_border', 'singleTB_doubleLR_border') if ($arg.Length -lt 3) { Show-ParamError 'min' '' $PSCompletions.info.sub_cmd $PSCompletions.info.menu.line_theme.example return } if ($arg[2] -notin $cmd_list) { Show-ParamError 'err' '' $PSCompletions.info.sub_cmd $PSCompletions.info.menu.line_theme.example return } if ($arg.Length -gt 3) { Show-ParamError 'max' '' '' $PSCompletions.info.menu.line_theme.example return } switch ($arg[2]) { # [int][char]"═" 'double_line_rect_border' { $PSCompletions.config.horizontal = [string][char]9552 # ═ $PSCompletions.config.vertical = [string][char]9553 # ║ $PSCompletions.config.top_left = [string][char]9556 # ╔ $PSCompletions.config.bottom_left = [string][char]9562 # ╚ $PSCompletions.config.top_right = [string][char]9559 # ╗ $PSCompletions.config.bottom_right = [string][char]9565 # ╝ } 'bold_line_rect_border' { $PSCompletions.config.horizontal = [string][char]9473 # ━ $PSCompletions.config.vertical = [string][char]9475 # ┃ $PSCompletions.config.top_left = [string][char]9487 # ┏ $PSCompletions.config.bottom_left = [string][char]9495 # ┗ $PSCompletions.config.top_right = [string][char]9491 # ┓ $PSCompletions.config.bottom_right = [string][char]9499 # ┛ } 'single_line_rect_border' { $PSCompletions.config.horizontal = [string][char]9472 # ─ $PSCompletions.config.vertical = [string][char]9474 # │ $PSCompletions.config.top_left = [string][char]9484 # ┌ $PSCompletions.config.bottom_left = [string][char]9492 # └ $PSCompletions.config.top_right = [string][char]9488 # ┐ $PSCompletions.config.bottom_right = [string][char]9496 # ┘ } 'single_line_round_border' { $PSCompletions.config.horizontal = [string][char]9472 # ─ $PSCompletions.config.vertical = [string][char]9474 # │ $PSCompletions.config.top_left = [string][char]9581 # ╭ $PSCompletions.config.bottom_left = [string][char]9584 # ╰ $PSCompletions.config.top_right = [string][char]9582 # ╮ $PSCompletions.config.bottom_right = [string][char]9583 # ╯ } # TB: top and bottom # LR: left and right 'boldTB_slimLR_border' { $PSCompletions.config.horizontal = [string][char]9473 # ━ $PSCompletions.config.vertical = [string][char]9474 # │ $PSCompletions.config.top_left = [string][char]9485 # ┍ $PSCompletions.config.bottom_left = [string][char]9493 # ┕ $PSCompletions.config.top_right = [string][char]9489 # ┑ $PSCompletions.config.bottom_right = [string][char]9497 # ┙ } 'slimTB_boldLR_border' { $PSCompletions.config.horizontal = [string][char]9472 # ─ $PSCompletions.config.vertical = [string][char]9475 # ┃ $PSCompletions.config.top_left = [string][char]9486 # ┎ $PSCompletions.config.bottom_left = [string][char]9494 # ┖ $PSCompletions.config.top_right = [string][char]9490 # ┒ $PSCompletions.config.bottom_right = [string][char]9498 # ┚ } 'doubleTB_singleLR_border' { $PSCompletions.config.horizontal = [string][char]9552 # ═ $PSCompletions.config.vertical = [string][char]9474 # │ $PSCompletions.config.top_left = [string][char]9554 # ╒ $PSCompletions.config.bottom_left = [string][char]9560 # ╘ $PSCompletions.config.top_right = [string][char]9557 # ╕ $PSCompletions.config.bottom_right = [string][char]9563 # ╛ } 'singleTB_doubleLR_border' { $PSCompletions.config.horizontal = [string][char]9472 # ─ $PSCompletions.config.vertical = [string][char]9553 # ║ $PSCompletions.config.top_left = [string][char]9555 # ╓ $PSCompletions.config.bottom_left = [string][char]9561 # ╙ $PSCompletions.config.top_right = [string][char]9558 # ╖ $PSCompletions.config.bottom_right = [string][char]9564 # ╜ } } $PSCompletions._need_update_data = $true $PSCompletions.write_with_color((_replace $PSCompletions.info.menu.line_theme.done)) } 'color_theme' { $cmd_list = @('default', 'magenta') if ($arg.Length -lt 3) { Show-ParamError 'min' '' $PSCompletions.info.sub_cmd $PSCompletions.info.menu.color_theme.example return } if ($arg[2] -notin $cmd_list) { Show-ParamError 'err' '' $PSCompletions.info.sub_cmd $PSCompletions.info.menu.color_theme.example return } if ($arg.Length -gt 3) { Show-ParamError 'max' '' '' $PSCompletions.info.menu.color_theme.example return } switch ($arg[2]) { 'default' { $PSCompletions.config.filter_color = 'Yellow' $PSCompletions.config.border_color = 'DarkGray' $PSCompletions.config.item_color = 'Blue' $PSCompletions.config.status_color = 'Blue' $PSCompletions.config.tip_color = 'Cyan' $PSCompletions.config.selected_color = 'white' $PSCompletions.config.selected_bgcolor = 'DarkGray' } 'magenta' { $PSCompletions.config.filter_color = 'Magenta' $PSCompletions.config.border_color = 'Magenta' $PSCompletions.config.item_color = 'Magenta' $PSCompletions.config.status_color = 'Magenta' $PSCompletions.config.tip_color = 'Magenta' $PSCompletions.config.selected_color = 'white' $PSCompletions.config.selected_bgcolor = 'DarkMagenta' } } $PSCompletions._need_update_data = $true $PSCompletions.write_with_color((_replace $PSCompletions.info.menu.color_theme.done)) } 'custom' { $cmd_list = @('color', 'line') if ($arg.Length -lt 3) { Show-ParamError 'min' '' $PSCompletions.info.sub_cmd $PSCompletions.info.menu.custom.example return } if ($arg[2] -notin $cmd_list) { Show-ParamError 'err' '' $PSCompletions.info.sub_cmd $PSCompletions.info.menu.custom.example return } if ($arg[2] -eq 'line' -and $PSEdition -eq 'Desktop') { $PSCompletions.write_with_color((_replace $PSCompletions.info.err.deny_change_menu_line_style)) return } if ($arg.Length -lt 4) { $cmd_list = $PSCompletions.menu.const."$($arg[2])_item" Show-ParamError 'min' '' $PSCompletions.info.sub_cmd $PSCompletions.info.menu.custom.example return } if ($arg[3] -notin $PSCompletions.menu.const."$($arg[2])_item") { $cmd_list = $PSCompletions.menu.const."$($arg[2])_item" Show-ParamError 'err' '' $PSCompletions.info.sub_cmd $PSCompletions.info.menu.custom.example return } if ($arg.Length -lt 5) { Write-Output $PSCompletions.wrap_whitespace($PSCompletions.config[$arg[3]]) return } if ($arg.Length -gt 5) { if ($arg[2] -eq 'color' -and $arg[4] -notin $PSCompletions.menu.const.color_value) { $cmd_list = $PSCompletions.menu.const.color_value Show-ParamError 'err' '' $PSCompletions.info.sub_cmd $PSCompletions.info.menu.custom.example return } } if ($arg[5]) { Show-ParamError 'max' '' '' $PSCompletions.info.menu.custom.example return } $config_item = $arg[3] $old_value = $PSCompletions.config.$config_item $new_value = $arg[4] $PSCompletions.config.$config_item = $new_value $PSCompletions._need_update_data = $true $PSCompletions.write_with_color((_replace $PSCompletions.info.menu.done)) } 'config' { $cmd_list = $PSCompletions.menu.const.config_item if ($arg.Length -lt 3) { Show-ParamError 'min' '' $PSCompletions.info.sub_cmd $PSCompletions.info.menu.config.example return } if ($arg[2] -notin $cmd_list) { Show-ParamError 'err' '' $PSCompletions.info.sub_cmd $PSCompletions.info.menu.config.example return } if ($arg.Length -eq 3) { Write-Output $PSCompletions.wrap_whitespace($PSCompletions.config.$($arg[2])) return } if ($arg.Length -gt 4) { Show-ParamError 'max' '' '' $PSCompletions.info.menu.config.example return } if ($arg[3] -match '^-?\d+$') { $value = [int]$arg[3] $is_num = $true } else { $value = $arg[3] $is_num = $false } switch ($arg[2]) { 'trigger_key' { try { if ($arg[3] -ne $PSCompletions.config.trigger_key) { Set-PSReadLineKeyHandler $arg[3] MenuComplete } } catch { Show-ParamError 'err' 'trigger_key' $PSCompletions.info.menu.config.err.trigger_key return } } { $_ -in @('list_min_width', 'list_max_count_when_above', 'list_max_count_when_below', 'completions_confirm_limit') } { $min = 0 if (!$is_num -or $value -lt $min) { $cmd_list = $null $sub_cmd = $value $cmd_info = _replace $PSCompletions.info.menu.config.err.v_ge Show-ParamError 'err' '' $PSCompletions.info.sub_cmd $PSCompletions.info.menu.config.example return } } { $_ -in 'height_from_menu_bottom_to_cursor_when_above', 'height_from_menu_top_to_cursor_when_below' } { $min = 0 $max = 5 if (!$is_num -or $value -lt $min -or $value -gt $max) { $cmd_list = $null $sub_cmd = $value $cmd_info = _replace $PSCompletions.info.menu.config.err.v_ge_le Show-ParamError 'err' '' $PSCompletions.info.sub_cmd $PSCompletions.info.menu.config.example return } } { $_ -in ($PSCompletions.menu.const.config_item | Where-Object { $_ -match "(enable_*)|(disable_*)" }) } { if (!$is_num -or $value -notin @(1, 0)) { $cmd_list = $null $sub_cmd = $value $cmd_info = $PSCompletions.info.menu.config.err.v_1_or_0 Show-ParamError 'err' '' $PSCompletions.info.sub_cmd $PSCompletions.info.menu.config.example return } } 'completion_suffix' { if ($value -notmatch '^\s*$') { Show-ParamError 'err' 'completion_suffix' $PSCompletions.info.menu.config.err.completion_suffix return } } } $config_item = $arg[2] $old_value = $PSCompletions.config.$config_item $new_value = $value $PSCompletions.config.$config_item = $new_value $PSCompletions._need_update_data = $true $PSCompletions.write_with_color((_replace $PSCompletions.info.menu.done)) if ($config_item -in @('enable_menu', 'enable_menu_enhance', 'trigger_key')) { $PSCompletions.write_with_color((_replace $PSCompletions.info.module.restart)) } } } } function _reset { $cmd_list = @('config', 'alias', 'order', 'completion', 'menu', '*') if ($arg.Length -lt 2) { Show-ParamError 'min' '' $PSCompletions.info.sub_cmd $PSCompletions.info.reset.example return } if ($arg[1] -notin $cmd_list) { $sub_cmd = $arg[1] Show-ParamError 'err' '' $PSCompletions.info.sub_cmd $PSCompletions.info.reset.example } if ($arg[1] -in @('alias', 'completion', 'menu')) { $cmd_list = @('*') if ($arg[1] -in @('alias', 'completion')) { $cmd_list += $PSCompletions.data.list } elseif ($arg[1] -eq 'menu') { $cmd_list += @('symbol', 'line', 'color', 'config') } if ($arg.Length -eq 2) { Show-ParamError 'min' '' $PSCompletions.info.sub_cmd $PSCompletions.info.reset.$($arg[1]).example return } if ($arg[1] -notin @('completion', 'alias') -and $arg.Length -gt 3) { Show-ParamError 'max' '' '' $PSCompletions.info.reset.$($arg[1]).example return } } else { if ($arg.Length -ne 2) { $cmd_list = $null Show-ParamError 'max' '' '' $PSCompletions.info.reset.example return } } function handle_reset { param([array]$configs) $change_list = [System.Collections.Generic.List[System.Object]]@() foreach ($_ in $configs) { $change_list.Add(@{ item = $_ old_value = $PSCompletions.config.$_ new_value = $PSCompletions.default_config.$_ }) $PSCompletions.config.$_ = $PSCompletions.default_config.$_ $PSCompletions._need_update_data = $true } return $change_list } switch ($arg[1]) { 'config' { $change_list = handle_reset $PSCompletions.config_item } 'alias' { $need_restart = $true $change_list = [System.Collections.Generic.List[System.Object]]@() $del_list = if ($arg[2] -eq '*') { , $PSCompletions.data.list }else { , $arg[2..($arg.Length - 1)] } foreach ($completion in $del_list) { if ($completion -in $PSCompletions.data.list) { $old_value = $PSCompletions.data.alias[$completion] -join ' ' $PSCompletions.data.alias.Remove($completion) $alias = ($PSCompletions.get_raw_content("$($PSCompletions.path.completions)/$completion/config.json") | ConvertFrom-Json).alias $new_value = if ($alias) { $alias }else { $completion } $PSCompletions.data.alias[$completion] = , $new_value $PSCompletions._need_update_data = $true $change_list.Add(@{ item = $completion old_value = $old_value new_value = $new_value }) } else { $PSCompletions.write_with_color((_replace $PSCompletions.info.no_completion)) } } $PSCompletions.data.aliasMap = [ordered]@{} foreach ($_ in $PSCompletions.data.list) { if ($PSCompletions.data.alias[$_]) { foreach ($a in $PSCompletions.data.alias.$_) { $PSCompletions.data.aliasMap.$a = $_ $PSCompletions._need_update_data = $true } } else { $PSCompletions.data.aliasMap.$_ = $_ $PSCompletions._need_update_data = $true } } } 'order' { foreach ($_ in Get-ChildItem $PSCompletions.path.order) { Remove-Item $_.FullName -Force -Recurse -ErrorAction SilentlyContinue } $PSCompletions.write_with_color((_replace $PSCompletions.info.reset.order.done)) return } 'completion' { function _do { param([string]$cmd) $PSCompletions.config.comp_config[$cmd] = @{} $path = "$($PSCompletions.path.completions)/$cmd/config.json" $json_config = $PSCompletions.get_raw_content($path) | ConvertFrom-Json $path = "$($PSCompletions.path.completions)/$cmd/language/$($json_config.language[0]).json" $json = $PSCompletions.ConvertFrom_JsonAsHashtable($PSCompletions.get_raw_content($path)) foreach ($item in $json.config) { $PSCompletions.config.comp_config[$cmd].$($item.name) = $item.value $change_list.Add(@{ cmd = $cmd item = $item.name old_value = $old_comp_config[$cmd].$($item.name) new_value = $item.value }) } foreach ($item in $PSCompletions.default_completion_item) { if ($null -ne $old_comp_config[$cmd].$item) { $change_list.Add(@{ cmd = $cmd item = $item old_value = $old_comp_config[$cmd].$($item) new_value = $null }) } } if ($null -ne $json_config.hooks) { $PSCompletions.config.comp_config[$cmd].enable_hooks = [int]$json_config.hooks $change_list.Add(@{ cmd = $cmd item = $item old_value = $old_comp_config[$cmd].enable_hooks new_value = [int]$json_config.hooks }) } $PSCompletions._need_update_data = $true } $old_comp_config = $PSCompletions.config.comp_config.Clone() $change_list = [System.Collections.Generic.List[System.Object]]@() if ($arg[2] -eq '*') { foreach ($_ in $PSCompletions.data.list) { _do $_ } } else { if ($arg[2] -notin $PSCompletions.data.list) { $completion = $arg[2] $PSCompletions.write_with_color((_replace $PSCompletions.info.no_completion)) return } if ($arg.Length -eq 3) { _do $arg[2] } elseif ($arg.Length -gt 3) { $config_list = $arg[3..($arg.Length - 1)] $path = "$($PSCompletions.path.completions)/$($arg[2])/config.json" $json = $PSCompletions.get_raw_content($path) | ConvertFrom-Json $path = "$($PSCompletions.path.completions)/$($arg[2])/language/$($json.language[0]).json" $json = $PSCompletions.ConvertFrom_JsonAsHashtable($PSCompletions.get_raw_content($path)) foreach ($config in $config_list) { if ($config -in $PSCompletions.default_completion_item) { $change_list.Add(@{ cmd = $arg[2] item = $config old_value = $PSCompletions.config.comp_config[$arg[2]].$config new_value = $null }) $PSCompletions.config.comp_config[$arg[2]].Remove($config) } else { $new_value = $json.config.Where({ $_.name -eq $config })[0].value $change_list.Add(@{ cmd = $arg[2] item = $config old_value = $PSCompletions.config.comp_config[$arg[2]].$config new_value = $new_value }) $PSCompletions.config.comp_config[$arg[2]].$config = $new_value } $PSCompletions._need_update_data = $true } } } } 'menu' { $cmd_list = @('*', 'symbol', 'line', 'color', 'config') if ($arg[2] -notin $cmd_list) { $sub_cmd = $arg[2] Show-ParamError 'min' '' $PSCompletions.info.sub_cmd $PSCompletions.info.reset.example return } switch ($arg[2]) { 'symbol' { $change_list = handle_reset $PSCompletions.menu.const.symbol_item } 'line' { $change_list = handle_reset $PSCompletions.menu.const.line_item } 'color' { $change_list = handle_reset $PSCompletions.menu.const.color_item } 'config' { $need_restart = $true $change_list = handle_reset $PSCompletions.menu.const.config_item } '*' { $need_restart = $true $change_list = [System.Collections.Generic.List[System.Object]]@() $change_list += handle_reset $PSCompletions.menu.const.symbol_item $change_list += handle_reset $PSCompletions.menu.const.line_item $change_list += handle_reset $PSCompletions.menu.const.color_item $change_list += handle_reset $PSCompletions.menu.const.config_item } } } '*' { $PSCompletions.write_with_color($PSCompletions.replace_content($PSCompletions.info.reset.init_confirm)) while (($PressKey = $host.UI.RawUI.ReadKey('NoEcho,IncludeKeyDown')).VirtualKeyCode) { if ($PressKey.ControlKeyState -notlike '*CtrlPressed*') { if ($PressKey.VirtualKeyCode -eq 13) { # 13: Enter Remove-Item $PSCompletions.path.temp -Force -Recurse -ErrorAction SilentlyContinue '{}' | Out-File $PSCompletions.path.data -Encoding utf8 -ErrorAction SilentlyContinue foreach ($_ in Get-ChildItem $PSCompletions.path.completions -Force -Recurse) { Remove-Item $_.FullName -Force -Recurse -ErrorAction SilentlyContinue } $PSCompletions.write_with_color((_replace $PSCompletions.info.reset.init_done)) $PSCompletions.init_data() } else { $PSCompletions.write_with_color($PSCompletions.replace_content($PSCompletions.info.confirm_cancel)) } break } } return } } $PSCompletions.write_with_color((_replace $PSCompletions.info.reset.done)) if ($need_restart) { $PSCompletions.write_with_color((_replace $PSCompletions.info.module.restart)) $need_restart = $null } } function _help { $json = $PSCompletions.completions.psc $info = $PSCompletions.info $PSCompletions.write_with_color((_replace $PSCompletions.info.description)) } $need_init = $true $PSCompletions._need_update_data = $null switch ($arg[0]) { 'list' { _list $need_init = $false } 'add' { _add } 'rm' { _rm } 'update' { _update } 'search' { _search $need_init = $false } 'which' { _which $need_init = $false } 'alias' { _alias } 'config' { _config } 'completion' { _completion } 'menu' { _menu } 'reset' { _reset } default { if ($arg[0]) { if ($arg[0] -in @('-?', '-h', '--help')) { _help } else { $sub_cmd = $arg[0] $cmd_list = @('list', 'add', 'rm', 'update', 'search', 'which', 'alias', 'config', 'completion', 'menu', 'reset') $PSCompletions.write_with_color((_replace $PSCompletions.info.sub_cmd)) } } else { _help } } } Out-Data if ($need_init) { $PSCompletions.init_data() } } Export-ModuleMember -Function $PSCompletions.config.function_name |