# Copyright 2017 ZTE Corporation.
#
# 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 json
import logging

import traceback
import sys
import os

from catalog.pub.database.models import NSDModel, NfPackageModel
from catalog.pub.utils.values import ignore_case_get
from catalog.pub.exceptions import NSLCMException
from catalog.pub.msapi import sdc
from catalog.pub.config.config import CATALOG_ROOT_PATH
from catalog.pub.utils import toscaparser
from catalog.pub.utils import fileutil
from catalog.pub.msapi import nfvolcm

logger = logging.getLogger(__name__)

STATUS_SUCCESS, STATUS_FAILED = "success", "failed"


def fmt_ns_pkg_rsp(status, desc, error_code="500"):
    return [0, {"status": status, "statusDescription": desc, "errorCode": error_code}]


def ns_on_distribute(csar_id):
    ret = None
    try:
        ret = NsPackage().on_distribute(csar_id)
    except NSLCMException as e:
        NsPackage().delete_catalog(csar_id)
        return fmt_ns_pkg_rsp(STATUS_FAILED, e.message)
    except:
        logger.error(traceback.format_exc())
        NsPackage().delete_catalog(csar_id)
        return fmt_ns_pkg_rsp(STATUS_FAILED, str(sys.exc_info()))
    return fmt_ns_pkg_rsp(STATUS_SUCCESS, ret[1], "")


def ns_delete_csar(csar_id, force_delete):
    ret = None
    try:
        ret = NsPackage().delete_csar(csar_id, force_delete)
    except NSLCMException as e:
        return fmt_ns_pkg_rsp(STATUS_FAILED, e.message)
    except:
        logger.error(traceback.format_exc())
        return fmt_ns_pkg_rsp(STATUS_FAILED, str(sys.exc_info()))
    return fmt_ns_pkg_rsp(STATUS_SUCCESS, ret[1], "")

def ns_get_csars():
    ret = None
    try:
        ret = NsPackage().get_csars()
    except NSLCMException as e:
        return [1, e.message]
    except:
        logger.error(traceback.format_exc())
        return [1, str(sys.exc_info())]
    return ret

def ns_get_csar(csar_id):
    ret = None
    try:
        ret = NsPackage().get_csar(csar_id)
    except NSLCMException as e:
        return [1, e.message]
    except:
        logger.error(traceback.format_exc())
        return [1, str(sys.exc_info())]
    return ret


###############################################################################################################

class NsPackage(object):
    """
    Actions for sdc ns package.
    """

    def __init__(self):
        pass

    def on_distribute(self, csar_id):
        if NSDModel.objects.filter(id=csar_id):
            raise NSLCMException("NS CSAR(%s) already exists." % csar_id)

        nsd,local_file_name,nsd_json = self.get_nsd(csar_id)

        nsd_id = nsd["metadata"]["id"]
        if NSDModel.objects.filter(nsd_id=nsd_id):
            raise NSLCMException("NSD(%s) already exists." % nsd_id)

        for vnf in nsd["vnfs"]:
            vnfd_id = vnf["properties"]["id"]
            pkg = NfPackageModel.objects.filter(vnfdid=vnfd_id)
            if not pkg:
                raise NSLCMException("VNF package(%s) is not distributed." % vnfd_id)

        NSDModel(
            id=csar_id,
            nsd_id=nsd_id,
            name=nsd["metadata"].get("name", nsd_id),
            vendor=nsd["metadata"].get("vendor", "undefined"),
            description=nsd["metadata"].get("description", ""),
            version=nsd["metadata"].get("version", "undefined"),
            nsd_path=local_file_name,
            nsd_model=nsd_json).save()

        return [0, "CSAR(%s) distributed successfully." % csar_id]

    def get_nsd(self, csar_id):
        artifact = sdc.get_artifact(sdc.ASSETTYPE_SERVICES, csar_id)
        local_path = os.path.join(CATALOG_ROOT_PATH, csar_id)
        local_file_name = sdc.download_artifacts(artifact["toscaModelURL"], local_path)

        nsd_json = toscaparser.parse_nsd(local_file_name)
        nsd = json.JSONDecoder().decode(nsd_json)

        return nsd,local_file_name,nsd_json

    def delete_csar(self, csar_id, force_delete):
        '''
        if force_delete:
            NSInstModel.objects.filter(nspackage_id=csar_id).delete()
        else:
            if NSInstModel.objects.filter(nspackage_id=csar_id):
                raise NSLCMException("CSAR(%s) is in using, cannot be deleted." % csar_id)
        '''
        nfvolcm.delete_ns_inst_mock()
        NSDModel.objects.filter(id=csar_id).delete()
        return [0, "Delete CSAR(%s) successfully." % csar_id]


    def get_csars(self):
        csars = []
        nss = NSDModel.objects.filter()
        for ns in nss:
            csars.append({
                "csarId": ns.id,
                "nsdId": ns.nsd_id
            })
        return [0, csars]

    def get_csar(self, csar_id):
        package_info = {}
        csars = NSDModel.objects.filter(id=csar_id)
        if csars:
            package_info["nsdId"] = csars[0].nsd_id
            package_info["nsdProvider"] = csars[0].vendor
            package_info["nsdVersion"] = csars[0].version

        #nss = NSInstModel.objects.filter(nspackage_id=csar_id)
        nss = nfvolcm.getNsInsts_mock()
        ns_instance_info = [{
            "nsInstanceId": ns.id, 
            "nsInstanceName": ns.name} for ns in nss]

        return [0, {"csarId": csar_id, 
            "packageInfo": package_info, 
            "nsInstanceInfo": ns_instance_info}]

    def delete_catalog(self, csar_id):
        local_path = os.path.join(CATALOG_ROOT_PATH, csar_id)
        fileutil.delete_dirs(local_path)