import json

import flask
from owsresponse.adaptors.flask_encoder import FlaskEncoder


def flaskify(response, headers=None, encoder=FlaskEncoder):
    """Format the response to be consumable by flask.

    The api returns mostly JSON responses. The format method converts the dicts
    into a json object (as a string), and the right response is returned (with
    the valid mimetype, charset and status.)

    Args:
        response (Response): The dictionary object to convert into a json
            object. If the value is a string, a dictionary is created with the
            key "message".
        headers (dict): optional headers for the flask response.
        encoder (Class): The class of the encoder (if any).

    Returns:
        flask.Response: The flask response with formatted data, headers, and
            mimetype.
    """

    status_code = response.status
    data = response.errors or response.message

    mimetype = 'text/plain'
    if isinstance(data, list) or isinstance(data, dict):
        mimetype = 'application/json'
        data = json.dumps(data, cls=encoder)

    return flask.Response(
        response=data, status=status_code, headers=headers, mimetype=mimetype)