from clvm.casts import int_from_bytes
from clvm_tools import binutils

from maize.consensus.block_rewards import calculate_base_farmer_reward, calculate_pool_reward
from maize.types.blockchain_format.program import Program
from maize.types.condition_opcodes import ConditionOpcode
from maize.util.bech32m import decode_puzzle_hash, encode_puzzle_hash
from maize.util.condition_tools import parse_sexp_to_conditions
from maize.util.ints import uint32

#address1 = "txch15gx26ndmacfaqlq8m0yajeggzceu7cvmaz4df0hahkukes695rss6lej7h"  # Gene wallet (m/12381/8444/2/42):
#address2 = "txch1c2cguswhvmdyz9hr3q6hak2h6p9dw4rz82g4707k2xy2sarv705qcce4pn"  # Mariano address (m/12381/8444/2/0)
address1 = "xmz15sjtarzr0h3xu58pu6yddyq3hu0ty0crdlyqpueg44nx4m5kxu2q3ntgl2"
address2 = "xmz1fy4qz73tcewgue47a708zrz46mr78dh4zgd9vquuy5mrsrkhtygq2vf3fv"

ph1 = decode_puzzle_hash(address1)
ph2 = decode_puzzle_hash(address2)

print("ph1", ph1.hex())
print("h1", encode_puzzle_hash(ph1, 'xmz'))
print("ph2", ph2.hex())
print("h2", encode_puzzle_hash(ph2, 'xmz'))


pool_amounts = int(calculate_pool_reward(uint32(0)) / 2)
farmer_amounts = int(calculate_base_farmer_reward(uint32(0)) / 2)

assert pool_amounts * 2 == calculate_pool_reward(uint32(0))
assert farmer_amounts * 2 == calculate_base_farmer_reward(uint32(0))


def make_puzzle(amount: int) -> int:
    puzzle = f"(q . ((51 0x{ph1.hex()} {amount}) (51 0x{ph2.hex()} {amount})))"
    # print(puzzle)

    puzzle_prog = Program.to(binutils.assemble(puzzle))
    print("Program: ", puzzle_prog)
    puzzle_hash = puzzle_prog.get_tree_hash()

    solution = "()"
    prefix = "xmz"
    print("PH", puzzle_hash)
    print(f"Address: {encode_puzzle_hash(puzzle_hash, prefix)}")

    result = puzzle_prog.run(solution)
    error, result_human = parse_sexp_to_conditions(result)

    total_maize = 0
    if error:
        print(f"Error: {error}")
    else:
        assert result_human is not None
        for cvp in result_human:
            assert len(cvp.vars) == 2
            total_maize += int_from_bytes(cvp.vars[1])
            print(
                f"{ConditionOpcode(cvp.opcode).name}: {encode_puzzle_hash(cvp.vars[0], prefix)},"
                f" amount: {int_from_bytes(cvp.vars[1])}"
            )
    return total_maize


total_maize = 0
print("Pool address: ")
#total_maize += make_puzzle(pool_amounts)
print("\nFarmer address: ")
#total_maize += make_puzzle(farmer_amounts)

#assert total_maize == calculate_base_farmer_reward(uint32(0)) + calculate_pool_reward(uint32(0))
