#!/usr/bin/env sh
#
# Author: Adrian Erik Hoemann
# Date: 9.18.2024

# Debug mode
# Default: true
DEBUG=true

# Debug mode check 
if [ $DEBUG ]; then
  set +x
fi

# Arguments
OPTION=$1
OS_NAME=$2

# Defining paths
ROOT_PATH="$HOME/Machines"
ISO_PATH="$ROOT_PATH/$OS_NAME/$OS_NAME.iso"
IMG_PATH="$ROOT_PATH/$OS_NAME/$OS_NAME.img"

# Sources default values
#. ".env"

# Sources virtual machine specific values
#. "$ROOT_PATH/$OS_NAME/.env"

# Functions

# Defines install image source
source() {
  while read LINE; do
    if [ $LINE == $OS_NAME* ]; then
      $OS_NAME=$LINE
    fi
  done < .sources
}

check() {
  if [ -z $OS_NAME ]; then
    error "Operating system is not defined"
  fi
  if [ -z $INSTALL_IMG ]; then
    info "$OS_NAME source not set, please update your .sources file"
  fi

}

error() {
  printf "$@" >&2
  exit 1
}

warn() {
  printf "$@" >&2
}

info() {
  printf "$@" >&1
}

# Virtual machine removal
remove() {
  rm -r $ROOT_PATH/$OS_NAME 
}

# Virtual machine start without a console
start() {
  qemu-system-x86_64 -boot c                            \
                       -display none                    \
                       -drive file=$IMG_PATH,format=raw \
                       -m $RAM -smp $CPU
}

# Virtual machine start with a console
console() {
  qemu-system-x86_64 -boot c                    \
                       -display curses            \
                       -drive file=$IMG_PATH,format=raw \
                       -m $RAM -smp $CPU
}

# Virtual machine initialization with a console
init() {
  def_os_source
  mkdir -p "$ROOT_PATH/$OS_NAME"
  qemu-img create $IMG_PATH $DISK_SIZE
  if [ ! -f $ISO_PATH ] && [ ! -z $INSTALL_IMG ]; then
    info "Fetching $OSNAME installer at: $INSTALL_IMG"
    curl -o $ISO_PATH $INSTALL_IMG
  elif [ ! -z $INSTALL_IMG ]; then
    info "Current INSTALL_IMG for $OS_NAME: $ISO_PATH"
  else
    error "Install image path is missing"
  fi
  qemu-system-x86_64 -boot d                          \
                    -display curses                   \
                    -drive file=$ISO_PATH,media=cdrom \
                    -drive file=$IMG_PATH,format=raw  \
                    -m $RAM -smp $CPU
}

# Display usage
usage() {
  printf "Usage: shemum [init] <virtual machine name> <disk size>"
  printf " init        initialize a virtual machine"
  printf " start       start a QEMU virtual machine without console output"
  printf " console     start a QEMU virtual machine with console output"
  printf " remove      remove virtual machine directory and all it's contents"
}

# Options
case $OPTION in
  "init")
    init
  ;;
  "console")
    console
  ;;
  "start")
    start
  ;;
  "remove")
    remove
  ;;
  *)
    usage
esac