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

_check_immich()
{
    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=()

    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)
            return
            ;;
    esac

    # the mode is the first word that is not an option; the options before
    # it take no value
    local i mode=
    for ((i = 1; i < COMP_CWORD; i++)); do
        if [[ ${COMP_WORDS[i]} != -* ]]; then
            mode=${COMP_WORDS[i]}
            break
        fi
    done

    local options
    case $mode in
        '')
            options='-h --help -V --version health stats storage rotate'
            ;;
        health | stats | storage)
            options='-h --help -H --hostname -p --port -S --ssl -k --insecure
                -t --timeout -T --token -f --token-file -w --warning -c --critical'
            ;;
        rotate)
            options='-h --help -H --hostname -p --port -S --ssl -k --insecure
                -t --timeout -f --token-file'
            ;;
        *)
            return
            ;;
    esac
    mapfile -t COMPREPLY < <(compgen -W "$options" -- "$cur")
}
complete -F _check_immich check_immich
