Private/completerScriptBlock.ps1
|
$Script:argCompFunction = { param($wordToComplete, $commandAst, $cursorPosition) <# [String]$completionLogFile = "$env:LOCALAPPDATA\WSLTabCompletion\completion.log" New-Item -ItemType Directory -Path "$env:LOCALAPPDATA\WSLTabCompletion" -ErrorAction SilentlyContinue | Out-Null #> # Adjust the cursor position in case there is extra whitespace # at the beginning of the command [int]$cursorPosition = $cursorPosition - $commandAst.Extent.StartOffset # The command string is the entire command line, minus leading whitespace [String]$commandString = $commandAst.Extent.ToString() # Array of each token on the command line [String[]]$compTokens = $commandAst.CommandElements.ForEach({$_.Extent.Text}) # Get the completion pattern string for the current commandline # See Get-CompletionPattern for the format [String]$compPattern = Get-CompletionPattern -Tokens $compTokens -WordToComplete $wordToComplete <# [String]$completionLogFile = "$env:LOCALAPPDATA\WSLTabCompletion\completion.log" New-Item -ItemType Directory -Path "$env:LOCALAPPDATA\WSLTabCompletion" -ErrorAction SilentlyContinue | Out-Null $astDump = ($commandAst | Out-String) -split "`r`n" | Where-Object { $_ -ne "" } $indentedAstString = ($astDump | ForEach-Object { " ${_}"}) -join "`r`n" ( "-----------------", "", "wordToComplete: '${wordToComplete}'", "cursorPosition: ${cursorPosition}", "Extent.StartOffset: $($commandAst.Extent.StartOffset)", "commandAst:", $indentedAstString, "commandString: '${commandString}'", "" ) -join "`r`n" | Out-File -Append -FilePath $completionLogFile #> if ($cursorPosition -lt $commandString.Length) { # Don't offer completions if the cursor isn't at the end of the line return } switch -Regex ($compPattern) { # WSL-only, no arguments or values yet # Then offer any matching flags or commands for completion '^e $' { return Get-WSLArgumentName } # Any command in the first position # Then only offer the completion function for # that command or the default completer '^ec.*$' { [String]$commandFlag = $compTokens[1] if ($flags[$commandFlag].completionFunction) { return (& $flags[$commandFlag].completionFunction) } else { return } } '^ep$' { # If the first token is the beginning of a flag or command # Then offer any matching flags or commands for completion return (& Get-WSLArgumentName $wordToComplete) } '^e.*fv$' { # If we're completing a partial value for a flag [String]$flag = $compTokens[-2] return (& Get-WSLArgumentValue $flag $wordToComplete) } '^e.*f $' { # If the last token is a flag followed by a space, # then we need to check whether it has a value. If # so, complete the value. [String]$flag = $compTokens[-1] if ($flags[$flag].hasValue) { if ($flags[$flag].completionFunction) { return (& $flags[$flag].completionFunction $wordToComplete) } } else { # Otherwise, offer possible flag completions. # Do not offer commands nor flags that have # already been used. return Get-WSLArgumentName -OnlyFlags -Tokens $compTokens -ArgumentNamePartial $wordToComplete } break } '^e.*fv $' { # If we're ending in a value, then the next token # should be a flag. Note that we've already # processed commands at this point. return Get-WSLArgumentName -OnlyFlags -Tokens $compTokens } '^e.*p$' { return Get-WSLArgumentName ` -OnlyFlags ` -Tokens $compTokens ` -ArgumentNamePartial $wordToComplete } } } |