find-ami

#!/bin/bash

# find-ami - Find the latest AMI based on region, description, name, etc.
#
# Author:   Michal Ludvig <mludvig@logix.net.nz>
# License:  BSD
# Homepage: https://aws.nz/aws-utils/find-ami

function show_help() {
	cat << __EOF__
Usage: $0 [-h] [options]

Purge AMI (Images) older than a given date, optionally matching certain
filters.

   -d AMI-ID    Describe AMI details (Name, Owner, etc)

   -l           List all AMIs matching the NAME and OWNER.
                By default only the newest one is returned.

   -n NAME      AMI Name, can be regexp.
                E.g: amzn-ami-hvm-[^-]*-gp2

   -o OWNER     AMI Owner, can be 'amazon' or numeric owner ID.

   -p PROFILE   AWS-CLI Profile name as defined in ~/.aws/config.
                Optional, will use default or instance meta-data if
                not specified.

   -r REGION    AWS Region where to look for the AMI.

   -t VIRT_TYPE Virtualization type:
                * hvm (default)
                * paravirtual

   -x PRESET    Name/Description/Owner preset for common AMIs:
                * amazon-linux
                * bitnami-wordpress
                You can override preset NAME with -n

   -h           Display help.

Visit https://aws.nz/aws-utils/find-ami for more info and examples.
__EOF__
	exit 1
}

function fatal() {
	echo $1 >&2
	exit 1
}

AWS=$(which aws 2>/dev/null)
test -z "${AWS}" && fatal "The 'aws' utility not found. Please install it or add its location to \$PATH"

JQ=$(which jq 2>/dev/null)
test -z "${JQ}" && fatal "The 'jq' utility not found. Please install it or add its location to \$PATH"

VIRT_TYPE="hvm"
LIST_ALL=0

while getopts "d:ln:o:p:r:x:h" OPT; do
	case "${OPT}" in
	d)
		DESCRIBE_AMI="${OPTARG}"
		;;
	l)
		LIST_ALL=1
		;;
	n)
		NAME="${OPTARG}"
    # Escape "()" to "\(\)"
    NAME="${NAME//\(/\\\\(}"
    NAME="${NAME//\)/\\\\)}"
		;;
	o)
		OWNER="${OPTARG}"
		;;
	p)
		AWS="${AWS} --profile=${OPTARG}"
		;;
	r)
		AWS="${AWS} --region=${OPTARG}"
		;;
	t)
		VIRT_TYPE="${OPTARG}"
		;;
	x)
		PRESET="${OPTARG}"
		;;
	h)
		show_help
		;;
	esac
done

if [ -n "${DESCRIBE_AMI}" ]; then
	IMAGES_JSON=$(${AWS} ec2 describe-images --filter Name=image-id,Values=${DESCRIBE_AMI})
	${JQ} -r ".Images[] |
		@text \"OWNER: \(.OwnerId)\",
		@text \"NAME:  \(.Name)\",
		@text \"DESC:  \(.Description)\",
		@text \"VIRT:  \(.VirtualizationType)\"
	" <<< ${IMAGES_JSON}
	exit 0
fi

if [ -n "${PRESET}" ]; then
	case "${PRESET}" in
	amazon-linux)
		OWNER=${OWNER:=amazon}
		NAME=${NAME:=amzn-ami-[^-]*-[^-]*-gp2}
		;;
	redhat-el7)
		OWNER=${OWNER:=309956199498}
		NAME=${NAME:=RHEL-7.*_HVM.*x86_64.*Hourly2-GP2}
		;;
	bitnami-wordpress)
		OWNER=${OWNER:=679593333241}
		NAME=${NAME:=bitnami-wordpress-[0-9.-]+-linux-ubuntu-.*-ebs}
		;;
	*)
		fatal "Preset ID '${PRESET}' isn't configured."
		;;
	esac
fi

echo "# Listing AMIs from owner ${OWNER} matching pattern: ${NAME}"
echo "# Please wait, this may take a long time..."
IMAGES_JSON=$(${AWS} ec2 describe-images --owners=${OWNER} --filter Name=virtualization-type,Values=${VIRT_TYPE})

if [ "${LIST_ALL}" = "1" ]; then
	# List all matching AMIs
	${JQ} -r "[.Images[]
		| select(.Name | test(\"${NAME}\"))
		] | sort_by(.CreationDate) | reverse | .[] |
		@text \"\(.ImageId)    \(.CreationDate)    \(.Name)\"
		" <<< ${IMAGES_JSON}
else
	# Sort AMIs by date and display only the newest one
	${JQ} -r "[.Images[]
		| select(.Name | test(\"${NAME}\"))
		] | max_by(.CreationDate) |
		@text \"export AMI_ID='\(.ImageId)'\",
		@text \"export AMI_NAME='\(.Name)'\",
		@text \"export AMI_OWNER='\(.OwnerId)'\",
		@text \"export AMI_DESCRIPTION='\(.Description)'\",
		@text \"export AMI_VIRT_TYPE='\(.VirtualizationType)'\"
		" <<< ${IMAGES_JSON}
fi