addsshkey

#!/usr/bin/env bash

set -e

USAGE="addsshkey [-h] [USER KEY] -- add a new ssh key to an account.
Creates and chmods .ssh if it doesn't exist, appends key to ~/.ssh/authorized_keys

    -h     show this help text 
    USER   the account username 
    KEY    the ssh key (in quotes, including ssh-rsa)"

# No options, show usage
if [[ $# == 0 ]] ; then
    echo "$USAGE"
    exit 0;
fi

# options
while getopts 'h' option; do
  case "$option" in
    h) echo "$USAGE"
       exit 0
       ;;
    *) echo "Error: invalid option to $0" >&2
       exit 1
       ;;
  esac
done

# no options, use two arguments
username=$1
ssh_key=$2

# become user, create ssh directory and add key
echo "Becoming ${username}, creating ssh dir and adding key"
become_output=$(sudo /shared/ucl/sysops/libexec/become "$username" <<EOF
echo "Beacon"
mkdir -p .ssh
chmod go-rwx .ssh
echo "${ssh_key}" >> ~/.ssh/authorized_keys
EOF
)
become_exit_status=$?
if [[ "${become_output:0:6}" != "Beacon" ]]; then
  echo "Error: could not become user ${username}" >&2
  exit 1
fi
if [[ "$become_exit_status" -ne 0 ]]; then
  echo "Warning: keydrop subprocess exited with a non-zero code. Check output." >&2
  exit "$become_exit_status"
fi