import random
from mersenne import BreakerPy
random.seed(31337)
arr = []
for _ in range(0, 624):
    arr.append(random.getrandbits(32))
breaker = BreakerPy()
breaker.get_seeds_python_fast(arr)

from rngeesus import Breaker
from random import Random
from secrets import randbits


def sample():
    print("Sampling...")

    correct = 0
    iterations = 10000
    for _ in range(iterations):
        seed = randbits(32)

        rng = Random(seed)

        # Empty the RNG that is filled from /dev/urandom
        # I think this is necessary to get a predictable RNG
        for _ in range(624):
            rng.getrandbits(32)

        b = Breaker()
        # Two values in the format (index, value)
        vals = [
            (0, rng.getrandbits(32)),
            (1, rng.getrandbits(32)),
            (2, rng.getrandbits(32)),
        ]

        print(vals)
        print("Predicting...")

        # This line always returns None for some reason?
        recovered_seed = b.get_seed_mt(vals)

        # Get the next value from the RNG
        # We will use this to check if the predictor is correct
        print(f"Actual seed: {seed}")
        print(f"Predicted seed: {recovered_seed}")

        if recovered_seed == seed:
            print("Correct!")
            correct += 1

        print(f"Current status: {correct}/{iterations}")

    print(f"Correct: {correct}/{iterations}")


if __name__ == "__main__":
    sample()