From 4cd8af1b4ef1fb9977d8977555ee44cce20798f4 Mon Sep 17 00:00:00 2001
From: Pino Toscano <ptoscano@redhat.com>
Date: Fri, 20 Mar 2015 13:26:02 +0100
Subject: [PATCH] Use the dib-utils repository too

Newer diskimage-builder versions require the external dib-utils
repository, as dib-run-parts has been moved there.

In preparation for using a newer diskimage-builder, checkout and use the
dib-utils repository; thus dib-run-parts from dib-utils will be used
instead of the one currently in diskimage-builder (if using a version
shipping it), which works the same way.
The dib-utils version used is the current last one, 0.0.9.

Change-Id: I7f537486279c40123b65d8206f9405774e9ebcbf
---
 diskimage-create/diskimage-create.sh          | 29 ++++++++++++++++++-
 .../install.d/01-sahara-version               |  5 ++--
 2 files changed, 31 insertions(+), 3 deletions(-)

diff --git a/diskimage-create/diskimage-create.sh b/diskimage-create/diskimage-create.sh
index 1e35486a..c3a6d468 100755
--- a/diskimage-create/diskimage-create.sh
+++ b/diskimage-create/diskimage-create.sh
@@ -9,6 +9,9 @@ unset DIB_IMAGE_SIZE
 # DEBUG_MODE is set by the -d flag, debug is enabled if the value is "true"
 DEBUG_MODE="false"
 
+# The default tag to use for the dib-utils repo
+DEFAULT_DIB_UTILS_REPO_BRANCH="0.0.9"
+
 # The default tag to use for the DIB repo
 DEFAULT_DIB_REPO_BRANCH="0.1.29"
 
@@ -33,7 +36,7 @@ usage() {
     echo "   '-v' is hadoop version (default: all supported by plugin)"
     echo "   '-r' is MapR Version (default: ${DIB_DEFAULT_MAPR_VERSION})"
     echo "   '-d' enable debug mode, root account will have password 'hadoop'"
-    echo "   '-m' set the diskimage-builder repo to the master branch (default: $DEFAULT_DIB_REPO_BRANCH)"
+    echo "   '-m' set the dib-utils and diskimage-builder repos to their master branches (default: dib-utils=$DEFAULT_DIB_UTILS_REPO_BRANCH, dib=$DEFAULT_DIB_REPO_BRANCH)"
     echo "   '-u' install missing packages necessary for building"
     echo
     echo "You shouldn't specify hadoop version and image type for spark plugin"
@@ -60,6 +63,12 @@ while getopts "p:i:v:dmur:" opt; do
             DEBUG_MODE="true"
         ;;
         m)
+            if [ -n "$DIB_UTILS_REPO_BRANCH" ]; then
+                echo "Error: DIB_UTILS_REPO_BRANCH set and -m requested, please choose one."
+                exit 3
+            else
+                DIB_UTILS_REPO_BRANCH="master"
+            fi
             if [ -n "$DIB_REPO_BRANCH" ]; then
                 echo "Error: DIB_REPO_BRANCH set and -m requested, please choose one."
                 exit 3
@@ -84,6 +93,10 @@ if [ "$1" ]; then
     usage
 fi
 
+if [ -z $DIB_UTILS_REPO_BRANCH ]; then
+    DIB_UTILS_REPO_BRANCH=$DEFAULT_DIB_UTILS_REPO_BRANCH
+fi
+
 if [ -z $DIB_REPO_BRANCH ]; then
     DIB_REPO_BRANCH=$DEFAULT_DIB_REPO_BRANCH
 fi
@@ -221,6 +234,20 @@ pushd $TEMP
 export DIB_IMAGE_CACHE=$TEMP/.cache-image-create
 
 # Working with repositories
+# dib-utils repo
+
+if [ -z $DIB_UTILS_REPO_PATH ]; then
+    git clone https://git.openstack.org/openstack/dib-utils
+    DIB_UTILS_REPO_PATH="$(pwd)/dib-utils"
+    git --git-dir=$DIB_UTILS_REPO_PATH/.git --work-tree=$DIB_UTILS_REPO_PATH checkout $DIB_UTILS_REPO_PATH
+fi
+
+export PATH=$PATH:$DIB_UTILS_REPO_PATH/bin
+
+pushd $DIB_UTILS_REPO_PATH
+export DIB_UTILS_COMMIT_ID=`git rev-parse HEAD`
+popd
+
 # disk-image-builder repo
 
 if [ -z $DIB_REPO_PATH ]; then
diff --git a/elements/sahara-version/install.d/01-sahara-version b/elements/sahara-version/install.d/01-sahara-version
index e40b9697..d82f96c9 100755
--- a/elements/sahara-version/install.d/01-sahara-version
+++ b/elements/sahara-version/install.d/01-sahara-version
@@ -1,9 +1,10 @@
 #!/bin/bash
 
-if [ -z "$SAHARA_ELEMENTS_COMMIT_ID" -o -z "$DIB_COMMIT_ID" ]; then
-    echo "Both SAHARA_ELEMENTS_COMMIT_ID and DIB_COMMIT_ID must be specified, exiting"
+if [ -z "$SAHARA_ELEMENTS_COMMIT_ID" -o -z "$DIB_UTILS_COMMIT_ID" -o -z "$DIB_COMMIT_ID" ]; then
+    echo "SAHARA_ELEMENTS_COMMIT_ID, DIB_UTILS_COMMIT_ID and DIB_COMMIT_ID must be specified, exiting"
     exit 3
 else
     echo -e "Sahara-elements-extra commit id: $SAHARA_ELEMENTS_COMMIT_ID,
+dib-utils commit id: $DIB_UTILS_COMMIT_ID,
 Diskimage-builder commit id: $DIB_COMMIT_ID" > /etc/sahara-extra.version
 fi