1
2using namespace System.Management.Automation
3using namespace System.Management.Automation.Language
4
5Register-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