#!/usr/bin/env bash
set -e

if ! command -v qemu-nbd &> /dev/null; then
	echo >&2 'error: "qemu-nbd" not found!'
	exit 1
fi

usage() {
	echo "Convert disk image to docker image"
	echo ""
	echo "usage: $0 image-name disk-image-file [ base-image ]"
	echo "   ie: $0 cirros:0.3.3 cirros-0.3.3-x86_64-disk.img"
	echo "       $0 ubuntu:cloud ubuntu-14.04-server-cloudimg-amd64-disk1.img ubuntu:14.04"
}

if [ "$#" -lt 2 ]; then
	usage
	exit 1
fi

CURDIR=$(pwd)

image_name="${1%:*}"
image_tag="${1#*:}"
if [ "$image_tag" == "$1" ]; then
	image_tag="latest"
fi

disk_image_file="$2"
docker_base_image="$3"

block_device=/dev/nbd0

builddir=$(mktemp -d)

cleanup() {
	umount "$builddir/disk_image" || true
	umount "$builddir/workdir" || true
	qemu-nbd -d $block_device &> /dev/null || true
	rm -rf $builddir
}
trap cleanup EXIT

# Mount disk image
modprobe nbd max_part=63
qemu-nbd -rc ${block_device} -P 1 "$disk_image_file"
mkdir "$builddir/disk_image"
mount -o ro ${block_device} "$builddir/disk_image"

mkdir "$builddir/workdir"
mkdir "$builddir/diff"

base_image_mounts=""

# Unpack base image
if [ -n "$docker_base_image" ]; then
	mkdir -p "$builddir/base"
	docker pull "$docker_base_image"
	docker save "$docker_base_image" | tar -xC "$builddir/base"

	image_id=$(docker inspect -f "{{.Id}}" "$docker_base_image")
	while [ -n "$image_id" ]; do
		mkdir -p "$builddir/base/$image_id/layer"
		tar -xf "$builddir/base/$image_id/layer.tar" -C "$builddir/base/$image_id/layer"

		base_image_mounts="${base_image_mounts}:$builddir/base/$image_id/layer=ro+wh"
		image_id=$(docker inspect -f "{{.Parent}}" "$image_id")
	done
fi

# Mount work directory
mount -t aufs -o "br=$builddir/diff=rw${base_image_mounts},dio,xino=/dev/shm/aufs.xino" none "$builddir/workdir"

# Update files
cd $builddir
LC_ALL=C diff -rq disk_image workdir \
	| sed -re "s|Only in workdir(.*?): |DEL \1/|g;s|Only in disk_image(.*?): |ADD \1/|g;s|Files disk_image/(.+) and workdir/(.+) differ|UPDATE /\1|g" \
	| while read action entry; do
		case "$action" in
			ADD | UPDATE)
				cp -a "disk_image$entry" "workdir$entry"
				;;
			DEL)
				rm -rf "workdir$entry"
				;;
			*)
				echo "Error: unknown diff line: $action $entry" >&2
				;;
		esac
	done

# Pack new image
new_image_id="$(for i in $(seq 1 32); do printf "%02x" $(($RANDOM % 256)); done)"
mkdir -p $builddir/result/$new_image_id
cd diff
tar -cf $builddir/result/$new_image_id/layer.tar *
echo "1.0" > $builddir/result/$new_image_id/VERSION
cat > $builddir/result/$new_image_id/json <<- EOS
	{ "docker_version": "1.4.1"
	, "id": "$new_image_id"
	, "created": "$(date -u +%Y-%m-%dT%H:%M:%S.%NZ)"
EOS

if [ -n "$docker_base_image" ]; then
	image_id=$(docker inspect -f "{{.Id}}" "$docker_base_image")
	echo ", \"parent\": \"$image_id\"" >> $builddir/result/$new_image_id/json
fi

echo "}" >> $builddir/result/$new_image_id/json

echo "{\"$image_name\":{\"$image_tag\":\"$new_image_id\"}}" > $builddir/result/repositories

cd $builddir/result

# mkdir -p $CURDIR/$image_name
# cp -r * $CURDIR/$image_name
tar -c * | docker load