#!/bin/bash
# Copyright 2013 Red Hat, Inc.  All rights reserved.
#
# 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, see <http://www.gnu.org/licenses/>.
#

__contains_word () {
        local word="$1"; shift
        for w in "$@"; do [[ $w = "$word" ]] && return 0; done
        return 1
}

_lsinitrd() {
        local cur=${COMP_WORDS[COMP_CWORD]} prev=${COMP_WORDS[COMP_CWORD-1]}
        local -A OPTS=(
                [STANDALONE]='-s --size -h --help'

                       [ARG]='-f --file -k --kver'
        )

        # shellcheck disable=SC2086
        if __contains_word "$prev" ${OPTS[ARG]}; then
                case $prev in
                        --file|-f)
                                comps=$(compgen -f -- "$cur")
                                compopt -o filenames
                        ;;
                        --kver|-k)
                                comps=$(cd /lib/modules || return 1; echo [0-9]*)
                        ;;
                        *)
                                return 0
                        ;;
                esac
                # shellcheck disable=SC2207
                # shellcheck disable=SC2016
                COMPREPLY=( $(compgen -W '$comps' -- "$cur") )
                return 0
        fi

        if [[ $cur = -* ]]; then
                # shellcheck disable=SC2207
                # shellcheck disable=SC2016
                COMPREPLY=( $(compgen -W '${OPTS[*]}' -- "$cur") )
                return 0
        fi

        # shellcheck disable=SC2034
        comps=$(compgen -f -- "$cur")
        compopt -o filenames
        # shellcheck disable=SC2207
        # shellcheck disable=SC2016
        COMPREPLY=( $(compgen -W '$comps' -- "$cur") )
        return 0
}

complete -F _lsinitrd lsinitrd