# dnf completion                                          -*- shell-script -*-

#

# This file is part of dnf.

#

# Copyright 2013 (C) Elad Alfassa <elad@fedoraproject.org>

# Copyright 2014-2015 (C) Igor Gnatenko <i.gnatenko.brain@gmail.com>

#

# This program is free software; you can redistribute it and/or modify

# it under the terms of the GNU General Public License as published by

# the Free Software Foundation; either version 2 of the License, or

# (at your option) any later version.

#

# This program is distributed in the hope that it will be useful,

# but WITHOUT ANY WARRANTY; without even the implied warranty of

# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

# GNU General Public License for more details.

#

# You should have received a copy of the GNU General Public License

# along with this program; if not, write to the Free Software

# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA

# 02110-1301  USA

__dnf_main_options="

    -4 -6 -b -c -C -d -e -h -q -R -v -x -y

    --advisory --advisories

    --allowerasing --assumeno

    --assumeyes

    --best --bugfix

    --bz

    --cacheonly --config

    --color --cve

    --debuglevel --debugsolver

    --disableexcludes --disableplugin

    --disableexcludepkgs --disablerepo

    --downloadonly --destdir

    --downloaddir

    --errorlevel --enableplugin

    --enablerepo --enhancement

    --exclude --excludepkgs

    --help --help-cmd

    --installroot

    --newpackage --nodocs

    --nogpgcheck --noplugins

    --noautoremove

    --obsoletes

    --quiet

    --randomwait --refresh

    --releasever --repofrompath

    --repo --repoid

    --rpmverbosity

    --sec-severity --secseverity

    --security --setopt

    --skip-broken --showduplicates

    --verbose --version

"

__dnf_history_subcmds="info list redo rollback undo userinstalled"

__dnf_secseverity_subcmds='Critical Important Low Moderate'

__dnf_clean_subcmds="all dbcache expire-cache metadata packages"

__dnf_repoquery_subcmds="

    -a -f -i -l -s

    --all --arch

    --archlist --available

    --alldeps

    --conflicts

    --deplist --duplicates

    --exactdeps --extras

    --envra --enhances

    --file

    --groupmember

    --installed --installonly

    --info

    --latest-limit --list

    --location

    --nevra  --nvr

    --provides

    --qf --queryformat

    --querytags

    --recent --requires-pre

    --recommends --requires

    --recursive --resolve

    --source --suggests

    --supplements --show-duplicates

    --srpm

    --tree

    --upgrades --unneeded

    --unsatisfied --userinstalled

    --whatconflicts --whatobsoletes

    --whatprovides --whatrequires

    --whatrecommends --whatenhances

    --whatsuggests --whatsupplements

"

__dnf_repopkgs_subcmds="

    check-update

    info install

    list

    move-to

    remove-or-reinstall remove-or-distro-sync

    reinstall reinstall-old

    upgrade upgrade-to

"

__dnf_python_exec=

_dnf_set_python_exec()

{

    if [[ "$( readlink /usr/bin/dnf )" == "dnf-2" ]]; then

        __dnf_python_exec="/usr/bin/python2"

    else

        if [ -x /usr/libexec/system-python ]; then

            __dnf_python_exec="/usr/libexec/system-python"

        else

            __dnf_python_exec="/usr/bin/python3"

        fi

    fi

}

_dnf_commands_helper()

{

    local helper_cmd="import sys; from dnf.cli import completion_helper as ch; ch.main(sys.argv[1:])"

    local helper_opts="-d 0 -q -C --assumeno --nogpgcheck"

    echo "$( ${__dnf_python_exec} -c "$helper_cmd" "$@" $helper_opts 2>/dev/null 

}

_dnf_is_path()

{

    if [[ "$1" == \.* ]] || [[ "$1" == \/* ]] || [[ "$1" == ~* ]]; then

        return 0

    else

        return 1

    fi

}

_dnf_show_packages()

{

    local args cmd=$1

    _count_args

    # do not print packages if command wasnt chosen yet

    # eg.: dnf inTAB

    [[ "$args" -eq 1 ]] && return

    # return when "-" follows command

    # eg.: dnf up -

    [[ "$cur" == -* ]] && return

    shift

    if ! _dnf_is_path "$cur"; then

        COMPREPLY+=( $(compgen -W "$( _dnf_commands_helper $cmd "$@" "$cur" )") )

    fi

    [[ $COMPREPLY ]] && return

    COMPREPLY+=( $(compgen -o default -f -X "!*.@(rpm)" -- "$cur" ) )

}

_dnf_get_first_command()

{

    local cmd i

    for (( i=1; i < ${#COMP_WORDS[@]}; i++ )); do

        if [[ "$1"  == *${COMP_WORDS[i]}* ]]; then

            cmd=${COMP_WORDS[i]}

            break

        fi

    done

    echo "$cmd"

}

_dnf()

{

    _dnf_set_python_exec

    local complete_commands complete_options extra_options command cmd_list

    local cur prev words cword split

    _init_completion -s || return

    cmd_list="$( _dnf_commands_helper "_cmds" "" )"

    command="$( _dnf_get_first_command "$cmd_list" )"

    case $prev in

    --version)

       return ;;

    -h|--help|--help-cmd)

        [[ $cword -eq 3 ]] && return ;;

    -d|--debuglevel|-e|--errorlevel)

        COMPREPLY=( $( compgen -W '0 1 2 3 4 5 6 7 8 9 10' -- "$cur" ) )

        return

        ;;

    --installroot)

        _filedir -d

        return ;;

    --enablerepo)

        COMPREPLY=( $(compgen -W "$( _dnf_commands_helper repolist disabled "$cur" )" -- "$cur") )

        return ;;

    --disablerepo)

        COMPREPLY=( $(compgen -W "$( _dnf_commands_helper repolist enabled "$cur" )" -- "$cur") )

        return ;;

    --color)

        COMPREPLY=( $(compgen -W "yes no true false 0 1" -- "$cur" ) )

        return ;;

    --destdir|--downloaddir)

        _filedir -d

        return ;;

    --sec*severity)

        COMPREPLY=( $( compgen -W "$__dnf_secseverity_subcmds" -- "$cur" ) )

        return ;;

    *)

        ;;

    esac

    complete_commands=

    case "$command" in

    autoremove|autoremove-n*)

        _dnf_show_packages list installed

        ;;

    check)

        extra_options="--all --dependencies --duplicates --obsoleted --provides"

        complete_commands="all dependencies duplicates obsoleted provides"

        ;;

    check-update|check-upgrade)

        _dnf_show_packages list updates

        ;;

    clean)

        complete_commands="$__dnf_clean_subcmds"

        ;;

    dup|dist-upgrade|dsync|distro*sync|distribution-synchronization)

        _dnf_show_packages list installed

        ;;

    dg|downgrade)

        _dnf_show_packages $command

        ;;

    group|groups|grouperase|groupinfo|groupinstall|grouplist|groupremove|groupupdate)

        extra_options="--with-optional --hidden --installed --available"

        ;;

    help)

        [[ $cword -eq 3 ]] && return 0

        complete_commands="$cmd_list"

        ;;

    history)

        complete_commands="$__dnf_history_subcmds"

        ;;

    info|list)

        local subcmd="available installed updates upgrades

                      --available --installed --updates --upgrades"

        case "$( _dnf_get_first_command "$subcmd" )" in

            --installed|installed)

                _dnf_show_packages list installed ;;

            --available|available)

                _dnf_show_packages list available ;;

            --updates|--upgrades|updates|upgrades)

                _dnf_show_packages list updates ;;

            *)

                _dnf_show_packages list

            ;;

        esac

        extra_options="--all --available --installed --extras --updates

                       --upgrades --autoremove --recent"

        ;;

    info-*|list-*|updateinfo|summary-updateinfo)

        extra_options="--summary --list --info"

        complete_commands="summary list info"

        ;;

    in|install*|localinstall)

        _dnf_show_packages $command

        ;;

    mark)

        _dnf_show_packages list installed

        ;;

    mc|ref|makecache|refresh)

        [[ "$prev" == *"timer" ]] && return

        extra_options="--timer"

        complete_commands="timer"

        ;;

    provides)

        ;;

    ri|rei|reinstall)

        _dnf_show_packages $command

        ;;

    rm|remove*|erase*)

        _dnf_show_packages $command

        extra_options="--oldinstallonly --duplicates --duplicated"

        ;;

    repoinfo|repolist)

        extra_options="--all --enabled --disabled"

        case $prev in

        enabled|--enabled)

            complete_commands="$( _dnf_commands_helper $command enabled "$cur" )"

            ;;

        disabled|--disabled)

            complete_commands="$( _dnf_commands_helper $command disabled "$cur" )"

            ;;

        all|--all|*)

            complete_commands="$( _dnf_commands_helper $command all "$cur" )"

        esac

        ;;

    repoquery*)

        extra_options="$__dnf_repoquery_subcmds"

        ;;

    repo*-pkgs|repo*-packages)

        complete_commands="$__dnf_repopkgs_subcmds"

        ;;

    se|search)

        [[ "$prev" == *"all" ]] && return

        extra_options="--all"

        complete_commands="all"

        ;;

    sh|shell) ;;

    swap) ;;

    um|u-m|upgrade|upgrade-n*)

        _dnf_show_packages $command

        ;;

    up|up-min|update|update-n*)

        _dnf_show_packages $command

        ;;

    whatprovides) ;;

    *)

        complete_commands="$cmd_list"

        ;;

    esac

    $split && return

    case $cur in

    -*)

        complete_options="$__dnf_main_options $extra_options"

        COMPREPLY=( $(compgen -W "$complete_options" -- "$cur" ))

        ;;

    *)

        if [[ $cur == $command ]]; then

            complete_commands="$cmd_list"

        fi

        if [[ "$complete_commands" ]]; then

            COMPREPLY=( $(compgen -W "$complete_commands" -- "$cur" ))

        fi

    esac

    [[ $COMPREPLY == *= ]] && compopt -o nospace

    return 0

}

complete -F _dnf dnf dnf-2 dnf-3