Very WIP

import subprocess
import sys
import pathlib
import binascii
settings = []
def flushSettings():
    with open("settings.ids", "w") as f:
        f.write("\n".join(settings))
try:
    with open("settings.ids") as f:
        [settings.append(line.rstrip("\n")) for line in f]
except:
    with open("settings.ids", "a") as f:
        [f.write("False\n") for i in range(10)]
        [settings.append("False") for i in range(10)]
if settings[0] == "False":
    commands = ["B", "3 005d1870.rec", "3 recrypt.sys", "q"]
    exe = subprocess.Popen("ique_diag.exe", stdin=subprocess.PIPE, stdout=subprocess.PIPE)
    toStdin = "\n".join(commands)
    print("Dumping necessary files... (This will only be done once)")
    returned = exe.communicate(input=bytes(toStdin.encode("utf-8")))
    for i in returned[0].decode("utf-8").split("\n"):
        if "__BBC_CheckHandle" in i:
            input("Make sure your iQue Player is connected to your VM. Press enter to exit...")
            sys.exit()
        elif "FFFFFFF" in i:
            input("Please play Dr. Mario at least once before using this program. Press enter to exit...")
    print("Necessary file dump finished.")
    plaintext = pathlib.Path("005d1870.z64")
    plaintextPath = "005d1870.z64"
    if not plaintext.is_file():
        while True:
            plaintextPath = str(input("Please enter the path to a Dr. Mario plaintext ROM, in .z64 format:\n"))
            if plaintextPath.split(".")[-1] == ".z64":
                break
            else:
                print("Invalid ROM")
    print("Patching Dr. Mario .rec... (This will only be done once)")
    subprocess.call(["ique_cbc_attack.exe", "-p", plaintextPath, "-r", "005d1870.rec", "-d", "081F0000000000000000000000000000", "-o", "1000"])
    commands = ["B", "4 005d1870.rec", "4 005d1870.sta", "q"]
    exe = subprocess.Popen("ique_diag.exe", stdin=subprocess.PIPE)
    toStdin = "\n".join(commands)
    print("Flashing patched Dr. Mario back to your iQue Player... (This will only be done once)")
    exe.communicate(input=bytes(toStdin.encode("utf-8")))
    print("Done!")
    settings[0] = "True"
    flushSettings()
if settings[1] == "False":
    input("You must now open Dr. Mario. Turn off your iQue Player, unplug the USB cable, turn your iQue Player on, open Dr. Mario. Wait approximately 10 seconds, then reset your iQue Player. When the menu appears, turn off your iQue Player, connect the USB cable, and turn it on. If you're using a VM, make sure that your iQue Player is correctly connected to your VM before continuing.")
    commands = ["B", "3 005d1870.sta", "q"]
    exe = subprocess.Popen("ique_diag.exe", stdin=subprocess.PIPE, stdout=subprocess.PIPE)
    toStdin = "\n".join(commands)
    print("Dumping file... (This will only be done once)")
    returned = exe.communicate(input=bytes(toStdin.encode("utf-8")))
    for i in returned[0].decode("utf-8").split("\n"):
        if "__BBC_CheckHandle" in i:
            input("Make sure your iQue Player is connected to your VM. Press enter to exit...")
            sys.exit()
    v2 = bytearray(open("005d1870.sta", 'rb').read())
    v2_bytes = []
    for i in v2:
        v2_bytes.append(hex(i).lstrip("0x").upper().zfill(2))
    with open("v2.bin", 'wb') as output:
        bytes_towrite = binascii.a2b_hex(''.join(v2_bytes[1536:1792]))
        output.write(bytes_towrite)
    settings[1] = "True"
    flushSettings()