PSFunctionInfo.psm1
|
Get-ChildItem $PSScriptRoot\functions\*.ps1 | ForEach-Object { . $_.FullName } #load defaults if found $defaults = Join-Path $home -ChildPath psfunctioninfo-defaults.json if (Test-Path -path $defaults) { $d = Get-Content -Path $defaults | ConvertFrom-JSON $d.PSObject.properties | Foreach-Object { $global:PSDefaultParameterValues["New-PSFunctionInfo:$($_.name)"] = $_.value } } #Add VSCode Shortcuts if ($host.name -eq 'Visual Studio Code Host') { $global:PSDefaultParameterValues["New-PSFunctionInfo:Path"] = {$PSEditor.GetEditorContext().CurrentFile.Path} #create an argument completer for the Name parameter Register-ArgumentCompleter -CommandName New-PSFunctionInfo -ParameterName Name -ScriptBlock { param($commandName, $parameterName, $WordToComplete, $commandAst, $fakeBoundParameter) #PowerShell code to populate $WordToComplete $ASTTokens = $PSEditor.GetEditorContext().CurrentFile.tokens $NameList = @() for ($i=0;$i -lt $ASTTokens.count;$i++) { if ($ASTTokens[$i].text -eq 'function') { $NameList+= $ASTTokens[$i+1].text } } $NameList | Where-Object {$_ -like "$WordToComplete*"} | ForEach-Object { # completion text,listItem text,result type,Tooltip [System.Management.Automation.CompletionResult]::new($_, $_, 'ParameterValue', $_) } } } elseif ($host.name -eq 'Windows PowerShell ISE Host') { #create shortcut for the ISE $global:PSDefaultParameterValues["New-PSFunctionInfo:Path"] = {$PSIse.CurrentFile.FullPath} #add a menu item $sb = { Function PickList { # A WPF function picker for the PowerShell ISE Param([string[]]$Name) $title = "New-PSFunctionInfo" $form = New-Object System.Windows.Window #define what it looks like $form.Title = $Title $form.Height = 200 $form.Width = 300 $form.WindowStartupLocation = [System.Windows.WindowStartupLocation]::CenterScreen $grid = New-Object System.Windows.Controls.Grid $label = New-Object System.Windows.Controls.Label $label.Content = "Select a function from the list and click OK." $label.HorizontalAlignment = "left" $label.VerticalAlignment = "top" $grid.AddChild($label) $combo = New-Object System.Windows.Controls.ListBox $combo.AllowDrop = $true $combo.Width = 160 $combo.Height = 75 $combo.HorizontalAlignment = "left" $combo.VerticalAlignment = "top" $combo.Margin = "15,25,0,0" foreach ($item in $Name) { [void]$combo.items.Add($item) } $combo.SelectedIndex = 0 $grid.AddChild($combo) #initialize an array to hold all processed objects $ok = New-Object System.windows.controls.button $ok.Content = "_OK" $ok.IsDefault = $True $ok.Width = 75 $ok.Height = 25 $ok.HorizontalAlignment = "left" $ok.Margin = "25,100,0,0" $OK.Add_Click( { $global:SelectedPickItem = $combo.SelectedItem #$combo.SelectedItem | Out-Default $form.Close() }) $grid.AddChild($ok) $cancel = New-Object System.Windows.Controls.Button $cancel.Content = "_Cancel" $cancel.Width = 75 $cancel.Height = 25 #$cancel.HorizontalAlignment = "right" $cancel.Margin = "150,100,0,0" $cancel.Add_Click( { $form.Close() }) $grid.AddChild($cancel) $form.AddChild($grid) [void]($form.ShowDialog()) } if ($PSIse.CurrentFile.IsSaved) { $Path = $PSIse.CurrentFile.FullPath New-Variable ASTTokens -Force New-Variable astErr -Force $AST = [System.Management.Automation.Language.Parser]::ParseFile($Path, [ref]$ASTTokens, [ref]$astErr) $file = [System.Collections.Generic.list[string[]]]::new() Get-Content $path | ForEach-Object { $file.Add($_) } $name = @() for ($i = 0; $i -lt $ASTTokens.count; $i++) { if ($ASTTokens[$i].text -eq 'function') { $name += $ASTTokens[$i + 1].text } } #for if ($name) { Remove-Variable -Name SelectedPickItem -ErrorAction SilentlyContinue PickList -name $name #| Out-GridView -Title "Select a function" -OutputMode Single if ($SelectedPickItem) { New-PSFunctionInfo -Name $SelectedPickItem -path $path $idx = $file.findIndex( { $args[0] -match "[Ff]unction(\s+)$SelectedPickItem" }) $idx++ #save and re-open the file [void]$PSIse.CurrentPowerShellTab.files.Remove($PSIse.CurrentFile) [void]$PSIse.CurrentPowerShellTab.Files.Add($path) $r = $PSIse.CurrentPowerShellTab.Files.where( { $_.FullPath -eq $path }) $r.Editor.Focus() if ($idx -ge 0) { $r.Editor.SetCaretPosition($idx, 1) $r.Editor.SelectCaretLine() $r.Editor.EnsureVisible($idx) } } } #if names found else { [System.Windows.MessageBox]::Show("No matching functions found in $Path", "New-PSFunctionInfo", "OK", "Information") } } #if saved else { [System.Windows.MessageBox]::Show("Please save your file first:$Path and then try again.", "New-PSFunctionInfo", "OK", "Warning") } } $PSIse.CurrentPowerShellTab.AddOnsMenu.Submenus.add("New-PSFunctionInfo", $sb, $Null) } #create an argument completer Register-ArgumentCompleter -CommandName Get-PSFunctionInfo -ParameterName FunctionName -ScriptBlock { param($commandName, $parameterName, $WordToComplete, $commandAst, $fakeBoundParameter) Get-ChildItem -path "Function:\$WordToComplete*" | ForEach-Object { # completion text,listItem text,result type,Tooltip [System.Management.Automation.CompletionResult]::new($_.name, $_.name, 'ParameterValue', $_.name) } } Register-ArgumentCompleter -CommandName Get-PSFunctionInfo -ParameterName Tag -ScriptBlock { param($commandName, $parameterName, $WordToComplete, $commandAst, $fakeBoundParameter) (Get-PSFunctionInfoTag).where({$_ -like "$WordToComplete*"}) | ForEach-Object { # completion text,listItem text,result type,Tooltip [System.Management.Automation.CompletionResult]::new($_, $_, 'ParameterValue', $_) } } |