import argparse
import json
import struct
import sys
from pathlib import Path

from unwrap_self import FIRST_REGION, SELF_MAGIC, SELF_MAGIC_KERNEL, Refused, parse_elf, read_records


def main(argv=None):
    for stream in (sys.stdout, sys.stderr):
        if hasattr(stream, "reconfigure"):
            stream.reconfigure(encoding="utf-8", errors="backslashreplace")
    parser = argparse.ArgumentParser(description="Inspect SELF record and original inner ELF header metadata as JSON; this does not validate conversion or decrypt segments")
    parser.add_argument("input", type=Path)
    args = parser.parse_args(argv)
    try:
        data = args.input.read_bytes()
        if len(data) < FIRST_REGION:
            raise Refused("file is smaller than the SELF header area")
        magic = struct.unpack_from("<I", data)[0]
        if magic not in (SELF_MAGIC, SELF_MAGIC_KERNEL):
            raise Refused("not a SELF container")
        records = read_records(data)
        inner, _, headers = parse_elf(data)
    except (OSError, ValueError, Refused, struct.error) as error:
        parser.error(str(error))
    fields = ("index", "type", "flags", "offset", "vaddr", "size", "memory", "align")
    print(json.dumps({"size": len(data), "magic": magic, "inner_elf": inner,
                      "records": [{"flags": flags, "offset": offset, "size": size} for flags, offset, size in records],
                      "program_headers": [{field: header[field] for field in fields} for header in headers]}, indent=2))
    return 0


if __name__ == "__main__":
    sys.exit(main())