import { assertEquals } from "@std/assert";
import {
  mbrSector,
  MBR_BOOT_SIGNATURE,
  MBR_PARTITION_TYPE,
  MBR_SIZE,
} from "@binstruct/mbr";

const coder = mbrSector();
const sector = {
  bootstrapCode: new Uint8Array(446),
  partitions: [
    {
      status: 0x80,
      chsFirst: new Uint8Array([0x00, 0x01, 0x00]),
      partitionType: MBR_PARTITION_TYPE.FAT32_LBA,
      chsLast: new Uint8Array([0xfe, 0xff, 0xff]),
      lbaFirstSector: 2048,
      sectorCount: 204800,
    },
    { status: 0, chsFirst: new Uint8Array(3), partitionType: MBR_PARTITION_TYPE.EMPTY, chsLast: new Uint8Array(3), lbaFirstSector: 0, sectorCount: 0 },
    { status: 0, chsFirst: new Uint8Array(3), partitionType: MBR_PARTITION_TYPE.EMPTY, chsLast: new Uint8Array(3), lbaFirstSector: 0, sectorCount: 0 },
    { status: 0, chsFirst: new Uint8Array(3), partitionType: MBR_PARTITION_TYPE.EMPTY, chsLast: new Uint8Array(3), lbaFirstSector: 0, sectorCount: 0 },
  ],
  bootSignature: MBR_BOOT_SIGNATURE,
};

const buffer = new Uint8Array(MBR_SIZE);
const written = coder.encode(sector, buffer);
const [decoded, read] = coder.decode(buffer);

assertEquals(written, MBR_SIZE);
assertEquals(read, MBR_SIZE);
assertEquals(decoded.partitions[0].partitionType, MBR_PARTITION_TYPE.FAT32_LBA);
assertEquals(decoded.partitions[0].sectorCount, 204800);
assertEquals(decoded.bootSignature, MBR_BOOT_SIGNATURE);