/*******************************************************************************
 *  ============LICENSE_START=======================================================
 *  slice-analysis-ms
 *  ================================================================================
 *   Copyright (C) 2020-2021 Wipro Limited.
 *   Copyright (C) 2022 Huawei Canada Limited.
 *   Copyright (C) 2022 CTC, 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.
 *     ============LICENSE_END=========================================================
 *
 *******************************************************************************/

package org.onap.slice.analysis.ms.service;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.UUID;

import javax.annotation.PostConstruct;

import org.onap.slice.analysis.ms.dmaap.PolicyDmaapClient;
import org.onap.slice.analysis.ms.models.Configuration;
import org.onap.slice.analysis.ms.models.policy.AAI;
import org.onap.slice.analysis.ms.models.policy.AdditionalProperties;
import org.onap.slice.analysis.ms.models.policy.OnsetMessage;
import org.onap.slice.analysis.ms.models.policy.Payload;
import org.onap.slice.analysis.ms.models.policy.Sla;
import org.onap.slice.analysis.ms.models.policy.TransportNetwork;
import org.onap.slice.analysis.ms.service.ccvpn.RequestOwner;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.stereotype.Component;

import com.fasterxml.jackson.databind.ObjectMapper;

/**
 * Serivce to generate and publish onsetMessage to ONAP/Policy
 */
@Component
public class PolicyService {
    private final static int SERVICE_RATE_INTERVAL = 5000; // in ms
    private PolicyDmaapClient policyDmaapClient;
    private static Logger log = LoggerFactory.getLogger(PolicyService.class);
    private ObjectMapper objectMapper = new ObjectMapper();
    private RateLimiter rateLimiter;

    /**
     * Initialization
     */
    @PostConstruct
    public void init() {
        Configuration configuration = Configuration.getInstance();
        policyDmaapClient = new PolicyDmaapClient(configuration);
        rateLimiter = new RateLimiter(1, SERVICE_RATE_INTERVAL);
    }

    protected <T> OnsetMessage formPolicyOnsetMessage(String snssai, AdditionalProperties<T> addProps, Map<String, String> serviceDetails) {
        OnsetMessage onsetmsg = new OnsetMessage();
        Payload payload = new Payload();
        payload.setGlobalSubscriberId(serviceDetails.get("globalSubscriberId"));
        payload.setSubscriptionServiceType(serviceDetails.get("subscriptionServiceType"));
        payload.setNetworkType("AN");
        payload.setName(serviceDetails.get("ranNFNSSIId"));
        payload.setServiceInstanceID(serviceDetails.get("ranNFNSSIId"));

        addProps.setModifyAction("");
        Map<String, String> nsiInfo = new HashMap<>();
        nsiInfo.put("nsiId", UUID.randomUUID().toString());
        nsiInfo.put("nsiName", "");
        addProps.setNsiInfo(nsiInfo);
        addProps.setScriptName("AN");
        addProps.setSliceProfileId(serviceDetails.get("sliceProfileId"));
        addProps.setModifyAction("reconfigure");
        List<String> snssaiList = new ArrayList<>();
        snssaiList.add(snssai);
        addProps.setSnssaiList(snssaiList);

        payload.setAdditionalProperties(addProps);
        try {
            onsetmsg.setPayload(objectMapper.writeValueAsString(payload));
        } catch (Exception e) {
            log.error("Error while mapping payload as string , {}",e.getMessage());
        }

        onsetmsg.setClosedLoopControlName("ControlLoop-Slicing-116d7b00-dbeb-4d03-8719-d0a658fa735b");
        onsetmsg.setClosedLoopAlarmStart(System.currentTimeMillis());
        onsetmsg.setClosedLoopEventClient("microservice.sliceAnalysisMS");
        onsetmsg.setClosedLoopEventStatus("ONSET");
        onsetmsg.setRequestID(UUID.randomUUID().toString());
        onsetmsg.setTarget("generic-vnf.vnf-id");
        onsetmsg.setTargetType("VNF");
        onsetmsg.setFrom("DCAE");
        onsetmsg.setVersion("1.0.2");
        AAI aai = new AAI();
        aai.setVserverIsClosedLoopDisabled("false");
        aai.setVserverProvStatus("ACTIVE");
        aai.setGenericVnfVNFId(serviceDetails.get("ranNFNSSIId"));
        onsetmsg.setAai(aai);
        return onsetmsg;
    }

    protected <T> void sendOnsetMessageToPolicy(String snssai, AdditionalProperties<T> addProps, Map<String, String> serviceDetails) {
        OnsetMessage onsetMessage = formPolicyOnsetMessage(snssai, addProps, serviceDetails);
        String msg =  "";
        try {
            msg = objectMapper.writeValueAsString(onsetMessage);
            log.info("Policy onset message for S-NSSAI: {} is {}", snssai, msg);
            policyDmaapClient.sendNotificationToPolicy(msg);
        }
        catch (Exception e) {
            log.error("Error sending notification to policy, {}",e.getMessage());
        }
    }

    /**
     * Generate onsetMessage for ccvpn service update operation
     * @param cllId cloud leased line Id (ethernet service id)
     * @param newBw new bandwidth value for bandwidth adjustment
     * @param <T> type for additionalPropert
     *           ies, can be omitted
     * @return OnsetMessage result
     */
    public <T> OnsetMessage formPolicyOnsetMessageForCCVPN(String cllId, Integer newBw, RequestOwner owner) {
        Sla sla = new Sla(2, newBw);
        String transportNetworkId = cllId;
        if (owner == RequestOwner.UUI) {
            transportNetworkId += "-network-001";
        } else if (owner == RequestOwner.DCAE) {
            transportNetworkId += "-network-002";
        }
        TransportNetwork transportNetwork = new TransportNetwork(transportNetworkId, sla);
        AdditionalProperties additionalProperties = new AdditionalProperties();
        additionalProperties.setModifyAction("bandwidth");
        additionalProperties.setEnableSdnc("true");
        List<TransportNetwork> transportNetworks = new ArrayList();
        transportNetworks.add(transportNetwork);
        additionalProperties.setTransportNetworks(transportNetworks);

        Payload payload = new Payload();
        payload.setGlobalSubscriberId("IBNCustomer");
        payload.setSubscriptionServiceType("IBN");
        payload.setServiceType("CLL");
        payload.setName("cloud-leased-line-101");
        payload.setServiceInstanceID(cllId);
        payload.setAdditionalProperties(additionalProperties);
        payload.setModelInvariantUuid("6790ab0e-034f-11eb-adc1-0242ac120002");
        payload.setModelUuid("6790ab0e-034f-11eb-adc1-0242ac120002");

        OnsetMessage onsetmsg = new OnsetMessage();
        try {
            onsetmsg.setPayload(objectMapper.writeValueAsString(payload));
        } catch (Exception e) {
            log.error("Error while mapping payload as string , {}",e.getMessage());
        }
        onsetmsg.setClosedLoopControlName("ControlLoop-CCVPN-CLL-227e8b00-dbeb-4d03-8719-d0a658fb846c");
        onsetmsg.setClosedLoopAlarmStart(System.currentTimeMillis());
        onsetmsg.setClosedLoopEventClient("microservice.sliceAnalysisMS");
        onsetmsg.setClosedLoopEventStatus("ONSET");
        onsetmsg.setRequestID(UUID.randomUUID().toString());
        onsetmsg.setTarget("generic-vnf.vnf-id");
        onsetmsg.setTargetType("VNF");
        onsetmsg.setFrom("DCAE");
        onsetmsg.setVersion("1.0.2");
        AAI aai = new AAI();
        aai.setGenericVnfIsClosedLoopDisabled("false");
        aai.setGenericVnfProvStatus("ACTIVE");
        aai.setGenericVnfVNFId("00000000-0000-0000-0000-000000000000");
        aai.setGenericVnfVnfName("00000");
        onsetmsg.setAai(aai);
        return onsetmsg;
    }

    /**
     * Sending the onsetMessage to Onap-Policy through PolicyDmaapClient
     * @param onsetMessage the onsetMessage about to send
     * @param <T> type inherent from previous implementation can be omitted
     */
    public <T> void sendOnsetMessageToPolicy(OnsetMessage onsetMessage){
        String msg =  "";
        try {
            msg = objectMapper.writeValueAsString(onsetMessage);
            rateLimiter.getToken();
            log.info("Sending onset message to Onap/Policy for ControlLoop-CCVPN-CLL, the msg: {}", msg);
            policyDmaapClient.sendNotificationToPolicy(msg);
        }
        catch (Exception e) {
            log.error("Error sending notification to policy, {}",e.getMessage());
        }
    }
}