1 files changed, 252 insertions, 0 deletions
diff --git a/common/openecomp-tosca-datatype/src/main/java/org/openecomp/sdc/tosca/services/YamlUtil.java b/common/openecomp-tosca-datatype/src/main/java/org/openecomp/sdc/tosca/services/YamlUtil.java
new file mode 100644
index 0000000000..7d0bcfeb83
--- /dev/null
+++ b/common/openecomp-tosca-datatype/src/main/java/org/openecomp/sdc/tosca/services/YamlUtil.java
@@ -0,0 +1,252 @@
+package org.openecomp.sdc.tosca.services;
+
+import org.yaml.snakeyaml.DumperOptions;
+import org.yaml.snakeyaml.TypeDescription;
+import org.yaml.snakeyaml.Yaml;
+import org.yaml.snakeyaml.constructor.Constructor;
+import org.yaml.snakeyaml.introspector.BeanAccess;
+import org.yaml.snakeyaml.introspector.Property;
+import org.yaml.snakeyaml.introspector.PropertyUtils;
+import org.yaml.snakeyaml.nodes.MappingNode;
+import org.yaml.snakeyaml.nodes.NodeTuple;
+import org.yaml.snakeyaml.nodes.Tag;
+import org.yaml.snakeyaml.parser.ParserException;
+import org.yaml.snakeyaml.representer.Representer;
+
+import java.beans.IntrospectionException;
+import java.io.IOException;
+import java.io.InputStream;
+import java.util.AbstractMap;
+import java.util.LinkedHashMap;
+import java.util.LinkedHashSet;
+import java.util.Map;
+import java.util.Set;
+
+/**
+ * The type Yaml util.
+ */
+@SuppressWarnings("unchecked")
+public class YamlUtil {
+
+  /**
+   * Yaml to object t.
+   *
+   * @param <T>         the type parameter
+   * @param yamlContent the yaml content
+   * @param typClass    the t class
+   * @return the t
+   */
+  public <T> T yamlToObject(String yamlContent, Class<T> typClass) {
+    Constructor constructor = getConstructor(typClass);
+    constructor.setPropertyUtils(getPropertyUtils());
+    TypeDescription yamlFileDescription = new TypeDescription(typClass);
+    constructor.addTypeDescription(yamlFileDescription);
+    Yaml yaml = new Yaml(constructor);
+    T yamlObj = (T) yaml.load(yamlContent);
+    //noinspection ResultOfMethodCallIgnored
+    yamlObj.toString();
+    return yamlObj;
+  }
+
+  /**
+   * Yaml to object t.
+   *
+   * @param <T>         the type parameter
+   * @param yamlContent the yaml content
+   * @param typClass    the t class
+   * @return the t
+   */
+  public <T> T yamlToObject(InputStream yamlContent, Class<T> typClass) {
+    try {
+      Constructor constructor = getConstructor(typClass);
+      constructor.setPropertyUtils(getPropertyUtils());
+      TypeDescription yamlFileDescription = new TypeDescription(typClass);
+      constructor.addTypeDescription(yamlFileDescription);
+      Yaml yaml = new Yaml(constructor);
+      T yamlObj = (T) yaml.load(yamlContent);
+      if (yamlObj != null) {
+        //noinspection ResultOfMethodCallIgnored
+        yamlObj.toString();
+        return yamlObj;
+      } else {
+        throw new RuntimeException();
+      }
+    } catch (Exception exception) {
+      throw new RuntimeException(exception);
+    } finally {
+      try {
+        if (yamlContent != null) {
+          yamlContent.close();
+        }
+      } catch (IOException ignore) {
+        //do nothing
+      }
+    }
+  }
+
+
+  /**
+   * Gets constructor.
+   *
+   * @param <T>      the type parameter
+   * @param typClass the t class
+   * @return the constructor
+   */
+  public <T> Constructor getConstructor(Class<T> typClass) {
+    return new StrictMapAppenderConstructor(typClass);
+  }
+
+  /**
+   * Gets property utils.
+   *
+   * @return the property utils
+   */
+  protected PropertyUtils getPropertyUtils() {
+    return new MyPropertyUtils();
+  }
+
+
+  /**
+   * Yaml to map map.
+   *
+   * @param yamlContent the yaml content
+   * @return the map
+   */
+  public Map<String, LinkedHashMap<String, Object>> yamlToMap(InputStream yamlContent) {
+    Yaml yaml = new Yaml();
+    @SuppressWarnings("unchecked") Map<String, LinkedHashMap<String, Object>> yamlData =
+        (Map<String, LinkedHashMap<String, Object>>) yaml.load(yamlContent);
+    return yamlData;
+  }
+
+  /**
+   * Object to yaml string.
+   *
+   * @param <T> the type parameter
+   * @param obj the obj
+   * @return the string
+   */
+  public <T> String objectToYaml(Object obj) {
+    DumperOptions options = new DumperOptions();
+    options.setPrettyFlow(true);
+    options.setDefaultFlowStyle(DumperOptions.FlowStyle.BLOCK);
+    Representer representer = new CustomRepresenter();
+    representer.addClassTag(obj.getClass(), Tag.MAP);
+    representer.setPropertyUtils(new MyPropertyUtils());
+
+    Yaml yaml = new Yaml(representer, options);
+    return yaml.dump(obj);
+  }
+
+  /**
+   * Is yaml file content valid boolean.
+   *
+   * @param yamlFullFileName the yaml full file name
+   * @return the boolean
+   */
+  public boolean isYamlFileContentValid(String yamlFullFileName) {
+    Yaml yaml = new Yaml();
+    try {
+      Object loadResult = yaml.load(yamlFullFileName);
+      return loadResult != null;
+    } catch (Exception exception) {
+      return false;
+    }
+  }
+
+
+  private class CustomRepresenter extends Representer {
+    @Override
+    protected MappingNode representJavaBean(Set<Property> properties, Object javaBean) {
+      //remove the bean type from the output yaml (!! ...)
+      if (!classTags.containsKey(javaBean.getClass())) {
+        addClassTag(javaBean.getClass(), Tag.MAP);
+      }
+
+      return super.representJavaBean(properties, javaBean);
+    }
+
+    @Override
+    protected NodeTuple representJavaBeanProperty(Object javaBean, Property property,
+                                                  Object propertyValue, Tag customTag) {
+      if (propertyValue == null) {
+        return null;
+      } else {
+        NodeTuple defaultNode =
+            super.representJavaBeanProperty(javaBean, property, propertyValue, customTag);
+
+        return property.getName().equals("_default")
+            ? new NodeTuple(representData("default"), defaultNode.getValueNode())
+            : defaultNode;
+      }
+    }
+  }
+
+
+  /**
+   * The type My property utils.
+   */
+  public class MyPropertyUtils extends PropertyUtils {
+    //Unsorted properties
+    @Override
+    protected Set<Property> createPropertySet(Class<? extends Object> type, BeanAccess bnAccess)
+        throws IntrospectionException {
+      return new LinkedHashSet<Property>(getPropertiesMap(type,
+          BeanAccess.FIELD).values());
+    }
+
+    @Override
+    public Property getProperty(Class<?> type, String name) throws IntrospectionException {
+      if (name.equals("default")) {
+        name = "_default";
+      }
+      return super.getProperty(type, name);
+    }
+
+  }
+
+  /**
+   * The type Strict map appender constructor.
+   */
+  protected class StrictMapAppenderConstructor extends Constructor {
+
+    /**
+     * Instantiates a new Strict map appender constructor.
+     *
+     * @param theRoot the the root
+     */
+    public StrictMapAppenderConstructor(Class<?> theRoot) {
+      super(theRoot);
+    }
+
+    @Override
+    protected Map<Object, Object> createDefaultMap() {
+      final Map<Object, Object> delegate = super.createDefaultMap();
+      return new AbstractMap<Object, Object>() {
+        @Override
+        public Object put(Object key, Object value) {
+          if (delegate.containsKey(key)) {
+            throw new IllegalStateException("duplicate key: " + key);
+          }
+          return delegate.put(key, value);
+        }
+
+        @Override
+        public Set<Entry<Object, Object>> entrySet() {
+          return delegate.entrySet();
+        }
+      };
+    }
+
+    @Override
+    protected Map<Object, Object> constructMapping(MappingNode node) {
+      try {
+        return super.constructMapping(node);
+      } catch (IllegalStateException exception) {
+        throw new ParserException("while parsing MappingNode",
+            node.getStartMark(), exception.getMessage(),
+            node.getEndMark());
+      }
+    }
+  }
+}