#!/bin/bash
SCHROOT_CHROOT_PATH="/var/lib/schroot/chroots"
USE_READY_IMAGE=false
CREATE_STATIC_IMAGE=false
TARGET_ARCH="armhf"
MACHINE_TYPE=`uname -m`
if [ ${MACHINE_TYPE} == 'x86_64' ]; then
  ARCH_TYPE="amd64"
else
  ARCH_TYPE="i386"
fi

STATIC_CHROOT_NAME="click-ubuntu-sdk-15.04-${TARGET_ARCH}"
PWD=`pwd`

# Set up this use for click schroots
CLICK_USER=`who am i|awk '{print $1}'`

couple_chroot() {
    echo "Set up the mounts"
    mount -o bind /dev ${PWD}/${STATIC_CHROOT_NAME}/dev
    mount -o bind /dev/pts ${PWD}/${STATIC_CHROOT_NAME}/dev/pts
    mount -o bind /sys ${PWD}/${STATIC_CHROOT_NAME}/sys
    mount -o bind /proc ${PWD}/${STATIC_CHROOT_NAME}/proc
    mount -o bind /tmp ${PWD}/${STATIC_CHROOT_NAME}/tmp
    mount -o bind /home ${PWD}/${STATIC_CHROOT_NAME}/home
    mkdir -p ${PWD}/${STATIC_CHROOT_NAME}/run/shm
    mount -o bind /run/shm ${PWD}/${STATIC_CHROOT_NAME}/run/shm
    echo "Fixing the nssdatabases"
    pushd "${PWD}/${STATIC_CHROOT_NAME}/etc"
        mv passwd passwd-
        cp /etc/passwd ./
        mv shadow shadow-
        cp /etc/shadow ./
        mv group group-
        cp /etc/group ./
        mv gshadow gshadow-
        cp /etc/gshadow ./
        mv services  services-
        cp /etc/services ./
        mv protocols protocols-
        cp /etc/protocols ./
        mv networks networks-
        cp /etc/networks ./
        mv hosts hosts-
        cp /etc/hosts ./
        mv resolv.conf resolv.conf-
        cp /etc/resolv.conf ./
    popd
}

decouple_chroot () {
    mount|grep ${STATIC_CHROOT_NAME}|awk '{print $3}'|xargs sudo umount -lf
    echo "Reverting the nssdatabases"
    pushd "${PWD}/${STATIC_CHROOT_NAME}/etc"
        mv passwd- passwd
        mv shadow- shadow
        mv group- group
        mv gshadow- gshadow
        mv services-  services
        mv protocols- protocols
        mv networks- networks
        mv hosts- hosts
        mv resolv.conf- resolv.conf
    popd
}

fix_core_chroot() {
    echo "Install dselect"
    chroot ${PWD}/${STATIC_CHROOT_NAME} apt-get -y install dselect 2>&1 > /dev/null
    echo "Fix the sources.list"
    cat >  ${PWD}/${STATIC_CHROOT_NAME}/etc/apt/sources.list << EOF
deb [arch=$TARGET_ARCH] http://ports.ubuntu.com/ubuntu-ports vivid main restricted universe multiverse
deb [arch=$TARGET_ARCH] http://ports.ubuntu.com/ubuntu-ports vivid-updates main restricted universe multiverse
deb [arch=$TARGET_ARCH] http://ports.ubuntu.com/ubuntu-ports vivid-security main restricted universe multiverse
deb [arch=$ARCH_TYPE] http://archive.ubuntu.com/ubuntu vivid main restricted universe multiverse
deb [arch=$ARCH_TYPE] http://archive.ubuntu.com/ubuntu vivid-updates main restricted universe multiverse
deb [arch=$ARCH_TYPE] http://archive.ubuntu.com/ubuntu vivid-security main restricted universe multiverse
deb-src http://archive.ubuntu.com/ubuntu vivid main restricted universe multiverse
deb-src http://archive.ubuntu.com/ubuntu vivid-updates main restricted universe multiverse
deb-src http://archive.ubuntu.com/ubuntu vivid-security main restricted universe multiverse
EOF
    chroot ${PWD}/${STATIC_CHROOT_NAME} apt-get -y  update 
    chroot ${PWD}/${STATIC_CHROOT_NAME} dselect update 
    chroot ${PWD}/${STATIC_CHROOT_NAME} dpkg --add-architecture $TARGET_ARCH
    chroot ${PWD}/${STATIC_CHROOT_NAME} apt-get -y update
    chroot ${PWD}/${STATIC_CHROOT_NAME} apt-get -y install ubuntu-sdk-libs:$TARGET_ARCH ubuntu-sdk-libs-dev:$TARGET_ARCH ubuntu-sdk-libs-tools 
    if [[ $TARGET_ARCH == armhf ]]; then
       chroot ${PWD}/${STATIC_CHROOT_NAME} apt-get -y install g++-arm-linux-gnueabihf pkg-config-arm-linux-gnueabihf
    else
       chroot ${PWD}/${STATIC_CHROOT_NAME} apt-get -y install g++ pkg-config
    fi
    chroot ${PWD}/${STATIC_CHROOT_NAME} apt-get -y clean
}

fix_new_schroot() {
    echo "Registering the new chroot to schroot"
    cat > /etc/schroot/chroot.d/${STATIC_CHROOT_NAME} << EOF
[${STATIC_CHROOT_NAME}]
description=Build chroot for click packages on ${TARGET_ARCH}
users=root,${CLICK_USER}
root-users=root,${CLICK_USER}
source-root-users=root,${CLICK_USER}
type=directory
profile=default
setup.fstab=click/fstab
# Not protocols or services see
# debian bug 557730
setup.nssdatabases=sbuild/nssdatabases
union-type=overlayfs
directory=${PWD}/${STATIC_CHROOT_NAME}
EOF
}

while getopts ":hc:a:f:" opt; do
    case $opt in
	a)
	    TARGET_ARCH=$OPTARG
            if [[ ${TARGET_ARCH} != i386 && ${TARGET_ARCH} != armhf ]]; then
                echo "${TARGET_ARCH} not a valid arch type. Use armhf or i386."
		exit
            fi
	    ;;
        f)
	    if [ ! -f ${IMAGE_FILE} ]; then
		echo "Error: " ${IMAGE_FILE} "does not exist."
                exit
            fi
	    USE_READY_IMAGE=true
            IMAGE_FILE=$OPTARG
	    TARGET_ARCH=${IMAGE_FILE/click-ubuntu-sdk-15\.04-/}
            TARGET_ARCH=${TARGET_ARCH/\.tar\.gz/}
            ;;
        c)
            CREATE_STATIC_IMAGE=true
            ;;
        h)
            echo "Usage: sudo create_static_schroot.sh -f [image file] -h"
            echo -e "\t-c : Create static click chroot image file for ${TARGET_ARCH}"
            echo -e "\t-f : Image file to set up as click chroot"
            echo -e "\t-h : Prints this help."
            echo ""
            echo "By default the create_static_schroot.sh downloads the ${ARCH_TYPE} image file"
            echo "and extend it with the Ubuntu 15.04 ${TARGET_ARCH} framework and SDK dev tools"
	    exit
            ;;
        :)
            echo "Option -$OPTARG requires an argument." >&2
            exit
    esac
done

if [ ${CREATE_STATIC_IMAGE} == true ]; then
    if [ ! -f vivid-core-${ARCH_TYPE}.tar.gz ]; then
        echo "Fetch the core"
        wget http://cdimage.ubuntu.com/ubuntu-core/daily/current/vivid-core-${ARCH_TYPE}.tar.gz
    fi
    if [ ! -f ${PWD}/${STATIC_CHROOT_NAME} ]; then
        mkdir ${PWD}/${STATIC_CHROOT_NAME}
        tar zxf vivid-core-${ARCH_TYPE}.tar.gz -C ${PWD}/${STATIC_CHROOT_NAME}
    fi
    couple_chroot
    fix_core_chroot
    decouple_chroot
    tar czf ${PWD}/${STATIC_CHROOT_NAME}.tar.gz ${PWD}/${STATIC_CHROOT_NAME}
    rm ${PWD}/${STATIC_CHROOT_NAME} -rf
    IMAGE_FILE=${PWD}/${STATIC_CHROOT_NAME}.tar.gz
fi 

if [ ${USE_READY_IMAGE} == true ]; then
    echo "Extracting ${IMAGE_FILE}"
    tar zxf ${IMAGE_FILE}
    if [ -f ${SCHROOT_CHROOT_PATH}/click-ubuntu-sdk-15.04-${TARGET_ARCH} ]; then
        mount|egrep "click-ubuntu-sdk-15\.04-${TARGET_ARCH}" |awk '{print $3}'|xargs sudo umount -lf
        mv ${SCHROOT_CHROOT_PATH}/click-ubuntu-sdk-15.04-${TARGET_ARCH} ${SCHROOT_CHROOT_PATH}/click-ubuntu-sdk-15.04-${TARGET_ARCH}-backup
    fi
    echo "Copy over the chroot"
    cp ${PWD}/${STATIC_CHROOT_NAME} ${SCHROOT_CHROOT_PATH}/ -r
    PWD="${SCHROOT_CHROOT_PATH}"
    STATIC_CHROOT_NAME="click-ubuntu-sdk-15.04-${TARGET_ARCH}"
    couple_chroot
    fix_new_schroot
fi