#include <pybind11/pybind11.h>
#include <pybind11/stl.h>
#include <pybind11/functional.h>
#include <pybind11/native_enum.h>

#include "icsneo/device/extensions/bootloader/api.h"
#include "icsneo/device/extensions/bootloader/parser/fwloader.h"

namespace icsneo {

struct BootloaderNamespace {};

void init_bootloader(pybind11::module_& m) {
	// Dummy class to act as Bootloader namespace
	pybind11::classh<BootloaderNamespace> bootloader(m, "Bootloader");

	// Opaque handle returned by FWLoader.load and consumed by update
	pybind11::classh<Bootloader::FWRepo>(bootloader, "FWRepo");

	pybind11::classh<Bootloader::FWLoader>(bootloader, "FWLoader")
		.def_static("load", &Bootloader::FWLoader::Load, pybind11::arg("path"), pybind11::call_guard<pybind11::gil_scoped_release>());

	pybind11::native_enum<Bootloader::StatusType>(bootloader, "StatusType", "enum.IntEnum")
		.value("Progress", Bootloader::StatusType::Progress)
		.value("Failed", Bootloader::StatusType::Failed)
		.finalize();

	bootloader.def_static("update",
		[](Device& device, const Bootloader::FWRepo& fw, Bootloader::StatusHandler statusHandler, bool alreadyInBootloader, bool forceFlash) {
			if(!statusHandler) {
				statusHandler = [](Bootloader::StatusType, const std::string&, std::optional<double>) {};
			}
			return Bootloader::update(device, fw, statusHandler, alreadyInBootloader, forceFlash);
		},
		pybind11::arg("device"),
		pybind11::arg("fw"),
		pybind11::arg("statusHandler") = pybind11::none(),
		pybind11::arg("already_in_bootloader") = false,
		pybind11::arg("force_flash") = false,
		pybind11::call_guard<pybind11::gil_scoped_release>());
}

} // namespace icsneo