# Copyright (C) 2018 Verizon. All Rights Reserved.
#
# 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 traceback

from drf_yasg.utils import swagger_auto_schema
from rest_framework import status
from rest_framework.response import Response
from rest_framework.views import APIView

from lcm.nf.biz.heal_vnf import HealVnf
from lcm.nf.serializers.heal_vnf_req import HealVnfRequestSerializer
from lcm.nf.serializers.response import ProblemDetailsSerializer
from lcm.pub.exceptions import NFLCMException, NFLCMExceptionNotFound, NFLCMExceptionConflict
from lcm.pub.utils.jobutil import JobUtil
from lcm.pub.database.models import NfInstModel
from lcm.nf.const import VNF_STATUS

logger = logging.getLogger(__name__)


class HealVnfView(APIView):
    @swagger_auto_schema(
        request_body=HealVnfRequestSerializer(),
        responses={
            status.HTTP_202_ACCEPTED: "Success",
            status.HTTP_404_NOT_FOUND: ProblemDetailsSerializer(),
            status.HTTP_409_CONFLICT: ProblemDetailsSerializer(),
            status.HTTP_500_INTERNAL_SERVER_ERROR: "Internal error"
        }
    )
    def post(self, request, instanceid):
        logger.debug("HealVnf--post::> %s" % request.data)
        try:
            heal_vnf_request_serializer = HealVnfRequestSerializer(data=request.data)
            if not heal_vnf_request_serializer.is_valid():
                raise NFLCMException(heal_vnf_request_serializer.errors)

            job_id = JobUtil.create_job('NF', 'HEAL', instanceid)
            JobUtil.add_job_status(job_id, 0, "HEAL_VNF_READY")
            self.heal_pre_check(instanceid, job_id)
            HealVnf(heal_vnf_request_serializer.data, instanceid, job_id).start()
            response = Response(data={"jobId": job_id}, status=status.HTTP_202_ACCEPTED)
            # todo, heal_vnf codes uses job as the status storage, not in VNFLcmOpOccModel.
            # response["Location"] = "/api/vnflcm/v1/vnf_lc_ops/%s" % lcmopoccid
            return response
        except NFLCMExceptionNotFound as e:
            probDetail = ProblemDetailsSerializer(data={"status": status.HTTP_404_NOT_FOUND, "detail": "VNF Instance not found"})
            resp_isvalid = probDetail.is_valid()
            if not resp_isvalid:
                raise NFLCMException(probDetail.errors)
            return Response(data=probDetail.data, status=status.HTTP_404_NOT_FOUND)
        except NFLCMExceptionConflict as e:
            probDetail = ProblemDetailsSerializer(data={"status": status.HTTP_409_CONFLICT, "detail": "VNF Instance not in Instantiated State"})
            resp_isvalid = probDetail.is_valid()
            if not resp_isvalid:
                    raise NFLCMException(probDetail.errors)
            return Response(data=probDetail.data, status=status.HTTP_409_CONFLICT)
        except NFLCMException as e:
            logger.error(e.message)
            return Response(data={'error': '%s' % e.message}, status=status.HTTP_500_INTERNAL_SERVER_ERROR)
        except Exception as e:
            logger.error(e.message)
            logger.error(traceback.format_exc())
            return Response(data={'error': 'unexpected exception'}, status=status.HTTP_500_INTERNAL_SERVER_ERROR)

    def heal_pre_check(self, nf_inst_id, job_id):
        vnf_insts = NfInstModel.objects.filter(nfinstid=nf_inst_id)
        if not vnf_insts.exists():
            raise NFLCMExceptionNotFound("VNF nf_inst_id does not exist.")

        if vnf_insts[0].status != 'INSTANTIATED':
            raise NFLCMExceptionConflict("VNF instantiationState is not INSTANTIATED.")

        NfInstModel.objects.filter(nfinstid=nf_inst_id).update(status=VNF_STATUS.HEALING)
        JobUtil.add_job_status(job_id, 15, 'Nf healing pre-check finish')
        logger.info("Nf healing pre-check finish")