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

# helper functions

_rpm_installed_packages()
{
    if [[ -r /var/log/rpmpkgs && \
        /var/log/rpmpkgs -nt /var/lib/rpm/Packages ]]; then
        # using RHL 7.2 or later - this is quicker than querying the DB
        COMPREPLY=( $( compgen -W "$( command sed -ne \
            's|^\([^[:space:]]\{1,\}\)-[^[:space:]-]\{1,\}-[^[:space:]-]\{1,\}\.rpm$|\1|p' \
            /var/log/rpmpkgs )" -- "$cur" ) )
    elif type rpmqpack &>/dev/null ; then
        # SUSE's rpmqpack is faster than rpm -qa
        COMPREPLY=( $( compgen -W '$( rpmqpack )' -- "$cur" ) )
    else
        COMPREPLY=( $( ${1:-rpm} -qa --nodigest --nosignature \
            --queryformat='%{NAME} ' "$cur*" 2>/dev/null ) )
    fi
}

_rpm_groups()
{
    local IFS=$'\n'
    COMPREPLY=( $( compgen -W "$( ${1:-rpm} -qa --nodigest --nosignature \
        --queryformat='%{GROUP}\n' 2>/dev/null )" -- "$cur" ) )
}

_rpm_macros()
{
    # get a list of macros
    COMPREPLY=( $( compgen -W "$( ${1:-rpm} --showrc | command sed -ne \
        's/^-\{0,1\}[0-9]\{1,\}[:=][[:space:]]\{1,\}\([^[:space:](]\{3,\}\).*/%\1/p' )" \
        -- "$cur" ) )
}

_rpm_buildarchs()
{
    COMPREPLY=( $( compgen -W "$( ${1:-rpm} --showrc | command sed -ne \
        's/^\s*compatible\s\s*build\s\s*archs\s*:\s*\(.*\)/\1/ p' )" \
        -- "$cur" ) )
}

# rpm(8) completion
#
_rpm()
{
    local cur prev words cword split
    _init_completion -s || return

    if [[ $cword -eq 1 ]]; then
        # first parameter on line
        case $cur in
            --*)
                COMPREPLY=( $( compgen -W '--help --version --initdb
                    --checksig --addsign --delsign --rebuilddb --showrc
                    --setperms --setugids --eval --install --upgrade --query
                    --freshen --erase --verify --querytags --import' \
                        -- "$cur" ) )
                ;;
            *)
                COMPREPLY=( $( compgen -W '-e -E -F -i -q -t -U -V' \
                    -- "$cur" ) )
                ;;
        esac
        return 0
    fi

    case $prev in
        --dbpath|--excludepath|--prefix|--relocate|--root|-r)
            _filedir -d
            return 0
            ;;
        --eval|-E)
            _rpm_macros $1
            return 0
            ;;
        --pipe)
            compopt -o filenames
            COMPREPLY=( $( compgen -c -- "$cur" ) )
            return 0
            ;;
        --rcfile)
            _filedir
            return 0
            ;;
        --specfile)
            # complete on .spec files
            _filedir spec
            return 0
            ;;
        --whatenchances|--whatprovides|--whatrecommends|--whatrequires|\
        --whatsuggests|--whatsupplements)
            if [[ "$cur" == */* ]]; then
                _filedir
            else
                # complete on capabilities
                local IFS=$'\n' fmt
                case $prev in
                    *enhances) fmt=ENHANCENAME ;;
                    *provides) fmt=PROVIDENAME ;;
                    *recommends) fmt=RECOMMENDNAME ;;
                    *requires) fmt=REQUIRENAME ;;
                    *suggests) fmt=SUGGESTNAME ;;
                    *supplements) fmt=SUPPLEMENTNAME ;;
                esac
                COMPREPLY=( $( compgen -W "$( $1 -qa --nodigest --nosignature \
                    --queryformat=\"%{$fmt}\\n\" 2>/dev/null |
                    command grep -vF '(none)' )" -- "$cur" ) )
            fi
            return 0
            ;;
        --define|-D|--fileid|--hdrid|--pkgid)
            # argument required but no completions available
            return 0
            ;;
    esac

    $split && return 0

    # options common to all modes
    local opts="--define= --eval= --macros= --nodigest --nosignature --rcfile=
        --quiet --pipe --verbose"

    case ${words[1]} in
        -[iFU]*|--install|--freshen|--upgrade)
            if [[ "$cur" == -* ]]; then
                COMPREPLY=( $( compgen -W "$opts --percent --force --test
                --replacepkgs --replacefiles --root --excludedocs --includedocs
                --noscripts --ignorearch --dbpath --prefix= --ignoreos --nodeps
                --allfiles --ftpproxy --ftpport --justdb --httpproxy --httpport
                --noorder --relocate= --badreloc --notriggers --excludepath=
                --ignoresize --oldpackage --queryformat --repackage
                --nosuggests" -- "$cur" ) )
            else
                _filedir '[rs]pm'
            fi
            ;;
        -e|--erase)
            if [[ "$cur" == -* ]]; then
                COMPREPLY=( $( compgen -W "$opts --allmatches --noscripts
                    --notriggers --nodeps --test --repackage" -- "$cur" ) )
            else
                _rpm_installed_packages $1
            fi
            ;;
        -q*|--query)
            # options common to all query types
            opts+=" --changelog --configfiles --conflicts --docfiles --dump
                --enhances --filesbypkg --filecaps --fileclass --filecolor
                --fileprovide --filerequire --filesbypkg --info --list
                --obsoletes --pipe --provides --queryformat= --requires
                --scripts --suggests --triggers --xml --recommends
                --supplements"

            if [[ ${words[@]} == *\ -@(*([^ -])f|-file )* ]]; then
                # -qf completion
                if [[ "$cur" == -* ]]; then
                    COMPREPLY=( $( compgen -W "$opts --dbpath --fscontext
                        --last --root --state" -- "$cur" ) )
                else
                    _filedir
                fi
            elif [[ ${words[@]} == *\ -@(*([^ -])g|-group )* ]]; then
                # -qg completion
                _rpm_groups $1
            elif [[ ${words[@]} == *\ -@(*([^ -])p|-package )* ]]; then
                # -qp; uninstalled package completion
                if [[ "$cur" == -* ]]; then
                    COMPREPLY=( $( compgen -W "$opts --ftpport --ftpproxy
                        --httpport --httpproxy --nomanifest" -- "$cur" ) )
                else
                    _filedir '[rs]pm'
                fi
            else
                # -q; installed package completion
                if [[ "$cur" == -* ]]; then
                    COMPREPLY=( $( compgen -W "$opts --all --file --fileid
                        --dbpath --fscontext --ftswalk --group --hdrid --last
                        --package --pkgid --root= --specfile --state
                        --triggeredby --whatenhances --whatprovides
                        --whatrecommends --whatrequires --whatsuggests
                        --whatsupplements" \
                            -- "$cur" ) )
                elif [[ ${words[@]} != *\ -@(*([^ -])a|-all )* ]]; then
                    _rpm_installed_packages $1
                fi
            fi
            ;;
        -K*|--checksig)
            if [[ "$cur" == -* ]]; then
                COMPREPLY=( $( compgen -W "$opts --nopgp --nogpg --nomd5" \
                    -- "$cur" ) )
            else
                _filedir '[rs]pm'
            fi
            ;;
        -[Vy]*|--verify)
            if [[ "$cur" == -* ]]; then
                COMPREPLY=( $( compgen -W "$opts --root= --dbpath --nodeps
                    --nogroup --nolinkto --nomode --nomtime --nordev --nouser
                    --nofiles --noscripts --nomd5 --querytags --specfile
                    --whatenhances --whatprovides --whatrecommends
                    --whatrequires --whatsuggests --whatsupplements" \
                        -- "$cur" ) )
            # check whether we're doing file completion
            elif [[ ${words[@]} == *\ -@(*([^ -])f|-file )* ]]; then
                _filedir
            elif [[ ${words[@]} == *\ -@(*([^ -])g|-group )* ]]; then
                _rpm_groups $1
            elif [[ ${words[@]} == *\ -@(*([^ -])p|-package )* ]]; then
                _filedir '[rs]pm'
            else
                _rpm_installed_packages $1
            fi
            ;;
        --resign|--addsign|--delsign)
            _filedir '[rs]pm'
            ;;
        --setperms|--setgids)
            _rpm_installed_packages $1
            ;;
        --import|--dbpath|--root)
            if [[ "$cur" == -* ]]; then
                COMPREPLY=( $( compgen -W '--import --dbpath --root=' \
                    -- "$cur" ) )
            else
                _filedir
            fi
            ;;
    esac
    [[ $COMPREPLY == *= ]] && compopt -o nospace

    return 0
} &&
complete -F _rpm rpm

_rpmbuild()
{
    local cur prev words cword split
    _init_completion -s || return

    local rpm="${1%build*}"
    [[ $rpm == $1 ]] || ! type $rpm &>/dev/null && rpm=

    case $prev in
        --buildroot|--root|-r|--dbpath)
            _filedir -d
            return 0
            ;;
        --target)
            _rpm_buildarchs
            return 0
            ;;
        --eval|-E)
            _rpm_macros $rpm
            return 0
            ;;
        --macros|--rcfile)
            _filedir
            return 0
            ;;
        --buildpolicy)
            local cfgdir=$( $rpm --eval '%{_rpmconfigdir}' 2>/dev/null )
            if [[ $cfgdir ]]; then
                COMPREPLY=( $( compgen -W "$( command ls $cfgdir 2>/dev/null \
                    | command sed -ne 's/^brp-//p' )" -- "$cur" ) )
            fi
            ;;
        --define|-D|--with|--without)
            return 0
            ;;
    esac

    $split && return 0

    if [[ $cur == -* ]]; then
        COMPREPLY=( $( compgen -W "$( _parse_help "$1" )" -- "$cur" ) )
        [[ $COMPREPLY == *= ]] && compopt -o nospace
        return 0
    fi

    # Figure out file extensions to complete
    local word ext
    for word in ${words[@]}; do
        case $word in
            -b?|--clean|--nobuild)
                ext=spec
                break
                ;;
            -t?|--tarbuild)
                ext='@(t?(ar.)@([gx]z|bz?(2))|tar?(.@(lzma|Z)))'
                break
                ;;
            --rebuild|--recompile)
                ext='@(?(no)src.r|s)pm'
                break
                ;;
        esac
    done
    [[ -n $ext ]] && _filedir $ext
} &&
complete -F _rpmbuild rpmbuild rpmbuild-md5

# ex: ts=4 sw=4 et filetype=sh
