From 55251c388b0d0571441f7f2b11bd5d2bd4c2dfda Mon Sep 17 00:00:00 2001
From: Wryn Wagner <wagner.wryn@gmail.com>
Date: Tue, 23 Feb 2021 13:38:09 -0700
Subject: [PATCH] add ssh utilities

---
 zsh/ssh | 63 +++++++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 63 insertions(+)
 create mode 100644 zsh/ssh

diff --git a/zsh/ssh b/zsh/ssh
new file mode 100644
index 0000000..6ded47f
--- /dev/null
+++ b/zsh/ssh
@@ -0,0 +1,63 @@
+#!/bin/zsh
+
+alias c='WRYN_REMOTE_CONNECTOR'
+alias sshc='WRYN_REMOTE_CONNECTOR ssh'
+alias sshe='$EDITOR $WRYN_SSH_CONNECTION_LIST'
+alias ssh-add-all='WRYN_SSH_ADD_CREDENTIALS'
+
+WRYN_SSH_CONNECTION_LIST="$HOME/.config/wryn/ssh-connections"
+WRYN_SSH_DELIMITER=" *| "
+[ -d "$HOME/.config/wryn" ] && [ ! -f "$WRYN_SSH_CONNECTION_LIST" ] && touch "$WRYN_SSH_CONNECTION_LIST";
+
+function WRYN_REMOTE_CONNECTOR() {
+	local COMMAND="$1"
+	local CONNECTION="$2"
+
+	local SSH_ID="$CONNECTION$WRYN_SSH_DELIMITER"
+
+	[ $2 ] \
+		|| { echo 'Must specify a connection'; return 1 };
+	grep -q "$SSH_ID" "$WRYN_SSH_CONNECTION_LIST" >/dev/null 2>&1 \
+		|| { echo 'No such connection'; return 2 };
+
+	local 
+	local PARSED_ARGS=$(sed -n "s/$SSH_ID\(.*\)$/\1/p" "$WRYN_SSH_CONNECTION_LIST" )
+
+	echo "$PARSED_ARGS:$3"
+	case $COMMAND in;
+		ssh     ) ssh $PARSED_ARGS ${@:3} ;;
+		sshfs   ) sshfs "$PARSED_ARGS:$3" ${@:4} ;;
+		* ) echo "$COMMAND not implemented" ;;
+	esac
+}
+_WRYN_REMOTE_CONNECTOR() {
+	local state;
+	_arguments \
+		'1: :->remote_command'\
+		'2: :->connections'\
+		':: :->remote_args'\
+	;
+	case "$state" in
+		remote_command)
+			compadd -X 'Choose a connection command:' ssh sshfs
+			;;
+		connections)
+			compadd -X 'Choose a conection:' $(sed "s/$WRYN_SSH_DELIMITER.*//" "$WRYN_SSH_CONNECTION_LIST")
+			;;
+		remote_args)
+			_files
+			case $words[2] in;
+				ssh     ) _ssh;;
+				sshfs   ) _sshfs;;
+				* ) ;;
+			esac
+			;;
+	esac
+}
+compdef _WRYN_REMOTE_CONNECTOR WRYN_REMOTE_CONNECTOR;
+
+WRYN_SSH_ADD_CREDENTIALS() {
+	for key in $(grep -slR "PRIVATE" "$HOME/.ssh/*"); do
+		ssh-add "$key" && { echo "Added $key" } || { echo "Failed to add $key" }
+	done
+}