#!/usr/bin/env python
# -*- coding: utf-8 -*-
#
# Copyright (c) 2014-2015, NORDUnet A/S.
# See LICENSE for licensing information.

import sys
import json
import base64
import requests
from time import sleep
from base64 import b64encode, b64decode
from certtools import timing_point, \
     create_ssl_context
from mergetools import get_curpos, get_logorder, chunks, get_missingentries, \
     publish_sth, sendlog, sendentries, parse_args, perm, get_frontend_verifiedsize, \
     frontend_verify_entries

def sendlog_helper(entries, curpos, nodename, nodeaddress, own_key, paths):
    print >>sys.stderr, "sending log:",
    sys.stderr.flush()
    for chunk in chunks(entries, 1000):
        for trynumber in range(5, 0, -1):
            sendlogresult = sendlog(nodename, nodeaddress,
                                    own_key, paths,
                                    {"start": curpos, "hashes": chunk})
            if sendlogresult == None:
                if trynumber == 1:
                    sys.exit(1)
                sleep(10)
                print >>sys.stderr, "tries left:", trynumber
                sys.stderr.flush()
                continue
            break
        if sendlogresult["result"] != "ok":
            print >>sys.stderr, "sendlog:", sendlogresult
            sys.exit(1)
        curpos += len(chunk)
        print >>sys.stderr, curpos,
        sys.stderr.flush()
    print >>sys.stderr
    print >>sys.stderr, "log sent"
    sys.stderr.flush()

def fill_in_missing_entries(nodename, nodeaddress, own_key, paths, chainsdb, timing):
    missingentries = get_missingentries(nodename, nodeaddress, own_key,
                                        paths)
    timing_point(timing, "get missing")

    while missingentries:
        print >>sys.stderr, "missing entries:", len(missingentries)
        sys.stderr.flush()

        sent_entries = 0
        print >>sys.stderr, "sending missing entries",
        sys.stderr.flush()
        with requests.sessions.Session() as session:
            for missingentry_chunk in chunks(missingentries, 100):
                missingentry_hashes = [base64.b64decode(missingentry) for missingentry in missingentry_chunk]
                hashes_and_entries = [(hash, chainsdb.get(hash)) for hash in missingentry_hashes]
                sendentryresult = sendentries(nodename, nodeaddress,
                                             own_key, paths,
                                             hashes_and_entries, session)
                if sendentryresult["result"] != "ok":
                    print >>sys.stderr, "sendentries:", sendentryresult
                    sys.exit(1)
                sent_entries += len(missingentry_hashes)
                print >>sys.stderr, sent_entries,
                sys.stderr.flush()
        print >>sys.stderr
        sys.stderr.flush()
        timing_point(timing, "send missing")

        missingentries = get_missingentries(nodename, nodeaddress,
                                                     own_key, paths)
        timing_point(timing, "get missing")
    

def merge_dist(args, localconfig, frontendnodes, timestamp):
    maxwindow = localconfig.get("maxwindow", 1000)
    paths = localconfig["paths"]
    own_key = (localconfig["nodename"],
               "%s/%s-private.pem" % (paths["privatekeys"],
                                      localconfig["nodename"]))
    mergedb = paths["mergedb"]
    chainsdb = perm(localconfig.get("dbbackend", "filedb"), mergedb + "/chains")
    logorderfile = mergedb + "/logorder"
    sthfile = mergedb + "/sth"
    create_ssl_context(cafile=paths["https_cacertfile"])
    timing = timing_point()

    try:
        sth = json.loads(open(sthfile, 'r').read())
    except (IOError, ValueError):
        print >>sys.stderr, "No valid STH file found in", sthfile
        return timestamp
    if sth['timestamp'] < timestamp:
        print >>sys.stderr, "New STH file older than the previous one:", \
          sth['timestamp'], "<", timestamp
        return timestamp
    if sth['timestamp'] == timestamp:
        return timestamp
    timestamp = sth['timestamp']

    logorder = get_logorder(logorderfile, sth['tree_size'])
    timing_point(timing, "get logorder")

    for frontendnode in frontendnodes:
        nodeaddress = "https://%s/" % frontendnode["address"]
        nodename = frontendnode["name"]
        timing = timing_point()

        print >>sys.stderr, "distributing for node", nodename
        sys.stderr.flush()
        curpos = get_curpos(nodename, nodeaddress, own_key, paths)
        timing_point(timing, "get curpos")
        print >>sys.stderr, "current position", curpos
        sys.stderr.flush()

        verifiedsize = get_frontend_verifiedsize(nodename, nodeaddress, own_key, paths)
        timing_point(timing, "get verified size")
        print >>sys.stderr, "verified size", verifiedsize

        assert verifiedsize >= curpos

        while verifiedsize < len(logorder):
            uptopos = min(verifiedsize + maxwindow, len(logorder))
        
            entries = [b64encode(entry) for entry in logorder[verifiedsize:uptopos]]
            sendlog_helper(entries, verifiedsize, nodename, nodeaddress, own_key, paths)
            timing_point(timing, "sendlog")

            fill_in_missing_entries(nodename, nodeaddress, own_key, paths, chainsdb, timing)

            verifiedsize = frontend_verify_entries(nodename, nodeaddress, own_key, paths, uptopos)
        
        print >>sys.stderr, "sending sth to node", nodename
        sys.stderr.flush()
        publishsthresult = publish_sth(nodename, nodeaddress, own_key, paths, sth)
        if publishsthresult["result"] != "ok":
            print >>sys.stderr, "publishsth:", publishsthresult
            sys.exit(1)
        timing_point(timing, "send sth")

        if args.timing:
            print >>sys.stderr, "timing: merge_dist:", timing["deltatimes"]
            sys.stderr.flush()

    return timestamp

def main():
    """
    Distribute missing entries and the STH to all frontend nodes.
    """
    args, config, localconfig = parse_args()
    timestamp = 0

    if len(args.node) == 0:
        nodes = config["frontendnodes"]
    else:
        nodes = [n for n in config["frontendnodes"] if n["name"] in args.node]

    while True:
        timestamp = merge_dist(args, localconfig, nodes, timestamp)
        if args.interval is None:
            break
        print >>sys.stderr, "sleeping", args.interval, "seconds"
        sleep(args.interval)

if __name__ == '__main__':
    sys.exit(main())