1 files changed, 269 insertions, 0 deletions
diff --git a/ecomp-sdk/quantum/src/main/java/org/openecomp/portalsdk/core/service/ElementLinkService.java b/ecomp-sdk/quantum/src/main/java/org/openecomp/portalsdk/core/service/ElementLinkService.java
new file mode 100644
index 00000000..948b635f
--- /dev/null
+++ b/ecomp-sdk/quantum/src/main/java/org/openecomp/portalsdk/core/service/ElementLinkService.java
@@ -0,0 +1,269 @@
+/*-
+ * ================================================================================
+ * eCOMP Portal SDK
+ * ================================================================================
+ * Copyright (C) 2017 AT&T Intellectual Property
+ * ================================================================================
+ * 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.
+ * ================================================================================
+ */
+package org.openecomp.portalsdk.core.service;
+
+import java.io.File;
+import java.io.IOException;
+import java.util.ArrayList;
+import java.util.HashMap;
+import java.util.List;
+import java.util.Map;
+
+import org.openecomp.portalsdk.core.util.SystemProperties;
+import org.openecomp.portalsdk.core.util.YamlUtils;
+
+
+
+public class ElementLinkService {
+	
+	
+	public String main1(String[] args) throws Exception{
+		String filePath;
+		if(args[0].startsWith("custom")){
+		   filePath = SystemProperties.getProperty("customCallFlow_path");
+		}else{
+		   filePath = SystemProperties.getProperty("element_map_file_path") + File.separator;
+		}
+
+		String callFlowBusinessYml = "";
+		String callFlowStep = "";
+		
+		if(args != null && args.length > 0 ) {
+			
+			if( args[0] != null) {
+				callFlowBusinessYml = args[0] + ".yml";
+			}
+			
+			if( args[1] != null) {
+				callFlowStep = args[1];
+			}
+		}
+		
+		
+		ElementLinkService mapper = new ElementLinkService();
+		
+		return mapper.createLinkFile(filePath, callFlowBusinessYml, callFlowStep);
+	}
+	
+    public String main2(String[] args) throws Exception{
+		
+		String filePath = SystemProperties.getProperty("element_map_file_path") + File.separator;
+		String callFlowBusinessYml = "";
+		String callFlowStep = "";
+		
+		if(args != null && args.length > 0 ) {
+			
+			if( args[0] != null) {
+				callFlowBusinessYml = args[0] + "-Override.yml";
+			}
+			
+			if( args[1] != null) {
+				callFlowStep = args[1];
+			}
+		}
+		
+		
+		ElementLinkService mapper = new ElementLinkService();
+		
+		return mapper.createLinkFileAdditional(filePath, callFlowBusinessYml, callFlowStep);
+	}	
+	
+
+	public static void main(String[] args) throws Exception{
+		
+		String filePath = "C:\\Users\\statta\\git\\D2Platform\\war\\WEB-INF\\resources\\trisim_files";
+		String callFlowBusinessYml = "call_flow_hc-origination-termination-to-volteue-3.3.16-Override.yml";
+
+		
+		ElementLinkService mapper = new ElementLinkService();
+		System.out.print(mapper.createLinkFileAdditional(filePath, callFlowBusinessYml, "Step_2"));
+	}
+	
+	@SuppressWarnings("unchecked")
+	protected String createLinkFile(String resourceFilePath, String callFLowBsFileName, String callFlowStep) throws Exception {
+		
+		Map<String, Object> callFlowBs = YamlUtils.readYamlFile(resourceFilePath, callFLowBsFileName);
+		
+		List<Map<String,Object>> callSteps  = (List<Map<String,Object>>) callFlowBs.get("callSequenceSteps");
+		String callFlowName = (String) callFlowBs.get("shortName");
+		return addLinks( resourceFilePath, callFlowName, callSteps, callFlowStep);
+		
+	
+	}
+	
+	@SuppressWarnings("unchecked")
+	protected String createLinkFileAdditional(String resourceFilePath, String callFLowBsFileName, String callFlowStep) throws Exception {
+	
+		Map<String, Object> callFlowBs;
+		
+		try{
+		callFlowBs = YamlUtils.readYamlFile(resourceFilePath, callFLowBsFileName);
+		
+		List<Map<String,Object>> callSteps  = (List<Map<String,Object>>) callFlowBs.get("callSequenceSteps");
+		String callFlowName = (String) callFlowBs.get("shortName");
+		return addLinksAdditional( resourceFilePath, callFlowName, callSteps, callFlowStep);
+		
+		} catch (Exception e) {
+			
+			return "";
+		}
+		
+	
+	}
+
+	@SuppressWarnings("unchecked")
+	protected String  addLinks(String filePath, String callFlowName,
+			List<Map<String, Object>> callSteps, String callFlowStep) throws IOException {
+		
+		Map<String,List<String>> checkDuplicateMap = new HashMap<String,List<String>>();
+		
+		for(Map<String,Object> callStep : callSteps) {
+			
+			if(((String)callStep.get("name")).split(":")[0].trim().replace(" ", "_").equals(callFlowStep)) {
+			
+			List<Map<String,Object>> links = new ArrayList<Map<String,Object>>();
+			
+			List<Map<String,Object>> subSteps = (List<Map<String,Object>>)callStep.get("subSteps");
+			
+			for(Map<String,Object> subStep : subSteps) {
+				Map<String,Object> link = new HashMap<String,Object>();
+				
+				String source = (String) subStep.get("source_tosca_id");
+				String destination = (String) subStep.get("destination_tosca_id");
+				
+				if((checkDuplicateMap.get(source) == null || checkDuplicateMap.get(source).isEmpty() || !checkDuplicateMap.get(source).contains(destination)) && !source.equals(destination)) {
+					if(checkDuplicateMap.get(destination) == null) {
+						List<String> toscaList = new ArrayList<String>();
+						checkDuplicateMap.put(destination, toscaList);
+					}
+					
+					if(checkDuplicateMap.get(source) == null) {
+						List<String> toscaList = new ArrayList<String>();
+						checkDuplicateMap.put(source, toscaList);
+					}
+					
+					
+					List<String> toscaSourceList = checkDuplicateMap.get(destination);
+					toscaSourceList.add(source);
+					
+					List<String> toscaDestinationList = checkDuplicateMap.get(source);
+					toscaDestinationList.add(destination);
+					
+					link.put("s", source);
+					link.put("d", destination);
+					links.add(link);
+					
+				}
+				
+			}
+			
+			Map<String,Object> callFlowUI = new HashMap<String,Object>();
+			callFlowUI.put("linkList", links);
+			
+			return YamlUtils.returnYaml(callFlowUI);
+			
+			}
+		
+			
+		}
+		return "";
+	}
+	
+	protected String addLinksAdditional(String filePath, String callFlowName,
+			List<Map<String, Object>> callSteps, String callFlowStep) throws IOException {
+		
+			
+		for(Map<String,Object> callStep : callSteps) {
+			
+			if(((String)callStep.get("name")).split(":")[0].trim().replace(" ", "_").equals(callFlowStep)) {
+			
+				Map<String,Object> callFlowUI = new HashMap<String,Object>();
+				try{
+				List<Map<String, Object>> links = addLinkVertices(callStep);
+				callFlowUI.put("linkList", links); }
+				catch(Exception e) {}
+				try{
+				List<String> activeIds = addActiveNodes(callStep);
+				callFlowUI.put("activeIds", activeIds);}
+				catch(Exception e) {}
+				try{
+				List<Map<String, Object>> disconnectLinks = addDisconnectLinks(callStep);
+				callFlowUI.put("disconnectLinks", disconnectLinks); }
+				catch(Exception e) {}
+				
+				return YamlUtils.returnYaml(callFlowUI);
+			}
+		}
+
+		return "";
+	}
+	
+	@SuppressWarnings("unchecked")
+	List<String> addActiveNodes(Map<String, Object> callStep) {
+		
+		List<String> activeIds = (List<String>)callStep.get("activeIds");
+		
+		return activeIds;
+	}
+	
+	@SuppressWarnings("unchecked")
+	List<Map<String,Object>> addDisconnectLinks(Map<String, Object> callStep) {
+		
+		List<Map<String,Object>> disconnectLinks = (List<Map<String,Object>>)callStep.get("disconnectLinks");
+		
+		return disconnectLinks;
+	}
+
+	@SuppressWarnings("unchecked")
+	List<Map<String, Object>> addLinkVertices(Map<String, Object> callStep) {
+		List<Map<String,Object>> links = new ArrayList<Map<String,Object>>();
+		
+		List<Map<String,Object>> vertices = (List<Map<String,Object>>)callStep.get("vertices");
+		
+			for(int i=0; i< vertices.size()-1;i++) {
+				Map<String,Object> vertex =  (Map<String,Object>) vertices.get(i);
+				Map<String,Object> vertexNext =  (Map<String,Object>) vertices.get(i+1);
+			
+				Integer sourceX = (Integer) vertex.get("x");
+				Integer sourceY = (Integer)  vertex.get("y");
+				String sourceD = (String)  vertex.get("D");
+				String sourceL = (vertex.get("L") != null) ? (String)  vertex.get("L") : "-";
+				
+				if(sourceX == -999) // there is a break in the linkage
+					continue;
+				
+				Integer destinationX = (Integer) vertexNext.get("x");
+				Integer destinationY = (Integer)  vertexNext.get("y");
+				String destinationD = (String)  vertexNext.get("D");
+				
+				if(destinationX == -999) // there is a break in the linkage
+					continue;
+				
+				Map<String,Object> link = new HashMap<String,Object>();
+							
+					link.put("s", sourceX + "," + sourceY +","+sourceD + "," + sourceL);
+					link.put("d", destinationX + "," + destinationY +"," + destinationD);
+					links.add(link);
+			}
+		return links;
+	}
+}
+
+