sh launch_ubuntu_on_aws.sh

Posted 2021-05-07

#!/bin/bash -e
## Launch the latest daily Ubuntu in AWS, optionally with user-data
## provided in a file and tagged with a name

function usage() {
    echo "$0: [-r region] [-V ubuntu_version] [-i instance_type] \\"
    echo "      [-u user_data_file] [-n instance_name] [-d disk_size] \\"
    echo "      [-D] to disable termination via the API"
    echo "      [-I] iam instance profile"
    echo ""
    echo "Defaults:     region: <default set in aws cli>"
    echo "              ubuntu_version: ${UBUNTU_VERSION}"
    echo "              instance_type: t2.nano"
}

function find_ami() {
    # Find the latest HVM EBS Ubuntu AMI from the official publication account
    REGION=$1
    UBUNTU_VERSION=$2

    aws --output json --region "${REGION}" ec2 describe-images \
         --owners 099720109477 \
        --filters "Name=name,Values=ubuntu/images-testing/hvm-ssd/ubuntu-${UBUNTU_VERSION}-daily-amd64-server-*" | \
      jq -r '.Images | sort_by(.["CreationDate"])[-1].ImageId'
}

function wait_instance_running() {
    # Wait for 2m for instance to enter running state
    REGION=$1
    INSTANCE_ID=$2

    echo -n "Waiting for running state..."
    for attempt in $(seq 12); do
        STATE=$(aws --output json --region "${REGION}" ec2 describe-instances \
                    --instance-ids "${INSTANCE_ID}" | \
                jq -r .Reservations[0].Instances[0].State.Name)
        if [ "${STATE}" = "running" ]; then
            echo " running."
            return
        else
            echo -n "."
            sleep 10
        fi
    done
    echo "failed."
    echo "Instance did not enter 'running' state after 2 minutes."
    return 1
}

# Check for awscli and 'jq'
set +e
aws --version 2>/dev/null >/dev/null
if [ "$?" != 0 ] ; then
    echo "This command requires the aws cli to be installed."
    exit 1
fi
jq --version 2>/dev/null >/dev/null
if [ "$?" != 0 ] ; then
    echo "This command requires the json query command 'jq' to be installed."
    exit 1
fi
set -e

REGION=$(aws configure get region)
UBUNTU_VERSION=xenial
INSTANCE_TYPE=t2.nano
USER_DATA_FILE=
INSTANCE_NAME=
DISK_SIZE=
API_TERM=
KEY_NAME="${AWS_KEY_NAME}"
IAM_PROFILE=

while getopts ":r:V:i:u:n:d:Dk:I:h" opt; do
    case "${opt}" in
    r)  REGION="${OPTARG}";;
    V)  UBUNTU_VERSION="${OPTARG}";;
    i)  INSTANCE_TYPE="${OPTARG}";;
    u)  USER_DATA_FILE="${OPTARG}";;
    n)  INSTANCE_NAME="${OPTARG}";;
    d)  DISK_SIZE="${OPTARG}";;
    D)  API_TERM="--disable-api-termination";;
    k)  KEY_NAME="${OPTARG}";;
    I)  IAM_PROFILE="${OPTARG}";;
    h)
        usage
        exit 0
        ;;
    \?)
        echo "Invalid option: -${OPTARG}" >&2
        usage
        exit 1
        ;;
    esac
done

echo "${KEY_NAME:?Key name must be specified via -k or envvar AWS_KEY_NAME}" > /dev/null

# Latest HVM EBS-SSD ${UBUNTU_VERSION} daily
echo -n "Finding an ami in ${REGION}... "
AMI=$(find_ami "${REGION}" "${UBUNTU_VERSION}")
echo "${AMI}"

USER_DATA=''
if [ -n "${USER_DATA_FILE}" ] ; then
    USER_DATA="--user-data file://${USER_DATA_FILE}"
fi

BDM=''
if [ -n "${DISK_SIZE}" ] ; then
    BDM="--block-device-mappings DeviceName=/dev/sda1,Ebs={VolumeSize=${DISK_SIZE},VolumeType=gp2}"
fi

IAM_INST_PROFILE=''
if [ -n "${IAM_PROFILE}" ] ; then
    IAM_INST_PROFILE="--iam-instance-profile Name=${IAM_PROFILE}"
fi

echo -n "Launching instance... "
INSTID=$(aws --output json --region "${REGION}" ec2 run-instances \
                 --key-name "${KEY_NAME}" \
                 --instance-type "${INSTANCE_TYPE}" \
                 --image-id "${AMI}" \
                 ${USER_DATA} ${BDM} ${API_TERM} ${IAM_INST_PROFILE} | \
             jq -r .Instances[0].InstanceId)
echo "${INSTID}"

wait_instance_running "${REGION}" "${INSTID}"

if [ -n "${INSTANCE_NAME}" ]; then
    echo "Tagging instance"
    aws --output json --region "${REGION}" ec2 create-tags \
        --resources "${INSTID}" \
        --tags "Key=Name,Value=${INSTANCE_NAME}"
fi

echo -n "IP address: "
aws --output json --region "${REGION}" ec2 describe-instances \
    --instance-ids "${INSTID}" | \
    jq -r '.Reservations[0].Instances[0].PublicIpAddress'