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

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

	FILE: DiskImage.cc
	PURPOSE: Disk Imaging framework.

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

#include <DiskImage.fwrk/headers/DiskImage.h>

#include <FirmwareKit/EPM.h>
#include <FSKit/NeFS.h>

/// @brief EPM format disk
/// @param img disk image structure.
/// @return Status code upon completion.
SInt32 DIFormatPartitionEPM(struct DI_DISK_IMAGE& img) noexcept
{
	if (!img.sector_sz || (img.sector_sz % 512 != 0))
		return kDIFailureStatus;

	if (*img.out_name == 0 ||
		*img.disk_name == 0)
		return kDIFailureStatus;

	struct ::EPM_PART_BLOCK block
	{
	};

	block.NumBlocks = img.block_cnt;
	block.SectorSz	= img.sector_sz;
	block.Version	= kEPMRevisionBcd;
	block.LbaStart	= sizeof(struct ::EPM_PART_BLOCK);
	block.LbaEnd	= img.disk_sz - block.LbaStart;
	block.FsVersion = kNeFSVersionInteger;

	::MmCopyMemory(block.Name, (VoidPtr)img.disk_name, ::MmStrLen(img.disk_name));
	::MmCopyMemory(block.Magic, (VoidPtr)kEPMMagic86, ::MmStrLen(kEPMMagic86));

	IOObject handle = IoOpenFile(img.out_name, nullptr);

	if (!handle)
		return kDIFailureStatus;

	::IoWriteFile(handle, (Char*)&block, sizeof(struct ::EPM_PART_BLOCK));

	::IoCloseFile(handle);

	handle = nullptr;

	return kDISuccessStatus;
}

/// @brief NeFS format over EPM.
/// @param img disk image structure.
/// @return Status code upon completion.
SInt32 DIFormatFilesystemNeFS(struct DI_DISK_IMAGE& img) noexcept
{
	if (!img.sector_sz || (img.sector_sz % 512 != 0))
		return kDIFailureStatus;

	if (*img.out_name == 0 ||
		*img.disk_name == 0)
		return kDIFailureStatus;

	struct ::NEFS_ROOT_PARTITION_BLOCK rpb
	{
	};

	::MmCopyMemory(rpb.PartitionName, (VoidPtr)img.disk_name, ::MmStrLen(img.disk_name));
	::MmCopyMemory(rpb.Ident, (VoidPtr)kNeFSIdent, ::MmStrLen(kNeFSIdent));

	rpb.Version	 = kNeFSVersionInteger;
	rpb.EpmBlock = kEPMBootBlockLba;

	rpb.StartCatalog = kNeFSCatalogStartAddress;
	rpb.CatalogCount = 0;

	rpb.DiskSize = img.disk_sz;

	rpb.SectorSize	= img.sector_sz;
	rpb.SectorCount = rpb.DiskSize / rpb.SectorSize;

	rpb.FreeSectors = rpb.SectorCount;
	rpb.FreeCatalog = rpb.DiskSize / sizeof(NEFS_CATALOG_STRUCT);

	IOObject handle = IoOpenFile(img.out_name, nullptr);

	if (!handle)
		return kDIFailureStatus;

	UInt64 p_prev = ::IoTellFile(handle);

	::IoWriteFile(handle, (Char*)&rpb, sizeof(struct ::NEFS_ROOT_PARTITION_BLOCK));

	::IoSeekFile(handle, p_prev);

	::IoCloseFile(handle);

	handle = nullptr;

	return kDISuccessStatus;
}