# systemd-nspawn(1) completion                  -*- shell-script -*-
#
# This file is part of systemd.
#
# Copyright 2014 Thomas H.P. Andersen
#
# systemd is free software; you can redistribute it and/or modify it
# under the terms of the GNU Lesser General Public License as published by
# the Free Software Foundation; either version 2.1 of the License, or
# (at your option) any later version.
#
# systemd 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 Lesser General Public License
# along with systemd; If not, see <http://www.gnu.org/licenses/>.

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

__get_users() {
        local a b
        loginctl list-users --no-legend --no-pager | { while read a b; do echo " $b"; done; };
}

__get_slices() {
        local a b
        systemctl list-units -t slice --no-legend --no-pager | { while read a b; do echo " $a"; done; };
}

__get_machines() {
        local a b
        machinectl list --no-legend --no-pager | { while read a b; do echo " $a"; done; };
}

__get_env() {
        local a
        env | { while read a; do echo " ${a%%=*}"; done; };
}

__get_interfaces(){
        { cd /sys/class/net && echo *; } | \
        while read -d' ' -r name; do
                [[ "$name" != "lo" ]] && echo "$name"
        done
}

_systemd_nspawn() {
        local cur=${COMP_WORDS[COMP_CWORD]} prev=${COMP_WORDS[COMP_CWORD-1]}
        local i verb comps

        local -A OPTS=(
               [STANDALONE]='-h --help --version --private-network -b --boot --read-only -q --quiet --share-system --keep-unit --network-veth -j'
                      [ARG]='-D --directory -u --user --uuid --capability --drop-capability --link-journal --bind --bind-ro -M --machine
                             -S --slice --setenv -Z --selinux-context -L --selinux-apifs-context --register --network-interface --network-bridge
                             --personality -i --image --tmpfs --volatile
                             --network-macvlan --kill-signal --template
                             --notify-ready'
        )

        _init_completion || return

        if __contains_word "$prev" ${OPTS[ARG]}; then
                case $prev in
                        --directory|-D|--template)
                                compopt -o nospace
                                comps=$(compgen -S/ -A directory -- "$cur" )
                        ;;
                        --user|-u)
                                comps=$( __get_users )
                        ;;
                        --uuid)
                                comps=''
                        ;;
                        --capability)
                                comps='CAP_BLOCK_SUSPEND CAP_IPC_LOCK CAP_MAC_ADMIN CAP_MAC_OVERRIDE CAP_SYS_MODULE CAP_SYS_PACCT CAP_SYS_RAWIO
                                       CAP_SYS_TIME CAP_SYSLOG CAP_WAKE_ALARM CAP_NET_ADMIN'
                        ;;
                        --drop-capability)
                                comps='CAP_AUDIT_CONTROL CAP_AUDIT_WRITE CAP_CHOWN CAP_DAC_OVERRIDE CAP_DAC_READ_SEARCH CAP_FOWNER CAP_FSETID
                                       CAP_IPC_OWNER CAP_KILL CAP_LEASE CAP_LINUX_IMMUTABLE CAP_MKNOD CAP_NET_ADMIN CAP_NET_BIND_SERVICE
                                       CAP_NET_BROADCAST CAP_NET_RAW CAP_SETFCAP CAP_SETGID CAP_SETPCAP CAP_SETUID CAP_SYS_ADMIN CAP_SYS_BOOT
                                       CAP_SYS_CHROOT CAP_SYS_NICE CAP_SYS_PTRACE CAP_SYS_RESOURCE CAP_SYS_TTY_CONFIG'
                        ;;
                        --link-journal)
                                comps='no auto guest try-guest host try-host'
                        ;;
                        --bind|--bind-ro)
                                compopt -o nospace
                                comps=$(compgen -S/ -A directory -- "$cur" )
                        ;;
                        --tmpfs)
                                compopt -o nospace
                                comps=$(compgen -S/ -A directory -- "$cur" )
                        ;;
                        --machine|-M)
                                comps=$( __get_machines )
                        ;;
                        --slice|-S)
                                comps=$( __get_slices )
                        ;;
                        --setenv)
                                comps=$( __get_env )
                        ;;
                        --selinux-context|-Z)
                                comps=''
                        ;;
                        --selinux-apifs-context|-L)
                                comps=''
                        ;;
                        --register)
                                comps='yes no'
                        ;;
                        --network-interface)
                                comps=$(__get_interfaces)
                        ;;
                        --network-bridge)
                                comps=''
                        ;;
                        --network-macvlan)
                                comps=''
                        ;;
                        --personality)
                                comps='x86 x86-64'
                        ;;
                        --volatile)
                                comps='yes state no'
                        ;;
                        --image|-i)
                                compopt -o nospace
                                comps=$( compgen -A file -- "$cur" )
                        ;;
                        --kill-signal)
                                _signals
                                return
                        ;;
                        --notify-ready)
                                comps='yes no'
                                return
                        ;;
                esac
                COMPREPLY=( $(compgen -W '$comps' -- "$cur") )
                return 0
        fi

        COMPREPLY=( $(compgen -W '${OPTS[*]}' -- "$cur") )
}

complete -F _systemd_nspawn systemd-nspawn
