import time
from webui import webui
from pathlib import Path
from serial import Serial
from utl import trans_list_to_str
from func_frame import make_frame_dlt645, check_frame_dlt645


def frame_read(daddr=0x60, dlen=0x50, dev_addr=[0xAA, 0xAA, 0xAA, 0xAA, 0xAA, 0xAA]):
    block_modbus = {
        'addr_dev' : 0x01,
        'data_addr' : daddr,
        'data_len' : dlen,
        'type' : 'read',
    }
    block_dlt645 = {
        'addr' : dev_addr,
        'type' : 'modbus',
        'data' : block_modbus,
    }
    frame2 = make_frame_dlt645(block_dlt645)
    if com is None:
        print(trans_list_to_str(frame2))
        return
    com.read_all()
    com.write(bytearray(frame2))
    time.sleep(0.5)
    frame3 = com.read_all()
    output_text = check_frame_dlt645(frame3, block=block_dlt645)
    print(output_text)

def frame_write_one(daddr=0x85, dval=-900, dev_addr=[0xAA, 0xAA, 0xAA, 0xAA, 0xAA, 0xAA]):
    block_modbus = {
        'addr_dev' : 0x01,
        'data_addr' : daddr,
        'data_val' : dval,
        'type' : 'write_one',
    }
    block_dlt645 = {
        'addr' : dev_addr,
        'type' : 'modbus',
        'data' : block_modbus,
    }
    frame2 = make_frame_dlt645(block_dlt645)
    if com is None:
        print(trans_list_to_str(frame2))
        return
    com.write(bytearray(frame2))

def frame_write_dual(daddr=0x91, dval=600, dev_addr=[0xAA, 0xAA, 0xAA, 0xAA, 0xAA, 0xAA]):
    block_modbus = {
        'addr_dev' : 0x01,
        'data_addr' : daddr,
        'data_val' : dval,
        'type' : 'write_dual',
    }
    block_dlt645 = {
        'addr' : dev_addr,
        'type' : 'modbus',
        'data' : block_modbus,
    }
    frame2 = make_frame_dlt645(block_dlt645)
    if com is None:
        print(trans_list_to_str(frame2))
        return
    com.write(bytearray(frame2))

def frame_write_str(daddr=0x82, dval=[0x06, 0x05, 0x04, 0x03, 0x02, 0x01], dev_addr=[0xAA, 0xAA, 0xAA, 0xAA, 0xAA, 0xAA]):
    block_modbus = {
        'addr_dev' : 0x01,
        'data_addr' : daddr,
        'data_val' : dval,
        'type' : 'write_str',
    }
    block_dlt645 = {
        'addr' : dev_addr,
        'type' : 'modbus',
        'data' : block_modbus,
    }
    frame2 = make_frame_dlt645(block_dlt645)
    if com is None:
        print(trans_list_to_str(frame2))
        return
    com.write(bytearray(frame2))

def frame_update(com, path_bin, dev_addr):
    """ 程序升级 """
    block_modbus = {
        'addr_dev' : 0x01,
        'type' : 'update',
        'step' : 'start',
        'index' : 0,
        'file' : Path(path_bin).read_bytes(),
        'header_offset': 128,
    }
    block_dlt645 = {
        'addr' : dev_addr,
        'type' : 'modbus',
        'data' : block_modbus,
    }
    # 启动帧
    frame_master = bytearray(make_frame_dlt645(block_dlt645))

    # 等待擦除完成返回
    try_times = 3000
    com.read_all()
    while try_times:
        time.sleep(0.4)
        com.write(frame_master)
        frame_slave = com.read_all()
        if not frame_slave:
            try_times -= 1
            continue
        
        block_dlt645["data"]['file_block_size'] = check_frame_dlt645(frame_slave, block_dlt645)
        break

    if block_dlt645["data"]['file_block_size'] == 0:
        raise Exception("Error slave response.")

    # 避免接收到延迟返回报文
    time.sleep(0.4)

    # 文件传输
    block_dlt645["data"]['step'] = 'trans'
    data_remain = len(block_dlt645["data"]['file']) - block_dlt645['data']['header_offset']
    while data_remain > 0:
        frame_master = bytearray(make_frame_dlt645(block_dlt645))

        com.read_all()
        com.write(frame_master)
        time.sleep(0.2)
        frame_slave = None
        while not frame_slave:
            frame_slave = com.read_all()
        check_frame_dlt645(frame_slave, block_dlt645)
        
        block_dlt645["data"]['index'] += 1
        data_remain -= block_dlt645["data"]['file_block_size']

    # 结束升级
    block_dlt645["data"]['step'] = 'end'
    frame_master = bytearray(make_frame_dlt645(block_dlt645))

    com.read_all()
    com.write(frame_master)
    time.sleep(0.1)
    frame_slave = None
    while not frame_slave:
        frame_slave = com.read_all()
    check_frame_dlt645(frame_slave[:18], block_dlt645)

    return com



def my_function(e : webui.event):
    """ WebUI回调函数 """
    global events
    events.append(e)
    print(len(events))

    print("Data from JavaScript: " + e.window.get_str(e, 0)) # Message from JS
    frame = e.window.get_str(e, 0)
    block_dlt645 = e.window.get_str(e, 1)
    output_text = check_frame_dlt645(frame, block=block_dlt645)
    
    return output_text

events = []
def main_webui():
    myWindow = webui.window()
    root = Path(".")

    file_main = root / 'resource\main.html'

    myWindow.bind("myID1", my_function)
    myWindow.bind("my_function", my_function)
    myWindow.show(str(file_main))

    while 1:
        """ 调试断点 """
        webui.wait()
        print(events)
        myWindow.show(str(file_main))

com = None
if __name__ == "__main__":
    com = Serial("Com16", baudrate=115200, parity='N', timeout=2)
    address = [0xAA, 0xAA, 0xAA, 0xAA, 0xAA, 0xAA]
    path_bin1 = "D:\\WorkingProject\\LightStackAdapter\\software\\lamina_adapter\\tools\\upgrade\\DGAPD_240415_1000_V1.05.bin"
    path_bin2 = "D:\\WorkingProject\\LightStackAdapter\\software\\lamina_adapter\\tools\\upgrade\\DGAPD_240416_1400_V1.05.bin"

    frame_read(0x100, 0x20)

    if not hasattr(__builtins__,"__IPYTHON__"):
        path_bin = path_bin1
        frame_update(com, path_bin, address)

        time.sleep(1)
        com.read_all()

        frame_read(0x100, 0x20)

    pass