valine-bash-completion.bash

# This file is part of valine, providing intelligent valine tab-completion for BASH
# Save it to: /etc/bash_completion.d/
#
# Revision date: 2016/03/29 matching up with valine v0.7.3
# 
# Copyright 2014, 2016 Ryan Sawhill Aroha <rsaw@redhat.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 3 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 <gnu.org/licenses/gpl.html> for more details.
#
#-------------------------------------------------------------------------------

# NOTE: For a non-root user to use valine and valine tab-completion they must be
#       a member of the 'libvirt' system group.

if groups | grep -qs '\<libvirt\>' && [[ -z $LIBVIRT_DEFAULT_URI ]]; then
    # If member of libvirt group and no LIBVIRT_DEFAULT_URI already set:
    export LIBVIRT_DEFAULT_URI=qemu:///system
fi

__v_VIRSH() {
    virsh "${@}" 2>/dev/null
}

__v_dom_is_off() {
    [[ $(__v_VIRSH domid ${1}) == - ]]
}

__v_dom_has_cdrom() {
    [[ -n $(__v_VIRSH domblklist ${1} --details | awk '$2=="cdrom" {print $NF}') ]]
    return
}

__v_get_domains() {
    __v_VIRSH list --all --name
}

__v_list_snapshots() {
    local cfgFile=/var/lib/valine/${1}
    if [[ -e ${cfgFile} ]]; then
        [[ -r ${cfgFile} ]] || return
        local desiredColumn=$(awk '/^# Columns:/ {print $3}' ${cfgFile} | awk -F❚ '{i = 1; while ($i != "LV") { i++ }; print i}')
        [[ ${desiredColumn} =~ [0-9]+ ]] || return
        awk -F❚ -v desiredColumn=${desiredColumn}  '(NF == 0 || $1 ~ /^\s*($|#)/) {next}; {print $desiredColumn}' ${cfgFile}
    else
        __v_VIRSH snapshot-list ${1} --name
    fi
}

_valine()  {
  
    # Variables
    local curr prev prevX2 virtDomains validsubcmds cdromSource
    
    # Wipe out COMPREPLY array
    COMPREPLY=()
  
    # Set cur & prev appropriately
    curr=${COMP_WORDS[COMP_CWORD]}
    prev=${COMP_WORDS[COMP_CWORD-1]}
    prevX2=${COMP_WORDS[COMP_CWORD-2]}
    
    # Check previous arg to see if we need to do anything special
    case "${prev}" in
        v|valine)
            COMPREPLY=( $(compgen -W "-a --all $(__v_get_domains)" -- "${curr}") )
            ;;
        --help|-h|--size|start|s|Shutdown|S|H|Hard-reboot|hibernate|h|destroy|d|console|c|loop-ssh|l|NUKE|N|set-maxmem|set-mem)
            ;;
        --off)
            if [[ ${prevX2} =~ ^n(ew-snap)?$ || ${COMP_WORDS[COMP_CWORD-3]} =~ ^n(ew-snap)?$ ]]; then
                COMPREPLY=( $(compgen -W "--size" -- "${curr}") )
            fi
            ;;
        Change-media|C)
            if __v_dom_has_cdrom "${prevX2}"; then
                compopt -o plusdirs  # Important!
                COMPREPLY=( $(compgen -f -- "${curr}") )
            fi
            ;;
        new-snap|n)
            COMPREPLY=( $(compgen -W "--off --size" -- "${curr}") )
            ;;
        revert-snap|r)
            case "${prevX2}" in
                --all|-a)
                    COMPREPLY=( $(compgen -W "--off" -- "${curr}") )  ;;
                *)
                    if __v_get_domains | grep -qs -- "^${prevX2}$"; then
                        COMPREPLY=( $(compgen -W "--off $(__v_list_snapshots ${prevX2})" -- "${curr}" ) )
                    fi
            esac
            ;;
        Delete-snap|D)
            case "${prevX2}" in
                --all|-a)
                    : ;;
                *)
                    if __v_get_domains | grep -qs -- "^${prevX2}$"; then
                        COMPREPLY=( $(compgen -W "$(__v_list_snapshots ${prevX2})" -- "${curr}" ) )
                    fi
            esac
            ;;
        --all|-a)
            COMPREPLY=( $(compgen -W "new-snap revert-snap start Shutdown Hard-reboot hibernate destroy" -- "${curr}") )
            ;;
        *)
            if [[ ${prevX2} == --size || ${prevX2} =~ ^r(evert-snap)?$ ]]; then
                COMPREPLY=( $(compgen -W "--off" -- "${curr}") )
            elif [[ ${prevX2} =~ ^n(ew-snap)?$ ]]; then
                COMPREPLY=( $(compgen -W "--off --size" -- "${curr}") )
            elif __v_get_domains | grep -qs -- "^${prev}$"; then
                validsubcmds="new-snap loop-ssh NUKE set-maxmem set-mem"
                [[ -n $(__v_list_snapshots "${prev}") ]] && validsubcmds+=" revert-snap Delete-snap"
                if __v_dom_is_off "${prev}"; then
                    validsubcmds+=" start"
                else
                    validsubcmds+=" Shutdown Hard-reboot hibernate destroy console"
                fi
                __v_dom_has_cdrom "${prev}" && validsubcmds+=" Change-media"
                COMPREPLY=( $(compgen -W "${validsubcmds}" -- "${curr}") )
            fi
    esac
    return 0

}

# Add the names of any valine aliases (or alternate file-names) to the end of the following line
complete -F _valine v valine