#!/usr/bin/env python3

# SPDX-FileCopyrightText: © 2023-2024 Decompollaborate
# SPDX-License-Identifier: MIT

from __future__ import annotations

import argparse
import ipl3checksum
from pathlib import Path


parser = argparse.ArgumentParser()
parser.add_argument("rom", help="Path to a big endian rom")
parser.add_argument("-c", "--cic", help="The cic to use. If unrecognized or missing then the script will default to 6102")
parser.add_argument("-a", "--autodetect", "--auto", help="Try to detect the cic automatically, ignoring the --cic parameter. If unable to detect then the script will default to 6102", action="store_true")

args = parser.parse_args()
romPathArg = args.rom

romPath = Path(romPathArg)
cic = int(args.cic if args.cic is not None else 0)

romBytes = romPath.read_bytes()

if args.autodetect:
    cickind = ipl3checksum.detectCIC(romBytes)
else:
    cickind = ipl3checksum.CICKind.fromValue(cic)

if cickind is None:
    cickind = ipl3checksum.CICKind.CIC_6102_7101

print(f"Using {cickind.name}")

checksum = ipl3checksum.calculateChecksum(romBytes, cickind)
assert checksum is not None

print(f"{checksum[0]:08X}")
print(f"{checksum[1]:08X}")