# this is a wrapper function to allow additional parsing and arguments such
# as a custom TLV with a hash to be added dynamically based on a key being 
# provided

import subprocess
import click
import json
import platform

@click.command()
@click.option('-s','--sdk', help='Path to the SDK to find additional scripts and tools as needed')
@click.option('-t','--toolchain',help='Path to the toolchain')
@click.option('-n','--name',help="name of project")
@click.option('-c','--color',help="color of the LED")
def make_binary(sdk,toolchain,name,color):
    # Select the proper default python executable depending on OS
    python_exe = "python3"
    if "Window" in platform.system():
        python_exe = "python"

    print("sdk:", sdk)
    print("toolchain:", toolchain)
    print("projectName:", name)

    f = open('../signingArgs.json')

    config = json.load(f)

    f.close()
    # config = {
    #     "version" : "1.0.0",
    #     "addKey" : True
    #     "privateKeyPath" : "/source/third_party/mcuboot/image_signTEST.pem"
    # }
    subprocess.run([toolchain+'/bin/tiarmobjcopy',name+'.out',
                '--output-target','binary','sample_image-unsigned.bin'],
                capture_output=True)


    if(type(config["version"]) is not list):
        versionList = [config["version"]]
    else:
        versionList = config["version"]

    retVal = config.get("slotSize")
    if(retVal is not None): 
        slotSize = retVal
    else:
        slotSize = 0x2800

    retVal = config.get("bankSize")
    if(retVal is not None): 
        bankSize = retVal
    else:
        bankSize = 0x20000
    
    retVal = config.get("offset")
    if(retVal is not None): 
        offset = retVal
    else:
        offset = 0x4800

    # iterate over all versions provided
    for ver in versionList:

        versionStr = ver.replace('.','_')

        imgtool_optional_args = ['--header-size','0x100','--align','4','--slot-size', hex(slotSize),
                '--pad','--version',ver,'--pad-header','--overwrite-only','--key',
                sdk+config["privateKeyPath"]]

        if "addKey" in config: 
            print("adding a key")
            imgtool_optional_args.extend(["--custom-tlv", "0xCC", "0xb93c72b74bc8487d29827005f80d635918f91bc22b14d7ed05714d58f96702dd"])
            keyFile = open("keyFile.bin", 'wb')
            keyArray = [0x603deb10, 0x15ca71be, 0x2b73aef0, 0x857d7781,
                        0x1f352c07, 0x3b6108d7, 0x2d9810a3, 0x0914dff4]
            keyString = ''.join(format(x, '08x') for x in keyArray)
            keyFile.write(bytearray.fromhex(keyString))
            keyFile.close()

        if "removeKey" in config:
            print("removing a key")
            imgtool_optional_args.extend(["--custom-tlv", "0xCD", "0x555555555555555555555555555555"])

        outFilename = 'sample_image_signed_' + hex(offset) + '_v'+versionStr+'_'+color

        imgtool_args = [python_exe,sdk+'/source/third_party/mcuboot/scripts/imgtool.py',
                    'sign'] + imgtool_optional_args + ['sample_image-unsigned.bin',
                        outFilename+'.bin']

        print(imgtool_args)
        proc = subprocess.run(imgtool_args, capture_output=True)
        print(proc)

        txtConversion_args = [python_exe,sdk+'/source/third_party/mcuboot/scripts/bin-to-ti-txt.py',
                            '-f',outFilename+'.bin','-o',outFilename+'.txt','--encrypt','-c',
                            outFilename+'_encrypted.txt','-s',hex(bankSize+offset)]

        proc = subprocess.run(txtConversion_args, capture_output=True)
        print(proc)

if __name__ == "__main__":
    make_binary()