# bash completion for check_gitlab                           -*- shell-script -*-

_check_gitlab()
{
    local cur=${COMP_WORDS[COMP_CWORD]} prev=${COMP_WORDS[COMP_CWORD-1]}
    # --option=value: bash splits at the =
    if [[ $cur == = ]]; then
        cur=
    elif [[ $prev == = && $COMP_CWORD -ge 2 ]]; then
        prev=${COMP_WORDS[COMP_CWORD-2]}
    fi
    COMPREPLY=()

    # the mode is the first word that is no option, before it come only -h and -V
    local i mode=
    for ((i = 1; i < COMP_CWORD; i++)); do
        if [[ ${COMP_WORDS[i]} != -* ]]; then
            mode=${COMP_WORDS[i]}
            break
        fi
    done
    if [[ -z $mode ]]; then
        mapfile -t COMPREPLY < <(compgen -W '-h --help -V --version
            health migrations sidekiq stats token rotate' -- "$cur")
        return
    fi

    case $prev in
        -H | --hostname)
            mapfile -t COMPREPLY < <(compgen -A hostname -- "$cur")
            return
            ;;
        -f | --token-file)
            compopt -o filenames 2>/dev/null
            mapfile -t COMPREPLY < <(compgen -f -- "$cur")
            return
            ;;
        -p | --port | -t | --timeout | -T | --token | -w | --warning | -c | --critical | \
            --lifetime | --rotate-before)
            return
            ;;
    esac

    local options='-h --help -H --hostname -p --port -S --ssl --no-ssl -k --insecure
        -t --timeout -f --token-file'
    case $mode in
        rotate) options+=' --lifetime --rotate-before' ;;
        *) options+=' -T --token -w --warning -c --critical' ;;
    esac
    mapfile -t COMPREPLY < <(compgen -W "$options" -- "$cur")
}
complete -F _check_gitlab check_gitlab
