# Debian aptitude(1) completion

have aptitude &&
_aptitude()
{
    local cur prev words cword
    _init_completion || return

    local dashoptions
    dashoptions='-S -u -i -h --help --version -s --simulate -d \
        --download-only -P --prompt -y --assume-yes -F \
        --display-format -O --sort -w --width -f -r -g \
        --with-recommends -R -G --without-recommends -t \
        --target-release -V --show-versions -D --show-deps\
        -Z -v --verbose --schedule-only'

    local special i
    for (( i=0; i < ${#words[@]}-1; i++ )); do
        if [[ ${words[i]} == @(@(|re)install|@(|un)hold|@(|un)markauto|@(dist|full)-upgrade|download|show|forbid-version|remove|changelog|why@(|-not)|keep@(|-all)|build-dep) ]]; then
            special=${words[i]}
        fi
        #exclude some mutually exclusive options
        [[ ${words[i]} == '-u' ]] && dashoptions=${dashoptions/-i}
        [[ ${words[i]} == '-i' ]] && dashoptions=${dashoptions/-u}
    done

    if [[ -n "$special" ]]; then
       case $special in
           install|hold|markauto|unmarkauto|dist-upgrade|full-upgrade| \
           download|show|changelog|why|why-not|build-dep)
               COMPREPLY=( $( apt-cache pkgnames $cur 2> /dev/null ) )
               return 0
               ;;
           remove|reinstall|forbid-version)
               if [ -f /etc/debian_version ]; then
                    # Debian system
                    COMPREPLY=( $( _comp_dpkg_installed_packages "$cur" ) )
               else
                    # assume RPM based
                    _rpm_installed_packages
               fi
               return 0
               ;;
       esac
    fi

    case $prev in
        # don't complete anything if these options are found
        autoclean|clean|forget-new|search|safe-upgrade|upgrade|update|keep-all)
            return 0
            ;;
        -S)
            _filedir
            return 0
            ;;
    esac

    if [[ "$cur" == -* ]]; then
        COMPREPLY=( $( compgen -W "$dashoptions" -- "$cur" ) )
    else
        COMPREPLY=( $( compgen -W 'update upgrade safe-upgrade forget-new \
            clean autoclean install reinstall remove \
            hold unhold markauto unmarkauto why why-not \
            dist-upgrade full-upgrade download search show \
            forbid-version changelog keep-all build-dep' -- "$cur" ) )
    fi

    return 0
} &&
complete -F _aptitude -o default aptitude

# Local variables:
# mode: shell-script
# sh-basic-offset: 4
# sh-indent-comment: t
# indent-tabs-mode: nil
# End:
# ex: ts=4 sw=4 et filetype=sh
