import json
import pprint
import requests

from orm.common.orm_common.injector import injector
from orm.services.image_manager.ims.logger import get_logger
from orm.services.image_manager.ims.logic.error_base import ErrorStatus

from pecan import conf, request

di = injector.get_di()

LOG = get_logger(__name__)

headers = {'content-type': 'application/json'}


def send_image(image_dict, transaction_id, action="put"):
    # action can be "post" for creating image or "delete" for deleting image

    data = {
        "service_template": {
            "resource": {
                "resource_type": "image"
            },
            "model": str(json.dumps(image_dict)),
            "tracking": {
                "external_id": "",
                "tracking_id": transaction_id
            }
        }
    }

    data_to_display = {
        "service_template": {
            "resource": {
                "resource_type": "image"
            },
            "model": image_dict,
            "tracking": {
                "external_id": "",
                "tracking_id": transaction_id
            }
        }
    }

    try:
        pp = pprint.PrettyPrinter(width=30)
        pretty_text = pp.pformat(data_to_display)
        wrapper_json = json.dumps(data)

        headers['X-RANGER-Client'] = request.headers[
            'X-RANGER-Client'] if 'X-RANGER-Client' in request.headers else \
            'NA'
        headers['X-RANGER-Requester'] = request.headers[
            'X-RANGER-Requester'] if 'X-RANGER-Requester' in request.headers else \
            ''
        headers['X-Auth-Region'] = request.headers[
            'X-Auth-Region'] if 'X-Auth-Region' in \
            request.headers else ''
        headers['X-Auth-Token'] = request.headers[
            'X-Auth-Token'] if 'X-Auth-Token' in \
            request.headers else ''

        LOG.debug("Wrapper JSON before sending action: {0} to Rds Proxy {1}".format(action, pretty_text))
        LOG.info("Sending to RDS Server: " + conf.api.rds_server.base + conf.api.rds_server.resources)
        if action == "post":
            resp = requests.post(conf.api.rds_server.base + conf.api.rds_server.resources,
                                 data=wrapper_json,
                                 headers=headers, verify=conf.verify)
        elif action == "put":
            resp = requests.put(conf.api.rds_server.base + conf.api.rds_server.resources,
                                data=wrapper_json,
                                headers=headers, verify=conf.verify)
        elif action == "delete":
            resp = requests.delete(conf.api.rds_server.base + conf.api.rds_server.resources,
                                   data=wrapper_json,
                                   headers=headers, verify=conf.verify)
        else:
            raise Exception("Invalid action in RdxProxy.send_image("
                            "image_dict, transaction_id, action) "
                            "action can be post or delete, "
                            "got {0}".format(action))

        content = resp.content
        LOG.debug("return from rds server status code: {0} "
                  "content: {1}".format(resp.status_code, resp.content))
        if resp.content and 200 <= resp.status_code < 300:
            content = resp.json()
        else:
            # In case of error from rds, the response is WSME format response.
            # the error message is within the 'faultstring'
            raise ErrorStatus(resp.status_code,
                              json.loads(content)["faultstring"])

    except Exception as exp:
        LOG.log_exception("ImageLogic - Failed to update image", exp)
        raise exp

    return content


def get_status(resource_id, json_convert=True):
    resp = requests.get(conf.api.rds_server.base + conf.api.rds_server.status + resource_id, verify=conf.verify)
    if json_convert:
        resp = resp.json()
    return resp