From 5634d522f9a851127c6a6cd37222216f165d1486 Mon Sep 17 00:00:00 2001
From: Anna Schumaker <Anna.Schumaker@Netapp.com>
Date: Thu, 2 Dec 2021 15:00:14 -0500
Subject: [PATCH] deploykernel.zsh: Take a list of machines to install on

Signed-off-by: Anna Schumaker <Anna.Schumaker@Netapp.com>
---
 completions/_deploykernel.zsh |  2 +-
 deploykernel.zsh              | 23 ++++++++++++++++-------
 2 files changed, 17 insertions(+), 8 deletions(-)

diff --git a/completions/_deploykernel.zsh b/completions/_deploykernel.zsh
index d830327..9ed7596 100644
--- a/completions/_deploykernel.zsh
+++ b/completions/_deploykernel.zsh
@@ -2,7 +2,7 @@
 
 function _deploykernel.zsh() {
 	_arguments \
-		':destination: _alternative
+		'*:destination: _alternative
 			"hosts:hosts: _ssh_hosts"
 			"domains:domains:($(virsh list --all --name))"'
 }
diff --git a/deploykernel.zsh b/deploykernel.zsh
index d829836..91a3af4 100755
--- a/deploykernel.zsh
+++ b/deploykernel.zsh
@@ -2,18 +2,27 @@
 autoload colors
 colors
 
-if [ "$#" -ne 1 ]; then
-	echo "Usage: $0 {remote host}"
+if [ "$#" -lt 1 ]; then
+	echo "Usage: $0 {remote host, ...}"
 	exit 1
 fi
 
 makepkg -f
-deploypkg.zsh $(makepkg --packagelist) $1
+deploypkg.zsh $(makepkg --packagelist) $*
 
 pkgname=$(grep ^pkgname PKGBUILD | awk -F= '{print $2}')
-title=$(ssh.zsh $1 grub-list.zsh -e | grep $pkgname$)
-ssh.zsh $1 "sudo grub-reboot \"$title\""
+
+for vm in $*; do
+	title=$(ssh.zsh $vm grub-list.zsh -e | grep $pkgname$)
+	ssh.zsh $vm "sudo grub-reboot \"$title\""
+done
 
 echo
-vm.zsh reboot $1
-echo $fg_bold[cyan]$(ssh.zsh $1 uname -rsn)$reset_color
+
+vm.zsh reboot $*
+echo
+
+tags=($(python /home/anna/bin/colors/vm.py $*))
+for ((i=1; i<=$#; i++)); do
+	echo "${tags[i]} $fg_bold[default]$(ssh.zsh ${@[i]} uname -rsn)$reset_color"
+done