/*-
 * ============LICENSE_START=======================================================
 * ONAP CLAMP
 * ================================================================================
 * Copyright (C) 2017 AT&T Intellectual Property. 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.
 * ============LICENSE_END============================================
 * ===================================================================
 * ECOMP is a trademark and service mark of AT&T Intellectual Property.
 */

package org.onap.clamp.clds.client.req;

import java.io.UnsupportedEncodingException;
import java.net.URLEncoder;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.Iterator;
import java.util.List;
import java.util.Map;

import org.jboss.resteasy.spi.BadRequestException;
import org.onap.clamp.clds.model.prop.Global;
import org.onap.clamp.clds.model.prop.ModelProperties;
import org.onap.clamp.clds.model.prop.PolicyChain;
import org.onap.clamp.clds.model.prop.PolicyItem;
import org.onap.clamp.clds.model.prop.Tca;
import org.onap.clamp.clds.model.refprop.RefProp;
import org.onap.policy.controlloop.policy.OperationsAccumulateParams;
import org.onap.policy.api.AttributeType;
import org.onap.policy.asdc.Resource;
import org.onap.policy.asdc.ResourceType;
import org.onap.policy.asdc.Service;
import org.onap.policy.controlloop.policy.Policy;
import org.onap.policy.controlloop.policy.PolicyResult;
import org.onap.policy.controlloop.policy.Target;
import org.onap.policy.controlloop.policy.TargetType;
import org.onap.policy.controlloop.policy.builder.BuilderException;
import org.onap.policy.controlloop.policy.builder.ControlLoopPolicyBuilder;
import org.onap.policy.controlloop.policy.builder.Message;
import org.onap.policy.controlloop.policy.builder.Results;

import com.att.eelf.configuration.EELFLogger;
import com.att.eelf.configuration.EELFLogger.Level;
import com.att.eelf.configuration.EELFManager;
import com.fasterxml.jackson.databind.JsonNode;
import com.fasterxml.jackson.databind.node.ObjectNode;

/**
 * Construct an Operational Policy request given CLDS objects.
 */
public class OperationalPolicyReq {
    protected static final EELFLogger logger        = EELFManager.getInstance().getLogger(OperationalPolicyReq.class);
    protected static final EELFLogger metricsLogger = EELFManager.getInstance().getMetricsLogger();

    /**
     * Format Operational Policy attributes.
     *
     * @param refProp
     * @param prop
     * @return
     * @throws BuilderException
     * @throws UnsupportedEncodingException
     */
    public static Map<AttributeType, Map<String, String>> formatAttributes(RefProp refProp, ModelProperties prop,
            String modelElementId, PolicyChain policyChain) throws BuilderException, UnsupportedEncodingException {
        Global global = prop.getGlobal();
        prop.setCurrentModelElementId(modelElementId);
        prop.setPolicyUniqueId(policyChain.getPolicyId());

        String templateName = "";
        String operationTopic = "";
        String notificationTopic = "";
        String controller = "";
        Tca tca = prop.getTca();
        if (tca.isFound()) {
            if (!global.getActionSet().equalsIgnoreCase("enbRecipe")) {
                throw new BadRequestException(
                        "Operation Policy validation problem: action set is not selected properly.");
            }
            templateName = refProp.getStringValue("op.eNodeB.templateName", global.getService());
            operationTopic = refProp.getStringValue("op.eNodeB.operationTopic", global.getService());
            notificationTopic = refProp.getStringValue("op.eNodeB.notificationTopic", global.getService());
            controller = refProp.getStringValue("op.eNodeB.controller", global.getService());
        } else {
            if (!global.getActionSet().equalsIgnoreCase("vnfRecipe")) {
                throw new BadRequestException(
                        "Operation Policy validation problem: Action set is not selected properly.");
            }
            templateName = refProp.getStringValue("op.templateName", global.getService());
            operationTopic = refProp.getStringValue("op.operationTopic", global.getService());
            notificationTopic = refProp.getStringValue("op.notificationTopic", global.getService());
            controller = refProp.getStringValue("op.controller", global.getService());
        }
        String recipeTopic = refProp.getStringValue("op.recipeTopic", global.getService());

        // ruleAttributes
        Map<String, String> ruleAttributes = new HashMap<String, String>();

        if (operationTopic == null || operationTopic.length() == 0) {
            logger.info("templateName=" + templateName);
            logger.info("recipeTopic=" + recipeTopic);
            logger.info("notificationTopic=" + notificationTopic);

            // if no operationTopic, then don't format yaml - use first policy
            // from list
            PolicyItem policyItem = policyChain.getPolicyItems().get(0);

            ruleAttributes.put("templateName", templateName);
            ruleAttributes.put("ClosedLoopControlName", prop.getControlNameAndPolicyUniqueId());
            ruleAttributes.put("RecipeTopic", recipeTopic);
            ruleAttributes.put("NotificationTopic", notificationTopic);

            String recipe = policyItem.getRecipe();
            String maxRetries = String.valueOf(policyItem.getMaxRetries());
            String retryTimeLimit = String.valueOf(policyItem.getRetryTimeLimit());
            logger.info("recipe=" + recipe);
            logger.info("maxRetries=" + maxRetries);
            logger.info("retryTimeLimit=" + retryTimeLimit);
            ruleAttributes.put("Recipe", recipe);
            ruleAttributes.put("MaxRetries", maxRetries);
            ruleAttributes.put("RetryTimeLimit", retryTimeLimit);
        } else {
            logger.info("templateName=" + templateName);
            logger.info("operationTopic=" + operationTopic);
            logger.info("notificationTopic=" + notificationTopic);

            // format yaml
            String yaml = tca.isFound() ? formateNodeBYaml(refProp, prop, modelElementId, policyChain)
                    : formatYaml(refProp, prop, modelElementId, policyChain);

            ruleAttributes.put("templateName", templateName);
            ruleAttributes.put("ClosedLoopControlName", prop.getControlNameAndPolicyUniqueId());
            ruleAttributes.put("OperationTopic", operationTopic);
            ruleAttributes.put("NotificationTopic", notificationTopic);

            ruleAttributes.put("ControlLoopYaml", yaml);
        }

        // matchingAttributes
        Map<String, String> matchingAttributes = new HashMap<String, String>();
        matchingAttributes.put("controller", controller);

        Map<AttributeType, Map<String, String>> attributes = new HashMap<AttributeType, Map<String, String>>();
        attributes.put(AttributeType.RULE, ruleAttributes);
        attributes.put(AttributeType.MATCHING, matchingAttributes);

        return attributes;
    }

    /**
     * Format Operational Policy yaml.
     *
     * @param refProp
     * @param prop
     * @return
     * @throws BuilderException
     * @throws UnsupportedEncodingException
     */
    private static String formatYaml(RefProp refProp, ModelProperties prop, String modelElementId,
            PolicyChain policyChain) throws BuilderException, UnsupportedEncodingException {

        // get property objects
        Global global = prop.getGlobal();
        prop.setCurrentModelElementId(modelElementId);
        prop.setPolicyUniqueId(policyChain.getPolicyId());

        // convert values to SDC objects
        Service service = new Service(global.getService());
        Resource[] vfResources = convertToResource(global.getResourceVf(), ResourceType.VF);
        Resource[] vfcResources = convertToResource(global.getResourceVfc(), ResourceType.VFC);

        // create builder
        ControlLoopPolicyBuilder builder = ControlLoopPolicyBuilder.Factory.buildControlLoop(prop.getControlName(),
                policyChain.getTimeout(), service, vfResources);
        builder.addResource(vfcResources);

        // process each policy
        HashMap<String, org.onap.policy.controlloop.policy.Policy> policyObjMap = new HashMap<String, org.onap.policy.controlloop.policy.Policy>();
        List<PolicyItem> policyItemList = orderParentFirst(policyChain.getPolicyItems());
        Target target = new Target();
        target.setType(TargetType.VM);
        for (int i = 0; i < policyItemList.size(); i++) {

            org.onap.policy.controlloop.policy.Policy policyObj;
            PolicyItem policyItem = policyItemList.get(i);
            String policyName = policyItem.getRecipe() + " Policy";
            if (i == 0) {
                String policyDescription = policyItem.getRecipe()
                        + " Policy - the trigger (no parent) policy - created by CLDS";
                policyObj = builder.setTriggerPolicy(policyName, policyDescription,
                        refProp.getStringValue("op.policy.appc"), target, policyItem.getRecipe(), null,
                        policyItem.getMaxRetries(), policyItem.getRetryTimeLimit());
            } else {
                org.onap.policy.controlloop.policy.Policy parentPolicyObj = policyObjMap
                        .get(policyItem.getParentPolicy());
                String policyDescription = policyItem.getRecipe() + " Policy - triggered conditionally by "
                        + parentPolicyObj.getName() + " - created by CLDS";
                policyObj = builder.setPolicyForPolicyResult(policyName, policyDescription,
                        refProp.getStringValue("op.policy.appc"), target, policyItem.getRecipe(), null,
                        policyItem.getMaxRetries(), policyItem.getRetryTimeLimit(), parentPolicyObj.getId(),
                        convertToPolicyResult(policyItem.getParentPolicyConditions()));

                logger.info("policyObj.id=" + policyObj.getId() + "; parentPolicyObj.id=" + parentPolicyObj.getId());
            }
            policyObjMap.put(policyItem.getId(), policyObj);
        }

        //
        // Build the specification
        //
        Results results = builder.buildSpecification();
        if (results.isValid()) {
            logger.info("results.getSpecification()=" + results.getSpecification());
        } else {
            // throw exception with error info
            StringBuffer sb = new StringBuffer();
            sb.append("Operation Policy validation problem: ControlLoopPolicyBuilder failed with following messages: ");
            for (Message message : results.getMessages()) {
                sb.append(message.getMessage());
                sb.append("; ");
            }
            throw new BadRequestException(sb.toString());
        }

        String encodedYaml = URLEncoder.encode(results.getSpecification(), "UTF-8");

        return encodedYaml;
    }

    /**
     * Format Operational Policy yaml.
     *
     * @param refProp
     * @param prop
     * @return
     * @throws BuilderException
     * @throws UnsupportedEncodingException
     */
    private static String formateNodeBYaml(RefProp refProp, ModelProperties prop, String modelElementId,
            PolicyChain policyChain) throws BuilderException, UnsupportedEncodingException {

        // get property objects
        Global global = prop.getGlobal();
        prop.setCurrentModelElementId(modelElementId);
        prop.setPolicyUniqueId(policyChain.getPolicyId());

        // convert values to SDC objects
        Service service = new Service(global.getService());
        Resource[] vfResources = convertToResource(global.getResourceVf(), ResourceType.VF);
        Resource[] vfcResources = convertToResource(global.getResourceVfc(), ResourceType.VFC);

        // create builder
        ControlLoopPolicyBuilder builder = ControlLoopPolicyBuilder.Factory.buildControlLoop(prop.getControlName(),
                policyChain.getTimeout(), service, vfResources);
        builder.addResource(vfcResources);

        // process each policy
        HashMap<String, Policy> policyObjMap = new HashMap<String, Policy>();
        List<PolicyItem> policyItemList = addAOTSActorRecipe(refProp, global.getService(),
                policyChain.getPolicyItems());
        Target target = new Target();
        target.setType(TargetType.VM);
        Policy lastPolicyObj = new Policy();
		for (int i = 0; i < policyItemList.size(); i++) {
            org.onap.policy.controlloop.policy.Policy policyObj;
            PolicyItem policyItem = policyItemList.get(i);
            String policyName = policyItem.getRecipe() + " Policy";
            if (i == 0) {
            	//To set up time window payload for trigger policy
				Map<String, String> payloadMap = new HashMap<String, String>();
				payloadMap.put("timeWindow", refProp.getStringValue("op.eNodeB.timeWindow"));
				String policyDescription = policyItem.getRecipe()
                        + " Policy - the trigger (no parent) policy - created by CLDS";
                policyObj = builder.setTriggerPolicy(policyName, policyDescription, policyItem.getActor(), target,
                        policyItem.getRecipe(), payloadMap, policyItem.getMaxRetries(), policyItem.getRetryTimeLimit());
            } else {
                Policy parentPolicyObj = policyObjMap
                        .get(policyItem.getParentPolicy());
                String policyDescription = policyItem.getRecipe() + " Policy - triggered conditionally by "
                        + parentPolicyObj.getName() + " - created by CLDS";
                policyObj = builder.setPolicyForPolicyResult(policyName, policyDescription, policyItem.getActor(),
                        target, policyItem.getRecipe(), null, policyItem.getMaxRetries(),
                        policyItem.getRetryTimeLimit(), parentPolicyObj.getId(),
                        convertToPolicyResult(policyItem.getParentPolicyConditions()));
                lastPolicyObj = policyObj;
                logger.info("policyObj.id=" + policyObj.getId() + "; parentPolicyObj.id=" + parentPolicyObj.getId());
            }
            policyObjMap.put(policyItem.getId(), policyObj);
        }
		//To set up operations accumulate params
		OperationsAccumulateParams operationsAccumulateParams = new OperationsAccumulateParams();
		operationsAccumulateParams.setLimit(Integer.valueOf(refProp.getStringValue("op.eNodeB.limit")));
		operationsAccumulateParams.setPeriod(refProp.getStringValue("op.eNodeB.period"));
		builder.addOperationsAccumulateParams(lastPolicyObj.getId(), operationsAccumulateParams);
				
        //
        // Build the specification
        //
        Results results = builder.buildSpecification();
        if (results.isValid()) {
            logger.info("results.getSpecification()=" + results.getSpecification());
        } else {
            // throw exception with error info
            StringBuffer sb = new StringBuffer();
            sb.append("Operation Policy validation problem: ControlLoopPolicyBuilder failed with following messages: ");
            for (Message message : results.getMessages()) {
                sb.append(message.getMessage());
                sb.append("; ");
            }
            throw new BadRequestException(sb.toString());
        }

        String encodedYaml = URLEncoder.encode(results.getSpecification(), "UTF-8");

        return encodedYaml;
    }

    /**
     * Adding AOTS actor and other recipe for yaml
     *
     * @param inOrigList
     * @return
     */
    private static List<PolicyItem> addAOTSActorRecipe(RefProp refProp, String service, List<PolicyItem> inOrigList) {
        List<PolicyItem> outList = new ArrayList<PolicyItem>();
        try {
            PolicyItem policyItem = inOrigList.get(0);
            ObjectNode rootNode = (ObjectNode) refProp.getJsonTemplate("op.eNodeB.recipe", service);
            Iterator<JsonNode> itr = rootNode.get("eNodeBRecipes").elements();
            while (itr.hasNext()) {
                PolicyItem policyItemObj = (PolicyItem) policyItem.clone();
                JsonNode recipeNode = itr.next();
                policyItemObj.setId(recipeNode.path("Recipe").asText());
                policyItemObj.setActor(recipeNode.path("Actor").asText());
                policyItemObj.setRecipe(recipeNode.path("Recipe").asText());
                policyItemObj.setParentPolicy(recipeNode.path("ParentPolicy").asText());
                if (!recipeNode.path("Retry").asText().isEmpty()) {
                    policyItemObj.setMaxRetries(Integer.parseInt(recipeNode.path("Retry").asText()));
                }
                if (!recipeNode.path("TimeLimit").asText().isEmpty()) {
                    policyItemObj.setRetryTimeLimit(Integer.parseInt(recipeNode.path("TimeLimit").asText()));
                }
                if (!recipeNode.path("PPConditions").asText().isEmpty()) {
                    List<String> parentPolicyConditions = new ArrayList<String>();
                    for (String ppCondition : recipeNode.path("PPConditions").asText().split(",")) {
                        parentPolicyConditions.add(ppCondition);
                    }
                    policyItemObj.setParentPolicyConditions(parentPolicyConditions);
                }
                outList.add(policyItemObj);
            }
        } catch (Exception e) {
            logger.log(Level.ERROR, "Error", e);
        }
        return outList;
    }

    /**
     * Order list of PolicyItems so that parents come before any of their
     * children
     *
     * @param inOrigList
     * @return
     */
    private static List<PolicyItem> orderParentFirst(List<PolicyItem> inOrigList) {
        List<PolicyItem> inList = new ArrayList<>();
        inList.addAll(inOrigList);
        List<PolicyItem> outList = new ArrayList<>();
        int prevSize = 0;
        while (!inList.isEmpty()) {
            // check if there's a loop in the policy chain (the inList should
            // have been reduced by at least one)
            if (inList.size() == prevSize) {
                throw new BadRequestException("Operation Policy validation problem: loop in Operation Policy chain");
            }
            prevSize = inList.size();
            // the following loop should remove at least one PolicyItem from the
            // inList
            Iterator<PolicyItem> inListItr = inList.iterator();
            while (inListItr.hasNext()) {
                PolicyItem inItem = inListItr.next();
                // check for trigger policy (no parent)
                String parent = inItem.getParentPolicy();
                if (parent == null || parent.length() == 0) {
                    if (outList.size() > 0) {
                        throw new BadRequestException(
                                "Operation Policy validation problem: more than one trigger policy");
                    } else {
                        outList.add(inItem);
                        inListItr.remove();
                    }
                } else {
                    // check if this PolicyItem's parent has been processed
                    for (PolicyItem outItem : outList) {
                        if (outItem.getId().equals(parent)) {
                            // if the inItem parent is already in the outList,
                            // then add inItem to outList and remove from inList
                            outList.add(inItem);
                            inListItr.remove();
                            break;
                        }
                    }
                }
            }
        }
        return outList;
    }

    /**
     * Convert a List of resource strings to an array of Resource objects.
     *
     * @param stringList
     * @param resourceType
     * @return
     */
    private static Resource[] convertToResource(List<String> stringList, ResourceType resourceType) {
        int size = 0;
        if (stringList != null) {
            size = stringList.size();
        }
        Resource[] resourceArray = new Resource[size];
        for (int i = 0; i < size; i++) {
            String rString = stringList.get(i);
            resourceArray[i] = new Resource(rString, resourceType);
        }
        return resourceArray;
    }

    /**
     * Convert a List of policy result strings to an array of PolicyResult
     * objects.
     *
     * @param prList
     * @return
     */
    private static PolicyResult[] convertToPolicyResult(List<String> prList) {
        int size = 0;
        if (prList != null) {
            size = prList.size();
        }
        PolicyResult[] prArray = new PolicyResult[size];
        for (int i = 0; i < size; i++) {
            String prString = prList.get(i);
            prArray[i] = PolicyResult.toResult(prString);
        }
        return prArray;
    }

}