# -*- mode: ruby -*-
# vim: set ft=ruby :

module VagrantConfig
require 'fileutils'
require 'rexml/document'
require 'etc'

cur_dir   = __dir__
vbox_name = 'fedora_40'
ssh_port  = 3333
cpus      = 4
memory    = 2048

linux_partitions = ''
linux = `uname -a | grep -i linux`

if not linux.empty?
  system = 'linux'
else
  system = 'macos'
end

if system == 'linux'
    sudo = false
    accelerator = 'kvm'
    qemu_dir = "#{cur_dir}/support_scripts"
    partition_msg = 'Running on Linux'
    ####################### ensure bios files are installed #######################
    edk2_aarch64_code = '/usr/share/edk2/aarch64/QEMU_EFI-silent-pflash.raw'
    edk2_arm_vars = '/usr/share/edk2/aarch64/vars-template-pflash.raw'
    support_dir = "#{cur_dir}/support_scripts"
    Dir.mkdir(support_dir) unless File.exist?(support_dir)

    if not File.exist?("#{support_dir}/edk2-aarch64-code.fd")
        FileUtils.cp(edk2_aarch64_code, "#{support_dir}/edk2-aarch64-code.fd")
    end

    if not File.exist?('#{support_dir}/edk2-arm-vars.fd')
        FileUtils.cp(edk2_arm_vars, "#{support_dir}/edk2-arm-vars.fd")
    end
    ###############################################################################
elsif system == 'macos'
    sudo = true
    accelerator='hvf'
    qemu_dir='/opt/homebrew/share/qemu'
    linux_disk_count = `ls /dev/disk0s* | grep -E '/dev/disk0s4|/dev/disk0s5|/dev/disk0s6' | wc -l | tr -d ' '`
    linux_disk_count = linux_disk_count.to_i
    if linux_disk_count == 3
        partition_msg = 'Linux partitions found'
        linux_partitions = '-drive if=virtio,format=raw,file=/dev/disk0s4 -drive if=virtio,format=raw,file=/dev/disk0s5 -drive if=virtio,format=raw,file=/dev/disk0s6'
    else
        partition_msg = 'Linux partitions not found'
    end
else
    puts 'system not supporrted'
    exit 1
end


qemu_plugin_installed = `vagrant plugin list | grep vagrant-qemu`
if qemu_plugin_installed.empty?
    puts 'the vagrant-qemu plugin is not installed'
    puts 'install it with:'
    puts 'vagrant plugin install vagrant-qemu'
    exit 1
end


if sudo == true && Etc.getpwuid.uid != 0
  puts "You need to run this script as root or sudo when running from macos"
  exit 1
end


Vagrant.configure("2") do |config|

  config.vm.boot_timeout   = 900
  config.ssh.username      = 'root'
  config.ssh.password      = 'fedora'
  config.ssh.keep_alive    = true
  config.vm.box = vbox_name
  config.vm.synced_folder cur_dir, '/vagrant', disabled: true
  config.trigger.after :up do |trigger|
      trigger.run_remote = {inline: "echo #{partition_msg}"}
  end

  config.vm.define 'fedora' do |fedora|
	  config.vm.provider "qemu" do |qe|
		  qe.name = 'fedora'
		  qe.qemu_dir = "#{qemu_dir}"
		  qe.arch = 'aarch64'
		  qe.machine = "virt,accel=#{accelerator},highmem=on"
		  qe.smp = cpus
		  qe.memory = memory
		  qe.cpu = "host"
		  qe.net_device = "virtio-net-pci"
		  qe.ssh_port = ssh_port
		  # uncomment if you need vnc
		  #qe.extra_netdev_args = "hostfwd=tcp::5900-:5900"

          qe.extra_qemu_args = "-fsdev local,id=cur_dir_dev,path=#{cur_dir},security_model=mapped-xattr
                                -device virtio-9p-pci,fsdev=cur_dir_dev,mount_tag=local_mnt #{linux_partitions}".split
  end
  end
  end
end