package org.onap.nbi.apis.serviceinventory;

import java.util.ArrayList;
import java.util.LinkedHashMap;
import java.util.List;
import org.onap.nbi.apis.serviceinventory.jolt.FindServiceInventoryJsonTransformer;
import org.onap.nbi.apis.serviceinventory.jolt.GetServiceInventoryJsonTransformer;
import org.onap.nbi.exceptions.BackendFunctionalException;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.http.HttpStatus;
import org.springframework.stereotype.Service;
import org.springframework.util.MultiValueMap;
import org.springframework.util.StringUtils;

@Service
public class ServiceInventoryService {

    @Autowired
    NbiClient nbiClient;

    @Autowired
    AaiClient aaiClient;

    @Autowired
    GetServiceInventoryJsonTransformer getServiceInventoryJsonTransformer;

    @Autowired
    FindServiceInventoryJsonTransformer findServiceInventoryJsonTransformer;

    public LinkedHashMap get(String serviceId, MultiValueMap<String, String> params) {

        String clientId = params.getFirst("relatedParty.id");
        String serviceSpecId = params.getFirst("serviceSpecification.id");
        String serviceSpecName = params.getFirst("serviceSpecification.name");

        if (StringUtils.isEmpty(serviceSpecId) && StringUtils.isEmpty(serviceSpecName)) {
            throw new BackendFunctionalException(HttpStatus.NOT_FOUND,
                    "serviceSpecName or serviceSpecId must be provided");
        }

        String customerId = getCustomerId(clientId);
        String serviceName = getServiceName(serviceSpecName, serviceSpecId);
        LinkedHashMap serviceResponse = aaiClient.getCatalogService(customerId, serviceName, serviceId);

        if (serviceResponse != null) {
            addVnfsToResponse(serviceResponse);
            LinkedHashMap serviceInventoryResponse =
                    (LinkedHashMap) getServiceInventoryJsonTransformer.transform(serviceResponse);
            addRelatedPartyId(customerId, serviceInventoryResponse);
            return serviceInventoryResponse;
        } else {
            throw new BackendFunctionalException(HttpStatus.NOT_FOUND, "no catalog service found");
        }

    }


    private String getCustomerId(String clientId) {

        if (StringUtils.isEmpty(clientId)) {
            return "generic";
        } else {
            return clientId;
        }

    }

    private String getServiceName(String serviceSpecificationName, String serviceSpecificationId) {

        if (StringUtils.isEmpty(serviceSpecificationName)) {
            LinkedHashMap serviceSpecification = nbiClient.getServiceSpecification(serviceSpecificationId);
            return (String) serviceSpecification.get("name");
        } else {
            return serviceSpecificationName;
        }

    }

    private void addRelatedPartyId(String customerId, LinkedHashMap serviceInventoryResponse) {

        LinkedHashMap relatedParty = (LinkedHashMap) serviceInventoryResponse.get("relatedParty");
        relatedParty.put("id", customerId);

    }

    private void addVnfsToResponse(LinkedHashMap serviceResponse) {

        List<LinkedHashMap> vnfs = new ArrayList<>();
        LinkedHashMap relationShip = (LinkedHashMap) serviceResponse.get("relationship-list");
        List<LinkedHashMap> relationsList = (List<LinkedHashMap>) relationShip.get("relationship");
        for (LinkedHashMap relation : relationsList) {
            String relatedLink = (String) relation.get("related-link");
            LinkedHashMap vnf = aaiClient.getVNF(relatedLink);
            if (vnf != null) {
                vnfs.add(vnf);
            }
        }
        serviceResponse.put("vnfs", vnfs);

    }


    public List<LinkedHashMap> find(MultiValueMap<String, String> params) {

        String clientId = params.getFirst("relatedParty.id");
        String serviceSpecId = params.getFirst("serviceSpecification.id");
        String serviceSpecName = params.getFirst("serviceSpecification.name");
        String customerId = getCustomerId(clientId);
        String serviceName;
        List<LinkedHashMap> serviceInstances = new ArrayList<>();
        if (StringUtils.isEmpty(serviceSpecId) && StringUtils.isEmpty(serviceSpecName)) {
            LinkedHashMap servicesInAaiForCustomer = aaiClient.getServicesInAaiForCustomer(customerId);
            List<LinkedHashMap> servicesInAAI =
                    (List<LinkedHashMap>) servicesInAaiForCustomer.get("service-subscription");
            for (LinkedHashMap service : servicesInAAI) {
                String serviceType = (String) service.get("service-type");
                buildServiceInstances(serviceInstances, customerId, serviceType);
            }
        } else {
            serviceName = getServiceName(serviceSpecName, serviceSpecId);
            buildServiceInstances(serviceInstances, customerId, serviceName);
        }

        List<LinkedHashMap> serviceInventoryResponse =
                (List<LinkedHashMap>) findServiceInventoryJsonTransformer.transform(serviceInstances);
        for (LinkedHashMap serviceInventory : serviceInventoryResponse) {
            LinkedHashMap party = (LinkedHashMap) serviceInventory.get("relatedParty");
            party.put("id", customerId);
        }
        return serviceInventoryResponse;

    }

    private void buildServiceInstances(List<LinkedHashMap> serviceInstances, String customerId, String serviceType) {

        LinkedHashMap serviceInstancesInAaiForCustomer =
                aaiClient.getServiceInstancesInAaiForCustomer(customerId, serviceType);
        List<LinkedHashMap> serviceInstancesForServiceType =
                (List<LinkedHashMap>) serviceInstancesInAaiForCustomer.get("service-instance");

        // add service type for jolt
        for (LinkedHashMap serviceInstanceForServiceType : serviceInstancesForServiceType) {
            serviceInstanceForServiceType.put("service-type", serviceType);
        }
        serviceInstances.addAll(serviceInstancesForServiceType);

    }

}