PSFzf.psm1
# AUTOGENERATED - DO NOT EDIT # PSFzf.Base.ps1 param( [parameter(Position=0,Mandatory=$false)][string]$PSReadlineChordProvider = 'Ctrl+t', [parameter(Position=1,Mandatory=$false)][string]$PSReadlineChordReverseHistory = 'Ctrl+r', [parameter(Position=2,Mandatory=$false)][string]$PSReadlineChordSetLocation = 'Alt+c', [parameter(Position=3,Mandatory=$false)][string]$PSReadlineChordReverseHistoryArgs = 'Alt+a') $script:IsWindows = ($PSVersionTable.PSVersion.Major -le 5) -or $IsWindows if ($script:IsWindows) { $script:ShellCmd = 'cmd.exe /S /C {0}' $script:DefaultFileSystemCmd = @" dir /s/b "{0}" "@ $script:DefaultFileSystemCmdDirOnly = @" dir /s/b/ad "{0}" "@ } else { $script:ShellCmd = '/bin/sh -c "{0}"' $script:DefaultFileSystemCmd = @" find -L '{0}' -path '*/\.*' -prune -o -type f -print -o -type l -print 2> /dev/null "@ $script:DefaultFileSystemCmdDirOnly = @" find -L '{0}' -path '*/\.*' -prune -o -type d -print 2> /dev/null "@ } $script:RunningInWindowsTerminal = [bool]($env:WT_Session) if ($script:RunningInWindowsTerminal) { $script:DefaultFileSystemFdCmd = "fd.exe --color always . `"{0}`"" } else { $script:DefaultFileSystemFdCmd = "fd.exe . `"{0}`"" } $script:UseFd = $false function Get-FileSystemCmd { param($dir, [switch]$dirOnly = $false) # Note that there is no way to know how to list only directories using # FZF_DEFAULT_COMMAND, so we never use it in that case. if ($dirOnly -or [string]::IsNullOrWhiteSpace($env:FZF_DEFAULT_COMMAND)) { if ($script:UseFd) { if ($dirOnly) { "$($script:DefaultFileSystemFdCmd -f $dir) --type directory" } else { $script:DefaultFileSystemFdCmd -f $dir } } else { $cmd = $script:DefaultFileSystemCmd if ($dirOnly) { $cmd = $script:DefaultFileSystemCmdDirOnly } $script:ShellCmd -f ($cmd -f $dir) } } else { $script:ShellCmd -f ($env:FZF_DEFAULT_COMMAND -f $dir) } } class FzfDefaultOpts { [bool]$UsePsFzfOpts [string]$PrevEnv [bool]$Restored FzfDefaultOpts([string]$tempVal) { $this.UsePsFzfOpts = -not [string]::IsNullOrWhiteSpace($env:_PSFZF_FZF_DEFAULT_OPTS) $this.PrevEnv = $env:FZF_DEFAULT_OPTS $env:FZF_DEFAULT_OPTS = $this.Get() + " " + $tempVal } [string]Get() { if ($this.UsePsFzfOpts) { return $env:_PSFZF_FZF_DEFAULT_OPTS; } else { return $env:FZF_DEFAULT_OPTS; } } [void]Restore() { $env:FZF_DEFAULT_OPTS = $this.PrevEnv } } function FixCompletionResult($str) { if ($str.Contains(" ") -or $str.Contains("`t")) { return "'{0}'" -f $str.Replace("`r`n","").Trim(@('''','"')) } else { return $str.Replace("`r`n","") } } #HACK: workaround for fact that PSReadLine seems to clear screen # after keyboard shortcut action is executed, and to work around a UTF8 # PSReadLine issue (GitHub PSFZF issue #71) function InvokePromptHack() { $previousOutputEncoding = [Console]::OutputEncoding [Console]::OutputEncoding = [Text.Encoding]::UTF8 try { [Microsoft.PowerShell.PSConsoleReadLine]::InvokePrompt() } finally { [Console]::OutputEncoding = $previousOutputEncoding } } $script:FzfLocation = $null $script:OverrideFzfDefaults = $null $script:PSReadlineHandlerChords = @() $script:TabContinuousTrigger = [IO.Path]::DirectorySeparatorChar.ToString() $MyInvocation.MyCommand.ScriptBlock.Module.OnRemove = { $PsReadlineShortcuts.Values | Where-Object Chord | ForEach-Object { Remove-PSReadlineKeyHandler $_.Chord } RemovePsFzfAliases RemoveGitKeyBindings } # if the quoted string ends with a '\', and we need to escape it for Windows: function script:PrepareArg($argStr) { if (-not $argStr.EndsWith("\\") -and $argStr.EndsWith('\')) { return $argStr + '\' } else { return $argStr } } function Set-PsFzfOption{ param( [switch] $TabExpansion, [string] $PSReadlineChordProvider, [string] $PSReadlineChordReverseHistory, [string] $PSReadlineChordSetLocation, [string] $PSReadlineChordReverseHistoryArgs, [switch] $GitKeyBindings, [switch] $EnableAliasFuzzyEdit, [switch] $EnableAliasFuzzyFasd, [switch] $EnableAliasFuzzyHistory, [switch] $EnableAliasFuzzyKillProcess, [switch] $EnableAliasFuzzySetLocation, [switch] $EnableAliasFuzzySetEverything, [switch] $EnableAliasFuzzyZLocation, [switch] $EnableAliasFuzzyGitStatus, [switch] $EnableFd ) if ($PSBoundParameters.ContainsKey('TabExpansion')) { SetTabExpansion $TabExpansion } if ($PSBoundParameters.ContainsKey('GitKeyBindings')) { SetGitKeyBindings $GitKeyBindings } $PsReadlineShortcuts.GetEnumerator() | ForEach-Object { if ($PSBoundParameters.ContainsKey($_.key)) { $info = $_.value $newChord = $PSBoundParameters[$_.key] $result = SetPsReadlineShortcut $newChord -Override $info.BriefDesc $info.Desc $info.ScriptBlock if ($result) { if (($null -ne $info.Chord) -and ($info.Chord.ToLower() -ne $newChord.ToLower())) { Remove-PSReadLineKeyHandler $info.Chord } $info.Chord = $newChord } } } if ($EnableAliasFuzzyEdit) { SetPsFzfAlias "fe" Invoke-FuzzyEdit} if ($EnableAliasFuzzyFasd) { SetPsFzfAlias "ff" Invoke-FuzzyFasd} if ($EnableAliasFuzzyHistory) { SetPsFzfAlias "fh" Invoke-FuzzyHistory } if ($EnableAliasFuzzyKillProcess) { SetPsFzfAlias "fkill" Invoke-FuzzyKillProcess } if ($EnableAliasFuzzySetLocation) { SetPsFzfAlias "fd" Invoke-FuzzySetLocation } if ($EnableAliasFuzzyZLocation) { SetPsFzfAlias "fz" Invoke-FuzzyZLocation } if ($EnableAliasFuzzyGitStatus) { SetPsFzfAlias "fgs" Invoke-FuzzyGitStatus } if ($EnableAliasFuzzySetEverything) { if (${function:Set-LocationFuzzyEverything}) { SetPsFzfAlias "cde" Set-LocationFuzzyEverything } } if ($PSBoundParameters.ContainsKey('EnableFd')) { $script:UseFd = $EnableFd } if ($PSBoundParameters.ContainsKey('TabContinuousTrigger')) { $script:TabContinuousTrigger = $TabContinuousTrigger } } function Stop-Pipeline { # borrowed from https://stackoverflow.com/a/34800670: (Add-Type -Passthru -TypeDefinition ' using System.Management.Automation; namespace PSFzf.IO { public static class CustomPipelineStopper { public static void Stop(Cmdlet cmdlet) { throw (System.Exception) System.Activator.CreateInstance(typeof(Cmdlet).Assembly.GetType("System.Management.Automation.StopUpstreamCommandsException"), cmdlet); } } }')::Stop($PSCmdlet) } function Invoke-Fzf { param( # Search [Alias("x")] [switch]$Extended, [Alias('e')] [switch]$Exact, [Alias('i')] [switch]$CaseInsensitive, [switch]$CaseSensitive, [Alias('d')] [string]$Delimiter, [switch]$NoSort, [Alias('tac')] [switch]$ReverseInput, [switch]$Phony, [ValidateSet('length','begin','end','index')] [string] $Tiebreak = $null, # Interface [Alias('m')] [switch]$Multi, [switch]$NoMouse, [string]$Bind, [switch]$Cycle, [switch]$KeepRight, [switch]$NoHScroll, [switch]$FilepathWord, # Layout [ValidatePattern("^[1-9]+[0-9]+$|^[1-9][0-9]?%?$|^100%?$")] [string]$Height, [ValidateRange(1,[int]::MaxValue)] [int]$MinHeight, [ValidateSet('default','reverse','reverse-list')] [string]$Layout = $null, [switch]$Border, [ValidateSet('rounded','sharp','horizontal')] [string]$BorderStyle, [ValidateSet('default','inline','hidden')] [string]$Info = $null, [string]$Prompt, [string]$Pointer, [string]$Marker, [string]$Header, [int]$HeaderLines = -1, # Display [switch]$Ansi, [int]$Tabstop = 8, [switch]$NoBold, # History [string]$History, [int]$HistorySize = -1, #Preview [string]$Preview, [string]$PreviewWindow, # Scripting [Alias('q')] [string]$Query, [Alias('s1')] [switch]$Select1, [Alias('e0')] [switch]$Exit0, [Alias('f')] [string]$Filter, [switch]$PrintQuery, [string]$Expect, [Parameter(ValueFromPipeline=$True)] [object[]]$Input ) Begin { # process parameters: $arguments = '' if ($PSBoundParameters.ContainsKey('Extended') -and $Extended) { $arguments += '--extended '} if ($PSBoundParameters.ContainsKey('Exact') -and $Exact) { $arguments += '--exact '} if ($PSBoundParameters.ContainsKey('CaseInsensitive') -and $CaseInsensitive) { $arguments += '-i '} if ($PSBoundParameters.ContainsKey('CaseSensitive') -and $CaseSensitive) { $arguments += '+i '} if ($PSBoundParameters.ContainsKey('Delimiter') -and ![string]::IsNullOrWhiteSpace($Delimiter)) { $arguments += "--delimiter=$Delimiter "} if ($PSBoundParameters.ContainsKey('NoSort') -and $NoSort) { $arguments += '--no-sort '} if ($PSBoundParameters.ContainsKey('ReverseInput') -and $ReverseInput) { $arguments += '--tac '} if ($PSBoundParameters.ContainsKey('Phony') -and $Phony) { $arguments += '--phony '} if ($PSBoundParameters.ContainsKey('Tiebreak') -and ![string]::IsNullOrWhiteSpace($Tiebreak)) { $arguments += "--tiebreak=$Tiebreak "} if ($PSBoundParameters.ContainsKey('Multi') -and $Multi) { $arguments += '--multi '} if ($PSBoundParameters.ContainsKey('NoMouse') -and $NoMouse) { $arguments += '--no-mouse '} if ($PSBoundParameters.ContainsKey('Bind') -and ![string]::IsNullOrWhiteSpace($Bind)) { $arguments += "--bind=$Bind "} if ($PSBoundParameters.ContainsKey('Reverse') -and $Reverse) { $arguments += '--reverse '} if ($PSBoundParameters.ContainsKey('Cycle') -and $Cycle) { $arguments += '--cycle '} if ($PSBoundParameters.ContainsKey('KeepRight') -and $KeepRight) { $arguments += '--keep-right '} if ($PSBoundParameters.ContainsKey('NoHScroll') -and $NoHScroll) { $arguments += '--no-hscroll '} if ($PSBoundParameters.ContainsKey('FilepathWord') -and $FilepathWord) { $arguments += '--filepath-word '} if ($PSBoundParameters.ContainsKey('Height') -and ![string]::IsNullOrWhiteSpace($Height)) { $arguments += "--height=$height "} if ($PSBoundParameters.ContainsKey('MinHeight') -and $MinHeight -ge 0) { $arguments += "--min-height=$MinHeight "} if ($PSBoundParameters.ContainsKey('Layout') -and ![string]::IsNullOrWhiteSpace($Layout)) { $arguments += "--layout=$Layout "} if ($PSBoundParameters.ContainsKey('Border') -and $Border) { $arguments += '--border '} if ($PSBoundParameters.ContainsKey('BorderStyle') -and ![string]::IsNullOrWhiteSpace($BorderStyle)) { $arguments += "--border=$BorderStyle "} if ($PSBoundParameters.ContainsKey('Info') -and ![string]::IsNullOrWhiteSpace($Info)) { $arguments += "--info=$Info "} if ($PSBoundParameters.ContainsKey('Prompt') -and ![string]::IsNullOrWhiteSpace($Prompt)) { $arguments += "--prompt='$Prompt' "} if ($PSBoundParameters.ContainsKey('Pointer') -and ![string]::IsNullOrWhiteSpace($Pointer)) { $arguments += "--pointer='$Pointer' "} if ($PSBoundParameters.ContainsKey('Marker') -and ![string]::IsNullOrWhiteSpace($Marker)) { $arguments += "--marker='$Marker' "} if ($PSBoundParameters.ContainsKey('Header') -and ![string]::IsNullOrWhiteSpace($Header)) { $arguments += "--header=""$Header"" "} if ($PSBoundParameters.ContainsKey('HeaderLines') -and $HeaderLines -ge 0) { $arguments += "--header-lines=$HeaderLines "} if ($PSBoundParameters.ContainsKey('Ansi') -and $Ansi) { $arguments += '--ansi '} if ($PSBoundParameters.ContainsKey('Tabstop') -and $Tabstop -ge 0) { $arguments += "--tabstop=$Tabstop "} if ($PSBoundParameters.ContainsKey('NoBold') -and $NoBold) { $arguments += '--no-bold '} if ($PSBoundParameters.ContainsKey('History') -and $History) { $arguments += "--history='$History' "} if ($PSBoundParameters.ContainsKey('HistorySize') -and $HistorySize -ge 1) { $arguments += "--history-size=$HistorySize "} if ($PSBoundParameters.ContainsKey('Preview') -and ![string]::IsNullOrWhiteSpace($Preview)) { $arguments += "--preview=""$Preview"" "} if ($PSBoundParameters.ContainsKey('PreviewWindow') -and ![string]::IsNullOrWhiteSpace($PreviewWindow)) { $arguments += "--preview-window=""$PreviewWindow"" "} if ($PSBoundParameters.ContainsKey('Query') -and ![string]::IsNullOrWhiteSpace($Query)) { $arguments += "--query=""{0}"" " -f $(PrepareArg $Query)} if ($PSBoundParameters.ContainsKey('Select1') -and $Select1) { $arguments += '--select-1 '} if ($PSBoundParameters.ContainsKey('Exit0') -and $Exit0) { $arguments += '--exit-0 '} if ($PSBoundParameters.ContainsKey('Filter') -and ![string]::IsNullOrEmpty($Filter)) { $arguments += "--filter=$Filter " } if ($PSBoundParameters.ContainsKey('PrintQuery') -and $PrintQuery) { $arguments += '--print-query '} if ($PSBoundParameters.ContainsKey('Expect') -and ![string]::IsNullOrWhiteSpace($Expect)) { $arguments += "--expect=""$Expect"" "} if (!$script:OverrideFzfDefaults) { $script:OverrideFzfDefaults = [FzfDefaultOpts]::new("") } if ($script:UseHeightOption -and [string]::IsNullOrWhiteSpace($Height) -and ` ([string]::IsNullOrWhiteSpace($script:OverrideFzfDefaults.Get()) -or ` (-not $script:OverrideFzfDefaults.Get().Contains('--height')))) { $arguments += "--height=40% " } if ($Border -eq $true -and -not [string]::IsNullOrWhiteSpace($BorderStyle)) { throw '-Border and -BorderStyle are mutally exclusive' } if ($script:UseFd -and $script:RunningInWindowsTerminal -and -not $arguments.Contains('--ansi')) { $arguments += "--ansi " } # prepare to start process: $process = New-Object System.Diagnostics.Process $process.StartInfo.FileName = $script:FzfLocation $process.StartInfo.Arguments = $arguments $process.StartInfo.StandardOutputEncoding = [System.Text.Encoding]::UTF8 $process.StartInfo.RedirectStandardInput = $true $process.StartInfo.RedirectStandardOutput = $true $process.StartInfo.UseShellExecute = $false if ($pwd.Provider.Name -eq 'FileSystem') { $process.StartInfo.WorkingDirectory = $pwd.ProviderPath } # Adding event handers for stdout: $stdOutEventId = "PsFzfStdOutEh-" + [System.Guid]::NewGuid() $stdOutEvent = Register-ObjectEvent -InputObject $process ` -EventName 'OutputDataReceived' ` -SourceIdentifier $stdOutEventId $processHasExited = new-object psobject -property @{flag = $false} # register on exit: $scriptBlockExited = { $Event.MessageData.flag = $true } $exitedEventId = "PsFzfExitedEh-" + [System.Guid]::NewGuid() $exitedEvent = Register-ObjectEvent -InputObject $process ` -Action $scriptBlockExited -EventName 'Exited' ` -SourceIdentifier $exitedEventId ` -MessageData $processHasExited $process.Start() | Out-Null $process.BeginOutputReadLine() | Out-Null $utf8Encoding = New-Object System.Text.UTF8Encoding -ArgumentList $false $script:utf8Stream = New-Object System.IO.StreamWriter -ArgumentList $process.StandardInput.BaseStream, $utf8Encoding $cleanup = [scriptblock] { if ($script:OverrideFzfDefaults) { $script:OverrideFzfDefaults.Restore() $script:OverrideFzfDefaults = $null } try { $process.StandardInput.Close() | Out-Null $process.WaitForExit() } catch { # do nothing } try { #$stdOutEventId,$exitedEventId | ForEach-Object { # Unregister-Event $_ -ErrorAction SilentlyContinue #} $stdOutEvent,$exitedEvent | ForEach-Object { Stop-Job $_ -ErrorAction SilentlyContinue Remove-Job $_ -Force -ErrorAction SilentlyContinue } } catch { } # events seem to be generated out of order - thereforce, we need sort by time created. For examp`le, # -print-query and -expect and will be outputted first if specified on the command line. Get-Event -SourceIdentifier $stdOutEventId | ` Sort-Object -Property TimeGenerated | ` Where-Object { $null -ne $_.SourceEventArgs.Data } | ForEach-Object { Write-Output $_.SourceEventArgs.Data Remove-Event -EventIdentifier $_.EventIdentifier } } $checkProcessStatus = [scriptblock] { if ($processHasExited.flag -or $process.HasExited) { $script:utf8stream = $null & $cleanup Stop-Pipeline } } } Process { $hasInput = $PSBoundParameters.ContainsKey('Input') # handle no piped input: if (!$hasInput) { # optimization for filesystem provider: if ($PWD.Provider.Name -eq 'FileSystem') { Invoke-Expression (Get-FileSystemCmd $PWD.ProviderPath) | ForEach-Object { try { $utf8Stream.WriteLine($_) } catch [System.Management.Automation.MethodInvocationException] { # Possibly broken pipe. Next clause will handle graceful shutdown. } finally { & $checkProcessStatus } } } else { Get-ChildItem . -Recurse -ErrorAction SilentlyContinue | ForEach-Object { $item = $_ if ($item -is [System.String]) { $str = $item } else { # search through common properties: $str = $item.FullName if ($null -eq $str) { $str = $item.Name if ($null -eq $str) { $str = $item.ToString() } } } if (![System.String]::IsNullOrWhiteSpace($str)) { try { $utf8Stream.WriteLine($str) } catch [System.Management.Automation.MethodInvocationException] { # Possibly broken pipe. We will shutdown the pipe below. } } & $checkProcessStatus } } } else { foreach ($item in $Input) { if ($item -is [System.String]) { $str = $item } else { # search through common properties: $str = $item.FullName if ($null -eq $str) { $str = $item.Name if ($null -eq $str) { $str = $item.ToString() } } } if (![System.String]::IsNullOrWhiteSpace($str)) { try { $utf8Stream.WriteLine($str) } catch [System.Management.Automation.MethodInvocationException] { # Possibly broken pipe. We will shutdown the pipe below. } } & $checkProcessStatus } } if ($null -ne $utf8Stream) { try { $utf8Stream.Flush() } catch [System.Management.Automation.MethodInvocationException] { # Possibly broken pipe, check process status. & $checkProcessStatus } } } End { & $cleanup } } function Find-CurrentPath { param([string]$line,[int]$cursor,[ref]$leftCursor,[ref]$rightCursor) if ($line.Length -eq 0) { $leftCursor.Value = $rightCursor.Value = 0 return $null } if ($cursor -ge $line.Length) { $leftCursorTmp = $cursor - 1 } else { $leftCursorTmp = $cursor } :leftSearch for (;$leftCursorTmp -ge 0;$leftCursorTmp--) { if ([string]::IsNullOrWhiteSpace($line[$leftCursorTmp])) { if (($leftCursorTmp -lt $cursor) -and ($leftCursorTmp -lt $line.Length-1)) { $leftCursorTmpQuote = $leftCursorTmp - 1 $leftCursorTmp = $leftCursorTmp + 1 } else { $leftCursorTmpQuote = $leftCursorTmp } for (;$leftCursorTmpQuote -ge 0;$leftCursorTmpQuote--) { if (($line[$leftCursorTmpQuote] -eq '"') -and (($leftCursorTmpQuote -le 0) -or ($line[$leftCursorTmpQuote-1] -ne '"'))) { $leftCursorTmp = $leftCursorTmpQuote break leftSearch } elseif (($line[$leftCursorTmpQuote] -eq "'") -and (($leftCursorTmpQuote -le 0) -or ($line[$leftCursorTmpQuote-1] -ne "'"))) { $leftCursorTmp = $leftCursorTmpQuote break leftSearch } } break leftSearch } } :rightSearch for ($rightCursorTmp = $cursor;$rightCursorTmp -lt $line.Length;$rightCursorTmp++) { if ([string]::IsNullOrWhiteSpace($line[$rightCursorTmp])) { if ($rightCursorTmp -gt $cursor) { $rightCursorTmp = $rightCursorTmp - 1 } for ($rightCursorTmpQuote = $rightCursorTmp+1;$rightCursorTmpQuote -lt $line.Length;$rightCursorTmpQuote++) { if (($line[$rightCursorTmpQuote] -eq '"') -and (($rightCursorTmpQuote -gt $line.Length) -or ($line[$rightCursorTmpQuote+1] -ne '"'))) { $rightCursorTmp = $rightCursorTmpQuote break rightSearch } elseif (($line[$rightCursorTmpQuote] -eq "'") -and (($rightCursorTmpQuote -gt $line.Length) -or ($line[$rightCursorTmpQuote+1] -ne "'"))) { $rightCursorTmp = $rightCursorTmpQuote break rightSearch } } break rightSearch } } if ($leftCursorTmp -lt 0 -or $leftCursorTmp -gt $line.Length-1) { $leftCursorTmp = 0} if ($rightCursorTmp -ge $line.Length) { $rightCursorTmp = $line.Length-1 } $leftCursor.Value = $leftCursorTmp $rightCursor.Value = $rightCursorTmp $str = -join ($line[$leftCursorTmp..$rightCursorTmp]) return $str.Trim("'").Trim('"') } function Invoke-FzfPsReadlineHandlerProvider { $leftCursor = $null $rightCursor = $null $line = $null $cursor = $null [Microsoft.PowerShell.PSConsoleReadline]::GetBufferState([ref]$line, [ref]$cursor) $currentPath = Find-CurrentPath $line $cursor ([ref]$leftCursor) ([ref]$rightCursor) $addSpace = $null -ne $currentPath -and $currentPath.StartsWith(" ") if ([String]::IsNullOrWhitespace($currentPath) -or !(Test-Path $currentPath)) { $currentPath = $PWD } $result = @() try { $script:OverrideFzfDefaults = [FzfDefaultOpts]::new($env:FZF_CTRL_T_OPTS) if (-not [System.String]::IsNullOrWhiteSpace($env:FZF_CTRL_T_COMMAND)) { Invoke-Expression ($env:FZF_CTRL_T_COMMAND) | Invoke-Fzf -Multi | ForEach-Object { $result += $_ } } else { if ([string]::IsNullOrWhiteSpace($currentPath)) { Invoke-Fzf -Multi | ForEach-Object { $result += $_ } } else { $resolvedPath = Resolve-Path $currentPath -ErrorAction SilentlyContinue $providerName = $null if ($null -ne $resolvedPath) { $providerName = $resolvedPath.Provider.Name } switch ($providerName) { # Get-ChildItem is way too slow - we optimize for the FileSystem provider by # using batch commands: 'FileSystem' { Invoke-Expression (Get-FileSystemCmd $resolvedPath.ProviderPath) | Invoke-Fzf -Multi | ForEach-Object { $result += $_ } } 'Registry' { Get-ChildItem $currentPath -Recurse -ErrorAction SilentlyContinue | Select-Object Name -ExpandProperty Name | Invoke-Fzf -Multi | ForEach-Object { $result += $_ } } $null { Get-ChildItem $currentPath -Recurse -ErrorAction SilentlyContinue | Select-Object FullName -ExpandProperty FullName | Invoke-Fzf -Multi | ForEach-Object { $result += $_ } } Default {} } } } } catch { # catch custom exception } finally { if ($script:OverrideFzfDefaults) { $script:OverrideFzfDefaults.Restore() $script:OverrideFzfDefaults = $null } } InvokePromptHack if ($null -ne $result) { # quote strings if we need to: if ($result -is [system.array]) { for ($i = 0;$i -lt $result.Length;$i++) { $result[$i] = FixCompletionResult $result[$i] } } else { $result = FixCompletionResult $result } $str = $result -join ',' if ($addSpace) { $str = ' ' + $str } $replaceLen = $rightCursor - $leftCursor if ($rightCursor -eq 0 -and $leftCursor -eq 0) { [Microsoft.PowerShell.PSConsoleReadLine]::Insert($str) } else { [Microsoft.PowerShell.PSConsoleReadLine]::Replace($leftCursor,$replaceLen+1,$str) } } } function Invoke-FzfPsReadlineHandlerHistory { $result = $null try { $script:OverrideFzfDefaults = [FzfDefaultOpts]::new($env:FZF_CTRL_R_OPTS) $line = $null $cursor = $null [Microsoft.PowerShell.PSConsoleReadline]::GetBufferState([ref]$line, [ref]$cursor) $reader = New-Object PSFzf.IO.ReverseLineReader -ArgumentList $((Get-PSReadlineOption).HistorySavePath) $fileHist = @{} $reader.GetEnumerator() | ForEach-Object { if (-not $fileHist.ContainsKey($_)) { $fileHist.Add($_,$true) $_ } } | Invoke-Fzf -NoSort -Query "$line" -Bind ctrl-r:toggle-sort | ForEach-Object { $result = $_ } } catch { # catch custom exception } finally { if ($script:OverrideFzfDefaults) { $script:OverrideFzfDefaults.Restore() $script:OverrideFzfDefaults = $null } # ensure that stream is closed: $reader.Dispose() } InvokePromptHack if (-not [string]::IsNullOrEmpty($result)) { [Microsoft.PowerShell.PSConsoleReadLine]::Replace(0,$line.Length,$result) } } function Invoke-FzfPsReadlineHandlerHistoryArgs { $result = @() try { $line = $null $cursor = $null [Microsoft.PowerShell.PSConsoleReadline]::GetBufferState([ref]$line, [ref]$cursor) $line = $line.Insert($cursor,"{}") # add marker for fzf $contentTable = @{} $reader = New-Object PSFzf.IO.ReverseLineReader -ArgumentList $((Get-PSReadlineOption).HistorySavePath) $fileHist = @{} $reader.GetEnumerator() | ForEach-Object { if (-not $fileHist.ContainsKey($_)) { $fileHist.Add($_,$true) [System.Management.Automation.PsParser]::Tokenize($_, [ref] $null) } } | Where-Object {$_.type -eq "commandargument" -or $_.type -eq "string"} | ForEach-Object { if (!$contentTable.ContainsKey($_.Content)) { $_.Content ; $contentTable[$_.Content] = $true } } | Invoke-Fzf -NoSort -Multi | ForEach-Object { $result += $_ } } catch { # catch custom exception } finally { $reader.Dispose() } InvokePromptHack [array]$result = $result | ForEach-Object { # add quotes: if ($_.Contains(" ") -or $_.Contains("`t")) { "'{0}'" -f $_.Replace("'","''") } else { $_ } } if ($result.Length -ge 0) { [Microsoft.PowerShell.PSConsoleReadLine]::Replace($cursor,0,($result -join ' ')) } } function Invoke-FzfPsReadlineHandlerSetLocation { $result = $null try { $script:OverrideFzfDefaults = [FzfDefaultOpts]::new($env:FZF_ALT_C_OPTS) if ($null -eq $env:FZF_ALT_C_COMMAND) { Invoke-Expression (Get-FileSystemCmd . -dirOnly) | Invoke-Fzf | ForEach-Object { $result = $_ } } else { Invoke-Expression ($env:FZF_ALT_C_COMMAND) | Invoke-Fzf | ForEach-Object { $result = $_ } } } catch { # catch custom exception } finally { if ($script:OverrideFzfDefaults) { $script:OverrideFzfDefaults.Restore() $script:OverrideFzfDefaults = $null } } if (-not [string]::IsNullOrEmpty($result)) { Set-Location $result [Microsoft.PowerShell.PSConsoleReadLine]::AcceptLine() } else { InvokePromptHack } } function SetPsReadlineShortcut($Chord,[switch]$Override,$BriefDesc,$Desc,[scriptblock]$scriptBlock) { if ([string]::IsNullOrEmpty($Chord)) { return $false } if ((Get-PSReadlineKeyHandler -Bound | Where-Object {$_.Key.ToLower() -eq $Chord}) -and -not $Override) { return $false } else { Set-PSReadlineKeyHandler -Key $Chord -Description $Desc -BriefDescription $BriefDesc -ScriptBlock $scriptBlock return $true } } function FindFzf() { if ($script:IsWindows) { $AppNames = @('fzf-*-windows_*.exe','fzf.exe') } else { if ($IsMacOS) { $AppNames = @('fzf-*-darwin_*','fzf') } elseif ($IsLinux) { $AppNames = @('fzf-*-linux_*','fzf') } else { throw 'Unknown OS' } } # find it in our path: $script:FzfLocation = $null $AppNames | ForEach-Object { if ($null -eq $script:FzfLocation) { $result = Get-Command $_ -ErrorAction SilentlyContinue $result | ForEach-Object { $script:FzfLocation = Resolve-Path $_.Source } } } if ($null -eq $script:FzfLocation) { throw 'Failed to find fzf binary in PATH. You can download a binary from this page: https://github.com/junegunn/fzf/releases' } } $PsReadlineShortcuts = @{ PSReadlineChordProvider = [PSCustomObject]@{ 'Chord' = "$PSReadlineChordProvider" 'BriefDesc' = 'Fzf Provider Select' 'Desc' = 'Run fzf for current provider based on current token' 'ScriptBlock' = { Invoke-FzfPsReadlineHandlerProvider } }; PSReadlineChordReverseHistory = [PsCustomObject]@{ 'Chord' = "$PSReadlineChordReverseHistory" 'BriefDesc' = 'Fzf Reverse History Select' 'Desc' = 'Run fzf to search through PSReadline history' 'ScriptBlock' = { Invoke-FzfPsReadlineHandlerHistory } }; PSReadlineChordSetLocation = @{ 'Chord' = "$PSReadlineChordSetLocation" 'BriefDesc' = 'Fzf Set Location' 'Desc' = 'Run fzf to select directory to set current location' 'ScriptBlock' = { Invoke-FzfPsReadlineHandlerSetLocation } }; PSReadlineChordReverseHistoryArgs = @{ 'Chord' = "$PSReadlineChordReverseHistoryArgs" 'BriefDesc' = 'Fzf Reverse History Arg Select' 'Desc' = 'Run fzf to search through command line arguments in PSReadline history' 'ScriptBlock' = { Invoke-FzfPsReadlineHandlerHistoryArgs } }; PSReadlineChordTabCompletion = [PSCustomObject]@{ 'Chord' = "Tab" 'BriefDesc' = 'Fzf Tab Completion' 'Desc' = 'Invoke Fzf for tab completion' 'ScriptBlock' = { Invoke-TabCompletion } }; } if (Get-Module -ListAvailable -Name PSReadline) { $PsReadlineShortcuts.GetEnumerator() | ForEach-Object { $info = $_.Value $result = SetPsReadlineShortcut $info.Chord -Override:$PSBoundParameters.ContainsKey($_.Key) $info.BriefDesc $info.Desc $info.ScriptBlock # store that the chord is not activated: if (-not $result) { $info.Chord = $null } } } else { Write-Warning "PSReadline module not found - keyboard handlers not installed" } FindFzf try { $fzfVersion = $(& $script:FzfLocation --version).Replace(' (devel)','').Split('.') $script:UseHeightOption = $fzfVersion.length -ge 2 -and ` ([int]$fzfVersion[0] -gt 0 -or ` [int]$fzfVersion[1] -ge 21) -and ` $script:RunningInWindowsTerminal } catch { # continue } # PSFzf.Functions.ps1 #.ExternalHelp PSFzf.psm1-help.xml $addedAliases = @() function script:SetPsFzfAlias { param($Name,$Function) New-Alias -Name $Name -Scope Global -Value $Function -ErrorAction Ignore $addedAliases += $Name } function script:SetPsFzfAliasCheck { param($Name,$Function) # prevent Get-Command from loading PSFzf $script:PSModuleAutoLoadingPreferencePrev=$PSModuleAutoLoadingPreference $PSModuleAutoLoadingPreference='None' if (-not (Get-Command -Name $Name -ErrorAction Ignore)) { SetPsFzfAlias $Name $Function } # restore module auto loading $PSModuleAutoLoadingPreference=$script:PSModuleAutoLoadingPreferencePrev } function script:RemovePsFzfAliases { $addedAliases | ForEach-Object { Remove-Item -Path Alias:$_ } } function Invoke-FuzzyEdit() { param($Directory=$null) $files = @() try { if ($Directory) { $prevDir = $PWD.ProviderPath cd $Directory } Invoke-Expression (Get-FileSystemCmd .) | Invoke-Fzf -Multi | ForEach-Object { $files += "$_" } } catch { } finally { if ($prevDir) { cd $prevDir } } # HACK to check to see if we're running under Visual Studio Code. # If so, reuse Visual Studio Code currently open windows: $editorOptions = '' if ($null -ne $env:VSCODE_PID) { $editor = 'code' $editorOptions += '--reuse-window' } else { $editor = $env:EDITOR if ($null -eq $editor) { if (!$IsWindows) { $editor = 'vim' } else { $editor = 'code' } } } if ($files.Count -gt 0) { try { if ($Directory) { $prevDir = $PWD.Path cd $Directory } # Not sure if being passed relative or absolute path $fileList = '"{0}"' -f ( (Resolve-Path $files) -join '" "' ) Invoke-Expression -Command ("$editor $editorOptions $fileList") } catch { } finally { if ($prevDir) { cd $prevDir } } } } #.ExternalHelp PSFzf.psm1-help.xml function Invoke-FuzzyFasd() { $result = $null try { if (Get-Command Get-Frecents -ErrorAction Ignore) { Get-Frecents | ForEach-Object { $_.FullPath } | Invoke-Fzf -ReverseInput -NoSort | ForEach-Object { $result = $_ } } elseif (Get-Command fasd -ErrorAction Ignore) { fasd -l | Invoke-Fzf -ReverseInput -NoSort | ForEach-Object { $result = $_ } } } catch { } if ($null -ne $result) { # use cd in case it's aliased to something else: cd $result } } #.ExternalHelp PSFzf.psm1-help.xml function Invoke-FuzzyHistory() { if (Get-Command Get-PSReadLineOption -ErrorAction SilentlyContinue) { $result = Get-Content (Get-PSReadLineOption).HistorySavePath | Invoke-Fzf -Reverse -NoSort } else { $result = Get-History | ForEach-Object { $_.CommandLine } | Invoke-Fzf -Reverse -NoSort } if ($null -ne $result) { Write-Output "Invoking '$result'`n" Invoke-Expression "$result" -Verbose } } #.ExternalHelp PSFzf.psm1-help.xml function GetProcessSelection() { param( [scriptblock] $ResultAction ) $header = "`n" + $("{0,-8} PROCESS NAME" -f "ID") + "`n" $result = Get-Process | Where-Object { ![string]::IsNullOrEmpty($_.ProcessName) } | ForEach-Object { "{0,-8} {1}" -f $_.Id,$_.ProcessName } | Invoke-Fzf -Multi -Header $header $result | ForEach-Object { &$ResultAction $_ } } function Invoke-FuzzyKillProcess() { GetProcessSelection -ResultAction { param($result) $id = $result -replace "([0-9]+)(.*)",'$1' Stop-Process $id -Verbose } } #.ExternalHelp PSFzf.psm1-help.xml function Invoke-FuzzySetLocation() { param($Directory=$null) if ($null -eq $Directory) { $Directory = $PWD.ProviderPath } $result = $null try { if ([string]::IsNullOrWhiteSpace($env:FZF_DEFAULT_COMMAND)) { Get-ChildItem $Directory -Recurse -ErrorAction Ignore | Where-Object{ $_.PSIsContainer } | Invoke-Fzf | ForEach-Object { $result = $_ } } else { Invoke-Fzf | ForEach-Object { $result = $_ } } } catch { } if ($null -ne $result) { Set-Location $result } } if ((-not $IsLinux) -and (-not $IsMacOS)) { #.ExternalHelp PSFzf.psm1-help.xml function Set-LocationFuzzyEverything() { param($Directory=$null) if ($null -eq $Directory) { $Directory = $PWD.ProviderPath $Global = $False } else { $Global = $True } $result = $null try { Search-Everything -Global:$Global -PathInclude $Directory -FolderInclude @('') | Invoke-Fzf | ForEach-Object { $result = $_ } } catch { } if ($null -ne $result) { # use cd in case it's aliased to something else: cd $result } } } #.ExternalHelp PSFzf.psm1-help.xml function Invoke-FuzzyZLocation() { $result = $null try { (Get-ZLocation).GetEnumerator() | Sort-Object { $_.Value } -Descending | ForEach-Object{ $_.Key } | Invoke-Fzf -NoSort | ForEach-Object { $result = $_ } } catch { } if ($null -ne $result) { # use cd in case it's aliased to something else: cd $result } } #.ExternalHelp PSFzf.psm1-help.xml function Invoke-FuzzyGitStatus() { $result = @() try { $headerStrings = Get-HeaderStrings $gitRoot = git rev-parse --show-toplevel git status --porcelain | Invoke-Fzf -Multi -Bind $headerStrings[1] -Header $headerStrings[0] | ForEach-Object { $result += Join-Path $gitRoot $('{0}' -f $_.Substring('?? '.Length)) } } catch { # do nothing } if ($null -ne $result) { $result } } function Enable-PsFzfAliases() { # set aliases: if (-not $DisableAliases) { SetPsFzfAliasCheck "fe" Invoke-FuzzyEdit SetPsFzfAliasCheck "fh" Invoke-FuzzyHistory SetPsFzfAliasCheck "ff" Invoke-FuzzyFasd SetPsFzfAliasCheck "fkill" Invoke-FuzzyKillProcess SetPsFzfAliasCheck "fd" Invoke-FuzzySetLocation if (${function:Set-LocationFuzzyEverything}) { SetPsFzfAliasCheck "cde" Set-LocationFuzzyEverything } SetPsFzfAliasCheck "fz" Invoke-FuzzyZLocation SetPsFzfAliasCheck "fgs" Invoke-FuzzyGitStatus } } # PSFzf.Git.ps1 $script:GitKeyHandlers = @() $script:gitPath = $null $script:bashPath = $null $script:gitPathLong = $null function SetGitKeyBindings($enable) { if ($IsLinux -or $IsMacOS) { Write-Error "Failed to register git key bindings - git bindings aren't supported on non-Windows platforms" } if ($enable) { if ($null -eq $gitPath) { $gitInfo = Get-Command git.exe -ErrorAction SilentlyContinue if ($null -ne $gitInfo) { $script:gitPathLong = Split-Path (Split-Path $gitInfo.Source -Parent) -Parent $a = New-Object -ComObject Scripting.FileSystemObject $f = $a.GetFolder($script:gitPathLong) $script:gitPath = $f.ShortPath $script:bashPath = $(Join-Path $script:gitPath "bin\bash.exe") $script:bashPath = Resolve-Path $script:bashPath } else { Write-Error "Failed to register git key bindings - git executable not found" return } } if (Get-Command Set-PSReadLineKeyHandler -ErrorAction SilentlyContinue) { @('ctrl+g,ctrl+f','Select Git Files', {Invoke-PsFzfGitFiles}), ` @('ctrl+g,ctrl+s','Select Git Hashes', {Invoke-PsFzfGitHashes}), ` @('ctrl+g,ctrl+b','Select Git Branches', {Invoke-PsFzfGitBranches}) | ForEach-Object { $script:GitKeyHandlers += $_[0] Set-PSReadLineKeyHandler -Chord $_[0] -Description $_[1] -ScriptBlock $_[2] } } else { Write-Error "Failed to register git key bindings - PSReadLine module not loaded" return } } } function RemoveGitKeyBindings() { $script:GitKeyHandlers | ForEach-Object { Remove-PSReadLineKeyHandler -Chord $_ } } function IsInGitRepo() { git rev-parse HEAD 2>&1 | Out-Null return $? } function Get-ColorAlways() { if ($RunningInWindowsTerminal) { ' --color=always' } else { '' } } function Get-HeaderStrings() { if ($RunningInWindowsTerminal) { $header = "`n`e[7mCTRL+A`e[0m Select All`t`e[7mCTRL+D`e[0m Deselect All`t`e[7mCTRL+T`e[0m Toggle All" } else { $header = "`nCTRL+A-Select All`tCTRL+D-Deselect All`tCTRL+T-Toggle All" } $keyBinds = 'ctrl-a:select-all,ctrl-d:deselect-all,ctrl-t:toggle-all' return $Header, $keyBinds } function Invoke-PsFzfGitFiles() { if (-not (IsInGitRepo)) { return } $previewCmd = "${script:bashPath} \""" + $(Join-Path $PsScriptRoot 'helpers/PsFzfGitFiles-Preview.sh') + "\"" {-1}" + $(Get-ColorAlways) + " \""$pwd\""" $result = @() $headerStrings = Get-HeaderStrings git status --short | ` Invoke-Fzf -Multi -Ansi ` -Preview "$previewCmd" -Header $headerStrings[0] -Bind $headerStrings[1] | foreach-object { $result += $_.Substring('?? '.Length) } InvokePromptHack if ($result.Length -gt 0) { $result = $result -join " " [Microsoft.PowerShell.PSConsoleReadLine]::Insert($result) } } function Invoke-PsFzfGitHashes() { if (-not (IsInGitRepo)) { return } $previewCmd = "${script:bashPath} \""" + $(Join-Path $PsScriptRoot 'helpers/PsFzfGitHashes-Preview.sh') + "\"" {}" + $(Get-ColorAlways) + " \""$pwd\""" $result = @() & git log --date=short --format="%C(green)%C(bold)%cd %C(auto)%h%d %s (%an)" $(Get-ColorAlways).Trim() | ` Invoke-Fzf -Ansi -NoSort -Multi -Bind ctrl-s:toggle-sort ` -Header 'Press CTRL-S to toggle sort' ` -Preview "$previewCmd" | ForEach-Object { if ($_ -match '\d\d-\d\d-\d\d\s+([a-f0-9]+)\s+') { $result += $Matches.1 } } InvokePromptHack if ($result.Length -gt 0) { $result = $result -join " " [Microsoft.PowerShell.PSConsoleReadLine]::Insert($result) } } function Invoke-PsFzfGitBranches() { if (-not (IsInGitRepo)) { return } $previewCmd = "${script:bashPath} \""" + $(Join-Path $PsScriptRoot 'helpers/PsFzfGitBranches-Preview.sh') + "\"" {}" + $(Get-ColorAlways) + " \""$pwd\""" $result = @() git branch -a | & "${script:gitPathLong}\usr\bin\grep.exe" -v '/HEAD\s' | ForEach-Object { $_.Substring('* '.Length) } | Sort-Object | ` Invoke-Fzf -Ansi -Multi -PreviewWindow "right:70%" -Preview "$previewCmd" | ForEach-Object { $result += $_ } [Microsoft.PowerShell.PSConsoleReadLine]::InvokePrompt() if ($result.Length -gt 0) { $result = $result -join " " [Microsoft.PowerShell.PSConsoleReadLine]::Insert($result) } } # gb() { # is_in_git_repo || return # git branch -a --color=always | grep -v '/HEAD\s' | sort | # fzf-down --ansi --multi --tac --preview-window right:70% \ # --preview 'git log --oneline --graph --date=short --color=always --pretty="format:%C(auto)%cd %h%d %s" $(sed s/^..// <<< {} | cut -d" " -f1) | head -'$LINES | # sed 's/^..//' | cut -d' ' -f1 | # sed 's#^remotes/##' # } # PSFzf.TabExpansion.ps1 # check if we're running on Windows PowerShell. This method is faster than Get-Command: if ($(get-host).Version.Major -le 5) { $script:PowershellCmd = 'powershell' } else { $script:PowershellCmd = 'pwsh' } # borrowed from https://github.com/dahlbyk/posh-git/blob/f69efd9229029519adb32e37a464b7e1533a372c/src/GitTabExpansion.ps1#L81 filter script:quoteStringWithSpecialChars { if ($_ -and ($_ -match '\s+|#|@|\$|;|,|''|\{|\}|\(|\)')) { $str = $_ -replace "'", "''" "'$str'" } else { $_ } } # taken from https://github.com/dahlbyk/posh-git/blob/2ad946347e7342199fd4bb1b42738833f68721cd/src/GitUtils.ps1#L407 function script:Get-AliasPattern($cmd) { $aliases = @($cmd) + @(Get-Alias | Where-Object { $_.Definition -eq $cmd } | Select-Object -Exp Name) "($($aliases -join '|'))" } function Expand-GitWithFzf($lastBlock) { $gitResults = Expand-GitCommand $lastBlock # if no results, invoke filesystem completion: if ($null -eq $gitResults) { $results = Invoke-Fzf -Multi | script:quoteStringWithSpecialChars } else { $results = $gitResults | Invoke-Fzf -Multi | script:quoteStringWithSpecialChars } if ($results.Count -gt 1) { $results -join ' ' } else { if (-not $null -eq $results) { $results } else { '' # output something to prevent default tab expansion } } InvokePromptHack } function Expand-FileDirectoryPath($lastWord) { # find dir and file pattern connected to the trigger: $lastWord = $lastWord.Substring(0, $lastWord.Length - 2) if ($lastWord.EndsWith('\')) { $dir = $lastWord.Substring(0, $lastWord.Length - 1) $file = $null } elseif (-not [string]::IsNullOrWhiteSpace($lastWord)) { $dir = Split-Path $lastWord -Parent $file = Split-Path $lastWord -Leaf } if (-not [System.IO.Path]::IsPathRooted($dir)) { $dir = Join-Path $PWD.ProviderPath $dir } $prevPath = $Pwd.ProviderPath try { if (-not [string]::IsNullOrEmpty($dir)) { Set-Location $dir } if (-not [string]::IsNullOrEmpty($file)) { Invoke-Fzf -Query $file } else { Invoke-Fzf } } finally { Set-Location $prevPath } InvokePromptHack } $script:TabExpansionEnabled = $false function SetTabExpansion($enable) { if ($enable) { if (-not $script:TabExpansionEnabled) { $script:TabExpansionEnabled = $true RegisterBuiltinCompleters Register-ArgumentCompleter -CommandName git,tgit,gitk -Native -ScriptBlock { param($wordToComplete, $commandAst, $cursorPosition) # The PowerShell completion has a habit of stripping the trailing space when completing: # git checkout <tab> # The Expand-GitCommand expects this trailing space, so pad with a space if necessary. $padLength = $cursorPosition - $commandAst.Extent.StartOffset $textToComplete = $commandAst.ToString().PadRight($padLength, ' ').Substring(0, $padLength) Expand-GitCommandPsFzf $textToComplete } } } else { if ($script:TabExpansionEnabled) { $script:TabExpansionEnabled = $false } } } function CheckFzfTrigger { param($commandName, $parameterName, $wordToComplete, $commandAst, $cursorPosition,$action) if ([string]::IsNullOrWhiteSpace($env:FZF_COMPLETION_TRIGGER)) { $completionTrigger = '**' } else { $completionTrigger = $env:FZF_COMPLETION_TRIGGER } if ($wordToComplete.EndsWith($completionTrigger)) { $wordToComplete = $wordToComplete.Substring(0, $wordToComplete.Length - $completionTrigger.Length) $wordToComplete } } function GetServiceSelection() { param( [scriptblock] $ResultAction ) $header = [System.Environment]::NewLine + $("{0,-24} NAME" -f "DISPLAYNAME") + [System.Environment]::NewLine $result = Get-Service | Where-Object { ![string]::IsNullOrEmpty($_.Name) } | ForEach-Object { "{0,-24} {1}" -f $_.DisplayName.Substring(0,[System.Math]::Min(24,$_.DisplayName.Length)),$_.Name } | Invoke-Fzf -Multi -Header $header $result | ForEach-Object { &$ResultAction $_ } } function RegisterBuiltinCompleters { $processIdOrNameScriptBlock = { param($commandName, $parameterName, $wordToComplete, $commandAst, $cursorPosition, $action) $wordToComplete = CheckFzfTrigger $commandName $parameterName $wordToComplete $commandAst $cursorPosition if ($null -ne $wordToComplete) { if ($parameterName -eq 'Name') { $group = '$2' } elseif ($parameterName -eq 'Id') { $group = '$1' } $script:resultArr = @() GetProcessSelection -ResultAction { param($result) $script:resultArr += $result -replace "([0-9]+)\s*(.*)",$group } if ($script:resultArr.Length -ge 1) { $script:resultArr -join ', ' } InvokePromptHack } else { # don't return anything - let normal tab completion work } } 'Get-Process','Stop-Process' | ForEach-Object { Register-ArgumentCompleter -CommandName $_ -ParameterName "Name" -ScriptBlock $processIdOrNameScriptBlock Register-ArgumentCompleter -CommandName $_ -ParameterName "Id" -ScriptBlock $processIdOrNameScriptBlock } $serviceNameScriptBlock = { param($commandName, $parameterName, $wordToComplete, $commandAst, $cursorPosition,$action) $wordToComplete = CheckFzfTrigger $commandName $parameterName $wordToComplete $commandAst $cursorPosition if ($null -ne $wordToComplete) { if ($parameterName -eq 'Name') { $group = '$2' } elseif ($parameterName -eq 'DisplayName') { $group = '$1' } $script:resultArr = @() GetServiceSelection -ResultAction { param($result) $script:resultArr += $result.Substring(24+1) } if ($script:resultArr.Length -ge 1) { $script:resultArr -join ', ' } InvokePromptHack } else { # don't return anything - let normal tab completion work } } 'Get-Service','Start-Service','Stop-Service' | ForEach-Object { Register-ArgumentCompleter -CommandName $_ -ParameterName "Name" -ScriptBlock $serviceNameScriptBlock Register-ArgumentCompleter -CommandName $_ -ParameterName "DisplayName" -ScriptBlock $serviceNameScriptBlock } } function Expand-GitCommandPsFzf($lastWord) { if ([string]::IsNullOrWhiteSpace($env:FZF_COMPLETION_TRIGGER)) { $completionTrigger = '**' } else { $completionTrigger = $env:FZF_COMPLETION_TRIGGER } if ($lastWord.EndsWith($completionTrigger)) { $lastWord = $lastWord.Substring(0, $lastWord.Length - $completionTrigger.Length) Expand-GitWithFzf $lastWord } else { Expand-GitCommand $lastWord } } function Invoke-FzfTabCompletion() { $script:continueCompletion = $true do { $script:continueCompletion = script:Invoke-FzfTabCompletionInner } while ($script:continueCompletion) } function script:Invoke-FzfTabCompletionInner() { $script:result = @() [void] [System.Reflection.Assembly]::LoadWithPartialName("System.Management.Automation") $line = $null $cursor = $null [Microsoft.PowerShell.PSConsoleReadline]::GetBufferState([ref]$line, [ref]$cursor) if ($cursor -lt 0 -or [string]::IsNullOrWhiteSpace($line)) { return $false } $completions = [System.Management.Automation.CommandCompletion]::CompleteInput($line, $cursor, @{}) $completionMatches = $completions.CompletionMatches if ($completionMatches.Count -le 0) { return $false } $script:continueCompletion = $false $addSpace = $null -ne $currentPath -and $currentPath.StartsWith(" ") if ($completionMatches.Count -eq 1) { $script:result = $completionMatches[0].CompletionText } elseif ($completionMatches.Count -gt 1) { $helpers = New-Object PSFzf.IO.CompletionHelpers $ambiguous = $false $addSpace = $false $prefix = $helpers.GetUnambiguousPrefix($completionMatches,[ref]$ambiguous) $script:result = @() $script:checkCompletion = $true $expectTrigger = $script:TabContinuousTrigger # need to escape the key if it's a forward slash: if ($expectTrigger -eq '\') { $expectTrigger += $expectTrigger } # normalize so path works correctly for Windows: $path = $PWD.ProviderPath.Replace('\','/') $previewScript = $(Join-Path $PsScriptRoot 'helpers/PsFzfTabExpansion-Preview.ps1') $additionalCmd = @{ Preview=$($script:PowershellCmd + " -NoProfile -NonInteractive -File \""$previewScript\"" \""" + $path + "\"" {}") } $script:fzfOutput = @() $completionMatches | ForEach-Object { $_.CompletionText } | Invoke-Fzf ` -Layout reverse ` -Expect $expectTrigger ` -Query "$prefix" ` -PrintQuery ` -Bind 'tab:down,btab:up' ` @additionalCmd | ForEach-Object { if ($null -eq $_) { $script:fzfOutput += "" } else { $script:fzfOutput += $_ } } # check if there's a selection: if ($script:fzfOutput.Length -gt 2) { $script:result = $script:fzfOutput[2..($script:fzfOutput.Length-1)] } # or just complete with the query string: else { $script:result = $script:fzfOutput[0] } # check if we should continue completion: $script:continueCompletion = $script:fzfOutput[1] -eq $script:TabContinuousTrigger InvokePromptHack } $result = $script:result if ($null -ne $result) { # quote strings if we need to: if ($result -is [system.array]) { for ($i = 0;$i -lt $result.Length;$i++) { $result[$i] = FixCompletionResult $result[$i] } $str = $result -join ',' } else { $str = FixCompletionResult $result } if ($script:continueCompletion) { $isQuoted = $str.EndsWith("'") $resultTrimmed = $str.Trim(@('''','"')) if (Test-Path "$resultTrimmed" -PathType Container) { if ($isQuoted) { $str = "'{0}{1}'" -f "$resultTrimmed",$script:TabContinuousTrigger } else { $str = "$resultTrimmed" + $script:TabContinuousTrigger } } else { # no more paths to complete, so let's stop completion: $str += ' ' $script:continueCompletion = $false } } if ($addSpace) { $str = ' ' + $str } $leftCursor = $completions.ReplacementIndex $replacementLength = $completions.ReplacementLength if ($leftCursor -le 0 -and $replacementLength -le 0) { [Microsoft.PowerShell.PSConsoleReadLine]::Insert($str) } else { [Microsoft.PowerShell.PSConsoleReadLine]::Replace($leftCursor,$replacementLength,$str) } } return $script:continueCompletion } |