/*-
 * ============LICENSE_START=======================================================
 * SDC
 * ================================================================================
 * Copyright (C) 2019 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=========================================================
 */

package org.openecomp.sdc.be.datamodel.utils;

import com.google.common.collect.ImmutableMap;
import com.google.common.collect.ImmutableSet;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.Set;
import org.openecomp.sdc.be.model.tosca.constraints.ConstraintType;
import org.openecomp.sdc.be.ui.model.UIConstraint;
import org.openecomp.sdc.tosca.datatypes.ToscaFunctions;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.yaml.snakeyaml.Yaml;

public class ConstraintConvertor {

    private static final Logger logger = LoggerFactory.getLogger(ConstraintConvertor.class);

    public static final String EQUAL_OPERATOR = ConstraintType.EQUAL.getTypes().get(1);
    public static final String GREATER_THAN_OPERATOR = ConstraintType.GREATER_THAN.getTypes().get(1);
    public static final String LESS_THAN_OPERATOR = ConstraintType.LESS_THAN.getTypes().get(1);
    public static final String STATIC_CONSTRAINT = "static";
    public static final String PROPERTY_CONSTRAINT = "property";
    public static final String SERVICE_INPUT_CONSTRAINT = "service_input";
    public static final String SELF = "SELF";
    private static Set<String> SUPPORTED_CONSTRAINT_LIST =
            ImmutableSet.of(EQUAL_OPERATOR, GREATER_THAN_OPERATOR, LESS_THAN_OPERATOR);

    private static Set<String> SUPPORTED_FUNCTIONS =
            ImmutableSet.of(ToscaFunctions.GET_INPUT.getFunctionName(), ToscaFunctions.GET_PROPERTY.getFunctionName());

    public UIConstraint convert(String inConstraint) {
        Yaml yamlSource = new Yaml();
        UIConstraint uiConstraint = new UIConstraint();
        Object content1 = yamlSource.load(inConstraint);
        if (!(content1 instanceof Map)) {
            return null;
        }
        Map map1 = (Map) content1;
        Object key = map1.keySet().iterator().next();
        uiConstraint.setServicePropertyName(key.toString());
        Object content2 = map1.get(key);
        if (!(content2 instanceof Map)) {
            return null;
        }
        Map map2 = (Map) content2;
        Object key2 = map2.keySet().iterator().next();
        final String operator = key2.toString();
        if (SUPPORTED_CONSTRAINT_LIST.contains(operator)) {
            uiConstraint.setConstraintOperator(operator);
        }
        Object content3 = map2.get(key2);
        if (content3 instanceof String || content3 instanceof Number || content3 instanceof Boolean) {
            uiConstraint.setValue(content3);
            uiConstraint.setSourceType(STATIC_CONSTRAINT);
            uiConstraint.setSourceName(STATIC_CONSTRAINT);
            return uiConstraint;
        } else if (content3 instanceof List) {
            List list1 = (List) content3;
            uiConstraint.setSourceType(STATIC_CONSTRAINT);
            uiConstraint.setSourceName(STATIC_CONSTRAINT);
            uiConstraint.setValue(list1);
            return uiConstraint;
        } else if (content3 instanceof Map) {
            return handleMap(uiConstraint, content3);
        }
        return null;
    }

    private UIConstraint handleMap(UIConstraint uiConstraint, Object content3) {
        Map map3 = (Map) content3;
        Map.Entry entry = (Map.Entry) map3.entrySet().iterator().next();
        final String firstKey = entry.getKey().toString().trim();
        if (!SUPPORTED_FUNCTIONS.contains(firstKey)) {
            uiConstraint.setValue(content3);
            return uiConstraint;
        }
        if (ToscaFunctions.GET_INPUT.getFunctionName().equals(firstKey)) {
            uiConstraint.setSourceType(SERVICE_INPUT_CONSTRAINT);
            uiConstraint.setValue(entry.getValue());
            return uiConstraint;
        } else if (ToscaFunctions.GET_PROPERTY.getFunctionName().equals(firstKey)) {
            uiConstraint.setSourceType(PROPERTY_CONSTRAINT);
            final List<String> value = (List<String>) entry.getValue();
            uiConstraint.setSourceName(value.get(0));
            uiConstraint.setValue(value.get(1));
            return uiConstraint;
        }
      	return null;
    }

    public List<String> convertToList(List<UIConstraint> uiConstraints) {
        List<String> retVal = new ArrayList<>();
        for (UIConstraint uiConstraint : uiConstraints) {
            String constraint = convert(uiConstraint);
            if (constraint != null) {
                retVal.add(constraint);
            }
        }
        return retVal;
    }

    public String convert(UIConstraint uiConstraint) {
        try {
            Map map1 = new HashMap();
            Map map2 = new HashMap();

            map1.put(uiConstraint.getServicePropertyName(), map2);
            if (uiConstraint.getSourceType().equals(STATIC_CONSTRAINT)) {
                Object value = uiConstraint.getValue();
                if (value instanceof String) {
                    value = new Yaml().load(value.toString());
                }
                map2.put(uiConstraint.getConstraintOperator(), value);
            } else if (uiConstraint.getSourceType().equals(PROPERTY_CONSTRAINT)) {
                List list1 = Arrays.asList(uiConstraint.getSourceName(), uiConstraint.getValue());
                Map map3 = ImmutableMap.of(ToscaFunctions.GET_PROPERTY.getFunctionName(), list1);
                map2.put(uiConstraint.getConstraintOperator(), map3);
            } else if (uiConstraint.getSourceType().equals(SERVICE_INPUT_CONSTRAINT)) {
                Map map3 = ImmutableMap.of(ToscaFunctions.GET_INPUT.getFunctionName(), uiConstraint.getValue());
                map2.put(uiConstraint.getConstraintOperator(), map3);
            }

            Yaml yamlSource = new Yaml();
            return yamlSource.dump(map1);
        } catch (NullPointerException ex) {
            logger.error(ex.getMessage(), ex);
        }
        return null;
    }

    public UIConstraint getUiConstraint(final String inConstraint, final UIConstraint uiConstraint) {
        final Object constraintObject = new Yaml().load(inConstraint);
        if (!(constraintObject instanceof Map)) {
            return null;
        }
        final Map constraintMap = (Map) constraintObject;
        final Object capabilityName = constraintMap.keySet().iterator().next();
        uiConstraint.setServicePropertyName(capabilityName.toString());
        Object capabilityProperties = constraintMap.get(capabilityName);
        if (!(capabilityProperties instanceof Map)) {
            return null;
        }
        final Map capabilityPropertiesMap = (Map) capabilityProperties;
        final Object constraintOperator = capabilityPropertiesMap.keySet().iterator().next();
        final String operator = constraintOperator.toString();
        if (SUPPORTED_CONSTRAINT_LIST.contains(operator)) {
            uiConstraint.setConstraintOperator(operator);
        }
        final Object constraintValue = capabilityPropertiesMap.get(constraintOperator);
        if (constraintValue instanceof String || constraintValue instanceof Number || constraintValue instanceof Boolean) {
            uiConstraint.setValue(constraintValue);
            uiConstraint.setSourceType(STATIC_CONSTRAINT);
            uiConstraint.setSourceName(STATIC_CONSTRAINT);
            return uiConstraint;
        } else if (constraintValue instanceof List) {
            final List constraintValueList = (List) constraintValue;
            uiConstraint.setSourceType(STATIC_CONSTRAINT);
            uiConstraint.setSourceName(STATIC_CONSTRAINT);
            uiConstraint.setValue(constraintValueList);
            return uiConstraint;
        } else if (constraintValue instanceof Map) {
            return handleMap(uiConstraint, constraintValue);
        }
        return null;

    }
}