configure do
  check_program "mformat"
  check_program "x86_64-w64-mingw32-gcc"
end

class Image < Builder
  def run(options)
    unless @cache.up_to_date?(@target, nil, @sources, @env)
      print_run_message("Image <target>#{@target}<reset>", nil)
      system(*%W[dd if=/dev/zero of=#{@target} bs=1k count=1440])
      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

env 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("*.c")
  env.Program("^/BOOTX64.EFI", "${sources}")
  env.Image("^/efi-loader.img", "^/BOOTX64.EFI")
end