core/completion/win.ps1
|
Add-Member -InputObject $PSCompletions -MemberType ScriptMethod generate_completion { if ($this.config.menu_enhance -and $this.config.menu_enable) { Add-Member -InputObject $PSCompletions -MemberType ScriptMethod handle_completion { Set-PSReadLineKeyHandler -Key $this.config.menu_trigger_key -ScriptBlock { $buffer = '' $cursorPosition = 0 [Microsoft.PowerShell.PSConsoleReadLine]::GetBufferState([ref]$buffer, [ref]$cursorPosition) if (!$cursorPosition) { return } # Windows PowerShell 5.x if ($PSEdition -ne "Core") { $PSCompletions.config.menu_line_horizontal = "-" $PSCompletions.config.menu_line_vertical = "|" $PSCompletions.config.menu_line_top_left = "+" $PSCompletions.config.menu_line_bottom_left = "+" $PSCompletions.config.menu_line_top_right = "+" $PSCompletions.config.menu_line_bottom_right = "+" } # 是否是按下空格键触发的补全 $space_tab = if ($buffer[-1] -eq " ") { 1 }else { 0 } # 使用正则表达式进行分割，将命令行中的每个参数分割出来，形成一个数组， 引号包裹的内容会被当作一个参数，且数组会包含 "--" $input_arr = @() $matches = [regex]::Matches($buffer, "(?:`"[^`"]*`"|'[^']*'|\S)+") foreach ($match in $matches) { $input_arr += $match.Value } # 触发补全的值，此值可能是别名或命令名 $alias = $input_arr[0] if ($PSCompletions.alias.$alias -and $input_arr[-1] -notmatch "^(?:\.\.?|~)?(?:[/\\]).*") { if ($buffer -eq $alias) { return } # 原始的命令名，也是 completions 目录下的命令目录名 $PSCompletions.current_cmd = $root = $PSCompletions.alias.$alias $input_arr = if ($input_arr.Count -le 1) { , @() } else { $input_arr[1..($input_arr.Count - 1)] } $filter_list = $PSCompletions.get_completion() $result = $PSCompletions.menu.show_module_menu($filter_list) if ($result) { if ($space_tab) { [Microsoft.PowerShell.PSConsoleReadLine]::Insert($result) } else { $input_arr = if ($input_arr.Count -le 1) { , @() } else { $input_arr[0..($input_arr.Count - 2)] } $result = if ($input_arr.Count -eq 0) { "$alias $result" }else { "$alias $($input_arr -join ' ') $result" } [Microsoft.PowerShell.PSConsoleReadLine]::Replace(0, $buffer.Length, $result) } } } else { try { $completion = TabExpansion2 $buffer $cursorPosition } catch { return } if (!$completion.CompletionMatches) { return } $filter_list = @() $runspacePool = [runspacefactory]::CreateRunspacePool(1, [Environment]::ProcessorCount) $runspacePool.Open() $runspaces = @() foreach ($completions in $PSCompletions.split_array($completion.CompletionMatches, [Environment]::ProcessorCount, $true)) { $runspace = [powershell]::Create().AddScript({ param($completions, $host_ui) $max_width = 0 $results = @() function get_length { param([string]$str) $host_ui.RawUI.NewBufferCellArray($str, $host_ui.RawUI.BackgroundColor, $host_ui.RawUI.BackgroundColor).LongLength } foreach ($completion in $completions) { if ($completion.ToolTip) { if ($completion.ResultType -in @("ParameterValue", "ParameterName")) { $tool_tip = $completion.ToolTip } else { # 如果是内置命令，由于其 ToolTip 写法很奇怪，会导致显示的内容有些问题 $tool_tip = @() foreach ($tip in $completion.ToolTip) { $tip = $tip -replace '\s{2}', ' ' $guid = [guid]::NewGuid() $tip = $tip -replace '\s{2}', $guid $tool_tip += $tip.Trim() -split $guid } $tool_tip = $tool_tip -join "`n`n" } $results += @{ ListItemText = $completion.ListItemText CompletionText = $completion.CompletionText ToolTip = $tool_tip } } else { $results += $completion } $max_width = [Math]::Max($max_width, (get_length $completion.ListItemText)) } @{ results = $results max_width = $max_width } }).AddArgument($completions).AddArgument($Host.UI) $runspace.RunspacePool = $runspacePool $runspaces += @{ Runspace = $runspace; Job = $runspace.BeginInvoke() } } foreach ($rs in $runspaces) { $result = $rs.Runspace.EndInvoke($rs.Job) $rs.Runspace.Dispose() $PSCompletions.menu.list_max_width = [Math]::Max($PSCompletions.menu.list_max_width, $result.max_width) if ($result.results) { $filter_list += $result.results } } $result = $PSCompletions.menu.show_module_menu($filter_list, $true) # apply the completion if ($result) { [Microsoft.PowerShell.PSConsoleReadLine]::Replace($completion.ReplacementIndex, $completion.ReplacementLength, $result) } } $PSCompletions.current_cmd = $null } } } else { Add-Member -InputObject $PSCompletions -MemberType ScriptMethod handle_completion { foreach ($_ in $this.alias.keys) { Register-ArgumentCompleter -CommandName $_ -ScriptBlock { param($word_to_complete, $command_ast, $cursor_position) $space_tab = if (!$word_to_complete.length) { 1 }else { 0 } $input_arr = @() $matches = [regex]::Matches($command_ast.CommandElements, "(?:`"[^`"]*`"|'[^']*'|\S)+") foreach ($match in $matches) { $input_arr += $match.Value } $alias = $input_arr[0] $PSCompletions.current_cmd = $root = $PSCompletions.alias.$alias $input_arr = if ($input_arr.Count -le 1) { , @() } else { $input_arr[1..($input_arr.Count - 1)] } $filter_list = $PSCompletions.get_completion() if ($PSCompletions.config.menu_enable) { # Windows PowerShell 5.x if ($PSEdition -ne "Core") { $PSCompletions.config.menu_line_horizontal = "-" $PSCompletions.config.menu_line_vertical = "|" $PSCompletions.config.menu_line_top_left = "+" $PSCompletions.config.menu_line_bottom_left = "+" $PSCompletions.config.menu_line_top_right = "+" $PSCompletions.config.menu_line_bottom_right = "+" } $PSCompletions.menu.show_module_menu($filter_list) } else { $PSCompletions.menu.show_powershell_menu($filter_list) } $PSCompletions.current_cmd = $null } } } } } |