from typing import List, Dict

def parse_hex(file: str) -> Dict:
    """
    TODO: Implement support for more record types
    """
    with open(file,
              "r", encoding='utf-8') as f:
        lines = f.readlines()

    memory_blocks = {}
    bytes = []
    block_offset = -1

    def add_memory_block():
        memory_blocks[block_offset] = bytes[:]

    is_error = False
    byte_len = 0

    next_line_offset = -1

    for i in range(lines.__len__()):
        line = lines[i]
        if line[0] != ':':
            print("Error: unexpected record mark on line %i, expect \':\', get \'%c\'" % (
                i+1, line[0]))
            is_error = True
            break

        reclen = int(line[1:3], base=16)        # Record length
        load_offset = int(line[3:7], base=16)   # Initial address of data byte
        rectype = int(line[7:9], base=16)       # Record type
        data_bytes: List[str] = []

        data_bytes_line = line[9:reclen*2 + 9]
        for i in range(reclen):
            data_bytes.append(data_bytes_line[i*2:i*2+2])
            byte_len += 1

        if rectype == 0:  # Data Record
            if next_line_offset == -1:
                next_line_offset = load_offset
                block_offset = load_offset

            if next_line_offset != load_offset:
                add_memory_block()
                bytes.clear()
                block_offset = load_offset
                next_line_offset = load_offset

            for i in range(reclen):
                byte = data_bytes[i]
                byte = int(f"0x{byte}", base=16)
                bytes.append(byte)

            next_line_offset += reclen

            # for i in range(data_len//4):
            #     data_bytes = word_bytes.reverse()
            # print("data words: ", data_words)
        elif rectype == 1:  # End of File Record
            # print("End of File")
            add_memory_block()
        elif rectype == 2:  # Extended Segment Address Record
            print("Record 2: Extended Segment Address Record")
            print("ERROR: unimplemented record type 2 on line %i" % (i+1))
            is_error = True
            break
        elif rectype == 3:  # Start Segment Address Record
            print("Start Segment Address Record")
            print("ERROR: unimplemented record type 3 on line %i" % (i+1))
            is_error = True
        elif rectype == 4:  # Extended Linear Address Record
            print("Extended Linear Address Record")
            print("ERROR: unimplemented record type 4 on line %i" % (i+1))
            # is_error = True
        elif rectype == 5:  # Start Linear Address Record
            print("Start Linear Address is 0x%s (line %i)" %
                    (data_bytes_line, (i+1)))
            print("MIK32 MCU does not support arbitrary start address")
        else:
            print("ERROR: unexpected record type %i on line %i" %
                    (rectype, i+1))
            is_error = True
            break
        # print("line %i data_bytes=%i line_addr=%i" % (i+1, data_bytes, line_addr))

    # for word in memory_blocks[0]:
    #     print(f"{word:#0x}")

    if is_error:
        print("ERROR: error while parsing")
        exit()

    return memory_blocks


def parse_bin(filename: str) -> List[int]:
    arr: List[int] = []
    with open(filename, "rb") as f:
        while byte := f.read(1):
            arr.append(byte[0])
    return arr


def bytes2words(arr: List[int]) -> List[int]:
    word = []
    words = []
    for byte in arr:
        word.append(byte)
        if word.__len__() == 4:
            words.append(word[0]+2**8*word[1]+2**16*word[2]+2**24*word[3])
            word = []
    return words

def get_content(filename: str) -> List[int]:
    content: List[int] = []

    if filename.endswith(".bin"):
        content = parse_bin(filename)
    elif filename.endswith(".hex"):
        content = parse_hex(filename)[0]
    else:
        raise Exception("Unsupported file format")
    
    return content

# parse_hex("mik32-uploader../test-roms/eeprom.hex")