Sync Integ to Master
[sdc.git] / catalog-be / src / main / java / org / openecomp / sdc / be / tosca / utils / ForwardingPathToscaUtil.java
1 package org.openecomp.sdc.be.tosca.utils;
2
3 import fj.data.Either;
4 import org.apache.commons.collections.MapUtils;
5 import org.openecomp.sdc.be.datatypes.elements.ForwardingPathDataDefinition;
6 import org.openecomp.sdc.be.datatypes.elements.ForwardingPathElementDataDefinition;
7 import org.openecomp.sdc.be.model.CapabilityDefinition;
8 import org.openecomp.sdc.be.model.Component;
9 import org.openecomp.sdc.be.model.ComponentInstance;
10 import org.openecomp.sdc.be.model.Service;
11 import org.openecomp.sdc.be.model.jsontitan.operations.ToscaOperationFacade;
12 import org.openecomp.sdc.be.tosca.CapabiltyRequirementConvertor;
13 import org.openecomp.sdc.be.tosca.model.ToscaNodeTemplate;
14 import org.openecomp.sdc.be.tosca.model.ToscaTemplateRequirement;
15
16 import java.util.*;
17
18 /**
19  * @author KATYR
20  * @since November 19, 2017
21  */
22
23 public class ForwardingPathToscaUtil {
24     public static final String FORWARDS_TO_TOSCA_NAME =
25             "org.openecomp.relationships.ForwardsTo";
26     public static final String PROTOCOL = "protocol";
27     public static final String PORTS_RANGE = "target_range";
28     public static final String FORWARDER = "forwarder";
29
30     public static void addForwardingPaths(Service service, Map<String, ToscaNodeTemplate>
31             nodeTemplates, CapabiltyRequirementConvertor capabiltyRequirementConvertor, Map<String, Component> originComponents, ToscaOperationFacade toscaOperationFacade) {
32         for (String forwardingPathName : service.getForwardingPaths().keySet()) {
33             ToscaNodeTemplate forwardingPathNodeTemplate =
34                     new ToscaNodeTemplate();
35             final ForwardingPathDataDefinition path =
36                     service.getForwardingPaths().get(forwardingPathName);
37             forwardingPathNodeTemplate.setType(path.getToscaResourceName());
38
39             if (Objects.nonNull(path.getDescription())) {
40                 forwardingPathNodeTemplate.setDescription(path
41                         .getDescription());
42             }
43             Map<String, Object> props = new HashMap<>();
44             if (Objects.nonNull(path.getDestinationPortNumber())) {
45                 props.put(PORTS_RANGE, Collections.singletonList(path.getDestinationPortNumber()));
46             }
47             if (Objects.nonNull(path.getProtocol())) {
48                 props.put(PROTOCOL, path.getProtocol());
49             }
50             if (MapUtils.isNotEmpty(props)) {
51                 forwardingPathNodeTemplate.setProperties(props);
52             }
53
54             final List<ForwardingPathElementDataDefinition> pathElements =
55                     path.getPathElements()
56                             .getListToscaDataDefinition();
57             forwardingPathNodeTemplate.setRequirements(convertPathElementsToRequirements(pathElements,
58                     service, capabiltyRequirementConvertor, originComponents, toscaOperationFacade));
59
60             nodeTemplates.put(path.getName(), forwardingPathNodeTemplate);
61         }
62
63     }
64
65     private static List<Map<String, ToscaTemplateRequirement>> convertPathElementsToRequirements(
66         List<ForwardingPathElementDataDefinition> pathElements, Service service, CapabiltyRequirementConvertor capabiltyRequirementConvertor, Map<String, Component> originComponents,ToscaOperationFacade toscaOperationFacade) {
67         List<Map<String, ToscaTemplateRequirement>> toscaRequirements = new ArrayList<>();
68         for (int i = 0; i <= pathElements.size() -1 ; i++) {
69                 final ForwardingPathElementDataDefinition element = pathElements.get(i);
70                 toscaRequirements.add(handleSingleReq(fetchCPName(service, element.getFromNode(), element.getFromCP(), capabiltyRequirementConvertor, originComponents, toscaOperationFacade), fetchNodeName(service, element.getFromNode())));
71                 if ( i == pathElements.size() -1) {
72                     toscaRequirements.add(handleSingleReq(fetchCPName(service, element.getToNode(), element.getToCP(), capabiltyRequirementConvertor, originComponents, toscaOperationFacade), fetchNodeName(service, element
73                             .getToNode())));
74                 }
75        }
76         return toscaRequirements;
77
78     }
79
80     private static String fetchNodeName(Service service, String nodeId) {
81         if (service.getComponentInstanceById(nodeId).isPresent()) {
82             return service.getComponentInstanceById(nodeId).get().getName();
83         } else {
84             return "";
85         }
86     }
87
88
89     private static Map<String, ToscaTemplateRequirement> handleSingleReq(
90             String fromCP, String fromNode) {
91         Map<String, ToscaTemplateRequirement> toscaReqMap = new HashMap<>();
92         ToscaTemplateRequirement firstReq = new ToscaTemplateRequirement();
93         firstReq.setRelationship(FORWARDS_TO_TOSCA_NAME); //todo
94         firstReq.setCapability(fromCP);
95         firstReq.setNode(fromNode);
96         toscaReqMap.put(FORWARDER, firstReq);
97
98         return toscaReqMap;
99     }
100
101     /**
102      * @todo handle errors.
103      */
104     private static String fetchCPName(Service service, String nodeID, String cpName, CapabiltyRequirementConvertor capabiltyRequirementConvertor, Map<String, Component> originComponents,  ToscaOperationFacade toscaOperationFacade) {
105         Optional<ComponentInstance> componentInstance = service.getComponentInstanceById(nodeID);
106         ComponentInstance componentInstanceVal = componentInstance.get();
107         String name = componentInstanceVal.getNormalizedName();
108         Component component = originComponents.get(componentInstanceVal.getComponentUid());
109         if(componentInstanceVal.getIsProxy()){
110             component = originComponents.get(componentInstanceVal.getSourceModelUid());
111             if (component == null) {
112                 component = toscaOperationFacade.getToscaFullElement(componentInstanceVal.getSourceModelUid()).left().value();
113             }
114
115         }
116         CapabilityDefinition capability = componentInstanceVal.getCapabilities().values().stream().flatMap(capabilityDefinitions -> capabilityDefinitions.stream())
117                 .filter(capabilityDefinition -> capabilityDefinition.getUniqueId().equals(cpName)).findAny().get();
118         List<String> path = capability.getPath();
119         List<String> reducedPath = new ArrayList<>(path);
120         reducedPath.remove(reducedPath.size() - 1);
121         Either<String, Boolean> stringBooleanEither = capabiltyRequirementConvertor.buildSubstitutedName(originComponents, component, reducedPath, capability.getName());
122         return name + "." + stringBooleanEither.left().value();
123     }
124 }