Support functions in TOSCA Simple Profile in YAML
[sdc.git] / catalog-be / src / main / java / org / openecomp / sdc / be / components / utils / ConsumptionUtils.java
1 /*
2  * Copyright © 2016-2018 European Support Limited
3  *
4  * Licensed under the Apache License, Version 2.0 (the "License");
5  * you may not use this file except in compliance with the License.
6  * You may obtain a copy of the License at
7  *
8  *      http://www.apache.org/licenses/LICENSE-2.0
9  *
10  * Unless required by applicable law or agreed to in writing, software
11  * distributed under the License is distributed on an "AS IS" BASIS,
12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13  * See the License for the specific language governing permissions and
14  * limitations under the License.
15  */
16
17 package org.openecomp.sdc.be.components.utils;
18
19 import com.google.gson.Gson;
20 import fj.data.Either;
21 import org.apache.commons.collections.CollectionUtils;
22 import org.openecomp.sdc.be.components.impl.ResponseFormatManager;
23 import org.openecomp.sdc.be.dao.api.ActionStatus;
24 import org.openecomp.sdc.be.datatypes.elements.OperationInputDefinition;
25 import org.openecomp.sdc.be.datatypes.elements.OperationOutputDefinition;
26 import org.openecomp.sdc.be.datatypes.enums.JsonPresentationFields;
27 import org.openecomp.sdc.be.model.CapabilityDefinition;
28 import org.openecomp.sdc.be.model.ComponentInstanceProperty;
29 import org.openecomp.sdc.be.model.Operation;
30 import org.openecomp.sdc.be.model.PropertyDefinition;
31 import org.openecomp.sdc.tosca.datatypes.ToscaFunctions;
32 import org.openecomp.sdc.be.model.tosca.ToscaPropertyType;
33 import org.openecomp.sdc.be.model.tosca.validators.PropertyTypeValidator;
34 import org.openecomp.sdc.be.types.ServiceConsumptionData;
35 import org.openecomp.sdc.exception.ResponseFormat;
36
37 import java.util.ArrayList;
38 import java.util.Collection;
39 import java.util.HashMap;
40 import java.util.List;
41 import java.util.Map;
42 import java.util.stream.Collectors;
43
44 public class ConsumptionUtils {
45
46     private ConsumptionUtils() {
47
48     }
49
50     public static Either<Operation, ResponseFormat> handleConsumptionInputMappedToCapabilityProperty(
51             Operation operation,
52             OperationInputDefinition operationInputDefinition, ServiceConsumptionData serviceConsumptionData,
53             Map<String, List<CapabilityDefinition>> capabilities, String componentName) {
54
55         List<CapabilityDefinition> componentCapabilityDefinitions = capabilities.values().stream()
56                 .flatMap(Collection::stream)
57                 .collect(Collectors.toList());
58         if (CollectionUtils.isEmpty(componentCapabilityDefinitions)) {
59             return Either.left(operation);
60         }
61
62         for (CapabilityDefinition capabilityDefinition : componentCapabilityDefinitions) {
63             String capabilityName = capabilityDefinition.getName();
64             List<ComponentInstanceProperty> capabilityProperties = capabilityDefinition.getProperties();
65             if (CollectionUtils.isEmpty(capabilityProperties)) {
66                 continue;
67             }
68             for (ComponentInstanceProperty capabilityProperty : capabilityProperties) {
69                 String capabilityPropertyName = capabilityProperty.getName();
70                 String capabilityPropertyIdentifier = capabilityName + "_" + capabilityPropertyName;
71                 if (capabilityPropertyIdentifier.equals(serviceConsumptionData.getValue())) {
72                     boolean isInputTypeSimilarToOperation =
73                             isAssignedValueFromValidType(operationInputDefinition.getType(), capabilityProperty);
74                     if (!isInputTypeSimilarToOperation) {
75                         return Either.right(getResponseFormatManager().getResponseFormat(
76                                 ActionStatus.INVALID_CONSUMPTION_TYPE, operationInputDefinition.getType()));
77                     }
78                     addCapabilityPropertyToInputValue(componentName, capabilityName, operation,
79                             operationInputDefinition, capabilityProperty);
80                 }
81             }
82         }
83         return Either.left(operation);
84     }
85
86     private static void addCapabilityPropertyToInputValue(String componentName, String capabilityName, Operation operation,
87                                                    OperationInputDefinition operationInputDefinition,
88                                                    PropertyDefinition capabilityProperty) {
89
90         List<String> getPropertyValues = new ArrayList<>();
91         getPropertyValues.add(componentName);
92         getPropertyValues.add(capabilityName);
93         getPropertyValues.add(capabilityProperty.getName());
94
95         Map<String, List<String>> getProperty = new HashMap<>();
96         getProperty.put(ToscaFunctions.GET_PROPERTY.getFunctionName(), getPropertyValues);
97
98         operationInputDefinition.setSourceProperty(capabilityProperty.getUniqueId());
99         operation.getInputs().delete(operationInputDefinition);
100         operationInputDefinition.setToscaPresentationValue(JsonPresentationFields.GET_PROPERTY,
101                 getPropertyValues);
102         operationInputDefinition.setValue((new Gson()).toJson(getProperty));
103         operation.getInputs().add(operationInputDefinition);
104     }
105
106     public static boolean isAssignedValueFromValidType(String operationInputType, Object actualValue) {
107         if (actualValue instanceof String) {
108             // validate static value
109             ToscaPropertyType actualType = ToscaPropertyType.isValidType(operationInputType);
110             PropertyTypeValidator validator = actualType.getValidator();
111             return validator.isValid((String)actualValue, operationInputType);
112         } else if (actualValue instanceof PropertyDefinition) {
113             // validate input / property value
114             String actualType = ((PropertyDefinition) actualValue).getType();
115             return actualType.equalsIgnoreCase(operationInputType);
116         } else if (actualValue instanceof OperationOutputDefinition) {
117             // validate input / output value
118             String actualType = ((OperationOutputDefinition) actualValue).getType();
119             return actualType.equalsIgnoreCase(operationInputType);
120         }
121         return false;
122     }
123
124     private static ResponseFormatManager getResponseFormatManager() {
125         return ResponseFormatManager.getInstance();
126     }
127 }