aboutsummaryrefslogtreecommitdiffstats
path: root/openecomp-be/lib/openecomp-tosca-lib/src/main/java/org/openecomp/sdc/tosca/services/yamlutil/ToscaExtensionYamlUtil.java
blob: 2b806fbea0f44f7a0dec46a2f6d722de1a339924 (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
/*-
 * ============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.tosca.services.yamlutil;

import org.openecomp.core.utilities.yaml.YamlUtil;
import org.yaml.snakeyaml.constructor.Constructor;
import org.yaml.snakeyaml.introspector.Property;
import org.yaml.snakeyaml.introspector.PropertyUtils;
import org.yaml.snakeyaml.nodes.MappingNode;
import org.yaml.snakeyaml.nodes.NodeId;

import java.beans.IntrospectionException;

public class ToscaExtensionYamlUtil extends YamlUtil {

  @Override
  public <T> Constructor getConstructor(Class<T> typClass) {
    return new ToscaWithHeatExtensionConstructor(typClass);
  }

  @Override
  protected PropertyUtils getPropertyUtils() {
    return new ToscaPropertyUtilsWithHeatExtension();
  }

  public class ToscaPropertyUtilsWithHeatExtension extends MyPropertyUtils {
    @Override
    public Property getProperty(Class<? extends Object> type, String name)
        throws IntrospectionException {
      try {
        if (type
            .equals(Class.forName("org.openecomp.sdc.tosca.datatypes.model.ParameterDefinition"))) {
          type = Class
              .forName("org.openecomp.sdc.tosca.datatypes.model.heatextend.ParameterDefinitionExt");
        }
      } catch (ClassNotFoundException ex) {
        throw new RuntimeException(ex);
      }
      return super.getProperty(type, name);
    }
  }

  protected class ToscaWithHeatExtensionConstructor extends StrictMapAppenderConstructor {
    public ToscaWithHeatExtensionConstructor(Class<?> theRoot) {
      super(theRoot);
      yamlClassConstructors.put(NodeId.mapping, new MyPersistentObjectConstruct());
    }

    class MyPersistentObjectConstruct extends ConstructMapping {
      @Override
      protected Object constructJavaBean2ndStep(MappingNode node, Object object) {
        Class type = node.getType();
        try {
          if (type.equals(
              Class.forName("org.openecomp.sdc.tosca.datatypes.model.ParameterDefinition"))) {
            Class extendHeatClass = Class.forName(
                "org.openecomp.sdc.tosca.datatypes.model.heatextend.ParameterDefinitionExt");
            Object extendHeatObject = extendHeatClass.newInstance();
            // create JavaBean
            return super.constructJavaBean2ndStep(node, extendHeatObject);
          } else {
            // create JavaBean
            return super.constructJavaBean2ndStep(node, object);
          }
        } catch (ClassNotFoundException | InstantiationException | IllegalAccessException ex) {
          throw new RuntimeException(ex);
        }
      }
    }
  }
}