feat: add completion scripts (#411)

1 files changed, 79 insertions, 0 deletions

diff --git a/scripts/completions/aichat.ps1 b/scripts/completions/aichat.ps1
new file mode 100644
index 0000000..7d5a7cf
--- /dev/null
+++ b/scripts/completions/aichat.ps1
@@ -0,0 +1,79 @@
+using namespace System.Management.Automation
+using namespace System.Management.Automation.Language
+
+Register-ArgumentCompleter -Native -CommandName 'aichat' -ScriptBlock {
+    param($wordToComplete, $commandAst, $cursorPosition)
+
+    $commandElements = $commandAst.CommandElements
+    $command = @(
+        'aichat'
+        for ($i = 1; $i -lt $commandElements.Count; $i++) {
+            $element = $commandElements[$i]
+            if ($element -isnot [StringConstantExpressionAst] -or
+                $element.StringConstantType -ne [StringConstantType]::BareWord -or
+                $element.Value.StartsWith('-') -or
+                $element.Value -eq $wordToComplete) {
+                break
+        }
+        $element.Value
+    }) -join ';'
+
+    $completions = @(switch ($command) {
+        'aichat' {
+            [CompletionResult]::new('-m', '-m', [CompletionResultType]::ParameterName, 'Choose a LLM model')
+            [CompletionResult]::new('--model', '--model', [CompletionResultType]::ParameterName, 'Choose a LLM model')
+            [CompletionResult]::new('-r', '-r', [CompletionResultType]::ParameterName, 'Choose a role')
+            [CompletionResult]::new('--role', '--role', [CompletionResultType]::ParameterName, 'Choose a role')
+            [CompletionResult]::new('-s', '-s', [CompletionResultType]::ParameterName, 'Create or reuse a session')
+            [CompletionResult]::new('--session', '--session', [CompletionResultType]::ParameterName, 'Create or reuse a session')
+            [CompletionResult]::new('-f', '-f', [CompletionResultType]::ParameterName, 'Attach files to the message')
+            [CompletionResult]::new('--file', '--file', [CompletionResultType]::ParameterName, 'Attach files to the message')
+            [CompletionResult]::new('-w', '-w', [CompletionResultType]::ParameterName, 'Specify the text-wrapping mode (no, auto, <max-width>)')
+            [CompletionResult]::new('--wrap', '--wrap', [CompletionResultType]::ParameterName, 'Specify the text-wrapping mode (no, auto, <max-width>)')
+            [CompletionResult]::new('--save-session', '--save-session', [CompletionResultType]::ParameterName, 'Whether to save the session')
+            [CompletionResult]::new('-e', '-e', [CompletionResultType]::ParameterName, 'Execute commands using natural language')
+            [CompletionResult]::new('--execute', '--execute', [CompletionResultType]::ParameterName, 'Execute commands using natural language')
+            [CompletionResult]::new('-c', '-c', [CompletionResultType]::ParameterName, 'Generate only code')
+            [CompletionResult]::new('--code', '--code', [CompletionResultType]::ParameterName, 'Generate only code')
+            [CompletionResult]::new('-H', '-H', [CompletionResultType]::ParameterName, 'Disable syntax highlighting')
+            [CompletionResult]::new('--no-highlight', '--no-highlight', [CompletionResultType]::ParameterName, 'Disable syntax highlighting')
+            [CompletionResult]::new('-S', '-S', [CompletionResultType]::ParameterName, 'No stream output')
+            [CompletionResult]::new('--no-stream', '--no-stream', [CompletionResultType]::ParameterName, 'No stream output')
+            [CompletionResult]::new('--light-theme', '--light-theme', [CompletionResultType]::ParameterName, 'Use light theme')
+            [CompletionResult]::new('--dry-run', '--dry-run', [CompletionResultType]::ParameterName, 'Run in dry run mode')
+            [CompletionResult]::new('--info', '--info', [CompletionResultType]::ParameterName, 'Print related information')
+            [CompletionResult]::new('--list-models', '--list-models', [CompletionResultType]::ParameterName, 'List all available models')
+            [CompletionResult]::new('--list-roles', '--list-roles', [CompletionResultType]::ParameterName, 'List all available roles')
+            [CompletionResult]::new('--list-sessions', '--list-sessions', [CompletionResultType]::ParameterName, 'List all available sessions')
+            [CompletionResult]::new('-h', '-h', [CompletionResultType]::ParameterName, 'Print help')
+            [CompletionResult]::new('--help', '--help', [CompletionResultType]::ParameterName, 'Print help')
+            [CompletionResult]::new('-V', '-V', [CompletionResultType]::ParameterName, 'Print version')
+            [CompletionResult]::new('--version', '--version', [CompletionResultType]::ParameterName, 'Print version')
+            break
+        }
+    })
+
+    function Get-AichatValues($arg) {
+        $(aichat $arg) -split '\n' | ForEach-Object { [CompletionResult]::new($_) }
+    }
+
+    if ($commandElements.Count -gt 1) {
+        $offset=2
+        if ($wordToComplete -eq "") {
+            $offset=1
+        }
+        $flag = $commandElements[$commandElements.Count-$offset].ToString()
+        if ($flag -eq "-m" -or $flag -eq "--model") {
+            $completions = Get-AichatValues "--list-models"
+        } elseif ($flag -eq "-r" -or $flag -eq "--role") {
+            $completions = Get-AichatValues "--list-roles"
+        } elseif ($flag -eq "-s" -or $flag -eq "--session") {
+            $completions = Get-AichatValues "--list-sessions"
+        } elseif ($flag -eq "-f" -or $flag -eq "--file") {
+            $completions = @()
+        }
+    }
+
+    $completions.Where{ $_.CompletionText -like "$wordToComplete*" } |
+        Sort-Object -Property ListItemText
+}