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

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

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

#include <KernelKit/DebugOutput.h>
#include <NewKit/Pmm.h>

#if defined(__NE_ARM64__)
#include <HALKit/ARM64/Processor.h>
#endif  // defined(__NE_ARM64__)

#if defined(__NE_AMD64__)
#include <HALKit/AMD64/Processor.h>
#endif  // defined(__NE_AMD64__)

namespace Kernel {
/***********************************************************************************/
/// @brief Pmm constructor.
/***********************************************************************************/
Pmm::Pmm() : fPageMgr() {
  kout << "[PMM] Allocate PageMemoryMgr.\r";
}

Pmm::~Pmm() = default;

/***********************************************************************************/
/// @param If this returns Null pointer, enter emergency mode.
/// @param user is this a user page?
/// @param readWrite is it r/w?
/***********************************************************************************/
Ref<PTEWrapper> Pmm::RequestPage(Boolean user, Boolean readWrite) {
  PTEWrapper pt = fPageMgr.Leak().Request(user, readWrite, false, kPageSize, 0);

  if (pt.fPresent) {
    kout << "[PMM]: Allocation failed.\r";
    return {};
  }

  return Ref<PTEWrapper>(pt);
}

Boolean Pmm::FreePage(Ref<PTEWrapper> PageRef) {
  if (!PageRef) return false;

  PageRef.Leak().fPresent = false;

  return true;
}

Boolean Pmm::TogglePresent(Ref<PTEWrapper> PageRef, Boolean Enable) {
  if (!PageRef) return false;

  PageRef.Leak().fPresent = Enable;

  return true;
}

Boolean Pmm::ToggleUser(Ref<PTEWrapper> PageRef, Boolean Enable) {
  if (!PageRef) return false;

  PageRef.Leak().fRw = Enable;

  return true;
}

Boolean Pmm::ToggleRw(Ref<PTEWrapper> PageRef, Boolean Enable) {
  if (!PageRef) return false;

  PageRef.Leak().fRw = Enable;

  return true;
}

Boolean Pmm::ToggleShare(Ref<PTEWrapper> PageRef, Boolean Enable) {
  if (!PageRef) return false;

  PageRef.Leak().fShareable = Enable;

  return true;
}
}  // namespace Kernel