From f4a1f40eff2ace8f9c92a5a24e44d6ca5de804a8 Mon Sep 17 00:00:00 2001
From: Anna Schumaker <Anna.Schumaker@Netapp.com>
Date: Mon, 29 Nov 2021 15:04:55 -0500
Subject: [PATCH] Create a wrapper around ssh

To check if we're trying to ssh to a VM so the VM can be booted if it is
off

Signed-off-by: Anna Schumaker <Anna.Schumaker@Netapp.com>
---
 completions/_ssh.zsh | 15 +++++++++++++++
 ssh.zsh              |  4 ++++
 2 files changed, 19 insertions(+)
 create mode 100644 completions/_ssh.zsh
 create mode 100755 ssh.zsh

diff --git a/completions/_ssh.zsh b/completions/_ssh.zsh
new file mode 100644
index 0000000..5f97d38
--- /dev/null
+++ b/completions/_ssh.zsh
@@ -0,0 +1,15 @@
+#compdef ssh.zsh
+
+function _ssh.zsh() {
+	_arguments \
+		':args: _alternative
+			"hosts:hosts: _ssh_hosts"
+			"users:users: _ssh.zsh_users -qS@"
+			"domains:domains:($(virsh list --all --name))"'
+}
+
+_ssh.zsh_users () {
+	_combination -s '[:@]' my-accounts users-hosts users "$@"
+} 
+
+_ssh.zsh "$@"
diff --git a/ssh.zsh b/ssh.zsh
new file mode 100755
index 0000000..c76b844
--- /dev/null
+++ b/ssh.zsh
@@ -0,0 +1,4 @@
+#!/bin/zsh
+
+vm.zsh boot $1
+ssh $*