#!/usr/bin/env python3
# Copyright (c) 2021, Arm Limited. All rights reserved.
#
# SPDX-License-Identifier: BSD-3-Clause

import argparse
import uuid
import zlib

def main(metadata_file, img_type_uuids, location_uuids, img_uuids):
    def add_field_to_metadata(value):
        # Write the integer values to file in little endian representation
        with open(metadata_file, "ab") as fp:
            fp.write(value.to_bytes(4, byteorder='little'))

    def add_uuid_to_metadata(uuid_str):
        # Validate UUID string and write to file in little endian representation
        uuid_val = uuid.UUID(uuid_str)
        with open(metadata_file, "ab") as fp:
            fp.write(uuid_val.bytes_le)

    # Fill metadata preamble
    add_field_to_metadata(1) #version=1
    add_field_to_metadata(0) #active_index=0
    add_field_to_metadata(0) #previous_active_index=0

    for img_type_uuid, location_uuid in zip(img_type_uuids, location_uuids):
        # Fill metadata image entry
        add_uuid_to_metadata(img_type_uuid) # img_type_uuid
        add_uuid_to_metadata(location_uuid) # location_uuid

        for img_uuid in img_uuids:
            # Fill metadata bank image info
            add_uuid_to_metadata(img_uuid) # image unique bank_uuid
            add_field_to_metadata(1)       # accepted=1
            add_field_to_metadata(0)       # reserved (MBZ)

    # Prepend CRC32
    with open(metadata_file, 'rb+') as fp:
        content = fp.read()
        crc = zlib.crc32(content)
        fp.seek(0)
        fp.write(crc.to_bytes(4, byteorder='little') + content)

if __name__ == "__main__":
    parser = argparse.ArgumentParser()
    parser.add_argument('--metadata_file', required=True,
                        help='Output binary file to store the metadata')
    parser.add_argument('--img_type_uuids', type=str, nargs='+', required=True,
                        help='A list of UUIDs identifying the image types')
    parser.add_argument('--location_uuids', type=str, nargs='+', required=True,
                        help='A list of UUIDs of the storage volumes where the images are located. '
                             'Must have the same length as img_type_uuids.')
    parser.add_argument('--img_uuids', type=str, nargs='+', required=True,
                        help='A list UUIDs of the images in a firmware bank')

    args = parser.parse_args()

    if len(args.img_type_uuids) != len(args.location_uuids):
        parser.print_help()
        raise argparse.ArgumentError(None, 'Arguments img_type_uuids and location_uuids must have the same length.')

    main(args.metadata_file, args.img_type_uuids, args.location_uuids, args.img_uuids)