'''
Proxy server for getting hyperthesis annotation
'''
import os

import bottle
import requests
from bottle import route, response

app = application = bottle.Bottle()

config_file_path = os.getenv('CONFIG_PATH', "hypothesis_proxy.conf")

app.config.load_config(config_file_path)

# From https://stackoverflow.com/a/17262900
class EnableCors(object):
    name = 'enable_cors'
    api = 2

    def apply(self, fn, context):
        def _enable_cors(*args, **kwargs):
            # set CORS headers
            response.headers['Access-Control-Allow-Origin'] = '*'
            response.headers['Access-Control-Allow-Methods'] = 'GET, POST, PUT, OPTIONS'
            response.headers['Access-Control-Allow-Headers'] = 'Origin, Accept, Content-Type, X-Requested-With, X-CSRF-Token'

            if bottle.request.method != 'OPTIONS':
                # actual request; reply with the actual response
                return fn(*args, **kwargs)

        return _enable_cors

app.install(EnableCors())

@app.route('/annotations')
def query_hypothesis():
    '''
    query hypothesis
    '''

    annotations_json = app.config.get('proxy.annotations_json', None)

    if annotations_json and os.path.isfile(annotations_json):
        import json
        with open(annotations_json, 'r') as f:
            return json.load(f)

    url = app.config.get('proxy.url', 'https://h.projet-episteme.org/api/search')
    headers = {}
    token = app.config.get('proxy.token')
    if token is not None:
        headers["Authorization"] = "Bearer " + token

    params = {}
    for config_tag in ['group', 'user', 'tag', 'limit']:
        config_val = app.config.get('proxy.%s' % config_tag)
        if config_val is not None:
            params[config_tag] = config_val

    offset = 0
    total = -1

    rows = []

    while total < 0 or offset < total:
        #TODO: handle errors
        res = requests.get(url, params=params, headers=headers)
        res_json = res.json()
        total = int(res_json['total'])
        rows += res_json['rows']
        offset = len(rows)
        params['offset'] = offset

    return {
        'rows': rows,
        'total': len(rows)
    }


if __name__ == "__main__":
    app.run(
        host=app.config.get('proxy.host', '127.0.0.1'),
        port=int(app.config.get('proxy.port', 8888))
    )