agentbox.completion

# Shell completion for agentbox
# Supports both bash and zsh

_agentbox_complete() {
  local cur prev words cword
  local -a opts agent_types

  # Initialize completion for bash
  if type _init_completion >/dev/null 2>&1; then
    _init_completion || return
  fi

  opts=(
    -v
    --verbose
    --docker
    -s
    --use-stash
    -a
    --agent
    --no-autostart
    --dangerously-skip-permissions
    --no-git-worktree
    --mount
    --block-folder
    --image
    --privileged
    --mount-docker-socket
    --refresh-cache
    --keep-container
    --no-devcontainer
    --help
    -h
  )
  agent_types=(claude-code qwen-code opencode-ai cursor)

  # Completing options or branch name
  if [[ "${cur}" == -* ]]; then
    # Completing options
    COMPREPLY=($(compgen -W "${opts[*]}" -- "${cur}"))
  elif [[ "${prev}" == "--agent" || "${prev}" == "-a" ]]; then
    # Completing agent type
    COMPREPLY=($(compgen -W "${agent_types[*]}" -- "${cur}"))
  elif [[ "${prev}" == "--image" ]]; then
    # Offer file paths (Containerfile/Dockerfile) as completions
    COMPREPLY=($(compgen -f -- "${cur}"))
    compopt -o filenames 2>/dev/null
  elif [[ "${prev}" == "--mount" ]]; then
    # Completing mount spec — offer file/directory path completion
    COMPREPLY=($(compgen -f -- "${cur}"))
    compopt -o filenames 2>/dev/null
  elif [[ "${prev}" == "--block-folder" ]]; then
    # Completing blocked folder — offer directory path completion
    COMPREPLY=($(compgen -d -- "${cur}"))
    compopt -o filenames 2>/dev/null
  else
    # Completing branch name or options
    COMPREPLY=($(compgen -W "${opts[*]}" -- "${cur}"))
  fi
}

# Bash completion
if type complete >/dev/null 2>&1; then
  complete -F _agentbox_complete agentbox
fi

# Zsh completion
if type compdef >/dev/null 2>&1; then
  _agentbox_zsh() {
    local context state state_descr line
    typeset -A opt_args
    local -a agent_types

    agent_types=(claude-code qwen-code opencode-ai cursor)

    _arguments \
      '(-v --verbose)'{-v,--verbose}'[Verbose output]' \
      '--docker[Use docker even if podman is available]' \
      '(-s --use-stash)'{-s,--use-stash}'[Stash current changes and apply to new worktree]' \
      '(-a --agent)'{-a,--agent}'[Agent type]:agent type:('"${agent_types[*]}"')' \
      '--no-autostart[Do not launch the agent CLI automatically]' \
      '--dangerously-skip-permissions[Run agent in yolo mode]' \
      '--no-git-worktree[Run without git, mount current directory]' \
      '--refresh-cache[Remove cached agent install and reinstall]' \
      '--privileged[Run container in privileged mode (enables Docker-in-Docker)]' \
      '--mount-docker-socket[Mount host docker/podman socket into container at /var/run/docker.sock]' \
      '--keep-container[Do not remove the container on exit]' \
      '--no-devcontainer[Skip automatic devcontainer.json detection]' \
      '--image[Use custom container image or Containerfile (glibc+bash required)]:image or file:_files' \
      '*--mount[Mount host path into container]:mount spec:_files' \
      '*--block-folder[Hide a directory from the agent]:folder:_directories' \
      ':branch name:' \
      && return
  }
  compdef _agentbox_zsh agentbox
fi