[aai-babel] Add support for sndc props parsing

Signed-off-by: efiacor <fiachra.corcoran@est.tech>
Issue-ID: AAI-3121
Change-Id: I9ed925e7e95d73b30942df895d19fba0c36500f0

3 files changed, 53 insertions, 0 deletions

diff --git a/src/main/java/org/onap/aai/babel/xml/generator/api/AaiArtifactGenerator.java b/src/main/java/org/onap/aai/babel/xml/generator/api/AaiArtifactGenerator.java
index 8da541f..c42518d 100644
--- a/src/main/java/org/onap/aai/babel/xml/generator/api/AaiArtifactGenerator.java
+++ b/src/main/java/org/onap/aai/babel/xml/generator/api/AaiArtifactGenerator.java
@@ -26,6 +26,7 @@ import java.nio.file.Files;
 import java.nio.file.Path;
 import java.util.ArrayList;
 import java.util.Collections;
+import java.util.HashMap;
 import java.util.List;
 import java.util.Map;
 import java.util.Optional;
@@ -51,17 +52,23 @@ import org.onap.aai.babel.xml.generator.model.Widget;
 import org.onap.aai.babel.xml.generator.model.WidgetType;
 import org.onap.aai.babel.xml.generator.types.ModelType;
 import org.onap.aai.cl.api.Logger;
+import org.onap.sdc.tosca.parser.api.IEntityDetails;
 import org.onap.sdc.tosca.parser.api.ISdcCsarHelper;
+import org.onap.sdc.tosca.parser.elements.queries.EntityQuery;
+import org.onap.sdc.tosca.parser.elements.queries.TopologyTemplateQuery;
 import org.onap.sdc.tosca.parser.enums.SdcTypes;
 import org.onap.sdc.tosca.parser.exceptions.SdcToscaParserException;
 import org.onap.sdc.tosca.parser.impl.SdcToscaParserFactory;
 import org.onap.sdc.toscaparser.api.Group;
 import org.onap.sdc.toscaparser.api.NodeTemplate;
+import org.onap.sdc.toscaparser.api.Property;
 import org.onap.sdc.toscaparser.api.elements.Metadata;
 import org.slf4j.MDC;
 
 public class AaiArtifactGenerator implements ArtifactGenerator {
 
+    private static final String SDNC_MODEL_VERSION = "sdnc_model_version";
+    private static final String SDNC_MODEL_NAME = "sdnc_model_name";
     private static Logger log = LogHelper.INSTANCE;
 
     private static final String MDC_PARAM_MODEL_INFO = "ARTIFACT_MODEL_INFO";
@@ -283,6 +290,12 @@ public class AaiArtifactGenerator implements ArtifactGenerator {
         Map<String, String> serviceMetadata = serviceVfNode.getMetaData().getAllProperties();
         resourceModel.populateModelIdentificationInformation(serviceMetadata);
 
+        Map<String, String> pnfProps = getResourceProperties(csarHelper, SdcTypes.PNF);
+        resourceModel.populateModelIdentificationInformation(pnfProps);
+
+        Map<String, String> vfProps = getResourceProperties(csarHelper, SdcTypes.VF);
+        resourceModel.populateModelIdentificationInformation(vfProps);
+
         parser.processResourceModels(resourceModel, getNonVnfChildren(serviceVfNode));
 
         List<NodeTemplate> serviceVfList = ToscaParser.getServiceNodeTemplates(csarHelper)
@@ -300,6 +313,21 @@ public class AaiArtifactGenerator implements ArtifactGenerator {
         resources.add(resourceModel);
     }
 
+    private Map<String, String> getResourceProperties(ISdcCsarHelper csarHelper, SdcTypes type) {
+        EntityQuery entityQuery = EntityQuery.newBuilder(type).build();
+        TopologyTemplateQuery topologyTemplateQuery = TopologyTemplateQuery.newBuilder(SdcTypes.SERVICE).build();
+        List<IEntityDetails> entityDetailsList = csarHelper.getEntity(entityQuery, topologyTemplateQuery, false);
+        Map<String, String> props = new HashMap<>();
+        for (IEntityDetails entityDetails : entityDetailsList) {
+            Map<String, Property> properties = entityDetails.getProperties();
+            if (properties.get(SDNC_MODEL_VERSION) != null && properties.get(SDNC_MODEL_NAME) != null) {
+                props.put(SDNC_MODEL_VERSION, String.valueOf(properties.get(SDNC_MODEL_VERSION).getValue()));
+                props.put(SDNC_MODEL_NAME, String.valueOf(properties.get(SDNC_MODEL_NAME).getValue()));
+            }
+        }
+        return props;
+    }
+
     /**
      * Return all child Node Templates (via Substitution Mappings) that do not have a type ending VnfConfiguration.
      *
diff --git a/src/main/java/org/onap/aai/babel/xml/generator/api/AaiModelGenerator.java b/src/main/java/org/onap/aai/babel/xml/generator/api/AaiModelGenerator.java
index 5247959..50c5d5b 100644
--- a/src/main/java/org/onap/aai/babel/xml/generator/api/AaiModelGenerator.java
+++ b/src/main/java/org/onap/aai/babel/xml/generator/api/AaiModelGenerator.java
@@ -103,6 +103,8 @@ public class AaiModelGenerator {
         modelVer.setModelName(model.getModelName());
         modelVer.setModelVersion(model.getModelVersion());
         modelVer.setModelVersionId(model.getModelNameVersionId());
+        modelVer.setSdncModelName(model.getSdncModelName());
+        modelVer.setSdncModelVersion(model.getSdncModelVersion());
         modelVer.setModelElements(new ModelElements());
         return modelVer;
     }
diff --git a/src/main/java/org/onap/aai/babel/xml/generator/model/Model.java b/src/main/java/org/onap/aai/babel/xml/generator/model/Model.java
index 558b561..ad852af 100644
--- a/src/main/java/org/onap/aai/babel/xml/generator/model/Model.java
+++ b/src/main/java/org/onap/aai/babel/xml/generator/model/Model.java
@@ -85,6 +85,18 @@ public abstract class Model {
             public void populate(Model model, String value) {
                 model.modelName = model.modelDescription = value;
             }
+        },
+        SDNC_MODEL_NAME("sdnc_model_name") {
+            @Override
+            public void populate(Model model, String value) {
+                model.sdncModelName = value;
+            }
+        },
+        SDNC_MODEL_VERSION("sdnc_model_version") {
+            @Override
+            public void populate(Model model, String value) {
+                model.sdncModelVersion = value;
+            }
         };
 
         private static final Map<String, ModelIdentification> propertyToModelIdent;
@@ -118,6 +130,8 @@ public abstract class Model {
     private String modelDescription;
     private String instantiationType;
     private String category;
+    private String sdncModelVersion;
+    private String sdncModelName;
     protected Set<Resource> resources = new HashSet<>();
     protected Set<Widget> widgets = new HashSet<>();
 
@@ -229,6 +243,15 @@ public abstract class Model {
     public String getCategory() {
     	return category;
     }
+
+    public String getSdncModelName() {
+        return sdncModelName;
+    }
+
+    public String getSdncModelVersion() {
+        return sdncModelVersion;
+    }
+
     /**
      * Gets widget version id.
      *