#!/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}/build/empty-hd.img
HD_GZ=${ARTIFACTS}/empty-hd.img.gz
INITRD_TMP=${BUILD}/$(sha1sum ${INITRD} | awk '{print $1}')
INITRD_TEST=${BUILD}/initrd.test

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

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

if [ "$1" == "--docker" ]; then
    docker build -t rancheros-run .
    docker run --rm --privileged -it rancheros-run
else
    find | cpio -H newc -o > ${INITRD_TEST}

    if [ ! -e ${HD} ]; then
        zcat ${HD_GZ} > ${HD}
    fi

    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 -drive if=virtio,file=${HD} -machine accel=kvm -cpu host -smp 4 -append "x $@" ${QEMU_ARGS}

fi