1 files changed, 157 insertions, 0 deletions
diff --git a/mod/bpgenerator/common/src/main/java/org/onap/blueprintgenerator/service/base/BlueprintService.java b/mod/bpgenerator/common/src/main/java/org/onap/blueprintgenerator/service/base/BlueprintService.java
new file mode 100644
index 0000000..83c43cf
--- /dev/null
+++ b/mod/bpgenerator/common/src/main/java/org/onap/blueprintgenerator/service/base/BlueprintService.java
@@ -0,0 +1,157 @@
+/*
+ *
+ *  * ============LICENSE_START=======================================================
+ *  *  org.onap.dcae
+ *  *  ================================================================================
+ *  *  Copyright (c) 2020  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.onap.blueprintgenerator.service.base;
+
+import org.onap.blueprintgenerator.model.base.Blueprint;
+import org.onap.blueprintgenerator.model.common.Input;
+import org.onap.blueprintgenerator.model.componentspec.base.ComponentSpec;
+import com.fasterxml.jackson.core.JsonProcessingException;
+import com.fasterxml.jackson.databind.ObjectMapper;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.beans.factory.annotation.Qualifier;
+import org.springframework.stereotype.Service;
+import org.springframework.util.StringUtils;
+
+
+import java.io.File;
+import java.io.IOException;
+import java.io.FileWriter;
+import java.io.BufferedWriter;
+import java.io.PrintWriter;
+import java.util.LinkedHashMap;
+import java.util.Map;
+import java.util.regex.Pattern;
+
+/**
+ * @author : Ravi Mantena
+ * @date 10/16/2020
+ * Application: DCAE/ONAP - Blueprint Generator
+ * Common Module: Used by both ONAp and DCAE Blueprint Applications
+ * Service: For Adding Quotes and Converting Blueprint to Yaml/String
+ */
+
+@Service
+public abstract class BlueprintService {
+
+  @Autowired
+  protected FixesService fixesService;
+
+  @Qualifier("objectMapper")
+  @Autowired
+  protected ObjectMapper objectMapper;
+
+  @Qualifier("yamlObjectMapper")
+  @Autowired
+  protected ObjectMapper yamlObjectMapper;
+
+  public void blueprintToYaml(ComponentSpec cs, Blueprint blueprint, Input input) {
+    String bluePrintName = input.getBluePrintName();
+    String outputPath = input.getOutputPath();
+    String comment = "# " + input.getComment() + '\n';
+
+    try {
+    File outputFile;
+    String name = StringUtils.isEmpty(bluePrintName) ? cs.getSelf().getName() : bluePrintName;
+    if(name.contains(".")) {
+      name = name.replaceAll(Pattern.quote("."), "_");
+    }
+    if(name.contains(" ")) {
+      name = name.replaceAll(" ", "");
+    }
+    String file = name + ".yaml";
+    outputFile = new File(outputPath, file);
+    outputFile.getParentFile().mkdirs();
+    try {
+      outputFile.createNewFile();
+    } catch (IOException e) {
+      throw new RuntimeException(e);
+    }
+
+    String version = "#blueprint_version: " + cs.getSelf().getVersion() + '\n';
+    String description = "#description: " + cs.getSelf().getDescription() + '\n';
+
+    BufferedWriter writer = new BufferedWriter(new FileWriter(outputFile, false));
+    writer.write(description);
+    writer.write(version);
+
+    if(input.getBpType().equals("dti") || input.getBpType().equals("m") || input.getBpType().equals("k"))
+      writer.write(comment);
+
+    writer.close();
+
+    PrintWriter out = new PrintWriter(new BufferedWriter(new FileWriter(outputFile, true)));
+    yamlObjectMapper.writeValue(out, blueprint);
+    out.close();
+
+    if(input.getBpType().equals("dti") || input.getBpType().equals("m") || input.getBpType().equals("k"))
+      fixesService.fixDcaeSingleQuotes(outputFile);
+    else
+      fixesService.fixOnapSingleQuotes(outputFile);
+
+    //new Yaml().load(new FileInputStream(outputFile));
+
+    System.out.println("Blueprint is created with valid YAML Format");
+    } catch (Exception ex) {
+    throw new RuntimeException("Unable to generate YAML file from Blueprint  or the generated YAML is not valid", ex);
+    }
+  }
+
+  // Used for DCAE
+  public Map<String, LinkedHashMap<String, Object>> addQuotes(Map<String, LinkedHashMap<String, Object>> inputs) {
+
+    inputs.forEach((key,value) -> {
+    if(value.get("type") != null){
+      if (value.get("type").equals("string") && value.get("default") != null && !key.contains("policies")){
+        value.replace("default",  "'" + value.get("default").toString() + "'");
+      } else if (value.get("type").equals("map") || value.get("type").equals("list")) {
+        // Commented the Code as we need to read the object as is for Map and List. If the List object is to be converted to string uncomment the below code.
+        /* if(inputs.get(s).get("type").equals("list")) {
+        String temp = inputs.get(s).get("default").toString();
+        inputs.get(s).replace("default", temp);
+        }*/
+        inputs.get(key).remove("type");
+      }
+    }
+    });
+
+   return inputs;
+  }
+
+  public String blueprintToString(ComponentSpec componentSpec, Blueprint blueprint, Input input) {
+    String ret = "";
+    try {
+    ret = yamlObjectMapper.writerWithDefaultPrettyPrinter().writeValueAsString(blueprint);
+    } catch (JsonProcessingException e) {
+    e.printStackTrace();
+    }
+    if(input.getBpType().equals("dti") || input.getBpType().equals("m") || input.getBpType().equals("k")){
+    ret = "# " + componentSpec.getSelf().getDescription() + "\n" + "# " + componentSpec.getSelf().getVersion() + "\n" + "# " + input.getComment() + "\n" + ret;
+    ret = fixesService.fixStringQuotes(ret);
+    }else
+    ret = fixesService.applyFixes(ret);
+    return ret;
+  }
+
+
+}