Private/ArgumentCompleters.ps1
|
using namespace System.Collections using namespace System.Collections.Generic using namespace System.Management.Automation using namespace System.Management.Automation.Language function Get-ModuleProjectArgumentCompleter { param ( [string] $CommandName , [string] $ParameterName, [string] $WordToComplete, [CommandAst] $CommandAst, [IDictionary] $FakeBoundParameters ) $Choices = [List[String]]::new() Get-ValidModuleProjectNames | Where-Object {$_ -like "$WordToComplete*"} | ForEach-Object { $Choices.Add("$_") } if (!$Choices) { throw [InvalidOperationException] 'No Modules Exist!' } else { return @($Choices) } } function Get-ApprovedVerbsArgumentCompleter { param ( [string] $CommandName , [string] $ParameterName, [string] $WordToComplete, [CommandAst] $CommandAst, [IDictionary] $FakeBoundParameters ) $Choices = [List[String]]::new() Get-ApprovedVerbs | Where-Object {$_ -like "$WordToComplete*"} | ForEach-Object { $Choices.Add("$_-") } return $Choices } function Get-CommandFromModuleArgumentCompleter { param ( [string] $CommandName , [string] $ParameterName, [string] $WordToComplete, [CommandAst] $CommandAst, [IDictionary] $FakeBoundParameters ) $ModuleProject = if ($FakeBoundParameters.Contains('ModuleProject')) { $FakeBoundParameters['ModuleProject'] } elseif ($FakeBoundParameters.Contains('SourceModuleProject')) { $FakeBoundParameters['SourceModuleProject'] } if($ModuleProject) { $Choices = @() $ModuleProjects = Get-ValidModuleProjectNames if ($ModuleProject -in $ModuleProjects) { [Array]$Functions = Get-ModuleProjectFunctionNames -ModuleProject $ModuleProject | Where-Object {$_ -like "$WordToComplete*"} if ($Functions) { $Choices += $Functions } [Array]$Aliases = Get-ModuleProjectAliasNames -ModuleProject $ModuleProject | Where-Object {$_ -like "$WordToComplete*"} if ($Aliases) { $Choices += $Aliases } if (!$Choices) { throw [InvalidOperationException] 'No Matching Commands Exist in Module!' } else { return @($Choices) } } else { throw [InvalidOperationException] 'No Modules Exist!' } } } <#TODO: Test#> function Get-NewCommandFromModuleArgumentCompleter { param ( [string] $CommandName , [string] $ParameterName, [string] $WordToComplete, [CommandAst] $CommandAst, [IDictionary] $FakeBoundParameters ) $ModuleProject = if ($FakeBoundParameters.Contains('ModuleProject')) { $FakeBoundParameters['ModuleProject'] } elseif ($FakeBoundParameters.Contains('SourceModuleProject')) { $FakeBoundParameters['SourceModuleProject'] } $CommandName = if ($FakeBoundParameters.Contains('CommandName')) { $FakeBoundParameters['CommandName'] } else {''} if($ModuleProject -and $CommandName) { $Choices = @() $ModuleProjects = Get-ValidModuleProjectNames if ($ModuleProject -in $ModuleProjects) { $Functions = Get-ModuleProjectFunctionNames -ModuleProject $ModuleProject | Where-Object {$_ -EQ $CommandName} if ($Functions) { Get-ApprovedVerbs | Where-Object {$_ -like "$WordToComplete*"} | ForEach-Object { $Choices += @("$_-") } } return @($Choices) } else { throw [InvalidOperationException] 'No Modules Exist!' } } } |