ltsdb/app.py

import hmac
import json
import logging
import os

from flask import (Flask, request, jsonify, abort)

from ltsdb_json import LTS

logging.basicConfig(level=logging.DEBUG)

app = Flask(__name__)

log = logging.getLogger()

@app.route("/")
def home():
    return jsonify({ "success": None })


@app.route("/report", methods=["POST"])
def report():
    data = request.get_json()
    n_ts = 0
    n_dp = 0
    for d in data:
        d["description"]["remote_addr"] = request.remote_addr
        d["description"]["node"] = verify_node(d)

        log.info("received %s", json.dumps(d))
        ts = LTS(d["description"])
        for dp in d["data"]:
            ts.add(*dp)
            ts.save()
            n_dp += 1
        n_ts += 1
    return jsonify({ "success": True, "timeseries": n_ts, "datapoints": n_dp })

@app.route("/ts/<id>")
def get_timeseries(id):
    try:
        ts = LTS(id=id)
    except FileNotFoundError:
        abort(404)
    return jsonify({"description": ts.description, "data": ts.data})

def verify_node(d):
    node = d["auth"]["node"]
    timestamp = d["auth"]["timestamp"]
    digest1 = d["auth"]["hmac"]
    if "/" in node:
        raise ValueError("invalid node name %s", node)
    try:
        log.info("getting client config from %s", "config/" + node)
        with open("config/" + node) as fh:
            node_conf = json.load(fh)
    except Exception as e:
        log.warning("got %s opening %s", e, "config/" + node)
        abort(401, "unknown client")
    last = node_conf["last"]
    for key in node_conf["keys"]:
        msg = (node + " " + str(timestamp)).encode("UTF-8")
        hmac2 = hmac.new(key.encode("UTF-8"), msg, "SHA256")
        digest2 = hmac2.hexdigest()
        if hmac.compare_digest(digest1, digest2):
            if timestamp > node_conf["last"]:
                node_conf["last"] = timestamp
                os.replace("config/" + node, "config/" + node + ".old")
                with open("config/" + node, "w") as fh:
                    json.dump(node_conf, fh) # XXX
                return node
            else:
                abort(409, "timestamp out of sync")
    abort(401, "auth failed")