From d11528ca98a9d76ab6e82fa11fb34abd4c709b75 Mon Sep 17 00:00:00 2001
From: Michael Marineau <michael.marineau@coreos.com>
Date: Thu, 8 May 2014 17:42:30 -0400
Subject: [PATCH] fix(image_set_group): Add support for uploading images.

---
 image_set_group | 13 +++++++++----
 1 file changed, 9 insertions(+), 4 deletions(-)

diff --git a/image_set_group b/image_set_group
index 8477e75732..d523fee87b 100755
--- a/image_set_group
+++ b/image_set_group
@@ -66,7 +66,7 @@ fi
 
 # Source should include version.txt, switch to its version information
 if [[ ! -f "${FLAGS_from}/version.txt" ]]; then
-  die_notrace "Source version info does not exist: ${SRC_IMAGE}"
+  die_notrace "Source version info does not exist: ${FLAGS_from}/version.txt"
 fi
 source "${FLAGS_from}/version.txt"
 COREOS_VERSION_STRING="${COREOS_VERSION}"
@@ -97,8 +97,8 @@ info "Copying from ${FLAGS_from}"
 cp "${SRC_IMAGE}" "${DST_IMAGE}"
 cp "${FLAGS_from}/version.txt" "${BUILD_DIR}/version.txt"
 UPDATE_PREFIX="${COREOS_PRODUCTION_IMAGE_NAME%_image.bin}_update"
-if [[ -e "${FLAGS_from}/${UPDATE_PREFIX}"* ]]; then
-    cp "${FLAGS_from}/${UPDATE_PREFIX}"* "${BUILD_DIR}/"
+if [[ -e "${FLAGS_from}/${UPDATE_PREFIX}.bin" ]]; then
+  cp "${FLAGS_from}/${UPDATE_PREFIX}".{bin,gz,meta,zip} "${BUILD_DIR}/"
 fi
 
 "${BUILD_LIBRARY_DIR}/disk_util" --disk_layout="${FLAGS_disk_layout}" \
@@ -114,7 +114,12 @@ EOF
 cleanup_mounts "${ROOT_FS_DIR}"
 trap - EXIT
 
-# TODO: Upload here
+upload_image "${DST_IMAGE}"
+upload_image "${BUILD_DIR}/version.txt"
+if [[ -e "${BUILD_DIR}/${UPDATE_PREFIX}.bin" ]]; then
+  upload_image -d "${BUILD_DIR}/${UPDATE_PREFIX}.DIGESTS" \
+      "${BUILD_DIR}/${UPDATE_PREFIX}".{bin,gz,meta,zip}
+fi
 
 set_build_symlinks "${FLAGS_group}-latest"