Create an install-scripts.zsh script

To copy over the things we need on remote machines. I also update
cthon.zsh to copy over a setup script.

Signed-off-by: Anna Schumaker <Anna.Schumaker@Netapp.com>

completions/_install-scripts.zsh

@@ -0,0 +1,8 @@
+#compdef install-scripts.zsh
+
+function _install_scripts.zsh() {
+	_arguments \
+		'*:remote host: _alternative
+			"hosts:hosts: _ssh_hosts"
+			"domains:domains:($(virsh list --all --name))"'
+}

cthon.zsh

@@ -32,18 +32,13 @@ fi
 if [ ${#KRB5} -gt 0 ]; then
 	RUN_CTHON="$RUN_CTHON --dokrb5"
 fi
-echo
 
-ssh ${CLIENT[-1]} "sudo mkdir -p ${MOUNTPOINT[-1]}"
-ssh ${CLIENT[-1]} "sudo mount -o sec=sys ${SERVER[-1]}:${EXPORT[-1]} ${MOUNTPOINT[-1]}"
-TRAPEXIT() { ssh ${CLIENT[-1]} "sudo umount ${MOUNTPOINT[-1]}" }
-ssh ${CLIENT[-1]} "$RUN_CTHON --mkdirs ${MOUNTPOINT[-1]}/$USER/ 2>&1" | python -u $COLOR
-ssh ${CLIENT[-1]} "sudo chmod -R 777 ${MOUNTPOINT[-1]}/$USER/"
-TRAPEXIT() { ; }
-ssh ${CLIENT[-1]} "sudo umount ${MOUNTPOINT[-1]}"
+echo
+$BIN/install-scripts.zsh ${CLIENT[-1]}
+ssh ${CLIENT[-1]} "sudo setup-cthon.zsh ${SERVER[-1]} ${EXPORT[-1]} ${MOUNTPOINT[-1]} $USER" | python -u $COLOR
 echo
 
 #
 # Run tests
 #
-ssh client "$RUN_CTHON --server ${SERVER[-1]} --serverdir ${EXPORT[-1]}/$USER 2>&1" | python -u $COLOR
+ssh ${CLIENT[-1]} "$RUN_CTHON --server ${SERVER[-1]} --serverdir ${EXPORT[-1]}/$USER 2>&1" | python -u $COLOR

install-scripts.zsh

@@ -0,0 +1,17 @@
+#!/bin/zsh
+BIN=$HOME/bin
+SCRIPTS=(grub-list.zsh setup-cthon.zsh)
+
+function install_script() {
+	ssh $1 mkdir -p bin/
+	echo -n "$2 Copying files ..."
+	for script in $SCRIPTS; do
+		scp -q $BIN/$script $1:bin/
+	done
+	echo " done"
+}
+
+tags=($(python $BIN/colors/vm.py $*))
+for ((i=1; i<=${#@}; i++)); do
+	install_script ${@[i]} ${tags[i]}
+done

setup-cthon.zsh

@@ -0,0 +1,19 @@
+#!/bin/zsh
+SERVER=$1
+EXPORT=$2
+MOUNTPOINT=$3
+USER=$4
+
+if [ "$#" -lt 4 ]; then
+	echo "Usage: $0 {server} {export} {mountpoint} {user}"
+	exit 1
+fi
+
+TRAPEXIT() {
+	umount $MOUNTPOINT
+}
+
+mkdir -p $MOUNTPOINT
+mount -o sec=sys $SERVER:$EXPORT $MOUNTPOINT
+cd /home/$USER/cthon04 && ./runcthon --rdma --mkdirs $MOUNTPOINT/$USER
+chmod 777 $MOUNTPOINT/$USER