import sys
import json
import logging
from flask import Flask, render_template, request, redirect, url_for

sys.path.append("../../Common/python")
import lockerfs

import client
import util

app = Flask(__name__)

@app.route("/setupAuth")
def setupAuth():
    return render_template("setupAuth.html")

@app.route("/save", methods=['POST'])
def saveAuth():
    logging.info("Saving auth")
    secrets = lockerfs.loadJsonFile("secrets.json")
    secrets["jid"] = request.form["jid"]
    secrets["password"] = request.form["password"]
    start(secrets)
    lockerfs.saveJsonFile("secrets.json", secrets)
    return json.dumps("started")

def start(secrets):
    logging.info("Starting")
    app.client = client.Client(app.info, jid=secrets["jid"], password=secrets["password"])
    address = (secrets["host"], secrets["port"]) if (secrets.has_key("host") and secrets.has_key("port")) else ()
    logging.info("XMPP connecting with address " + str(address))
    if app.client.connect(address):
        app.client.process(threaded=True)
        app.started = True
    else:
        # XXX We shouldn't die here, we should still serve existing data and try again.  
        # We could also prompt for credentials again
        util.die("XMPP connection failed")

@app.route("/")
def index():
    if app.started:
        return json.dumps({
                "/messages" : "All messages received. Filter by: body, from, mucnick, mucroom, to, type, id, subject",
                "/statuses" : "All status updates received. Filter by: status, from, show, priority, type, id",
                "/roster" : "Current roster (at time of login)"
                })
    else:
        return redirect(app.me_info["uri"] + "setupAuth")

def matches_arg(value, arg):
    # either a literal match or a range [lo,hi]
    if type(arg) is list and len(arg) is 2:
        (lo, hi) = arg
        return (lo <= value) and (value < hi)
    else:
        return (value == arg)

@app.route("/messages")
def messages():
    if not app.client or len(app.client.messages) == 0: return "[]"
    messages = app.client.messages
    for key, value in request.args.items():
        messages = [msg for msg in messages if matches_arg(msg[key], json.loads(value))]
    return json.dumps(messages)

@app.route("/statuses")
def statuses():
    if not app.client or len(app.client.statuses) == 0: return "[]"
    statuses = app.client.statuses
    for key, value in request.args.items():
        statuses = [sts for sts in statuses if matches_arg(sts[key], json.loads(value))]
    return json.dumps(statuses)

@app.route("/roster")
def roster():
    return json.dumps(app.client.fetch_roster())

def runService(info):
    app.info = info
    app.client = None
    app.started = False

    app.me_info = lockerfs.loadJsonFile("me.json")

    secrets = lockerfs.loadJsonFile("secrets.json")
    if "jid" in secrets and "password" in secrets:
        start(secrets)
    else:
        logging.info("No auth details available")
    app.debug = True
    app.run(port=app.info["port"], use_reloader=False)

if __name__ == "__main__":
    logging.basicConfig(level=logging.INFO,
                        format='%(levelname)-8s %(message)s')

    runService({"port": 7474})