# 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 uuid

from lcm.pub.config.config import AAI_BASE_URL, AAI_USER, AAI_PASSWORD
from lcm.pub.exceptions import NFLCMException
from lcm.pub.utils import restcall

logger = logging.getLogger(__name__)


def call_aai(resource, method, data=''):
    additional_headers = {
        'X-FromAppId': 'VFC-GVNFM-VNFLCM',
        'X-TransactionId': str(uuid.uuid1())
    }
    return restcall.call_req(AAI_BASE_URL,
                    AAI_USER,
                    AAI_PASSWORD,
                    restcall.rest_no_auth,
                    resource,
                    method,
                    data,
                    additional_headers)


def create_customer_aai(global_customer_id, data):
    resource = "/business/customers/customer/%s" % global_customer_id
    ret = call_aai(resource, "PUT", data)
    if ret[0] != 0:
        logger.error("Status code is %s, detail is %s.", ret[2], ret[1])
        raise NFLCMException("Customer creation exception in AAI")
    return json.JSONDecoder().decode(ret[1]), ret[2]


def get_customer_aai(global_customer_id):
    resource = "/business/customers/customer/%s?depth=all" % global_customer_id
    ret = call_aai(resource, "GET")
    if ret[0] != 0:
        logger.error("Status code is %s, detail is %s.", ret[2], ret[1])
        raise NFLCMException("Get customer info exception in AAI")
    return json.JSONDecoder().decode(ret[1])


def delete_customer_aai(global_customer_id, resource_version=""):
    resource = "/business/customers/customer/%s" % global_customer_id
    if resource_version:
        resource = resource + "?resource-version=%s" % resource_version
    ret = call_aai(resource, "DELETE")
    if ret[0] != 0:
        logger.error("Status code is %s, detail is %s.", ret[2], ret[1])
        raise NFLCMException("Customer delete exception in AAI")
    return json.JSONDecoder().decode(ret[1]), ret[2]


def put_customer_relationship(global_customer_id, data):
    resource = "/business/customers/customer/{global-customer-id}/relationship-list/relationship" % global_customer_id
    ret = call_aai(resource, "PUT", data)
    if ret[0] != 0:
        logger.error("Status code is %s, detail is %s.", ret[2], ret[1])
        raise NFLCMException("Put or update customer relationship exception in AAI")
    return json.JSONDecoder().decode(ret[1])


def delete_customer_relationship(global_customer_id):
    resource = "/business/customers/customer/{global-customer-id}/relationship-list/relationship" % global_customer_id
    ret = call_aai(resource, "DELETE")
    if ret[0] != 0:
        logger.error("Status code is %s, detail is %s.", ret[2], ret[1])
        raise NFLCMException("Delete customer relationship exception in AAI")
    return json.JSONDecoder().decode(ret[1])


def create_ns_aai(global_customer_id, service_type, service_instance_id, data):
    resource = "/business/customers/customer/%s/service-subscriptions/service-subscription/" \
               "%s/service-instances/service-instance/%s" % \
               (global_customer_id, service_type, service_instance_id)
    ret = call_aai(resource, "PUT", data)
    if ret[0] != 0:
        logger.error("Status code is %s, detail is %s.", ret[2], ret[1])
        raise NFLCMException("Ns instance creation exception in AAI")
    return json.JSONDecoder().decode(ret[1]), ret[2]


def query_ns_aai(global_customer_id, service_type, service_instance_id):
    resource = "/business/customers/customer/%s/service-subscriptions/service-subscription/" \
               "%s/service-instances/service-instance/%s?depth=all" % \
               (global_customer_id, service_type, service_instance_id)
    ret = call_aai(resource, "GET")
    if ret[0] != 0:
        logger.error("Status code is %s, detail is %s.", ret[2], ret[1])
        raise NFLCMException("Ns instance query exception in AAI")
    return json.JSONDecoder().decode(ret[1])


def delete_ns_aai(global_customer_id, service_type, service_instance_id, resource_version=""):
    resource = "/business/customers/customer/%s/service-subscriptions/service-subscription/" \
               "%s/service-instances/service-instance/%s" % \
               (global_customer_id, service_type, service_instance_id)
    if resource_version:
        resource = resource + "?resource-version=%s" % resource_version
    ret = call_aai(resource, "DELETE")
    if ret[0] != 0:
        logger.error("Status code is %s, detail is %s.", ret[2], ret[1])
        raise NFLCMException("Ns instance delete exception in AAI")
    return json.JSONDecoder().decode(ret[1]), ret[2]


def put_ns_relationship(global_customer_id, service_type, service_instance_id, data):
    resource = "/business/customers/customer/%s/service-subscriptions/service-subscription/" \
               "%s/service-instances/service-instance/%s/relationship-list/relationship" % \
               (global_customer_id, service_type, service_instance_id)
    ret = call_aai(resource, "PUT", data)
    if ret[0] != 0:
        logger.error("Status code is %s, detail is %s.", ret[2], ret[1])
        raise NFLCMException("Put or update ns instance relationship exception in AAI")
    return json.JSONDecoder().decode(ret[1])


def delete_ns_relationship(global_customer_id, service_type, service_instance_id):
    resource = "/business/customers/customer/%s/service-subscriptions/service-subscription/" \
               "%s/service-instances/service-instance/%s/relationship-list/relationship" % \
               (global_customer_id, service_type, service_instance_id)
    ret = call_aai(resource, "DELETE")
    if ret[0] != 0:
        logger.error("Status code is %s, detail is %s.", ret[2], ret[1])
        raise NFLCMException("Delete ns instance relationship exception in AAI")
    return json.JSONDecoder().decode(ret[1])


def create_vnf_aai(vnf_id, data):
    resource = "/network/generic-vnfs/generic-vnf/%s" % vnf_id
    ret = call_aai(resource, "PUT", data)
    if ret[0] != 0:
        logger.error("Status code is %s, detail is %s.", ret[2], ret[1])
        raise NFLCMException("Vnf instance creation exception in AAI")
    return json.JSONDecoder().decode(ret[1]), ret[2]


def query_vnf_aai(vnf_id):
    resource = "/network/generic-vnfs/generic-vnf/%s?depth=all" % vnf_id
    ret = call_aai(resource, "GET")
    if ret[0] != 0:
        logger.error("Status code is %s, detail is %s.", ret[2], ret[1])
        raise NFLCMException("Vnf instance query exception in AAI")
    return json.JSONDecoder().decode(ret[1])


def delete_vnf_aai(vnf_id, resource_version=""):
    resource = "/network/generic-vnfs/generic-vnf/%s" % vnf_id
    if resource_version:
        resource = resource + "?resource-version=%s" % resource_version
    ret = call_aai(resource, "DELETE")
    if ret[0] != 0:
        logger.error("Status code is %s, detail is %s.", ret[2], ret[1])
        raise NFLCMException("Vnf instance delete exception in AAI")
    return json.JSONDecoder().decode(ret[1]), ret[2]


def put_vnf_relationship(vnf_id, data):
    resource = "/network/generic-vnfs/generic-vnf/%s/relationship-list/relationship" % vnf_id
    ret = call_aai(resource, "PUT", data)
    if ret[0] != 0:
        logger.error("Status code is %s, detail is %s.", ret[2], ret[1])
        raise NFLCMException("Put or update vnf instance relationship exception in AAI")
    return json.JSONDecoder().decode(ret[1])


def delete_vnf_relationship(vnf_id):
    resource = "/network/generic-vnfs/generic-vnf/%s/relationship-list/relationship" % vnf_id
    ret = call_aai(resource, "DELETE")
    if ret[0] != 0:
        logger.error("Status code is %s, detail is %s.", ret[2], ret[1])
        raise NFLCMException("Delete vnf instance relationship exception in AAI")
    return json.JSONDecoder().decode(ret[1])


def create_vserver_aai(cloud_owner, cloud_region_id, tenant_id, vserver_id, data):
    resource = "/cloud-infrastructure/cloud-regions/cloud-region/%s/" \
               "%s/tenants/tenant/%s/vservers/vserver/%s" % \
               (cloud_owner, cloud_region_id, tenant_id, vserver_id)
    ret = call_aai(resource, "PUT", data)
    if ret[0] != 0:
        logger.error("Status code is %s, detail is %s.", ret[2], ret[1])
        raise NFLCMException("Vserver creation exception in AAI")
    return json.JSONDecoder().decode(ret[1]), ret[2]


def query_vserver_aai(cloud_owner, cloud_region_id, tenant_id, vserver_id):
    resource = "/cloud-infrastructure/cloud-regions/cloud-region/%s/" \
               "%s/tenants/tenant/%s/vservers/vserver/%s?depth=all" % \
               (cloud_owner, cloud_region_id, tenant_id, vserver_id)
    ret = call_aai(resource, "GET")
    if ret[0] != 0:
        logger.error("Status code is %s, detail is %s.", ret[2], ret[1])
        raise NFLCMException("Vserver query exception in AAI")
    return json.JSONDecoder().decode(ret[1])


def delete_vserver_aai(cloud_owner, cloud_region_id, tenant_id, vserver_id, resource_version=""):
    resource = "/cloud-infrastructure/cloud-regions/cloud-region/%s/" \
               "%s/tenants/tenant/%s/vservers/vserver/%s" % \
               (cloud_owner, cloud_region_id, tenant_id, vserver_id)
    if resource_version:
        resource = resource + "?resource-version=%s" % resource_version
    ret = call_aai(resource, "DELETE")
    if ret[0] != 0:
        logger.error("Status code is %s, detail is %s.", ret[2], ret[1])
        raise NFLCMException("Vserver delete exception in AAI")
    return json.JSONDecoder().decode(ret[1]), ret[2]


def put_vserver_relationship(cloud_owner, cloud_region_id, tenant_id, vserver_id, data):
    resource = "/cloud-infrastructure/cloud-regions/cloud-region/%s/" \
               "%s/tenants/tenant/%s/vservers/vserver/%s/relationship-list/relationship" % \
               (cloud_owner, cloud_region_id, tenant_id, vserver_id)
    ret = call_aai(resource, "PUT", data)
    if ret[0] != 0:
        logger.error("Status code is %s, detail is %s.", ret[2], ret[1])
        raise NFLCMException("Put or update vserver relationship exception in AAI")
    return json.JSONDecoder().decode(ret[1])


def delete_vserver_relationship(cloud_owner, cloud_region_id, tenant_id, vserver_id):
    resource = "/cloud-infrastructure/cloud-regions/cloud-region/%s/" \
               "%s/tenants/tenant/%s/vservers/vserver/%s/relationship-list/relationship" % \
               (cloud_owner, cloud_region_id, tenant_id, vserver_id)
    ret = call_aai(resource, "DELETE")
    if ret[0] != 0:
        logger.error("Status code is %s, detail is %s.", ret[2], ret[1])
        raise NFLCMException("Delete vserver relationship exception in AAI")
    return json.JSONDecoder().decode(ret[1])