#!/usr/bin/env python

from electrum_nvc import Interface, SimpleConfig, set_verbosity
from electrum_nvc.network import DEFAULT_SERVERS, filter_protocol
import time, Queue
from collections import defaultdict

set_verbosity(False)

config = SimpleConfig()
servers = filter_protocol(DEFAULT_SERVERS,'t')
interfaces = map ( lambda server: Interface(server, config), servers )

q = Queue.Queue()
results_queue = Queue.Queue()

for i in interfaces: i.start(q)

reached_servers = []

while servers:
    i = q.get(timeout=1000)
    servers.remove(i.server)
    if i.is_connected:
        i.send([('blockchain.headers.subscribe',[])], lambda i,x: results_queue.put((i,x)))
        reached_servers.append(i.server)
        i.status = "ok"
    else:
        i.status = "unreachable"

d = defaultdict(int)

while reached_servers:
    i, r = results_queue.get(timeout=1000)
    i.blocks = r.get('result').get('block_height')
    d[i.blocks] += 1
    reached_servers.remove(i.server)

v = d.values()
numblocks = d.keys()[v.index(max(v))]

for i in interfaces:
    if i.status == "ok":
        if abs(i.blocks-numblocks)>1: i.status = "lagging" 
    else:
        i.blocks = 0

    print "%30s   %d   %s   "%(i.host, i.blocks, i.status)