setup-user: prompt user for ssh key

2 files changed, 38 insertions, 3 deletions

diff --git a/setup-user.in b/setup-user.in
index 8b9e6cf..c02c1a2 100644
--- a/setup-user.in
+++ b/setup-user.in
@@ -6,23 +6,26 @@ PREFIX=@PREFIX@
 
 usage() {
 	cat <<-__EOF__
-		usage: setup-user [-h] [-f FULLNAME] [USERNAME]
+		usage: setup-user [-h] [-f FULLNAME] [-k SSHKEY] [USERNAME]
 
 		Create user account
 
 		options:
 		 -h  Show this help
 		 -f  Set full name for user
+		 -k  ssh key or URL to ssh key (eg. https://gitlab.alpinelinux.org/user.keys)
+		     or 'none' for no key
 
 		If USERNAME is not specified user will be prompted.
 	__EOF__
 	exit $1
 }
 
-while getopts "f:h" opt; do
+while getopts "f:hk:" opt; do
 	case $opt in
 		h) usage 0;;
 		f) fullname="$OPTARG";;
+		k) keysopt="$OPTARG";;
 		'?') usage "1" >&2;;
 	esac
 done
@@ -63,4 +66,35 @@ else
 	fi
 fi
 
+if [ -n "$interactive" ] && [ -z "$keysopt" ]; then
+	while true; do
+		ask "Enter ssh key or URL for $username (or 'none')"
+		case "$resp" in
+			none) break;;
+			https://*|http://*) sshkeys=$(wget -q -O- $resp | grep ^ssh-);;
+			*) sshkeys="$resp";;
+		esac
+		if echo "$sshkeys" | grep -q ^ssh-; then
+			break
+		fi
+		echo "Did not find any key in '$resp'"
+	done
+else
+	case "$keysopt" in
+		https://*|http://*)
+			sshkeys=$(wget -q -O- "$sshkeys" | grep ^ssh-);;
+		none)
+			sshkeys="" ;;
+		*)
+			sshkeys="$keysopt";;
+	esac
+	if [ -n "$sshkeys" ] && ! echo "$sshkeys" | grep -q ^ssh-; then
+		echo "Could not find any keys in '$resp'" >&2
+		exit 1
+	fi
+fi
 
+if [ -n "$sshkeys" ] && [ "$sshkeys" != "none" ]; then
+	mkdir -p "$ROOT"/home/$username/.ssh
+	echo "$sshkeys" > "$ROOT"/home/$username/.ssh/authorized_keys
+fi
diff --git a/tests/setup_user_test b/tests/setup_user_test
index c2c7ce0..c67f495 100755
--- a/tests/setup_user_test
+++ b/tests/setup_user_test
@@ -33,6 +33,7 @@ setup_user_interactive_body() {
 	(
 		echo "Joe User"
 		echo "testuser"
+		echo "none"
 	) | atf_check -s exit:0 \
 		-o match:"Enter full name" \
 		-o match:"Enter username" \
@@ -52,6 +53,6 @@ setup_user_interactive_fullname_body() {
 		-o not-match:"adduser.* -D .*testuser" \
 		-o match:"adduser.* -g Joe User .*testuser" \
 		-e empty \
-		setup-user -f "Joe User"
+		setup-user -f "Joe User" -k none
 }