# jq(1) completion                                         -*- shell-script -*-

_jq()
{
    local cur prev words cword
    _init_completion || return

    case $prev in
        --help | --version | --arg | --argjson | --slurpfile | --argfile)
            return
            ;;
        --indent)
            COMPREPLY=($(compgen -W '{1..8}' -- "$cur"))
            return
            ;;
        --from-file | --run-tests | -!(-*)f)
            _filedir
            return
            ;;
        -!(-*)L)
            _filedir -d
            return
            ;;
    esac

    ((cword > 2)) &&
        case ${words[cword - 2]} in
            --arg | --argjson)
                return
                ;;
            --slurpfile | --argfile)
                _filedir json
                return
                ;;
        esac

    if [[ $cur == -* ]]; then
        COMPREPLY=($(compgen -W '$(_parse_help "$1")' -- "$cur"))
        return
    fi

    local args
    # TODO: DTRT with args taking 2 options
    _count_args "" "@(--arg|--arg?(json|file)|--?(slurp|from-)file|--indent|--run-tests|-!(-*)[fL])"

    # 1st arg is filter
    ((args == 1)) && return
    # 2... are input files
    _filedir json

} &&
    complete -F _jq jq

# ex: filetype=sh
