#!/usr/bin/env python
# -*- coding: utf-8 -*-

# Dieses Programm sollte auf einem Gateway laufen
# Es liefert Informationen über die Tunnel nach Redis

import os  
import sys   
import socket
import json
import subprocess
import datetime

from rediscluster import StrictRedisCluster
from rediscluster.exceptions import ClusterDownError
from redis.exceptions import ConnectionError

def get_gate_nodeid():
    # Die ID kann statisch angegeben werden, falls das nicht
    # der Fall ist, wird der Hostname angenommen
    try:
        with open('/etc/alfred/statics.json', 'r') as fh:
            statics = json.load(fh)
    except (IOError, ValueError):
        return socket.gethostname()
    try:
        nodeid = statics['node']['node_id']
    except:
        nodeid = socket.gethostname()
    return nodeid


def main(rc):

    # fastd-Socket auslesen, liefert ein JSON-Objekt
    client = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM)
    try:
        client.connect("/var/run/fastd/ffpi.sock")
    except socket.error, msg:  
        print >>sys.stderr, msg
        sys.exit(1)
    data = json.loads(client.makefile('r').read())
    client.close()
    # Ermittelte Daten aufbereiten und nach Redis schreiben
    gate_id = get_gate_nodeid()
    now = datetime.datetime.now().replace(microsecond=0).isoformat()
    peers = {}
    for key, peer in data['peers'].iteritems():
        if peer['connection']:
            try:
                tunnel_id = peer['connection']['mac_addresses'][0].replace(':', '')
            except IndexError:
                continue
            rckey = "fastd:tunnel:%s" % tunnel_id
            try:
                rc.hset(rckey, 'key', key)
                rc.hset(rckey, 'last_seen', now)
                rc.hset(rckey, 'last_gate', gate_id)
            except ClusterDownError:
                pass

if __name__ == '__main__':
    startup_nodes = [{"host": "127.0.0.1", "port": "7000"}]
    try:
        rc = StrictRedisCluster(startup_nodes=startup_nodes, decode_responses=True)
    except ConnectionError:
        # Kann z.B. bei lokalen Netzwerkproblemen auftreten
        sys.exit(1)
    main(rc)