From 19008a43a722327f9a0e45b249880641eea0580c Mon Sep 17 00:00:00 2001
From: Alex Polvi <alex@polvi.net>
Date: Mon, 14 Oct 2013 14:03:36 -0700
Subject: [PATCH] feat(all-amis): add uploading off all ami regions

---
 oem/ami/upload_ami_txt.sh | 55 +++++++++++++++++++++++++++++----------
 1 file changed, 41 insertions(+), 14 deletions(-)

diff --git a/oem/ami/upload_ami_txt.sh b/oem/ami/upload_ami_txt.sh
index 05e1d8b63d..acb8054c7f 100755
--- a/oem/ami/upload_ami_txt.sh
+++ b/oem/ami/upload_ami_txt.sh
@@ -3,6 +3,19 @@
 # Set pipefail along with -e in hopes that we catch more errors
 set -e -o pipefail
 
+# we just use this for the list of regions
+# should a copy/paste from build_ebs_on_ec2.sh
+declare -A AKI
+AKI["us-east-1"]=aki-b4aa75dd
+AKI["us-west-1"]=aki-eb7e26ae
+AKI["us-west-2"]=aki-f837bac8
+AKI["eu-west-1"]=aki-8b655dff
+AKI["ap-southeast-1"]=aki-fa1354a8
+AKI["ap-southeast-2"]=aki-3d990e07
+AKI["ap-northeast-1"]=aki-40992841
+AKI["sa-east-1"]=aki-c88f51d5
+# AKI["gov-west-1"]=aki-75a4c056
+
 USAGE="Usage: $0 -a ami-id
     -V VERSION  Find AMI by CoreOS version. (required)
     -K KEY      Path to Amazon API private key.
@@ -35,26 +48,40 @@ if [[ $(id -u) -eq 0 ]]; then
     exit 1
 fi
 
-if [[ -n "$VER" ]]; then
-    AMI=$(ec2-describe-images -F name="CoreOS-$VER" | grep -m1 ^IMAGE \
-        | cut -f2) || true # Don't die silently, error messages are good
-    if [[ -z "$AMI" ]]; then
-        echo "$0: Cannot find an AMI for CoreOS $VER" >&2
-        exit 1
-    fi
-else
+if [[ ! -n "$VER" ]]; then
     echo "$0: AMI version required via -V" >&2
     echo "$USAGE" >&2
     exit 1
 fi
 
-zoneurl=http://instance-data/latest/meta-data/placement/availability-zone
-zone=$(curl --fail -s $zoneurl)
-region=$(echo $zone | sed 's/.$//')
-url=$(printf "$URL_FMT" "$VER" "$region")
+declare -A AMIS
+for r in "${!AKI[@]}"; do
+    AMI=$(ec2-describe-images --region=${r} -F name="CoreOS-$VER" | grep -m1 ^IMAGE \
+        | cut -f2) || true # Don't die silently, error messages are good
+    if [[ -z "$AMI" ]]; then
+        echo "$0: Cannot find ${r} AMI for CoreOS $VER" >&2
+	continue
+    fi
+    AMIS[${r}]=$AMI
+done
 
+OUT=
+for r in "${!AMIS[@]}"; do
+    url=$(printf "$URL_FMT" "$VER" "$r")
+    tmp=$(mktemp --suffix=.txt)
+    trap "rm -f '$tmp'" EXIT
+    echo "${AMIS[$r]}" > "$tmp"
+    gsutil cp "$tmp" "$url"
+    echo "OK, $r ${AMIS[$r]}, $url"
+    if [[ -z "$OUT" ]]; then
+        OUT="${r}=${AMIS[$r]}"
+    else
+        OUT="${OUT}|${r}=${AMIS[$r]}"
+    fi
+done
+url=$(printf "$URL_FMT" "$VER" "all")
 tmp=$(mktemp --suffix=.txt)
 trap "rm -f '$tmp'" EXIT
-echo "$AMI" > "$tmp"
+echo "$OUT" > "$tmp"
 gsutil cp "$tmp" "$url"
-echo "OK"
+echo "OK, all, $url"