diff options
Diffstat (limited to 'scripts/jv_completion.ps1')
| -rw-r--r-- | scripts/jv_completion.ps1 | 204 |
1 files changed, 204 insertions, 0 deletions
diff --git a/scripts/jv_completion.ps1 b/scripts/jv_completion.ps1 new file mode 100644 index 0000000..79d8df3 --- /dev/null +++ b/scripts/jv_completion.ps1 @@ -0,0 +1,204 @@ +# JustEnoughVCS PowerShell Completion +Register-ArgumentCompleter -Native -CommandName jv -ScriptBlock { + param($wordToComplete, $commandAst, $cursorPosition) + + $words = $commandAst.CommandElements | ForEach-Object { $_.ToString() } + $currentIndex = $words.IndexOf($wordToComplete) + if ($currentIndex -lt 0) { $currentIndex = $words.Count } + + $cmd = "jv" + $subcmd = if ($words.Count -gt 1) { $words[1] } else { $null } + $subsubcmd = if ($words.Count -gt 2) { $words[2] } else { $null } + + # Base commands + $baseCommands = @( + "create", "init", "direct", "unstain", "account", "update", + "sheet", "status", "here", "import", "export", "in", "out", + "move", "mv", "docs", "exit", "use", "sheets", "accounts", + "as", "make", "drop", "track", "hold", "throw", "login", + "jump", "align" + ) + + # Account subcommands + $accountCommands = @("list", "as", "add", "remove", "movekey", "mvkey", "mvk", "genpub", "help") + + # Sheet subcommands + $sheetCommands = @("list", "use", "exit", "make", "drop", "help", "align") + + # Completion for main command + if ($currentIndex -eq 1) { + return $baseCommands | Where-Object { $_ -like "$wordToComplete*" } + } + + # Completion for account command + if ($subcmd -eq "account" -or $subcmd -eq "acc") { + if ($currentIndex -eq 2) { + return $accountCommands | Where-Object { $_ -like "$wordToComplete*" } + } + + switch ($subsubcmd) { + { @("as", "remove", "mvkey", "mvk", "movekey", "genpub") -contains $_ } { + if ($currentIndex -eq 3) { + $accounts = & $cmd account list --raw 2>$null + return $accounts | Where-Object { $_ -like "$wordToComplete*" } + } elseif ($currentIndex -eq 4 -and (@("mvkey", "mvk", "movekey", "genpub") -contains $subsubcmd)) { + # File completion for key files + return Get-ChildItem -Name -File -Path "." | Where-Object { $_ -like "$wordToComplete*" } + } + } + "add" { + if ($currentIndex -eq 3) { + # No completion for account name + return @() + } elseif ($currentIndex -eq 4 -and $wordToComplete.StartsWith("-")) { + # Complete --keygen option + if ("--keygen" -like "$wordToComplete*") { + return "--keygen" + } + } + } + { @("-", "rm") -contains $_ } { + if ($currentIndex -eq 3) { + $accounts = & $cmd account list --raw 2>$null + return $accounts | Where-Object { $_ -like "$wordToComplete*" } + } + } + } + return @() + } + + # Completion for sheet command + if ($subcmd -eq "sheet" -or $subcmd -eq "sh") { + if ($currentIndex -eq 2) { + return $sheetCommands | Where-Object { $_ -like "$wordToComplete*" } + } + + switch ($subsubcmd) { + { @("use", "drop") -contains $_ } { + if ($currentIndex -eq 3) { + $sheets = & $cmd sheet list --raw 2>$null + return $sheets | Where-Object { $_ -like "$wordToComplete*" } + } + } + "make" { + if ($currentIndex -eq 3) { + $allSheets = & $cmd sheet list --all --raw 2>$null + return $allSheets | Where-Object { $_ -like "$wordToComplete*" } + } + } + "align" { + if ($currentIndex -eq 3) { + $alignItems = @("lost", "moved") + $unsolvedItems = & $cmd sheet align --unsolved --raw 2>$null + $completions = $alignItems + $unsolvedItems + return $completions | Where-Object { $_ -like "$wordToComplete*" } + } elseif ($currentIndex -eq 4) { + $item = $words[3] + $alignOperations = @() + $createdItems = & $cmd sheet align --created --raw 2>$null + + if ($item -eq "lost") { + $alignOperations = @("confirm") + } elseif ($item -like "lost:*") { + $alignOperations = @("confirm") + $createdItems + } elseif ($item -eq "moved" -or $item -like "moved:*") { + $alignOperations = @("local", "remote") + } else { + $alignOperations = @("local", "remote", "confirm") + $createdItems + } + + return $alignOperations | Where-Object { $_ -like "$wordToComplete*" } + } + } + } + return @() + } + + # Completion for align command + if ($subcmd -eq "align") { + if ($currentIndex -eq 2) { + $alignItems = @("lost", "moved") + $unsolvedItems = & $cmd sheet align --unsolved --raw 2>$null + $completions = $alignItems + $unsolvedItems + return $completions | Where-Object { $_ -like "$wordToComplete*" } + } elseif ($currentIndex -eq 3) { + $item = $words[2] + $alignOperations = @() + $createdItems = & $cmd sheet align --created --raw 2>$null + + if ($item -eq "lost") { + $alignOperations = @("confirm") + } elseif ($item -like "lost:*") { + $alignOperations = @("confirm") + $createdItems + } elseif ($item -eq "moved" -or $item -like "moved:*") { + $alignOperations = @("local", "remote") + } else { + $alignOperations = @("local", "remote", "confirm") + $createdItems + } + + return $alignOperations | Where-Object { $_ -like "$wordToComplete*" } + } + return @() + } + + # Completion for login command + if ($subcmd -eq "login") { + if ($currentIndex -eq 2) { + $accounts = & $cmd account list --raw 2>$null + return $accounts | Where-Object { $_ -like "$wordToComplete*" } + } elseif ($currentIndex -eq 3) { + $ipHistory = & $cmd _ip_history 2>$null + return $ipHistory | Where-Object { $_ -like "$wordToComplete*" } + } + return @() + } + + # Completion for direct command + if ($subcmd -eq "direct") { + if ($currentIndex -eq 2) { + $ipHistory = & $cmd _ip_history 2>$null + return $ipHistory | Where-Object { $_ -like "$wordToComplete*" } + } + return @() + } + + # Aliases completion + switch ($subcmd) { + "as" { + if ($currentIndex -eq 2) { + $accounts = & $cmd account list --raw 2>$null + return $accounts | Where-Object { $_ -like "$wordToComplete*" } + } + } + "use" { + if ($currentIndex -eq 2) { + $sheets = & $cmd sheet list --raw 2>$null + return $sheets | Where-Object { $_ -like "$wordToComplete*" } + } + } + "make" { + if ($currentIndex -eq 2) { + $allSheets = & $cmd sheet list --all --raw 2>$null + return $allSheets | Where-Object { $_ -like "$wordToComplete*" } + } + } + "drop" { + if ($currentIndex -eq 2) { + $sheets = & $cmd sheet list --raw 2>$null + return $sheets | Where-Object { $_ -like "$wordToComplete*" } + } + } + "docs" { + if ($currentIndex -eq 2) { + $docs = & $cmd docs list --raw 2>$null + return $docs | Where-Object { $_ -like "$wordToComplete*" } + } + } + { @("move", "mv", "import", "export", "in", "out", "track", "hold", "throw") -contains $_ } { + # File completion for file operations + return Get-ChildItem -Name -File -Path "." | Where-Object { $_ -like "$wordToComplete*" } + } + } + + return @() +}
\ No newline at end of file |