configure do
  check_program "mformat"
  check_program "x86_64-w64-mingw32-gcc"
  check_cfg package: "gnu-efi", use: false
end

class Image < Builder
  def run(options)
    unless @cache.up_to_date?(@target, nil, @sources, @env)
      print_run_message("Image <target>#{@target}<reset>", nil)
      File.binwrite(@target, "\0" * (1440 * 1024))
      system(*%W[mformat -i #{@target} -f 1440 ::])
      system(*%W[mmd -i #{@target} ::/EFI])
      system(*%W[mmd -i #{@target} ::/EFI/BOOT])
      system(*%W[mcopy -i #{@target} #{@sources.first} ::/EFI/BOOT])
      @cache.register_build(@target, nil, @sources, @env)
    end
    true
  end
end

uefi_env = env "uefi" do |env|
  env.add_builder(Image)
  env["CC"] = "x86_64-w64-mingw32-gcc"
  env["CPPPATH"] += %w[/usr/include/efi /usr/include/efi/x86_64]
  env["CFLAGS"] += %w[-ffreestanding]
  env["LDFLAGS"] += %w[-nostdlib -Wl,-dll -shared -Wl,--subsystem,10 -e efi_main]
  env["LIBS"] += %w[gcc]
  env["sources"] = glob("src/**/*.c")
  env.Program("^/BOOTX64.EFI", "${sources}")
  env.Image("^/efi-loader.img", "^/BOOTX64.EFI")
end

task "run" do
	sh %W[qemu-system-x86_64 -bios OVMF.fd -drive file=#{uefi_env.expand("^/efi-loader.img")},format=raw]
end