1 2 using namespace System.Management.Automation 3 using namespace System.Management.Automation.Language 4 5 Register-ArgumentCompleter -Native -CommandName 'my-app' -ScriptBlock { 6 param($wordToComplete, $commandAst, $cursorPosition) 7 8 $commandElements = $commandAst.CommandElements 9 $command = @( 10 'my-app' 11 for ($i = 1; $i -lt $commandElements.Count; $i++) { 12 $element = $commandElements[$i] 13 if ($element -isnot [StringConstantExpressionAst] -or 14 $element.StringConstantType -ne [StringConstantType]::BareWord -or 15 $element.Value.StartsWith('-') -or 16 $element.Value -eq $wordToComplete) { 17 break 18 } 19 $element.Value 20 }) -join ';' 21 22 $completions = @(switch ($command) { 23 'my-app' { 24 [CompletionResult]::new('-c', 'c', [CompletionResultType]::ParameterName, 'some config file') 25 [CompletionResult]::new('-C', 'C', [CompletionResultType]::ParameterName, 'some config file') 26 [CompletionResult]::new('--config', 'config', [CompletionResultType]::ParameterName, 'some config file') 27 [CompletionResult]::new('--conf', 'conf', [CompletionResultType]::ParameterName, 'some config file') 28 [CompletionResult]::new('-h', 'h', [CompletionResultType]::ParameterName, 'Print help') 29 [CompletionResult]::new('--help', 'help', [CompletionResultType]::ParameterName, 'Print help') 30 [CompletionResult]::new('-V', 'V', [CompletionResultType]::ParameterName, 'Print version') 31 [CompletionResult]::new('--version', 'version', [CompletionResultType]::ParameterName, 'Print version') 32 [CompletionResult]::new('test', 'test', [CompletionResultType]::ParameterValue, 'tests things') 33 [CompletionResult]::new('some_cmd', 'some_cmd', [CompletionResultType]::ParameterValue, 'top level subcommand') 34 [CompletionResult]::new('help', 'help', [CompletionResultType]::ParameterValue, 'Print this message or the help of the given subcommand(s)') 35 break 36 } 37 'my-app;test' { 38 [CompletionResult]::new('--case', 'case', [CompletionResultType]::ParameterName, 'the case to test') 39 [CompletionResult]::new('-h', 'h', [CompletionResultType]::ParameterName, 'Print help') 40 [CompletionResult]::new('--help', 'help', [CompletionResultType]::ParameterName, 'Print help') 41 [CompletionResult]::new('-V', 'V', [CompletionResultType]::ParameterName, 'Print version') 42 [CompletionResult]::new('--version', 'version', [CompletionResultType]::ParameterName, 'Print version') 43 break 44 } 45 'my-app;some_cmd' { 46 [CompletionResult]::new('-h', 'h', [CompletionResultType]::ParameterName, 'Print help') 47 [CompletionResult]::new('--help', 'help', [CompletionResultType]::ParameterName, 'Print help') 48 [CompletionResult]::new('-V', 'V', [CompletionResultType]::ParameterName, 'Print version') 49 [CompletionResult]::new('--version', 'version', [CompletionResultType]::ParameterName, 'Print version') 50 [CompletionResult]::new('sub_cmd', 'sub_cmd', [CompletionResultType]::ParameterValue, 'sub-subcommand') 51 [CompletionResult]::new('help', 'help', [CompletionResultType]::ParameterValue, 'Print this message or the help of the given subcommand(s)') 52 break 53 } 54 'my-app;some_cmd;sub_cmd' { 55 [CompletionResult]::new('--config', 'config', [CompletionResultType]::ParameterName, 'the other case to test') 56 [CompletionResult]::new('-h', 'h', [CompletionResultType]::ParameterName, 'Print help (see more with ''--help'')') 57 [CompletionResult]::new('--help', 'help', [CompletionResultType]::ParameterName, 'Print help (see more with ''--help'')') 58 [CompletionResult]::new('-V', 'V', [CompletionResultType]::ParameterName, 'Print version') 59 [CompletionResult]::new('--version', 'version', [CompletionResultType]::ParameterName, 'Print version') 60 break 61 } 62 'my-app;some_cmd;help' { 63 [CompletionResult]::new('sub_cmd', 'sub_cmd', [CompletionResultType]::ParameterValue, 'sub-subcommand') 64 [CompletionResult]::new('help', 'help', [CompletionResultType]::ParameterValue, 'Print this message or the help of the given subcommand(s)') 65 break 66 } 67 'my-app;some_cmd;help;sub_cmd' { 68 break 69 } 70 'my-app;some_cmd;help;help' { 71 break 72 } 73 'my-app;help' { 74 [CompletionResult]::new('test', 'test', [CompletionResultType]::ParameterValue, 'tests things') 75 [CompletionResult]::new('some_cmd', 'some_cmd', [CompletionResultType]::ParameterValue, 'top level subcommand') 76 [CompletionResult]::new('help', 'help', [CompletionResultType]::ParameterValue, 'Print this message or the help of the given subcommand(s)') 77 break 78 } 79 'my-app;help;test' { 80 break 81 } 82 'my-app;help;some_cmd' { 83 [CompletionResult]::new('sub_cmd', 'sub_cmd', [CompletionResultType]::ParameterValue, 'sub-subcommand') 84 break 85 } 86 'my-app;help;some_cmd;sub_cmd' { 87 break 88 } 89 'my-app;help;help' { 90 break 91 } 92 }) 93 94 $completions.Where{ $_.CompletionText -like "$wordToComplete*" } | 95 Sort-Object -Property ListItemText 96 } 97