import networksim
from casper import Validator
import casper
from ethereum.parse_genesis_declaration import mk_basic_state
from ethereum.config import Env
from ethereum.casper_utils import RandaoManager, generate_validation_code, call_casper, \
    get_skips_and_block_making_time, sign_block, get_contract_code, \
    casper_config, get_casper_ct, get_casper_code, get_rlp_decoder_code, \
    get_hash_without_ed_code, make_casper_genesis
from ethereum.utils import sha3, privtoaddr
from ethereum.transactions import Transaction
from ethereum.state_transition import apply_transaction

from ethereum.slogging import LogRecorder, configure_logging, set_level
# config_string = ':info,eth.vm.log:trace,eth.vm.op:trace,eth.vm.stack:trace,eth.vm.exit:trace,eth.pb.msg:trace,eth.pb.tx:debug'
config_string = ':info,eth.vm.log:trace'
configure_logging(config_string=config_string)

n = networksim.NetworkSimulator(latency=150)
n.time = 2
print 'Generating keys'
keys = [sha3(str(i)) for i in range(20)]
print 'Initializing randaos'
randaos = [RandaoManager(sha3(k)) for k in keys]
deposit_sizes = [128] * 15 + [256] * 5

print 'Creating genesis state'
s = make_casper_genesis(validators=[(generate_validation_code(privtoaddr(k)), ds * 10**18, r.get(9999))
                                    for k, ds, r in zip(keys, deposit_sizes, randaos)],
                        alloc={privtoaddr(k): {'balance': 10**18} for k in keys},
                        timestamp=2,
                        epoch_length=50)
g = s.to_snapshot()
print 'Genesis state created'

validators = [Validator(g, k, n, Env(config=casper_config), time_offset=4) for k in keys]
n.agents = validators
n.generate_peers()
lowest_shared_height = -1
made_101_check = 0

for i in range(100000):
    # print 'ticking'
    n.tick()
    if i % 100 == 0:
        print '%d ticks passed' % i
        print 'Validator heads:', [v.chain.head.header.number if v.chain.head else None for v in validators]
        print 'Total blocks created:', casper.global_block_counter
        print 'Dunkle count:', call_casper(validators[0].chain.state, 'getTotalDunklesIncluded', [])
        lowest_shared_height = min([v.chain.head.header.number if v.chain.head else -1 for v in validators])
        if lowest_shared_height >= 101 and not made_101_check:
            made_101_check = True
            print 'Checking that withdrawn validators are inactive'
            assert len([v for v in validators if v.active]) == len(validators) - 5, len([v for v in validators if v.active])
            print 'Check successful'
            break
    if i == 1:
        print 'Checking that all validators are active'
        assert len([v for v in validators if v.active]) == len(validators)
        print 'Check successful'
    if i == 2000:
        print 'Withdrawing a few validators'
        for v in validators[:5]:
            v.withdraw()
    if i == 4000:
        print 'Checking that validators have withdrawn'
        for v in validators[:5]:
            assert v.call_casper('getEndEpoch', [v.vchash]) <= 2
        for v in validators[5:]:
            assert v.call_casper('getEndEpoch', [v.vchash]) > 2
        print 'Check successful'