# shellcheck disable=SC2154
declare version="${args[--version]}"
# shellcheck disable=SC2154
declare disk_size="${args[--disk-size]}"
# shellcheck disable=SC2154
declare ram_size="${args[--ram-size]}"
# shellcheck disable=SC2154
declare cpu_cores="${args[--cpu-cores]}"
# shellcheck disable=SC2154
declare share_directory="${args[--share-directory]}"
# shellcheck disable=SC2154
declare persistent_dir_prefix="${args[--persistent-dir-prefix]:-$version}"
# shellcheck disable=SC2154
declare usb="${args[--usb]}"
declare -a flags=()

if [[ -n "$usb" ]]; then
  vendor_id="$(udevadm info --query=all --name="$usb" | grep ID_VENDOR_ID | awk -F= '{print $2}')"
  product_id="$(udevadm info --query=all --name="$usb" | grep ID_MODEL_ID | awk -F= '{print $2}')"
  flags+=("--device=/dev/bus/usb")
  flags+=("-e")
  flags+=(ARGUMENTS="-device usb-host,vendorid=0x${vendor_id},productid=0x${product_id}")
fi

if [[ "${args[--wipe-persistent-data]}" == 1 ]]; then
  declare persistent_data_dir="$HOME/.vm/mac/$persistent_dir_prefix"
  if [[ -d "$persistent_data_dir" ]]; then
    yellow "Removing persisted session data for Mac OS ${version}..."
    rm -rf "$persistent_data_dir"
  else
    red "There's no persisted MacOS ${version} session in the specified persistent directory: $persistent_data_dir"
    exit 1
  fi
fi

if [[ "${args[--persistent]}" == 1 ]]; then
  [[ -d "$HOME/.vm/mac/$persistent_dir_prefix" ]] || mkdir -p "$HOME/.vm/mac/$persistent_dir_prefix"
fi

if [[ "${args[--persistent]}" == 1 ]]; then
  container_id=$(docker run -it --rm \
    -v "$HOME/.vm/mac/$persistent_dir_prefix:/storage:rw" \
    -p 8006:8006 \
    -p 5900:5900 \
    -p 2222:22 \
    --device=/dev/kvm \
    -e "VERSION=$version" \
    -e "DISK_SIZE=${disk_size}G" \
    -e "RAM_SIZE=${ram_size}G" \
    -e "CPU_CORES=$cpu_cores" \
    -v "$share_directory:/shared" \
    --cap-add NET_ADMIN \
    --stop-timeout 120 \
    "${flags[@]}" \
    -d \
    dockurr/macos)
else
  container_id=$(docker run -it --rm \
    -p 8006:8006 \
    -p 5900:5900 \
    -p 2222:22 \
    --device=/dev/kvm \
    -e "VERSION=$version" \
    -e "DISK_SIZE=${disk_size}G" \
    -e "RAM_SIZE=${ram_size}G" \
    -e "CPU_CORES=$cpu_cores" \
    -v "$share_directory:/shared" \
    --cap-add NET_ADMIN \
    --stop-timeout 120 \
    "${flags[@]}" \
    -d \
    dockurr/macos)
fi

cleanup() {
  cyan "Stopping Mac container $container_id..."
  docker stop "$container_id" > /dev/null
}

trap cleanup EXIT

until (docker logs "$container_id" 2>&1 | grep -qi "starting Boot.*from PciRoot"); do
  blue "Waiting for Mac to boot..."
  sleep 3
done

remmina -c vnc://localhost:5900 --set-option quality=2 --set-option scale=2 --set-option keyboard_grab=1