#!/bin/bash

set -e

if [ x`whoami` != x"root" ]; then
    echo "ERROR: Need to run $0 as root"
    exit 1
fi

board="tk1"
disk=/dev/sda
fs_type=ext4
mnt=/mnt
rootfs_tar="default"

OPTS="`getopt -o v -l board: -- "$@"`"
while test $# -gt 0; do
    case $1 in
	--board) board="$2"; shift ;;
	--rootfs) rootfs_tar="$2"; shift ;;
	-v) set -x ;;
    esac
    shift
done

if [ x"$rootfs_tar" = x"default" ]; then
    rootfs_tar="http://images.validation.linaro.org/nvidia/$board/trusty.tar.gz"
fi

echo "ABOUT TO PARTITION AND FORMAT $disk"

parted -s $disk mklabel gpt
parted -s $disk mkpart primary $fs_type 1MB 116GB
parted -s $disk mkpart primary linux-swap 116GB 120GB
mkfs.$fs_type ${disk}1
mkswap ${disk}2

echo "DONE PARTITIONING $disk:"
parted -s $disk print

mount ${disk}1 $mnt

case "$rootfs_tar" in
    "http"*) cat_cmd="wget -O -" ;;
    *) cat_cmd="cat" ;;
esac

case "$rootfs_tar" in
    *".tar.gz") tar_opt="z" ;;
    *".tar.bz2") tar_opt="j" ;;
    *".tar.xz") tar_opt="J" ;;
esac

echo "WRITING ROOTFS TO ${disk}2"

$cat_cmd $rootfs_tar | (cd $mnt; tar x$tar_opt)

rm -f $mnt/README
echo "${disk}2 none swap sw 0 0" >> $mnt/etc/fstab
sed -i -e "s/nameserver 127.0.1.1/nameserver 8.8.8.8/" $mnt/etc/resolv.conf
rsync -a "$(dirname $0)/" $mnt/root/lab-ansible/

umount $mnt

echo "ALL DONE"