# Copyright (c) 2017-2018 Wind River Systems, Inc.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#         http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
import logging
import json
import traceback

from keystoneauth1.exceptions import HttpError
from rest_framework import status
from rest_framework.response import Response
from rest_framework.views import APIView

from common.exceptions import VimDriverNewtonException
from newton_base.util import VimDriverUtils

logger = logging.getLogger(__name__)


class Flavors(APIView):
    service = {'service_type': 'compute',
               'interface': 'public'}
    keys_mapping = [
        ("project_id", "tenantId"),
        ("ram", "memory"),
        ("vcpus", "vcpu"),
        ("OS-FLV-EXT-DATA:ephemeral", "ephemeral"),
        ("os-flavor-access:is_public", "isPublic"),
        ("extra_specs", "extraSpecs"),
    ]

    def _convert_extra_specs(self, extraSpecs, extra_specs, reverse=False):
       if reverse == False:
          #from extraSpecs to extra_specs
          for spec in extraSpecs:
              extra_specs[spec['keyName']] = spec['value']
       else:
          for k,v in extra_specs.items():
              spec={}
              spec['keyName']=k
              spec['value']=v
              extraSpecs.append(spec)

    def get(self, request, vimid="", tenantid="", flavorid=""):
        logger.info("vimid, tenantid, flavorid = %s,%s,%s" % (vimid, tenantid, flavorid))
        if request.data:
            logger.debug("With data = %s" % request.data)
            pass

        try:
            # prepare request resource to vim instance
            query = VimDriverUtils.get_query_part(request)

            vim = VimDriverUtils.get_vim_info(vimid)
            sess = VimDriverUtils.get_session(vim, tenantid)
            resp = self._get_flavor(sess, request, flavorid)
            content = resp.json()

            if flavorid:
                flavor = content.pop("flavor", None)
                extraResp = self._get_flavor_extra_specs(sess, flavor["id"])
                extraContent = extraResp.json()
                if extraContent["extra_specs"]:
                    extraSpecs = []
                    self._convert_extra_specs(extraSpecs, extraContent["extra_specs"], True)
                    flavor["extraSpecs"] = extraSpecs
                VimDriverUtils.replace_key_by_mapping(flavor,
                                                   self.keys_mapping)
                content.update(flavor)

            else:
                wanted = None
                #check if query contains name="flavorname"
                if query:
                    for queryone in query.split('&'):
                        k,v = queryone.split('=')
                        if k == "name":
                            wanted = v
                            break

                if wanted:
                   oldFlavors = content.pop("flavors", None)
                   content["flavors"] = []
                   for flavor in oldFlavors:
                       if wanted == flavor["name"]:
                           content["flavors"].append(flavor)

                #iterate each flavor to get extra_specs
                for flavor in content["flavors"]:
                    extraResp = self._get_flavor_extra_specs(sess, flavor["id"])
                    extraContent = extraResp.json()
                    if extraContent["extra_specs"]:
                        extraSpecs = []
                        self._convert_extra_specs(extraSpecs, extraContent["extra_specs"], True)
                        flavor["extraSpecs"] = extraSpecs
                    VimDriverUtils.replace_key_by_mapping(flavor,
                                                   self.keys_mapping)

            #add extra keys
            vim_dict = {
                "vimName": vim["name"],
                "vimId": vim["vimId"],
                "tenantId": tenantid,
            }
            content.update(vim_dict)

            logger.info("response with status = %s" % resp.status_code)

            return Response(data=content, status=resp.status_code)
        except VimDriverNewtonException as e:
            logger.error("response with status = %s" % e.status_code)
            return Response(data={'error': e.content}, status=e.status_code)
        except HttpError as e:
            logger.error("HttpError: status:%s, response:%s" % (e.http_status, e.response.json()))
            return Response(data=e.response.json(), status=e.http_status)
        except Exception as e:
            logger.error(traceback.format_exc())
            return Response(data={'error': str(e)},
                            status=status.HTTP_500_INTERNAL_SERVER_ERROR)

    def _get_flavor_extra_specs(self, sess, flavorid):
        if flavorid:
            # prepare request resource to vim instance
            req_resouce = "/flavors/%s/os-extra_specs" % flavorid

            logger.info("making request with URI:%s" % req_resouce)

            resp = sess.get(req_resouce, endpoint_filter=self.service)

            logger.info("request returns with status %s" % resp.status_code)
            if resp.status_code == status.HTTP_200_OK:
                logger.debug("with content:%s" % resp.json())
                pass

            return resp
        return {}

    def _get_flavor(self, sess, request, flavorid=""):
        if sess:
            # prepare request resource to vim instance
            req_resouce = "/flavors"
            if flavorid:
                req_resouce += "/%s" % flavorid
            else:
                req_resouce += "/detail"

            query = VimDriverUtils.get_query_part(request)
            if query:
                req_resouce += "?%s" % query

            logger.info("making request with URI:%s" % req_resouce)

            resp = sess.get(req_resouce, endpoint_filter=self.service)

            logger.info("request returns with status %s" % resp.status_code)
            if resp.status_code == status.HTTP_200_OK:
                logger.debug("with content:%s" % resp.json())
                pass

            return resp
        return {}

    def post(self, request, vimid="", tenantid="", flavorid=""):
        logger.info("vimid, tenantid, flavorid = %s,%s,%s" % (vimid, tenantid, flavorid))
        if request.data:
            logger.debug("With data = %s" % request.data)
            pass

        sess = None
        resp = None
        resp_body = None
        try:
            # prepare request resource to vim instance
            vim = VimDriverUtils.get_vim_info(vimid)
            sess = VimDriverUtils.get_session(vim, tenantid)

            #check if the flavor is already created: name or id
            tmpresp = self._get_flavor(sess, request)
            content = tmpresp.json()
            #iterate each flavor to get extra_specs
            existed = False
            for flavor in content["flavors"]:
                if flavor["name"] == request.data["name"]:
                   existed = True
                   break
                elif hasattr(request.data, "id") and flavor["id"] == request.data["id"]:
                   existed = True
                   break

            if existed:
                extraResp = self._get_flavor_extra_specs(sess, flavor["id"])
                extraContent = extraResp.json()
                if extraContent["extra_specs"]:
                    extraSpecs = []
                    self._convert_extra_specs(extraSpecs, extraContent["extra_specs"], True)
                    flavor["extraSpecs"] = extraSpecs
                VimDriverUtils.replace_key_by_mapping(flavor,
                                               self.keys_mapping)
                vim_dict = {
                    "vimName": vim["name"],
                    "vimId": vim["vimId"],
                    "tenantId": tenantid,
                     "returnCode": 0,
                }
                flavor.update(vim_dict)
                return Response(data=flavor, status=tmpresp.status_code)

            extraSpecs = request.data.pop("extraSpecs", None)
            #create flavor first
            resp = self._create_flavor(sess, request)
            if resp.status_code == 202:
                resp_body = resp.json()["flavor"]
            else:
                return resp


            flavorid = resp_body['id']
            if extraSpecs:
                extra_specs={}
                self._convert_extra_specs(extraSpecs, extra_specs, False)

                extraResp = self._create_flavor_extra_specs(sess, extra_specs, flavorid)
                if extraResp.status_code == 200:
                    #combine the response body and return
                    tmpSpecs = []
                    tmpRespBody = extraResp.json()
                    self._convert_extra_specs(tmpSpecs, tmpRespBody['extra_specs'], True)

                    resp_body.update({"extraSpecs":tmpSpecs})
                else:
                    #rollback
                    self._delete_flavor(self, request, vimid, tenantid, flavorid)
                    return extraResp

            VimDriverUtils.replace_key_by_mapping(resp_body, self.keys_mapping)
            vim_dict = {
                "vimName": vim["name"],
                "vimId": vim["vimId"],
                "tenantId": tenantid,
                 "returnCode": 1,
            }
            resp_body.update(vim_dict)
            return Response(data=resp_body, status=resp.status_code)
        except VimDriverNewtonException as e:
            logger.error("response with status = %s" % e.status_code)
            if sess and resp and resp.status_code == 200:
                self._delete_flavor(sess, flavorid)

            return Response(data={'error': e.content}, status=e.status_code)
        except HttpError as e:
            logger.error("HttpError: status:%s, response:%s" % (e.http_status, e.response.json()))
            return Response(data=e.response.json(), status=e.http_status)
        except Exception as e:
            logger.error(traceback.format_exc())

            if sess and resp and resp.status_code == 200:
                self._delete_flavor(sess, flavorid)

            return Response(data={'error': str(e)},
                            status=status.HTTP_500_INTERNAL_SERVER_ERROR)

    def _create_flavor(self, sess, request):
        # prepare request resource to vim instance
        req_resouce = "/flavors"

        flavor = request.data

        VimDriverUtils.replace_key_by_mapping(flavor,
                                              self.keys_mapping, True)
        req_body = json.JSONEncoder().encode({"flavor": flavor})

        logger.info("making request with URI:%s" % req_resouce)
        logger.debug("with data:%s" % req_body)

        resp = sess.post(req_resouce, data=req_body,
                         endpoint_filter=self.service)

        logger.info("request returns with status %s" % resp.status_code)

        return resp

    def _create_flavor_extra_specs(self, sess, extraspecs, flavorid):
        # prepare request resource to vim instance
        req_resouce = "/flavors"
        if flavorid:
            req_resouce += "/%s/os-extra_specs" % flavorid
        else:
            raise VimDriverNewtonException(message="VIM newton exception",
                       content="internal bug in creating flavor extra specs",
                       status_code=500)

        req_body = json.JSONEncoder().encode({"extra_specs": extraspecs})

        logger.info("making request with URI:%s" % req_resouce)
        logger.debug("with data:%s" % req_body)

        resp = sess.post(req_resouce, data=req_body,
                         endpoint_filter=self.service)

        logger.info("request returns with status %s" % resp.status_code)

        return resp

    def delete(self, request, vimid="", tenantid="", flavorid=""):
        logger.info("vimid, tenantid, flavorid = %s,%s,%s" % (vimid, tenantid, flavorid))
        if request.data:
            logger.debug("With data = %s" % request.data)
            pass

        try:
            # prepare request resource to vim instance
            vim = VimDriverUtils.get_vim_info(vimid)
            sess = VimDriverUtils.get_session(vim, tenantid)

            #delete extra specs one by one
            resp = self._delete_flavor_extra_specs(sess, flavorid)

            #delete flavor
            resp = self._delete_flavor(sess, flavorid)

            #return results
            return Response(status=resp.status_code)
        except VimDriverNewtonException as e:
            logger.error("response with status = %s" % e.status_code)
            return Response(data={'error': e.content}, status=e.status_code)
        except HttpError as e:
            logger.error("HttpError: status:%s, response:%s" % (e.http_status, e.response.json()))
            return Response(data=e.response.json(), status=e.http_status)
        except Exception as e:
            logger.error(traceback.format_exc())
            return Response(data={'error': str(e)},
                            status=status.HTTP_500_INTERNAL_SERVER_ERROR)

    def _delete_flavor_extra_specs(self, sess, flavorid):
        #delete extra specs one by one
        resp = self._get_flavor_extra_specs(sess, flavorid)
        extra_specs = resp.json()
        if extra_specs and extra_specs["extra_specs"]:
            for k, _ in extra_specs["extra_specs"].items():
                # just try to delete extra spec, but do not care if succeeded
                self._delete_flavor_one_extra_spec(sess, flavorid, k)
        return resp

    def _delete_flavor_one_extra_spec(self, sess, flavorid, extra_spec_key):
        # prepare request resource to vim instance
        try:
            req_resouce = "/flavors"
            if flavorid and extra_spec_key:
                req_resouce += "/%s" % flavorid
                req_resouce += "/os-extra_specs/%s" % extra_spec_key
            else:
                raise VimDriverNewtonException(message="VIM newton exception",
                       content="internal bug in deleting flavor extra specs: %s" % extra_spec_key,
                       status_code=status.HTTP_500_INTERNAL_SERVER_ERROR)

            logger.info("making request with URI:%s" % req_resouce)

            resp = sess.delete(req_resouce, endpoint_filter=self.service)

            logger.info("request returns with status %s" % resp.status_code)

            return resp

        except HttpError as e:
            logger.error("HttpError: status:%s, response:%s" % (e.http_status, e.response.json()))
            return Response(data=e.response.json(), status=e.http_status)
        except Exception as e:
            logger.error(traceback.format_exc())
            return Response(data={'error': str(e)},
                            status=status.HTTP_500_INTERNAL_SERVER_ERROR)

    def _delete_flavor(self, sess, flavorid):
        # prepare request resource to vim instance
        req_resouce = "/flavors"
        if flavorid:
            req_resouce += "/%s" % flavorid
        else:
            raise VimDriverNewtonException(message="VIM newton exception",
                   content="internal bug in deleting flavor",
                   status_code=500)

        logger.info("making request with URI:%s" % req_resouce)

        resp = sess.delete(req_resouce, endpoint_filter=self.service)

        logger.info("request returns with status %s" % resp.status_code)

        return resp