"""Apply K Rool Phase order."""

from randomizer.Enums.Maps import Maps
from randomizer.Patching.Patcher import LocalROM
from randomizer.Patching.Library.Assets import getPointerLocation, TableNames


def randomize_krool(spoiler, ROM_COPY: LocalROM):
    """Apply K Rool Phase order based on krool_order from spoiler."""
    varspaceOffset = spoiler.settings.rom_data
    # /* 0x058 */ char k_rool_order[5]; // Order of K. Rool phases: [0,1,2,3,4] dictates DK->Diddy->Lanky->Tiny->Chunky. If K. Rool is being shortened to less than 5 phases, put the unused phases as -1
    kroolOffset = 0x058
    ROM_COPY.seek(varspaceOffset + kroolOffset)
    phaseCount = len(spoiler.settings.krool_order)
    ROM_COPY.writeBytes(bytearray(spoiler.settings.krool_order))
    for i in range(5 - phaseCount):
        ROM_COPY.write(255)

    firstPhase = spoiler.settings.krool_order[0]
    if firstPhase != 0:  # If not starting with DK

        # Find Isles->DK Phase loading zone in Pointer table 18 and write new destination map
        cont_map_lzs_address = getPointerLocation(TableNames.Triggers, Maps.Isles)
        ROM_COPY.seek(cont_map_lzs_address)
        lz_count = int.from_bytes(ROM_COPY.readBytes(2), "big")
        for lz_id in range(lz_count):
            start = (lz_id * 0x38) + 2
            ROM_COPY.seek(cont_map_lzs_address + start + 0x12)
            lz_map = int.from_bytes(ROM_COPY.readBytes(2), "big")
            if lz_map == Maps.KroolDonkeyPhase:
                ROM_COPY.seek(cont_map_lzs_address + start + 0x12)
                ROM_COPY.writeMultipleBytes(firstPhase, 2)


def randomize_helm(spoiler, ROM_COPY: LocalROM):
    """Apply Helm Room order based on helm_order from spoiler."""
    varspaceOffset = spoiler.settings.rom_data
    helmOffset = 0x190
    ROM_COPY.seek(varspaceOffset + helmOffset)
    roomCount = len(spoiler.settings.helm_order)
    ROM_COPY.writeBytes(bytearray(spoiler.settings.helm_order))
    for i in range(5 - roomCount):
        ROM_COPY.write(255)