from webagg.inputrequest import DirectWSGIInputRequest, POSTInputRequest
from bottle import route, request, response, abort, Bottle, debug as bottle_debug

import requests
import traceback
import json

JSON_CT = 'application/json; charset=utf-8'


#=============================================================================
class ResAggApp(object):
    def __init__(self, *args, **kwargs):
        self.application = Bottle()
        self.application.default_error_handler = self.err_handler
        self.route_dict = {}
        self.debug = kwargs.get('debug', False)

        if self.debug:
            bottle_debug(True)

        @self.application.route('/')
        def list_routes():
            return self.route_dict

    def add_route(self, path, handler):
        @self.application.route([path, path + '/<mode:path>'], 'ANY')
        @self.wrap_error
        def direct_input_request(mode=''):
            params = dict(request.query)
            params['mode'] = mode
            params['_input_req'] = DirectWSGIInputRequest(request.environ)
            return handler(params)

        @self.application.route([path + '/postreq', path + '/<mode:path>/postreq'], 'POST')
        @self.wrap_error
        def post_fullrequest(mode=''):
            params = dict(request.query)
            params['mode'] = mode
            params['_input_req'] = POSTInputRequest(request.environ)
            return handler(params)

        handler_dict = handler.get_supported_modes()
        self.route_dict[path] = handler_dict
        self.route_dict[path + '/postreq'] = handler_dict

    def err_handler(self, exc):
        if self.debug:
            print(exc)
            traceback.print_exc()
        response.status = exc.status_code
        response.content_type = JSON_CT
        err_msg = json.dumps({'message': exc.body})
        response.headers['ResErrors'] = err_msg
        return err_msg

    def wrap_error(self, func):
        def wrap_func(*args, **kwargs):
            try:
                out_headers, res, errs = func(*args, **kwargs)

                if out_headers:
                    for n, v in out_headers.items():
                        response.headers[n] = v

                if res:
                    if errs:
                        response.headers['ResErrors'] = json.dumps(errs)
                    return res

                last_exc = errs.pop('last_exc', None)
                if last_exc:
                    if self.debug:
                        traceback.print_exc()

                    response.status = last_exc.status()
                    message = last_exc.msg
                else:
                    response.status = 404
                    message = 'No Resource Found'

                response.content_type = JSON_CT
                res = {'message': message}
                if errs:
                    res['errors'] = errs

                err_msg = json.dumps(res)
                response.headers['ResErrors'] = err_msg
                return err_msg

            except Exception as e:
                if self.debug:
                    traceback.print_exc()
                abort(500, 'Internal Error: ' + str(e))

        return wrap_func