__init__.py

from __future__ import annotations
import os
import sys
import subprocess
import time

sys.path.append(os.path.dirname(os.path.realpath(__file__)))

from decompileAll import decompileAll
from mrbParser import RiteFile
from mrbToRb.mrbToRb import mrbToRb

def decompileFile(file: str, outFile: str|None = None):
    with open(file, "rb") as f:
        riteFile = RiteFile(f)
        codesRes = mrbToRb(riteFile)
        # print(codesRes.toStr())
    with open(outFile or f"{file}.rb", "wb") as f:
        f.write(codesRes.toStr().encode("utf-8", "ignore"))

def compileFile(file: str, outFile: str|None = None):
    WIN_BIN = "bins\\windows\\mrbc.exe"
    LINUX_BIN = "bins/linux/mrbc.exe"
    curDir = os.path.dirname(os.path.realpath(__file__))
    binAbs = os.path.join(curDir, WIN_BIN if os.name == "nt" else LINUX_BIN)
    outFile = outFile or file + ".mrb"
    cmd = f"\"{binAbs}\" -o \"{outFile}\" \"{file}\""
    subprocess.call(cmd)

if __name__ == "__main__":
    mrbFiles = [f for f in sys.argv[1:] if os.path.exists(f)]
    t1 = time.time()

    if "--decompileAll" in sys.argv:
        decompileAll(mrbFiles[0])
    else:
        for file in mrbFiles:
            if os.path.isdir(file):
                print(f"Decompiling all files in {file}")
                decompileAll(file)
            elif file.endswith(".mrb") or file.endswith("_scp.bin"):
                print(f"Decompiling {file}")
                decompileFile(file)
            elif file.endswith(".rb"):
                print(f"Compiling {file}")
                compileFile(file)
            else:
                print(f"Unknown file type: {file}")

    tD = time.time() - t1
    if tD < 0.5:
        print(f"Time: {(tD*1000):.1f}ms")
    else:
        print(f"Time: {tD:.1f}s")