#!/usr/bin/env python

import sys, hashlib
from electrum import Interface

"""validate a transaction (SPV)"""


Hash = lambda x: hashlib.sha256(hashlib.sha256(x).digest()).digest()

def rev_hex(s):
    return s.decode('hex')[::-1].encode('hex')

def int_to_hex(i, length=1):
    s = hex(i)[2:].rstrip('L')
    s = "0"*(2*length - len(s)) + s
    return rev_hex(s)


i = Interface({'server':'ecdsa.org:50002:s'})
i.start()


def merkle_root(merkle):

    merkle = map (lambda tx_hash: tx_hash.decode('hex')[::-1], merkle)
    while len(merkle) != 1:
        if len(merkle)%2: merkle.append( merkle[-1] )
        n = []
        while merkle:
            n.append( Hash( merkle[0] + merkle[1] ) )
            merkle = merkle[2:]

        merkle = n
            
    return merkle[0][::-1].encode('hex')


def hash_header(res):
    header = int_to_hex(res.get('version'),4) \
        + rev_hex(res.get('prev_block_hash')) \
        + rev_hex(res.get('merkle_root')) \
        + int_to_hex(int(res.get('timestamp')),4) \
        + int_to_hex(int(res.get('bits')),4) \
        + int_to_hex(int(res.get('nonce')),4)
    return rev_hex(Hash(header.decode('hex')).encode('hex'))


def verify_tx(tx_hash):
    
    res = i.synchronous_get([ ('blockchain.transaction.get_merkle',[tx_hash]) ])[0]
    assert tx_hash in res.get('merkle')
    assert res.get('merkle_root') == merkle_root(res['merkle'])
    block_height = res.get('block_height')
    print block_height

    headers_requests = []
    for height in range(block_height-10,block_height+10):
        headers_requests.append( ('blockchain.block.get_header',[height]) )
    res = i.synchronous_get(headers_requests)

    _hash = None
    for header in res:
        if _hash: assert _hash == header.get('prev_block_hash')
        _hash = hash_header(header)
        print _hash
        if height==block_height:
            assert header.get('merkle_root') == res.get('merkle_root')
        


try:
    tx = sys.argv[1]
except:
    tx = '587430e52af2cec98b3fd543083469ffa7a5f5dd2bd569898a7897a64e2eb031'

verify_tx(tx)