1 files changed, 255 insertions, 0 deletions
diff --git a/blueprint-generator/src/main/java/org/onap/blueprintgenerator/models/blueprint/Blueprint.java b/blueprint-generator/src/main/java/org/onap/blueprintgenerator/models/blueprint/Blueprint.java
new file mode 100644
index 0000000..861c820
--- /dev/null
+++ b/blueprint-generator/src/main/java/org/onap/blueprintgenerator/models/blueprint/Blueprint.java
@@ -0,0 +1,255 @@
+/**============LICENSE_START======================================================= 
+ org.onap.dcae 
+ ================================================================================ 
+ Copyright (c) 2019 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.models.blueprint;
+
+import java.io.BufferedWriter;
+import java.io.File;
+import java.io.FileWriter;
+import java.io.IOException;
+import java.io.PrintWriter;
+import java.util.ArrayList;
+import java.util.LinkedHashMap;
+import java.util.TreeMap;
+import java.util.regex.Pattern;
+
+import org.onap.blueprintgenerator.core.Fixes;
+import org.onap.blueprintgenerator.models.componentspec.ComponentSpec;
+import org.onap.blueprintgenerator.models.componentspec.Parameters;
+import org.onap.blueprintgenerator.models.componentspec.Publishes;
+import org.onap.blueprintgenerator.models.componentspec.Subscribes;
+import org.onap.blueprintgenerator.models.onapbp.OnapBlueprint;
+
+import com.fasterxml.jackson.annotation.JsonInclude;
+import com.fasterxml.jackson.core.JsonProcessingException;
+//import com.fasterxml.jackson.databind.ObjectMapper;
+import com.fasterxml.jackson.databind.ObjectMapper;
+import com.fasterxml.jackson.dataformat.yaml.YAMLFactory;
+import com.fasterxml.jackson.dataformat.yaml.YAMLGenerator;
+
+import lombok.AllArgsConstructor;
+import lombok.Builder;
+import lombok.Getter;
+import lombok.NoArgsConstructor;
+import lombok.Setter;
+
+
+
+@Getter @Setter
+@JsonInclude(JsonInclude.Include.NON_NULL)
+
+public class Blueprint {
+
+
+	private String tosca_definitions_version;
+
+
+	private ArrayList<String> imports;
+
+
+	private TreeMap<String, LinkedHashMap<String, Object>> inputs;
+
+
+	private TreeMap<String, Node> node_templates;
+	
+	public Blueprint createBlueprint(ComponentSpec cs, String name, char bpType, String importPath) {
+		Blueprint bp = new Blueprint();
+		
+		if(bpType == 'o') {
+			OnapBlueprint onap = new OnapBlueprint();
+			bp = onap.createOnapBlueprint(cs, importPath);
+			bp = bp.setQuotations(bp);
+		}
+//		if(bpType == 't') {
+//			bp.createBlueprintTemplate();
+//		}
+		return bp;
+	}
+	
+	public Blueprint setQuotations(Blueprint bp) {
+		for(String s: bp.getInputs().keySet()) {
+			LinkedHashMap<String, Object> temp = bp.getInputs().get(s);
+			if(temp.get("type") == "string") {
+				String def = (String) temp.get("default");
+				def = '"' + def + '"';
+				temp.replace("default", def);
+				bp.getInputs().replace(s, temp);
+			}
+		}
+		
+		return bp;
+	}
+	
+//	public void createBlueprintTemplate() {
+//		//set the tosca definition
+//		this.setTosca_definitions_version("cloudify_dsl_1_3");
+//		
+//		//set the imports
+//		Imports imps = new Imports();
+//		this.setImports(imps.createOnapImports());
+//		
+//		//create the needed inputs and just add the default ones
+//		TreeMap<String, LinkedHashMap<String, Object>> inputs = createTemplateInputs();
+//		this.setInputs(inputs);
+//		
+//		//create a node template
+//		TreeMap<String, Node> nodeTemplate = new TreeMap<String, Node>();
+//		TemplateNode template = new TemplateNode();
+//		template.createTemplateNode();
+//		nodeTemplate.put("Blueprint_Template", template);
+//		this.setNode_template(nodeTemplate);
+//		
+//		
+//	}
+//	//add tag, externam port, and replicas since they are in all the bps
+//	public TreeMap<String, LinkedHashMap<String, Object>> createTemplateInputs() {
+//		TreeMap<String, LinkedHashMap<String, Object>> inputs = new TreeMap<String, LinkedHashMap<String, Object>>();
+//		
+//		LinkedHashMap<String, Object> tag = new LinkedHashMap<String, Object>();
+//		tag.put("type", "string");
+//		tag.put("default", "{{ ONAPTEMPLATE_DOCKERREGURL_org_onap_dcaegen2_releases }}/onap/org.onap.dcaegen2.collectors.ves.vescollector:1.3.1");
+//		inputs.put("tag_version", tag);
+//		
+//		LinkedHashMap<String, Object> port = new LinkedHashMap<String, Object>();
+//		port.put("type", "string");
+//		port.put("description", "Kubernetes node port on which collector is exposed");
+//		port.put("default", "30235");
+//		inputs.put("external_port", port);
+//		
+//		LinkedHashMap<String, Object> rep = new LinkedHashMap<String, Object>();
+//		rep.put("type", "integer");
+//		rep.put("description", "number of instances");
+//		rep.put("default", 1);
+//		inputs.put("replicas", rep);
+//		
+//		return inputs;
+//	}
+
+	public void blueprintToYaml(String outputPath, String bluePrintName, ComponentSpec cs) {
+		File outputFile;
+
+		if(bluePrintName.equals("")) {
+			String name = cs.getSelf().getName();
+			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);
+			}
+		} else {
+			if(bluePrintName.contains(" ") || bluePrintName.contains(".")) {
+				bluePrintName = bluePrintName.replaceAll(Pattern.quote("."), "_");
+				bluePrintName = bluePrintName.replaceAll(" ", "");
+			}
+			String file = bluePrintName + ".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 = null;
+		try {
+			writer = new BufferedWriter(new FileWriter(outputFile, false));
+		} catch (IOException e1) {
+			throw new RuntimeException(e1);
+		}
+		if(writer != null) {
+			try {
+				writer.write(description);
+			} catch (IOException e) {
+				throw new RuntimeException(e);
+			}
+			try {
+				writer.write(version);
+			} catch (IOException e) {
+				throw new RuntimeException(e);
+			}
+			try {
+				writer.close();
+			} catch (IOException e) {
+				throw new RuntimeException(e);
+			}
+		}
+
+
+		//read the translated blueprint into the file
+		ObjectMapper blueprintMapper = new ObjectMapper(new YAMLFactory().configure(YAMLGenerator.Feature.MINIMIZE_QUOTES, true));
+
+		PrintWriter out = null;
+		try {
+			out = new PrintWriter(new BufferedWriter(new FileWriter(outputFile, true)));
+		} catch (IOException e) {
+			throw new RuntimeException(e);
+		}
+
+		try {
+			if(out != null) {
+				blueprintMapper.writeValue(out, this);
+				out.close();
+			}
+		} catch (IOException e) {
+			
+			throw new RuntimeException(e);
+		}
+
+
+		Fixes fix = new Fixes();
+		try {
+			fix.fixSingleQuotes(outputFile);
+		} catch (IOException e) {
+			throw new RuntimeException(e);
+		}
+
+		System.out.println("Blueprint created");
+	}
+
+
+	public String blueprintToString() {
+		String ret = "";
+
+		ObjectMapper blueprintMapper = new ObjectMapper(new YAMLFactory().configure(YAMLGenerator.Feature.MINIMIZE_QUOTES, true));
+		try {
+			ret = blueprintMapper.writerWithDefaultPrettyPrinter().writeValueAsString(this);
+		} catch (JsonProcessingException e) {
+			throw new RuntimeException(e);
+		}
+
+
+		return ret;
+	}
+}