# bash completion for pi-worktree
_pi_worktree() {
  local cur first
  cur="${COMP_WORDS[COMP_CWORD]}"
  first="${COMP_WORDS[1]}"

  local SUBS="list ls info out remove rm prune -l --list -i --info -o --out -r --remove -p --prune -h --help"

  # existing worktree names (only linked worktrees, i.e. under .worktrees/)
  local wts=""
  if git rev-parse --is-inside-work-tree >/dev/null 2>&1; then
    local main_dir
    main_dir=$(git rev-parse --path-format=absolute --git-common-dir 2>/dev/null | sed 's|/.git$||')
    wts=$(git worktree list --porcelain 2>/dev/null \
      | sed -n 's|^worktree ||p' \
      | while read -r p; do [[ "$p" != "$main_dir" ]] && basename "$p"; done || true)
  fi

  if (( COMP_CWORD == 1 )); then
    # first word: subcommands, flags, and existing worktree names (to enter)
    COMPREPLY=($(compgen -W "$SUBS $wts" -- "$cur"))
    return
  fi

  case "$first" in
    -i|--info|info|-r|--remove|remove|rm)
      # second word of info/remove: worktree names only
      COMPREPLY=($(compgen -W "$wts" -- "$cur"))
      ;;
    *)
      # after a name: default (file) completion for pi args
      COMPREPLY=()
      ;;
  esac
}
complete -F _pi_worktree pi-worktree