/* -------------------------------------------

	Copyright (C) 2024-2025, Amlal EL Mahrouss, all rights reserved.

------------------------------------------- */

#ifndef __EFI_API__
#define __EFI_API__

#include <FirmwareKit/EFI/EFI.h>
#include <FirmwareKit/Handover.h>
#include <KernelKit/MSDOS.h>
#include <KernelKit/PE.h>
#include <BootKit/Qr.h>

#define kNeKernelSubsystem (17)
#define kNeWebsiteMacro	   "https://vswirl.com/help"

#ifdef __BOOTZ__
// forward decl.
class BTextWriter;

#define __BOOTKIT_NO_INCLUDE__ 1

#include <BootKit/BootKit.h>
#include <Mod/CoreGfx/FBMgr.h>
#endif // ifdef __BOOTZ__

inline EfiSystemTable*	ST = nullptr;
inline EfiBootServices* BS = nullptr;

EXTERN_C void rt_cli();
EXTERN_C void rt_hlt();

namespace EFI
{
	/// @brief Halt and clear interrupts.
	/// @return
	inline Void Stop() noexcept
	{
		while (YES)
		{
			rt_cli();
			rt_hlt();
		}
	}

	/**
@brief Exit EFI API to let the OS load correctly.
Bascially frees everything we have in the EFI side.
*/
	inline void ExitBootServices(UInt64 MapKey, EfiHandlePtr ImageHandle) noexcept
	{
		if (!ST)
			return;

		ST->BootServices->ExitBootServices(ImageHandle, MapKey);
	}

	inline UInt32 Platform() noexcept
	{
		return kPeMachineAMD64;
	}

	/***
	 * @brief Throw an error, stop execution as well.
	 * @param ErrorCode error code to be print.
	 * @param Reason reason to be print.
	 */
	inline void ThrowError(const EfiCharType* ErrorCode,
						   const EfiCharType* Reason) noexcept
	{
		ST->ConOut->OutputString(ST->ConOut, L"\r*** STOP ***\r");

		ST->ConOut->OutputString(ST->ConOut, L"*** ERROR: ");
		ST->ConOut->OutputString(ST->ConOut, ErrorCode);

		ST->ConOut->OutputString(ST->ConOut, L" ***\r *** REASON: ");
		ST->ConOut->OutputString(ST->ConOut, Reason);

		ST->ConOut->OutputString(ST->ConOut, L" ***\r");

		constexpr auto ver	 = 4;
		auto		   ecc	 = qr::Ecc::H;
		auto		   input = kNeWebsiteMacro;
		auto		   len	 = StrLen(kNeWebsiteMacro);

		qr::Qr<ver>	   encoder;
		qr::QrDelegate del;

		encoder.encode(input, len, ecc, 0); // Manual mask 0

		constexpr auto whereX = 10;
		constexpr auto whereY = 10;

		/// tell delegate to draw encoded QR.
		del.draw<ver>(encoder, whereX,
					  whereY);

		EFI::Stop();
	}
} // namespace EFI

inline void InitEFI(EfiSystemTable* SystemTable) noexcept
{
	if (!SystemTable)
		return;

	ST = SystemTable;
	BS = ST->BootServices;

	ST->ConOut->ClearScreen(SystemTable->ConOut);
	ST->ConOut->SetAttribute(SystemTable->ConOut, kEFIYellow);

	ST->BootServices->SetWatchdogTimer(0, 0, 0, nullptr);
	ST->ConOut->EnableCursor(ST->ConOut, false);
}

#ifdef __BOOTZ__

#include <BootKit/Platform.h>

#endif // ifdef __BOOTZ__

#endif /* ifndef __EFI_API__ */