#!/bin/bash
set -ex

SOURCE="${BASH_SOURCE[0]}"
while [ -h "$SOURCE" ]; do # resolve $SOURCE until the file is no longer a symlink
  DIR="$( cd -P "$( dirname "$SOURCE" )" >/dev/null 2>&1 && pwd )"
  SOURCE="$(readlink "$SOURCE")"
  [[ $SOURCE != /* ]] && SOURCE="$DIR/$SOURCE" # if $SOURCE was a relative symlink, we need to resolve it relative to the path where the symlink file was located
done
BASEDIR="$( cd -P "$( dirname "$SOURCE" )" >/dev/null 2>&1 && pwd )"

# Whether to build an 'iso' or 'qcow'
build_type="${1:-qcow}"
# The host mount to use to exchange data with this container
host_mount_directory="${2:-$BASEDIR/../config}"
# Docker image to use when launching this container
image="${3:-port/image-builder:latest-ubuntu_focal}"
# proxy to use, if applicable
proxy="$4"
# noproxy to use, if applicable
noproxy="$5"

workdir="$(realpath ${host_mount_directory})"

# Overrides
: ${user_data:=$workdir/iso/user_data}
: ${network_config:=$workdir/iso/network_data.json}
: ${osconfig_params:=$workdir/control-plane/osconfig-control-plane-vars.yaml}
: ${qcow_params:=$workdir/control-plane/qcow-control-plane-vars.yaml}

if [ -n "$proxy" ]; then
  export http_proxy=$proxy
  export https_proxy=$proxy
  export HTTP_PROXY=$proxy
  export HTTPS_PROXY=$proxy
fi

if [ -n "$noproxy" ]; then
  export no_proxy=$noproxy
  export NO_PROXY=$noproxy
fi

# Install pre-requisites
install_pkg(){
  dpkg -l $1 2> /dev/null | grep ^ii > /dev/null || sudo -E apt-get -y install $1
}

if [ ! -f /var/lib/apt/periodic/update-success-stamp ] || \
     sudo find /var/lib/apt/periodic/update-success-stamp -mtime +1 | grep update-success-stamp; then
     sudo -E apt -y update
fi

install_pkg qemu-kvm
install_pkg virtinst
install_pkg libvirt-bin
install_pkg cloud-image-utils
install_pkg ovmf
type docker >& /dev/null || install_pkg docker.io

if [ -d /sys/firmware/efi ]; then
  uefi_mount='--volume /sys/firmware/efi:/sys/firmware/efi:rw'
  uefi_boot_arg='--boot uefi'
fi

if [[ $build_type = iso ]]; then
  : ${img_name:=ephemeral.iso}
  iso_config=/tmp/${img_name}_config
  echo "user_data:
$(cat $user_data | sed 's/^/    /g')
network_config:
$(cat $network_config | sed 's/^/    /g')
outputFileName: $img_name" > ${iso_config}
  sudo -E docker run -i --rm \
   --volume $workdir:/config \
   --env BUILDER_CONFIG=/config/${build_type}.yaml \
   --env IMAGE_TYPE="iso" \
   --env VERSION="v2" \
   --env http_proxy=$proxy \
   --env https_proxy=$proxy \
   --env HTTP_PROXY=$proxy \
   --env HTTPS_PROXY=$proxy \
   --env no_proxy=$noproxy \
   --env NO_PROXY=$noproxy \
   ${image} < ${iso_config}
  disk1="--disk path=${workdir}/${img_name},device=cdrom"
elif [[ $build_type == qcow ]]; then
  : ${img_name:=airship-ubuntu.qcow2}
  sudo -E modprobe nbd
  qcow_config=/tmp/${img_name}_config
  echo "osconfig:
$(cat $osconfig_params | sed 's/^/    /g')
qcow:
$(cat $qcow_params | sed 's/^/    /g')
outputFileName: $img_name" > ${qcow_config}
  echo "Note: This step can be slow if you don't have an SSD."
  sudo -E docker run -i --rm \
   --privileged \
   --volume /dev:/dev:rw  \
   --volume /dev/pts:/dev/pts:rw \
   --volume /proc:/proc:rw \
   --volume /sys:/sys:rw \
   --volume /lib/modules:/lib/modules:rw \
   --volume $workdir:/config \
   ${uefi_mount} \
   --env BUILDER_CONFIG=/config/${build_type}.yaml \
   --env IMAGE_TYPE="qcow" \
   --env VERSION="v2" \
   --env http_proxy=$proxy \
   --env https_proxy=$proxy \
   --env HTTP_PROXY=$proxy \
   --env HTTPS_PROXY=$proxy \
   --env no_proxy=$noproxy \
   --env NO_PROXY=$noproxy \
   ${image} < ${qcow_config}
  cloud_init_config_dir='assets/tests/qcow/cloud-init'
  sudo -E cloud-localds -v --network-config="${cloud_init_config_dir}/network-config" "${workdir}/${img_name}_config.iso" "${cloud_init_config_dir}/user-data" "${cloud_init_config_dir}/meta-data"
  disk1="--disk path=${workdir}/${img_name}"
  disk2="--disk path=${workdir}/${img_name}_config.iso,device=cdrom"
else
  echo Unknown build type: $build_type, exiting.
  exit 1
fi

imagePath=$(echo $disk1 | cut -d'=' -f2 | cut -d',' -f1)
echo Image successfully written to $imagePath

sudo -E virsh destroy ${img_name} 2> /dev/null || true
sudo -E virsh undefine ${img_name} --nvram 2> /dev/null || true

cpu_type=''
kvm-ok >& /dev/null && cpu_type='--cpu host-passthrough' || true

network='--network network=default'
if ! sudo -E virsh net-list | grep default | grep active > /dev/null; then
  network='--network none'
fi

xml=$(mktemp)
sudo -E virt-install --connect qemu:///system \
 --name ${img_name} \
 --memory 1536 \
 ${network} \
 ${cpu_type} \
 --vcpus 4 \
 --import \
 ${disk1} \
 ${disk2} \
 ${virt_type} \
 ${uefi_boot_arg} \
 --noautoconsole \
 --graphics vnc,listen=0.0.0.0 \
 --print-xml > $xml
virsh define $xml

echo Virsh definition accepted
echo Image artifact located at $imagePath