#!/usr/bin/env python3
#
# Copyright (c) 2024 Raspberry Pi (Trading) Ltd.
#
# SPDX-License-Identifier: BSD-3-Clause
"""Generate a header that provides a binary file as an array."""

import argparse
from pathlib import Path
import sys


_BYTES_PER_LINE = 32


def _parse_args():
    parser = argparse.ArgumentParser(
        description=__doc__,
    )
    parser.add_argument(
        "data",
        type=argparse.FileType("rb"),
        help="Path to data file to generate a header for",
    )
    parser.add_argument(
        "-o",
        "--output",
        type=argparse.FileType("wb"),
        default=sys.stdout.buffer,
        help="Output file path. Defaults to stdout.",
    )
    return parser.parse_args()


def generate_header(data, output):
    var_name = Path(output.name).stem.replace(".", "_")
    include_guard = f"_{var_name.upper()}_H"
    prefix_lines = (
        f"#ifndef {include_guard}",
        f"#define {include_guard}",
        "",
        "#include <stddef.h>",
        "",
        f"const unsigned char {var_name}[] = {{",
    )
    output.write("\n".join(prefix_lines).encode())

    bytes_written = 0
    while True:
        b = data.read(1)
        if b == b"":
            break
        if bytes_written % _BYTES_PER_LINE == 0:
            output.write("\n  ".encode())
        output.write(f"0x{int.from_bytes(b, 'little'):02x}, ".encode())
        bytes_written += 1

    suffix_lines = (
        "",
        "};",
        "",
        f"const size_t {var_name}_SIZE = {bytes_written};",
        "",
        f"#endif  // {include_guard}",
        "",
    )
    output.write("\n".join(suffix_lines).encode())


if __name__ == "__main__":
    sys.exit(generate_header(**vars(_parse_args())))