#!/bin/bash
set -e
set -x

cd $(dirname $0)/..

source scripts/build-common

BASE=$(pwd)

KERNEL=${BASE}/dist/artifacts/vmlinuz
INITRD=${BASE}/dist/artifacts/initrd
NO_COMPRESS_INITRD=${INITRD}.none
HD=${BASE}/state/empty-hd.img
HD_GZ=${ARTIFACTS}/empty-hd.img.gz
INITRD_TMP=${BUILD}/$(sha1sum ${INITRD} | awk '{print $1}')
INITRD_CURRENT=${BUILD}/initrd-current
INITRD_TEST=${BUILD}/initrd.test
USER_DATA=cloud-init/openstack/latest/user_data
CLOUD_CONFIG=$(readlink -f "$1") || :

if [[ ! -e ${KERNEL} || ! -e ${INITRD} ]]; then
    echo "Failed to find ${KERNEL} or ${INITRD}" 1>&2
    exit 1
fi

if [ ! -d ${INITRD_TMP} ]; then
    mkdir -p ${INITRD_TMP}
    pushd ${INITRD_TMP}
    if [ -e ${NO_COMPRESS_INITRD} ]; then
        cat ${NO_COMPRESS_INITRD} | sudo cpio -idmv
    else
        xz -dc ${INITRD} | sudo cpio -idmv
    fi
    rm -f init
    popd
fi

if [ -e $INITRD_CURRENT ]; then
    rm -f ${INITRD_CURRENT}
fi
ln -s ${INITRD_TMP} ${INITRD_CURRENT}

if [ ! -e bin/rancheros ]; then
    ./scripts/build
fi

cp bin/rancheros ${INITRD_TMP}/init
cd ${INITRD_TMP}

find | cpio -H newc -o > ${INITRD_TEST}

if [ ! -e ${HD} ]; then
    mkdir -p $(dirname ${HD})
    zcat ${HD_GZ} > ${HD}
fi

mkdir -p $(dirname $USER_DATA)

if [ -n "$CLOUD_CONFIG" ]; then
    cat ${CLOUD_CONFIG} > ${USER_DATA}
else
    echo "#cloud-config" > ${USER_DATA}
    echo "ssh_authorized_keys:" >> ${USER_DATA}
    for i in ${HOME}/.ssh/*.pub; do
        if [ -e $i ]; then
            echo "  - $(<$i)" >> ${USER_DATA}
        fi
    done
fi

while [ "$#" -gt 0 ]; do
    case $1 in
        --append)
            shift 1
            QEMU_APPEND="${QEMU_APPEND} $1"
            ;;
        *)
            break
            ;;
    esac
    shift 1
done

qemu-system-x86_64 -serial stdio \
    -kernel ${KERNEL} \
    -initrd ${INITRD_TEST} \
    -m 1024 \
    -net nic,vlan=0,model=virtio \
    -net user,vlan=0,hostfwd=tcp::2222-:22,hostname=rancher-dev \
    -drive if=virtio,file=${HD} \
    -machine accel=kvm \
    -cpu host \
    -smp 4 \
    -fsdev local,id=conf,security_model=none,readonly,path=$(pwd)/cloud-init \
    -device virtio-9p-pci,fsdev=conf,mount_tag=config-2 \
    -append "rancher.password=rancher console=ttyS0 ${QEMU_APPEND}" \
    -serial mon:telnet:localhost:4444,server,nowait \
    ${QEMU_ARGS} "${@}"