aboutsummaryrefslogtreecommitdiffstats
path: root/openecomp-be/lib/openecomp-sdc-translator-lib/openecomp-sdc-translator-core/src/main/java/org/openecomp/sdc/translator/services/heattotosca/impl/functiontranslation/FunctionTranslationGetParamImpl.java
blob: 877659ee109fce2ef0c2c99c12bb0f5ae9b9341f (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
/*-
 * ============LICENSE_START=======================================================
 * SDC
 * ================================================================================
 * 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=========================================================
 */

package org.openecomp.sdc.translator.services.heattotosca.impl.functiontranslation;

import org.openecomp.sdc.heat.datatypes.model.HeatOrchestrationTemplate;
import org.openecomp.sdc.heat.datatypes.model.HeatPseudoParameters;
import org.openecomp.sdc.tosca.datatypes.ToscaFunctions;
import org.openecomp.sdc.tosca.datatypes.model.ServiceTemplate;
import org.openecomp.sdc.tosca.datatypes.model.Template;
import org.openecomp.sdc.translator.services.heattotosca.FunctionTranslation;
import org.openecomp.sdc.translator.services.heattotosca.FunctionTranslationFactory;
import org.openecomp.sdc.translator.datatypes.heattotosca.TranslationContext;

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

/**
 * @author SHIRIA
 * @since December 15, 2016.
 */
public class FunctionTranslationGetParamImpl implements FunctionTranslation {
  @Override
  public Object translateFunction(ServiceTemplate serviceTemplate, String resourceId,
                                  String propertyName, String functionKey,
                                  Object functionValue, String heatFileName,
                                  HeatOrchestrationTemplate heatOrchestrationTemplate,
                                  Template toscaTemplate, TranslationContext context) {
    Map returnValue = new HashMap<>();
    returnValue.put(ToscaFunctions.GET_INPUT.getDisplayName(),
        translateGetParamFunctionExpression(serviceTemplate, resourceId, propertyName,
            functionValue, heatFileName, heatOrchestrationTemplate, context));
    return returnValue;
  }

  private static Object translateGetParamFunctionExpression(ServiceTemplate serviceTemplate,
                                                            String resourceId,
                                                            String propertyName,Object functionValue,
                                                            String heatFileName,
                                                            HeatOrchestrationTemplate
                                                                heatOrchestrationTemplate,
                                                            TranslationContext context) {
    Object returnValue = null;
    if (functionValue instanceof String) {
      returnValue = functionValue;
      if (HeatPseudoParameters.getPseudoParameterNames().contains(functionValue)) {
        context
            .addUsedHeatPseudoParams(heatFileName, (String) functionValue, (String) functionValue);
      }
    } else if (functionValue instanceof List) {
      returnValue = new ArrayList<>();
      for (int i = 0; i < ((List) functionValue).size(); i++) {
        Object paramValue = ((List) functionValue).get(i);
        if ((paramValue instanceof Map && !((Map) paramValue).isEmpty())) {
          Map<String, Object> paramMap = (Map) paramValue;
          ((List) returnValue).add(translatedInnerMap(serviceTemplate, resourceId,
              propertyName, paramMap, heatFileName, heatOrchestrationTemplate, context));
        } else {
          ((List) returnValue).add(paramValue);
        }
      }
    }

    return returnValue;
  }

  private static Object translatedInnerMap(ServiceTemplate serviceTemplate, String resourceId,
                                           String propertyName, Map<String, Object> paramMap,
                                           String heatFileName,HeatOrchestrationTemplate
                                               heatOrchestrationTemplate,
                                           TranslationContext context) {

    Map<String, Object> translatedInnerMapValue = new HashMap<>();
    for (Map.Entry<String, Object> entry : paramMap.entrySet()) {
      if (FunctionTranslationFactory.getInstance(entry.getKey()).isPresent()) {
        return FunctionTranslationFactory.getInstance(entry.getKey()).get()
            .translateFunction(serviceTemplate, resourceId, propertyName, entry.getKey(),
                entry.getValue(), heatFileName, heatOrchestrationTemplate, null, context);
      } else {
        translatedInnerMapValue.put(entry.getKey(),
            translatedInnerValue(serviceTemplate, resourceId, propertyName,entry.getValue(),
                heatFileName, heatOrchestrationTemplate, context));

      }
    }
    return translatedInnerMapValue;
  }

  private static Object translatedInnerValue(ServiceTemplate serviceTemplate, String resourceId,
                                             String propertyName,Object value, String heatFileName,
                                             HeatOrchestrationTemplate heatOrchestrationTemplate,
                                             TranslationContext context) {
    if (value instanceof String) {
      return value;
    } else if (value instanceof Map) {
      return translatedInnerMap(serviceTemplate, resourceId, propertyName,(Map<String, Object>)
          value, heatFileName, heatOrchestrationTemplate, context);
    } else if (value instanceof List) {
      List returnedList = new ArrayList();
      for (int i = 0; i < ((List) value).size(); i++) {
        returnedList.add(translatedInnerValue(serviceTemplate, resourceId, propertyName,
            ((List) value).get(i), heatFileName, heatOrchestrationTemplate, context));
      }
      return returnedList;
    }

    return value;
  }
}