1 files changed, 86 insertions, 0 deletions
diff --git a/openecomp-be/lib/openecomp-sdc-translator-lib/openecomp-sdc-translator-core/src/main/java/org/openecomp/sdc/translator/services/heattotosca/ConsolidationEntityType.java b/openecomp-be/lib/openecomp-sdc-translator-lib/openecomp-sdc-translator-core/src/main/java/org/openecomp/sdc/translator/services/heattotosca/ConsolidationEntityType.java
new file mode 100644
index 0000000000..c2de58fd5a
--- /dev/null
+++ b/openecomp-be/lib/openecomp-sdc-translator-lib/openecomp-sdc-translator-core/src/main/java/org/openecomp/sdc/translator/services/heattotosca/ConsolidationEntityType.java
@@ -0,0 +1,86 @@
+package org.openecomp.sdc.translator.services.heattotosca;
+
+import static org.openecomp.sdc.translator.services.heattotosca.ConsolidationDataUtil.isComputeResource;
+import static org.openecomp.sdc.translator.services.heattotosca.ConsolidationDataUtil.isPortResource;
+import static org.openecomp.sdc.translator.services.heattotosca.ConsolidationDataUtil.isVolumeResource;
+
+import org.apache.commons.lang3.ObjectUtils;
+import org.openecomp.core.utilities.yaml.YamlUtil;
+import org.openecomp.sdc.heat.datatypes.manifest.FileData;
+import org.openecomp.sdc.heat.datatypes.model.HeatOrchestrationTemplate;
+import org.openecomp.sdc.heat.datatypes.model.Resource;
+import org.openecomp.sdc.translator.datatypes.heattotosca.TranslationContext;
+
+import java.util.Collection;
+import java.util.Objects;
+import java.util.Optional;
+
+/**
+ * The enum Entity type.
+ */
+public enum ConsolidationEntityType {
+  COMPUTE,
+  PORT,
+  VOLUME,
+  NESTED,
+  VFC_NESTED,
+  SUB_PORT,
+  OTHER;
+
+  private ConsolidationEntityType sourceEntityType;
+  private ConsolidationEntityType targetEntityType;
+
+  public ConsolidationEntityType getSourceEntityType() {
+    return sourceEntityType;
+  }
+
+  public ConsolidationEntityType getTargetEntityType() {
+    return targetEntityType;
+  }
+
+  /**
+   * Sets entity type.
+   *
+   * @param heatOrchestrationTemplate the heat orchestration template
+   * @param sourceResource          the source resource
+   * @param targetResource          the target resource
+   */
+  public void setEntityType(HeatOrchestrationTemplate heatOrchestrationTemplate,
+                            Resource sourceResource,
+                            Resource targetResource,
+                            TranslationContext context) {
+    targetEntityType =
+        getEntityType(heatOrchestrationTemplate, targetResource, context);
+    sourceEntityType =
+        getEntityType(heatOrchestrationTemplate, sourceResource, context);
+  }
+
+  private ConsolidationEntityType getEntityType(HeatOrchestrationTemplate heatOrchestrationTemplate,
+                                                Resource resource, TranslationContext context) {
+    if (isComputeResource(resource)) {
+      return ConsolidationEntityType.COMPUTE;
+    } else if (isPortResource(resource)) {
+      return ConsolidationEntityType.PORT;
+    } else if (isVolumeResource(resource)) {
+      return ConsolidationEntityType.VOLUME;
+    } else if (HeatToToscaUtil.isNestedResource(resource)) {
+      Optional<String> nestedHeatFileName = HeatToToscaUtil.getNestedHeatFileName(resource);
+      if (nestedHeatFileName.isPresent()) {
+        HeatOrchestrationTemplate nestedHeatOrchestrationTemplate = new YamlUtil()
+            .yamlToObject(context.getFileContent(nestedHeatFileName.get()),
+                HeatOrchestrationTemplate.class);
+        if (Objects.nonNull(nestedHeatOrchestrationTemplate.getResources())) {
+          for (String innerResourceId : nestedHeatOrchestrationTemplate.getResources().keySet()) {
+            if (ConsolidationDataUtil
+                .isComputeResource(nestedHeatOrchestrationTemplate, innerResourceId)) {
+              return ConsolidationEntityType.VFC_NESTED;
+            }
+          }
+        }
+      }
+      return ConsolidationEntityType.NESTED;
+    } else {
+      return ConsolidationEntityType.OTHER;
+    }
+  }
+}
+\ No newline at end of file