#!/usr/bin/env python3
import serial
import argparse
import os
from time import sleep
from math import ceil

#we send syn, recv ack
SYN_MAGIC = 0xdeadbeef 
ACK_MAGIC = 0xf00dd00d

EEPROM_PAGESIZE = 64
EEPROM_FILE = "./compiled_eeprom"


parser = argparse.ArgumentParser(description="sends binary file over serial so the avr can write it to eeprom")
parser.add_argument('port', type=str)
parser.add_argument('baud', type=int)
args = parser.parse_args()

eeprom_file = open("./compiled_eeprom", "rb")
eeprom_filesize = os.fstat(eeprom_file.fileno()).st_size
page_len = ceil(eeprom_filesize / EEPROM_PAGESIZE)

s = serial.Serial(port=args.port, baudrate=args.baud, parity=serial.PARITY_NONE, timeout=0, dsrdtr=1)

#slow method
def write_hwf(data):
    debugval = 0
    print("writing")
    data_parsed = [data[b:b+1] for b in range(len(data))]
    for c in data_parsed:
        print(debugval)
        debugval += 1
        s.rts = 1 
        while not s.cts:
            pass
        s.rts = 0
        s.write(c)

def read_hwf(length):
    print("reading")
    msg = b''
    for c in range(length):
        s.rts = 1
        while not s.cts:
            pass
        s.rts = 0
        msg += s.read(size=1)
    return(msg)

def handle_writer_error(write_status):
    if write_status == 1: # verification failed
        print("Verification failed on page {}!".format(page))
    elif write_status == 0xff:
        print("Unknown error writing page {}!".format(page))

#while True:
#    write_hwf(b"abcdef\r\n")
#    print(read_hwf(10))

write_hwf(page_len.to_bytes(1, 'little'))

for page in range(0, page_len):
    print(page)
    if EEPROM_PAGESIZE > (eeprom_filesize - eeprom_file.tell()):
        write_hwf(eeprom_file.read())
        [write_hwf(b'\x00') for b in range(0, eeprom_filesize % EEPROM_PAGESIZE)]
    else:
        write_hwf(eeprom_file.read(EEPROM_PAGESIZE))
    write_status = read_hwf(1)
    print("WRITE STATUS: {}".format(write_status))
    if not write_status == 0:
        handle_writer_error(write_status)

print("Done writing to EEPROM!\n")