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

_python_modules()
{
    COMPREPLY+=( $( compgen -W \
        "$( ${1:-python} ${BASH_SOURCE[0]%/*}/../helpers/python 2>/dev/null )" \
        -- "$cur" ) )
}

_python()
{
    local cur prev words cword prefix
    _init_completion || return

    case $cur in
        -[QWX]?*)
            prefix=${cur:0:2}
            prev=$prefix
            cur=${cur:2}
        ;;
    esac

    case $prev in
        -'?'|-h|--help|-V|--version|-c|-X)
            return
            ;;
        -m)
            _python_modules "$1"
            return
            ;;
        -Q)
            COMPREPLY=( $( compgen -W "old new warn warnall" -P "$prefix" \
                -- "$cur" ) )
            return
            ;;
        -W)
            COMPREPLY=( $( compgen -W "ignore default all module once error" \
                -P "$prefix" -- "$cur" ) )
            return
            ;;
        --jit)
            # TODO: quite a few others, parse from "--jit help" output?
            COMPREPLY=( $( compgen -W "help off" -- "$cur" ) )
            return
            ;;
        !(?(*/)python*([0-9.])|?(*/)pypy*([0-9.])|-?))
            [[ $cword -lt 2 || ${words[cword-2]} != -[QWX] ]] && _filedir
            ;;
    esac


    # if -c or -m is already given, complete all kind of files.
    if [[ "${words[@]::$cword}" == *\ -[cm]\ * ]]; then
        _filedir
    elif [[ "$cur" != -* ]]; then
        _filedir 'py?([cowz])'
    else
        COMPREPLY=( $( compgen -W '$( _parse_help "$1" -h )' -- "$cur" ) )
    fi
} &&
complete -F _python python python2 python3 pypy pypy3 micropython

# ex: filetype=sh