#!/usr/bin/env python
#
# This file serves the static files in the
# web directory and provides API endpoints
# to read/write the existing markers.
#

import os
from cStringIO import StringIO
import hashlib
import threading
import json
import time

import cherrypy
from cherrypy.lib import auth_digest

class SynchronizedJSON(object):
    def __init__(self, filename):
        self._filename = filename
        self._new = {}
        if os.path.exists(self._filename):
            with open(self._filename, 'rb') as f:
                self._new['data'] = f.read()
        else:
            self._new['data'] = '{}'

        self._update_sync_id()
        self.cur = self._new

        self.lock = threading.Lock()

    def _update_sync_id(self):
        doc = json.loads(self._new['data'])
        if 'sync-id' in doc:
            del doc['sync-id']
        hashed_data = json.dumps(doc)
        h = hashlib.sha256()
        h.update(hashed_data)
        doc['sync-id'] = h.hexdigest()
        self._new['data'] = json.dumps(doc, indent=4)
        self._new['sync-id'] = h.hexdigest()

    def set_data(self, data):
        assert self.lock.locked()

        if self.cur['data'] == data:
            return

        self._new = {}
        self._new['data'] = data
        self._update_sync_id()

        with open(self._filename + '.new', 'wb') as f:
            f.write(self._new['data'])
            f.flush()
            os.fsync(f.fileno())
        os.rename(self._filename + '.new', self._filename)

        self.cur = self._new

class EventMapMarkerApi(object):
    def __init__(self, path):
        self.marker_doc = SynchronizedJSON(os.path.join(path, 'markers.json'))

    @cherrypy.expose
    def get(self):
        cherrypy.response.headers['Content-Type']= 'application/json'
        return self.marker_doc.cur['data']

    @cherrypy.expose
    def poll(self, current):
        cherrypy.response.headers['Content-Type']= 'application/json'
        counter = 0
        while self.marker_doc.cur['sync-id'] == current:
            time.sleep(1)
            counter += 1
            if counter >= 20:
                yield ' '
                counter = 0
            if cherrypy.engine.state != cherrypy.engine.states.STARTED:
                break
        yield self.marker_doc.cur['data']

    @cherrypy.expose
    def post(self):
        content_length = cherrypy.request.headers['Content-Length']
        data = cherrypy.request.body.read(int(content_length))

        doc = json.loads(data)

        with self.marker_doc.lock:
            if self.marker_doc.cur['sync-id'] != doc['sync-id']:
                raise cherrypy.HTTPError(503, "Sorry, but the server database changed in between.")
            self.marker_doc.set_data(data)
        cherrypy.response.headers['Content-Type']= 'application/json'
        return '{}'

class EventMapApi(object):
    def __init__(self, path):
        self.markers = EventMapMarkerApi(path)

def test_log(msg, level):
    print "%s, %s" % (msg, level)

if __name__ == '__main__':

    USERS = {'username' : 'password'}

    current_dir = os.path.dirname(os.path.abspath(__file__))
    cherrypy.engine.subscribe('log', test_log)
    cherrypy.config.update({
        'server.socket_host': '127.0.0.1',
        'server.socket_port': 8023,
        'server.thread_pool_max': 500,
        'server.thread_pool': 100,
        'log.screen': True
    })

    cherrypy.tree.mount(None, '/', {
        '/': {
            'tools.staticdir.on': True,
            'tools.staticdir.dir': os.path.join(current_dir, 'web'),
            'tools.staticdir.index': 'index.html',
            'tools.auth_digest.on': True,
            'tools.auth_digest.realm': 'eventmap',
            'tools.auth_digest.get_ha1': auth_digest.get_ha1_dict_plain(USERS),
            'tools.auth_digest.key': '640f0d950828cc65'
        }
    })

    cherrypy.tree.mount(EventMapApi(current_dir), '/api', {
        '/': {
            'response.timeout': 600,
            'response.stream': True
        }
    })

    cherrypy.engine.signals.subscribe()
    cherrypy.engine.start()
    cherrypy.engine.block()