runtime-sdk 0.4.49__py3-none-win_amd64.whl

runtime_sdk/completion.py

@@ -0,0 +1,348 @@
+from __future__ import annotations
+
+import sys
+
+from runtime_sdk import output as cli_output
+
+
+_COMPLETION_BASH = r"""# bash completion for runtime
+_runtime_complete() {
+  local cur sub
+  COMPREPLY=()
+  cur="${COMP_WORDS[COMP_CWORD]}"
+  if [[ ${COMP_CWORD} -eq 1 ]]; then
+    COMPREPLY=( $(compgen -W "create list ls info url start enter ssh run exec files fs file publish delete rm share network service goal proxy switch checkout github integrate secrets codex claude grok xai api-keys login whoami logout completion help" -- "$cur") )
+    return
+  fi
+  sub="${COMP_WORDS[1]}"
+  case "$sub" in
+    list|ls)
+      COMPREPLY=( $(compgen -W "--json --watch -w" -- "$cur") )
+      ;;
+    info|url|start|enter|ssh|run|exec|publish|delete|rm)
+      local ids
+      ids=$(runtime _slugs 2>/dev/null)
+      COMPREPLY=( $(compgen -W "$ids" -- "$cur") )
+      ;;
+    files|fs|file)
+      if [[ ${COMP_CWORD} -eq 2 ]]; then
+        COMPREPLY=( $(compgen -W "read write list ls stat mkdir rm delete upload download" -- "$cur") )
+      else
+        local ids
+        ids=$(runtime _slugs 2>/dev/null)
+        COMPREPLY=( $(compgen -W "$ids" -- "$cur") )
+      fi
+      ;;
+    service)
+      if [[ ${COMP_CWORD} -eq 2 ]]; then
+        COMPREPLY=( $(compgen -W "run show status logs restart stop clear" -- "$cur") )
+      else
+        local ids
+        ids=$(runtime _slugs 2>/dev/null)
+        COMPREPLY=( $(compgen -W "$ids" -- "$cur") )
+      fi
+      ;;
+    goal)
+      if [[ ${COMP_CWORD} -eq 2 ]]; then
+        COMPREPLY=( $(compgen -W "watch" -- "$cur") )
+      else
+        local ids
+        ids=$(runtime _slugs 2>/dev/null)
+        COMPREPLY=( $(compgen -W "$ids" -- "$cur") )
+      fi
+      ;;
+    proxy)
+      COMPREPLY=( $(compgen -W "start ls stop" -- "$cur") )
+      ;;
+    github)
+      COMPREPLY=( $(compgen -W "list ls disconnect rm" -- "$cur") )
+      ;;
+    api-keys)
+      COMPREPLY=( $(compgen -W "list ls create revoke" -- "$cur") )
+      ;;
+    secrets)
+      COMPREPLY=( $(compgen -W "list ls show set import delete rm" -- "$cur") )
+      ;;
+    integrate)
+      COMPREPLY=( $(compgen -W "github" -- "$cur") )
+      ;;
+    completion)
+      COMPREPLY=( $(compgen -W "bash zsh fish powershell" -- "$cur") )
+      ;;
+    codex)
+      COMPREPLY=( $(compgen -W "login status logout" -- "$cur") )
+      ;;
+    claude)
+      COMPREPLY=( $(compgen -W "login status logout" -- "$cur") )
+      ;;
+    grok)
+      COMPREPLY=( $(compgen -W "login status logout" -- "$cur") )
+      ;;
+    xai)
+      COMPREPLY=( $(compgen -W "login status logout" -- "$cur") )
+      ;;
+    share)
+      if [[ ${COMP_CWORD} -eq 2 ]]; then
+        COMPREPLY=( $(compgen -W "public private" -- "$cur") )
+      else
+        local ids
+        ids=$(runtime _slugs 2>/dev/null)
+        COMPREPLY=( $(compgen -W "$ids" -- "$cur") )
+      fi
+      ;;
+	network)
+		if [[ ${COMP_CWORD} -eq 2 ]]; then
+			COMPREPLY=( $(compgen -W "default show set denials" -- "$cur") )
+		fi
+		;;
+  esac
+}
+complete -F _runtime_complete runtime
+"""
+
+
+_COMPLETION_ZSH = r"""#compdef runtime
+_runtime() {
+  local -a _runtime_commands
+  _runtime_commands=(
+    'create:Create a new computer'
+    'list:List computers'
+    'ls:List computers'
+    'info:Show details'
+    'url:Show public URL'
+    'start:Wake a cold computer'
+    'enter:Enter console'
+    'ssh:SSH through Runtime'
+    'run:Run a command'
+    'exec:Stream a command'
+    'files:Manage files'
+    'fs:Manage files'
+    'file:Manage files'
+    'publish:Publish a port'
+    'delete:Delete a computer'
+    'rm:Delete a computer'
+    'share:Set visibility'
+	'network:Manage outbound network access'
+    'service:Service config'
+    'goal:Watch goal runs'
+    'proxy:Port proxies'
+    'switch:Open a GitHub branch workspace'
+    'checkout:Open a GitHub branch workspace'
+    'github:GitHub connections'
+    'integrate:Connect GitHub'
+    'secrets:Secret sets'
+    'codex:Codex login'
+    'claude:Claude login'
+    'grok:Grok login'
+    'xai:xAI login'
+    'api-keys:API keys'
+    'login:Sign in with WorkOS'
+    'whoami:Show account'
+    'logout:Log out'
+    'completion:Shell completion'
+    'help:Extended help'
+  )
+  local state
+  _arguments -C '1: :->cmd' '*:: :->args'
+  case $state in
+    cmd)
+      _describe -t commands 'runtime command' _runtime_commands
+      ;;
+    args)
+      case ${words[1]} in
+        list|ls)
+          _arguments '--json[write machine-readable JSON]' '-w[refresh every N seconds]' '--watch[refresh every N seconds]'
+          ;;
+        info|url|start|enter|ssh|run|exec|publish|delete|rm)
+          local -a ids
+          ids=(${(f)"$(runtime _slugs 2>/dev/null)"})
+          _describe -t slugs 'computer' ids
+          ;;
+        files|fs|file)
+          if (( CURRENT == 2 )); then
+            _values 'file action' read write list ls stat mkdir rm delete upload download
+          else
+            local -a ids
+            ids=(${(f)"$(runtime _slugs 2>/dev/null)"})
+            _describe -t slugs 'computer' ids
+          fi
+          ;;
+        service)
+          if (( CURRENT == 2 )); then
+            _values 'service action' run show status logs restart stop clear
+          else
+            local -a ids
+            ids=(${(f)"$(runtime _slugs 2>/dev/null)"})
+            _describe -t slugs 'computer' ids
+          fi
+          ;;
+        goal)
+          if (( CURRENT == 2 )); then
+            _values 'goal action' watch
+          else
+            local -a ids
+            ids=(${(f)"$(runtime _slugs 2>/dev/null)"})
+            _describe -t slugs 'computer' ids
+          fi
+          ;;
+        proxy)
+          _values 'proxy action' start ls stop
+          ;;
+        github)
+          _values 'github action' list ls disconnect rm
+          ;;
+        api-keys)
+          _values 'api key action' list ls create revoke
+          ;;
+        secrets)
+          _values 'secret action' list ls show set import delete rm
+          ;;
+        integrate)
+          _values 'provider' github
+          ;;
+        share)
+          if (( CURRENT == 2 )); then
+            _values 'visibility' public private
+          else
+            local -a ids
+            ids=(${(f)"$(runtime _slugs 2>/dev/null)"})
+            _describe -t slugs 'computer' ids
+          fi
+          ;;
+		network)
+			_values 'network action' default show set denials
+			;;
+        completion)
+          _values 'shell' bash zsh fish powershell
+          ;;
+        codex)
+          _values 'codex action' login status logout
+          ;;
+        claude)
+          _values 'claude action' login status logout
+          ;;
+        grok)
+          _values 'grok action' login status logout
+          ;;
+        xai)
+          _values 'xai action' login status logout
+          ;;
+      esac
+      ;;
+  esac
+}
+_runtime "$@"
+"""
+
+
+_COMPLETION_FISH = r"""# fish completion for runtime
+function __runtime_slugs
+  runtime _slugs 2>/dev/null
+end
+
+function __runtime_is_command
+  set -l tokens (commandline -opc)
+  test (count $tokens) -ge 2; and contains -- $tokens[2] $argv
+end
+
+complete -c runtime -f
+complete -c runtime -n '__fish_use_subcommand' -a create -d 'Create'
+complete -c runtime -n '__fish_use_subcommand' -a list -d 'List'
+complete -c runtime -n '__fish_use_subcommand' -a ls -d 'List'
+complete -c runtime -n '__fish_use_subcommand' -a info -d 'Show details'
+complete -c runtime -n '__fish_use_subcommand' -a url -d 'Show URL'
+complete -c runtime -n '__fish_use_subcommand' -a start -d 'Wake'
+complete -c runtime -n '__fish_use_subcommand' -a enter -d 'Console'
+complete -c runtime -n '__fish_use_subcommand' -a ssh -d 'SSH'
+complete -c runtime -n '__fish_use_subcommand' -a run -d 'Run command'
+complete -c runtime -n '__fish_use_subcommand' -a exec -d 'Stream command'
+complete -c runtime -n '__fish_use_subcommand' -a files -d 'Manage files'
+complete -c runtime -n '__fish_use_subcommand' -a fs -d 'Manage files'
+complete -c runtime -n '__fish_use_subcommand' -a file -d 'Manage files'
+complete -c runtime -n '__fish_use_subcommand' -a publish -d 'Publish port'
+complete -c runtime -n '__fish_use_subcommand' -a delete -d 'Delete'
+complete -c runtime -n '__fish_use_subcommand' -a rm -d 'Delete'
+complete -c runtime -n '__fish_use_subcommand' -a share -d 'Set visibility'
+complete -c runtime -n '__fish_use_subcommand' -a network -d 'Manage outbound network access'
+complete -c runtime -n '__fish_use_subcommand' -a service -d 'Service config'
+complete -c runtime -n '__fish_use_subcommand' -a goal -d 'Watch goal runs'
+complete -c runtime -n '__fish_use_subcommand' -a proxy -d 'Port proxies'
+complete -c runtime -n '__fish_use_subcommand' -a switch -d 'Open branch workspace'
+complete -c runtime -n '__fish_use_subcommand' -a checkout -d 'Open branch workspace'
+complete -c runtime -n '__fish_use_subcommand' -a github -d 'GitHub connections'
+complete -c runtime -n '__fish_use_subcommand' -a integrate -d 'Connect GitHub'
+complete -c runtime -n '__fish_use_subcommand' -a secrets -d 'Secret sets'
+complete -c runtime -n '__fish_use_subcommand' -a codex -d 'Codex login'
+complete -c runtime -n '__fish_use_subcommand' -a claude -d 'Claude login'
+complete -c runtime -n '__fish_use_subcommand' -a grok -d 'Grok login'
+complete -c runtime -n '__fish_use_subcommand' -a xai -d 'xAI login'
+complete -c runtime -n '__fish_use_subcommand' -a api-keys -d 'API keys'
+complete -c runtime -n '__fish_use_subcommand' -a login -d 'Sign in with WorkOS'
+complete -c runtime -n '__fish_use_subcommand' -a whoami -d 'Show account'
+complete -c runtime -n '__fish_use_subcommand' -a logout -d 'Log out'
+complete -c runtime -n '__fish_use_subcommand' -a help -d 'Extended help'
+complete -c runtime -n '__fish_use_subcommand' -a completion -d 'Completion'
+
+complete -c runtime -n '__runtime_is_command info url start enter ssh run exec publish delete rm' -a '(__runtime_slugs)'
+complete -c runtime -n '__runtime_is_command list ls' -l json -d 'Write machine-readable JSON'
+complete -c runtime -n '__runtime_is_command list ls' -s w -l watch -d 'Refresh every N seconds'
+complete -c runtime -n '__runtime_is_command files fs file; and not __fish_seen_subcommand_from read write list ls stat mkdir rm delete upload download' -a 'read write list ls stat mkdir rm delete upload download'
+complete -c runtime -n '__runtime_is_command service; and not __fish_seen_subcommand_from run show status logs restart stop clear' -a 'run show status logs restart stop clear'
+complete -c runtime -n '__runtime_is_command goal; and not __fish_seen_subcommand_from watch' -a watch
+complete -c runtime -n '__runtime_is_command proxy; and not __fish_seen_subcommand_from start ls stop' -a 'start ls stop'
+complete -c runtime -n '__runtime_is_command github; and not __fish_seen_subcommand_from list ls disconnect rm' -a 'list ls disconnect rm'
+complete -c runtime -n '__runtime_is_command api-keys; and not __fish_seen_subcommand_from list ls create revoke' -a 'list ls create revoke'
+complete -c runtime -n '__runtime_is_command secrets; and not __fish_seen_subcommand_from list ls show set import delete rm' -a 'list ls show set import delete rm'
+complete -c runtime -n '__runtime_is_command integrate' -a github
+complete -c runtime -n '__fish_seen_subcommand_from completion' -a 'bash zsh fish powershell'
+complete -c runtime -n '__fish_seen_subcommand_from share' -a 'public private'
+complete -c runtime -n '__runtime_is_command network; and not __fish_seen_subcommand_from default show set denials' -a 'default show set denials'
+complete -c runtime -n '__fish_seen_subcommand_from codex' -a 'login status logout'
+complete -c runtime -n '__fish_seen_subcommand_from claude' -a 'login status logout'
+complete -c runtime -n '__fish_seen_subcommand_from grok' -a 'login status logout'
+complete -c runtime -n '__fish_seen_subcommand_from xai' -a 'login status logout'
+"""
+
+
+_COMPLETION_POWERSHELL = r"""Register-ArgumentCompleter -Native -CommandName runtime -ScriptBlock {
+  param($wordToComplete, $commandAst, $cursorPosition)
+  $commands = 'create list ls info url start enter ssh run exec files fs file publish delete rm share network service goal proxy switch checkout github integrate secrets codex claude grok xai api-keys login whoami logout completion help'
+  $tokens = @($commandAst.CommandElements | ForEach-Object { $_.ToString() })
+  if ($tokens.Count -le 2) {
+    $candidates = $commands -split ' '
+  } else {
+    $candidates = switch ($tokens[1]) {
+      { $_ -in 'list', 'ls' } { '--json --watch -w' -split ' ' }
+      'service' { 'run show status logs restart stop clear' -split ' ' }
+      'network' { 'default show set denials' -split ' ' }
+      'goal' { 'watch' }
+      'proxy' { 'start ls stop' -split ' ' }
+      'github' { 'list ls disconnect rm' -split ' ' }
+      'api-keys' { 'list ls create revoke' -split ' ' }
+      'secrets' { 'list ls show set import delete rm' -split ' ' }
+      'integrate' { 'github' }
+      'share' { 'public private' -split ' ' }
+      'completion' { 'bash zsh fish powershell' -split ' ' }
+      { $_ -in 'codex', 'claude', 'grok', 'xai' } { 'login status logout' -split ' ' }
+      default { @() }
+    }
+  }
+  $candidates | Where-Object { $_ -like "$wordToComplete*" } | ForEach-Object {
+    [System.Management.Automation.CompletionResult]::new($_, $_, 'ParameterValue', $_)
+  }
+}
+"""
+
+
+def handle_completion(shell: str) -> int:
+    scripts = {
+        "bash": _COMPLETION_BASH,
+        "zsh": _COMPLETION_ZSH,
+        "fish": _COMPLETION_FISH,
+        "powershell": _COMPLETION_POWERSHELL,
+    }
+    script = scripts.get(shell)
+    if script is None:
+        return cli_output.report_error(f"unsupported shell: {shell}")
+    sys.stdout.write(script)
+    return 0