# pydoc completion                                         -*- shell-script -*-

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

    case $prev in
        -k|-p)
            return
            ;;
        -w)
            _filedir
            return
            ;;
    esac

    if [[ "$cur" == -* ]]; then
        COMPREPLY=( $(compgen -W \
            '$("$1" | command sed -e "s/^pydoc3\{0,1\} //" | _parse_help -)' \
            -- "$cur") )
        return
    fi

    COMPREPLY=( $(compgen -W 'keywords topics modules' -- "$cur") )

    if [[ $cur != @(.|*/)* ]]; then
        local python=python; [[ ${1##*/} == *3* ]] && python=python3
        _xfunc python _python_modules $python
    fi

    # Note that we don't do "pydoc modules" as it is known to hang on
    # some systems; _python_modules tends to work better and faster.
    COMPREPLY+=( $(compgen -W \
        '$($1 keywords topics | command sed -e /^Here/d)' -- "$cur") )

    _filedir py
} &&
complete -F _pydoc pydoc pydoc3

# ex: filetype=sh