import sys
import function_puller
from optparse import OptionParser

def build_phase0_spec(sourcefile, outfile):
    code_lines = []
    code_lines.append("""

from typing import (
    Any,
    Dict,
    List,
    NewType,
    Tuple,
)
from eth2spec.utils.minimal_ssz import (
    SSZType,
    SSZTypeExtension,
    hash_tree_root,
    signing_root,
)
from eth2spec.utils.bls_stub import (
    bls_aggregate_pubkeys,
    bls_verify,
    bls_verify_multiple,
)
from eth2spec.utils.hash_function import hash


# stub, will get overwritten by real var
SLOTS_PER_EPOCH = 64

Slot = NewType('Slot', int)  # uint64
Epoch = NewType('Epoch', int)  # uint64
Shard = NewType('Shard', int)  # uint64
ValidatorIndex = NewType('ValidatorIndex', int)  # uint64
Gwei = NewType('Gwei', int)  # uint64
Bytes32 = NewType('Bytes32', bytes)  # bytes32
BLSPubkey = NewType('BLSPubkey', bytes)  # bytes48
BLSSignature = NewType('BLSSignature', bytes)  # bytes96
Store = None
""")

    code_lines += function_puller.get_spec(sourcefile)

    code_lines.append("""
# Monkey patch validator compute committee code
_compute_committee = compute_committee
committee_cache = {}


def compute_committee(indices: List[ValidatorIndex], seed: Bytes32, index: int, count: int) -> List[ValidatorIndex]:
    param_hash = (hash_tree_root(indices), seed, index, count)

    if param_hash in committee_cache:
        return committee_cache[param_hash]
    else:
        ret = _compute_committee(indices, seed, index, count)
        committee_cache[param_hash] = ret
        return ret


# Monkey patch hash cache
_hash = hash
hash_cache = {}


def hash(x):
    if x in hash_cache:
        return hash_cache[x]
    else:
        ret = _hash(x)
        hash_cache[x] = ret
        return ret


# Access to overwrite spec constants based on configuration
def apply_constants_preset(preset: Dict[str, Any]):
    global_vars = globals()
    for k, v in preset.items():
        global_vars[k] = v

    # Deal with derived constants
    global_vars['GENESIS_EPOCH'] = slot_to_epoch(GENESIS_SLOT)

    # Initialize SSZ types again, to account for changed lengths
    init_SSZ_types()
""")

    with open(outfile, 'w') as out:
        out.write("\n".join(code_lines))


def build_phase1_spec(sourcefile, outfile):
    code_lines = []
    code_lines.append("""
from eth2spec.phase0.spec import *
from eth2spec.phase0.spec import apply_constants_preset as apply_constants_preset_phase0

""")

    code_lines += function_puller.get_spec(sourcefile)

    code_lines.append("""
# Access to overwrite spec constants based on configuration
def apply_constants_preset(preset: Dict[str, Any]):

    apply_constants_preset_phase0(preset)

    global_vars = globals()
    for k, v in preset.items():
        global_vars[k] = v

    # Deal with derived constants
    global_vars['GENESIS_EPOCH'] = slot_to_epoch(GENESIS_SLOT)

    # Initialize SSZ types again, to account for changed lengths
    init_SSZ_types()
""")

    with open(outfile, 'w') as out:
        out.write("\n".join(code_lines))


if __name__ == '__main__':
    parser = OptionParser()
    parser.add_option("-p", "--phase", dest="phase", type="int", default=0,
                  help="Build for phase #")

    (options, args) = parser.parse_args()

    if len(args) < 2:
        parser.print_help()
    
    if options.phase == 0:
        build_phase0_spec(args[0], args[1])
        print(args[0])
        print(args[1])
    elif options.phase == 1:
        build_phase1_spec(args[0], args[1])
    else:
        print("Invalid phase: {0}".format(options["phase"]))