get-live-images.sh

#!/bin/bash -xe

# Check and set http(s)_proxy. Required for cURL to use a proxy
export http_proxy=${http_proxy:-$HTTP_PROXY}
export https_proxy=${https_proxy:-$HTTPS_PROXY}
export no_proxy=${no_proxy:-$NO_PROXY}
export CURL_CA_BUNDLE=${CURL_CA_BUNDLE:-/etc/pki/ca-trust/extracted/pem/tls-ca-bundle.pem}
export IP_OPTIONS=${IP_OPTIONS:-}

# Which image should we use
export RHCOS_IMAGE_URL=${1:-$RHCOS_IMAGE_URL}
if [ -z "$RHCOS_IMAGE_URL" ] ; then
    echo "No image URL provided"
    exit 1
fi

function download_image() {
	CONNECT_TIMEOUT=120
	MAX_ATTEMPTS=5

	for i in $(seq ${MAX_ATTEMPTS}); do
		if ! curl -v -g --compressed -L --fail --connect-timeout ${CONNECT_TIMEOUT} -o "$1" "$2"; then
			if (( ${i} == ${MAX_ATTEMPTS} )); then
				echo "Download failed."
				exit 1
			else
				SLEEP_TIME=$((i*i))
				echo "Download failed, retrying after ${SLEEP_TIME} seconds..."
				sleep ${SLEEP_TIME}
			fi
		else
			break
		fi
	done
}

function cache_image() {
	FILENAME=$(basename $1)
	FILENAME_CACHED=cached-${FILENAME}
	TMPDIR=$(mktemp -d -p /shared/tmp)
	trap "rm -fr $TMPDIR" EXIT
	cd $TMPDIR

	# We have a file in the cache that matches the one we want, use it
	if [ -s "/shared/html/images/$FILENAME/$FILENAME_CACHED" ]; then
		echo "Cached file: $FILENAME/$FILENAME_CACHED found"
	else
		download_image "${FILENAME_CACHED}" "${URL_STRIPPED}"
	fi

	cd /shared/html/images
	chmod 755 $TMPDIR
	mv $TMPDIR $FILENAME
	if [[ $FILENAME =~ live-initramfs ]]; then
		ln -sf "$FILENAME/$FILENAME_CACHED" "$FFILENAME.initramfs"
	elif [[ $FILENAME =~ live-rootfs ]]; then
		ln -sf "$FILENAME/$FILENAME_CACHED" "$FFILENAME.rootfs"
	elif [[ $FILENAME =~ live-kernel ]]; then
		ln -sf "$FILENAME/$FILENAME_CACHED" "$FFILENAME.kernel"
	elif [[ $FILENAME =~ -live ]] && [[ $FILENAME =~ .iso ]]; then
		ln -sf "$FILENAME/$FILENAME_CACHED" "$FFILENAME.iso"
		# Append dhcp options for dualstack installs
		if [ -n "$IP_OPTIONS" ] ; then
			coreos-installer iso kargs modify -a "$IP_OPTIONS" "$FILENAME/$FILENAME_CACHED"
		fi
	fi
	# We need to create a symlink so that the second-level image cache
	# can download and prefix 'cached-' to the filename correctly
	cd $FILENAME
	ln -sf "$FILENAME_CACHED" "$FILENAME"
}

mkdir -p /shared/html/images /shared/tmp

URLS=($(echo $RHCOS_IMAGE_URL | tr "," "\n"))

# This name will be used in the ironic image to embed the agent ignition
FFILENAME="ironic-python-agent"
for URL in "${URLS[@]}"
do
	URL_STRIPPED=`echo $URL | cut -f 1 -d \?`
	cache_image $URL_STRIPPED
done