#!/bin/bash
set -euo pipefail

# USAGE
# awsssh <instance-id>
# awsssh <ssh key> <instance-id>
#
# Looks for AWS credentials in ~/.aws/credentials

SSH_ID=""
#if [ $# -eq 2 ]; then
#  SSH_ID="-i ${1}"
#  shift
#fi

IP=$(aws ec2 describe-instances \
              --filters "Name=instance-id,Values=${1}" \
              --query 'Reservations[0].Instances[0].PrivateIpAddress' \
              --output text)

if [ $? != 0 ]; then
  exit 1
fi

if [ -z ${IP} -o ${IP} == "None" ]; then
  echo "Invalid Instance ID"
  exit 1
fi

shift

ssh ${SSH_ID} ${USER}@${IP} $@
