#!/usr/bin/python

import json

from flask import Flask, jsonify, request
from flask.json import JSONEncoder
from DLIESConnector import DLIESConnector
from DLIQueryResolver import QueryResolver

from logger import dlilogger


class MyJSONEncoder(JSONEncoder):
    def default(self, obj):
        return obj.__dict__



app = Flask(__name__)
app.debug = False
app.json_encoder = MyJSONEncoder

host = "http://localhost:8983/solr/DLIF-index-cleaned"
base_dnet_url = "http://localhost:8280/app"


# q = QueryResolver("http://localhost:8983/solr/DLIF-index-cleaned_shard1_replica1")

class InvalidUsage(Exception):
    status_code = 400

    def __init__(self, message, status_code=None, payload=None):
        Exception.__init__(self)
        self.message = message
        if status_code is not None:
            self.status_code = status_code
        self.payload = payload

    def to_dict(self):
        rv = dict(self.payload or ())
        rv['message'] = self.message
        return rv


# Routes
@app.route('/')
def root():
    return app.send_static_file('index.html')


@app.route('/<path:path>')
def static_proxy(path):
    # send_static_file will guess the correct MIME type
    return app.send_static_file(path)


@app.route('/api/item/<path:identifier>', methods=['post', 'get'])
def getItem(identifier):
    if identifier:
        type  = request.form.get('type')
        start = request.form.get('from')
        if len(type)== 0:
            type =None
        if len(start) == 0:
            start=None
        else:
            start=int(start)
        connector = DLIESConnector('ip-90-147-167-27.ct1.garrservices.it', 'dli')
        return jsonify(result=connector.item_by_id(identifier, type, start))
    else:
        raise InvalidUsage('This view is gone', status_code=410)


@app.route('/api/stats/', methods=['post', 'get'])
def stats():
    q = QueryResolver(host, base_dnet_url)
    return jsonify(stats=q.get_stats())


@app.route('/api/ds_info/', methods=['post', 'get'])
def info():
    q = QueryResolver(host, base_dnet_url)
    if 'id' in request.form:
        id = request.form['id']
        name = request.form['name']
        result = q.get_ds_info(id, name)
        return jsonify(info=result)
    result = q.get_all_ds()
    return jsonify(info=result)


@app.route('/api/stats_detail/', methods=['post', 'get'])
def stats_detail():
    return ""

    # q = QueryResolver(host, base_dnet_url)
    # return jsonify(stats=q.get_stats_detail())


@app.route('/api/post/', methods=['post', 'get'])
def query_post():
    action = None
    query = None
    filterKey = None
    filterValue = None

    start = 0
    if 'action' in request.form:
        action = request.form['action']
    if 'query' in request.form:
        query = request.form['query']
    if 'start' in request.form:
        start = int(request.form['start'])
    if 'filter' in request.form:
        filterKey = request.form['filter']
    if 'value' in request.form:
        filterValue = request.form['value']

    if action == 'query' and query is not None:
        connector = DLIESConnector('ip-90-147-167-27.ct1.garrservices.it', 'dli')
        try:
            result = connector.simple_query(query, start=start, filter_key=filterKey, filter_value=filterValue)
        except Exception as e:
            print e
        return jsonify(result=result)
    else:
        return jsonify(error={'error_code': 'an error occur during query on server'})


if __name__ == '__main__':
    app.logger.addHandler(dlilogger)
    app.run(debug=False, host='0.0.0.0', processes=5)