From 65cf60871009b3cead0818724adefa026a264dcf Mon Sep 17 00:00:00 2001
From: Dean Troyer <dtroyer@gmail.com>
Date: Fri, 16 Sep 2011 12:22:21 -0500
Subject: [PATCH] Add image scripts

---
 make_image.sh   | 88 +++++++++++++++++++++++++++++++++++++++++++++++++
 upload_image.sh | 82 +++++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 170 insertions(+)
 create mode 100755 make_image.sh
 create mode 100755 upload_image.sh

diff --git a/make_image.sh b/make_image.sh
new file mode 100755
index 0000000..9b5c10c
--- /dev/null
+++ b/make_image.sh
@@ -0,0 +1,88 @@
+#!/bin/bash
+# make_image.sh - Create Ubuntu images in various formats
+#
+# make_image.sh release format
+#
+# Supported formats: qcow (kvm), vmdk (vmserver), vdi (vbox), vhd (vpc)
+#
+# Requires sudo to root
+
+ROOTSIZE=${ROOTSIZE:-8192}
+SWAPSIZE=${SWAPSIZE:-1024}
+
+usage() {
+    echo "$0 - Create Ubuntu images"
+    echo ""
+    echo "$0 [-r rootsize] [-s swapsize] release format"
+    exit 1
+}
+
+while getopts hm:r:s: c; do
+    case $c in
+        h)	usage
+		;;
+	m)	MIRROR=$OPTARG
+		;;
+	r)	ROOTSIZE=$OPTARG
+		;;
+	s)	SWAPSIZE=$OPTARG
+		;;
+    esac
+done
+shift `expr $OPTIND - 1`
+
+RELEASE=$1
+FORMAT=$2
+
+case $RELEASE in
+    natty)	;;
+    maverick)	;;
+    lucid)	;;
+    karmic)	;;
+    jaunty)	;;
+    *)		echo "Unknown release: $RELEASE"
+		usage
+esac
+
+case $FORMAT in
+    kvm|qcow2)	FORMAT=qcow2
+		TARGET=kvm
+		;;
+    vmserver|vmdk)	FORMAT=vmdk
+		TARGET=vmserver
+		;;
+    vbox|vdi)	FORMAT=qcow2
+		TARGET=kvm
+		FINAL_FORMAT=vdi
+		;;
+    vhd|vpc)	FORMAT=qcow2
+		TARGET=kvm
+		FINAL_FORMAT=vpc
+		;;
+    *)		echo "Unknown format: $FORMAT"
+		usage
+esac
+
+# Install stuff if necessary
+if [ -z `which vmbuilder` ]; then
+	sudo apt-get install ubuntu-vm-builder
+fi
+
+# Build the image
+sudo vmbuilder $TARGET ubuntu --suite $RELEASE \
+  -o \
+  --rootsize=$ROOTSIZE \
+  --swapsize=$SWAPSIZE \
+  --tmpfs - \
+  --addpkg=openssh-server \
+
+#  --mirror=$MIRROR \
+
+if [ -z "$FINAL_FORMAT" ]; then
+    # Get image
+    mv ubuntu-$TARGET/tmp*.$FORMAT $RELEASE.$FORMAT
+else
+    # Convert image
+    qemu-img convert -O $FINAL_FORMAT ubuntu-$TARGET/tmp*.$FORMAT $RELEASE.$FINAL_FORMAT
+fi
+rm -rf ubuntu-$TARGET
diff --git a/upload_image.sh b/upload_image.sh
new file mode 100755
index 0000000..8436b76
--- /dev/null
+++ b/upload_image.sh
@@ -0,0 +1,82 @@
+#!/bin/bash
+# upload_image.sh - Upload Ubuntu images (create if necessary) in various formats
+#
+# upload_image.sh release format
+#
+# format   target
+# qcow2    kvm,qemu
+# vmdk     vmw6
+# vbox     vdi
+# vhd      vpc
+
+HOST=${HOST:-demo.rcb.me}
+PORT=${PORT:-9292}
+
+usage() {
+    echo "$0 - Upload Ubuntu images"
+    echo ""
+    echo "$0 [-h host] [-p port] release format"
+    exit 1
+}
+
+while getopts h:p: c; do
+    case $c in
+        h)      HOST=$OPTARG
+                ;;
+	p)	PORT=$OPTARG
+		;;
+    esac
+done
+shift `expr $OPTIND - 1`
+
+RELEASE=$1
+FORMAT=$2
+
+case $RELEASE in
+    natty)	;;
+    maverick)	;;
+    lucid)	;;
+    karmic)	;;
+    jaunty)	;;
+    *)		echo "Unknown release: $RELEASE"
+		usage
+esac
+
+case $FORMAT in
+    kvm|qcow2)	FORMAT=qcow2
+		TARGET=kvm
+		;;
+    vmserver|vmdk)	FORMAT=vmdk
+		TARGET=vmserver
+		;;
+    vbox|vdi)	TARGET=kvm
+		FORMAT=vdi
+		;;
+    vhd|vpc)	TARGET=kvm
+		FORMAT=vpc
+		;;
+    *)		echo "Unknown format: $FORMAT"
+		usage
+esac
+
+GLANCE=`which glance`
+if [ -z "$GLANCE" ]; then
+	echo "Glance not found, must install client"
+	sudo apt-get install python-pip python-eventlet python-routes python-greenlet python-argparse python-sqlalchemy python-wsgiref python-pastedeploy python-xattr
+	sudo pip install kombu
+	git clone https://github.com/cloudbuilders/glance.git
+	cd glance
+	sudo python setup.py develop
+	cd ..
+	GLANCE=`which glance`
+fi
+
+# Create image if it doesn't exist
+if [ ! -r $RELEASE.$FORMAT ]; then
+	DIR=`dirname $0`
+	echo "$RELEASE.$FORMAT not found, creating...must be root to do this:"
+	$DIR/make_image.sh $RELEASE $FORMAT
+fi
+
+# Upload the image
+$GLANCE add name=$RELEASE.$FORMAT is_public=true disk_format=$FORMAT --host $HOST --port $PORT <$RELEASE.$FORMAT