# virt-tools bash completion script -*- shell-script -*-
# Copyright (C) 2010-2013 Red Hat Inc.
#
# 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.

# List all local libvirt domains.
_guestfs_virsh_list ()
{
    local flag_ro=$1 flags

    if [ "$flag_ro" -eq 1 ]; then
        flags="--all"
    else
        flags="--inactive"
    fi
    virsh list $flags | head -n -1 | tail -n +3 | awk '{print $2}'
}

_guestfs_virttools ()
{
    local cur prev words cword split
    local longopts="$1" flag_ro="$2" doms

    _init_completion -s || return

    case "$prev" in
        -d|--domain)
            doms=$(_guestfs_virsh_list "$flag_ro")
            COMPREPLY=( $(compgen -W "$doms" -- "$cur") )
            return ;;
    esac

    case "$cur" in
        --*)
            # --options
            COMPREPLY=( $(compgen -W "$longopts" -- "$cur") )
            return ;;
        *)
            COMPREPLY=( $(compgen "$cur") )
            return ;;
    esac
}

_virt_alignment_scan ()
{
    _guestfs_virttools "$(virt-alignment-scan --long-options)" 1
} &&
complete -o default -F _virt_alignment_scan virt-alignment-scan

_virt_cat ()
{
    _guestfs_virttools "$(virt-cat --long-options)" 1
} &&
complete -o default -F _virt_cat virt-cat

_virt_df ()
{
    _guestfs_virttools "$(virt-df --long-options)" 1
} &&
complete -o default -F _virt_df virt-df

_virt_edit ()
{
    _guestfs_virttools "$(virt-edit --long-options)" 0
} &&
complete -o default -F _virt_edit virt-edit

_virt_filesystems ()
{
    _guestfs_virttools "$(virt-filesystems --long-options)" 1
} &&
complete -o default -F _virt_filesystems virt-filesystems

_virt_format ()
{
    _guestfs_virttools "$(virt-format --long-options)" 0
} &&
complete -o default -F _virt_format virt-format

_virt_inspector ()
{
    _guestfs_virttools "$(virt-inspector --long-options)" 1
} &&
complete -o default -F _virt_inspector virt-inspector

_virt_ls ()
{
    _guestfs_virttools "$(virt-ls --long-options)" 1
} &&
complete -o default -F _virt_ls virt-ls

_virt_sysprep ()
{
    _guestfs_virttools "$(virt-sysprep --long-options)" 0
} &&
complete -o default -F _virt_sysprep virt-sysprep
