1 files changed, 208 insertions, 0 deletions
diff --git a/openecomp-be/lib/openecomp-sdc-model-lib/openecomp-sdc-model-impl/src/main/java/org/openecomp/sdc/model/impl/zusammen/ServiceModelDaoZusammenImpl.java b/openecomp-be/lib/openecomp-sdc-model-lib/openecomp-sdc-model-impl/src/main/java/org/openecomp/sdc/model/impl/zusammen/ServiceModelDaoZusammenImpl.java
new file mode 100644
index 0000000000..530ad44574
--- /dev/null
+++ b/openecomp-be/lib/openecomp-sdc-model-lib/openecomp-sdc-model-impl/src/main/java/org/openecomp/sdc/model/impl/zusammen/ServiceModelDaoZusammenImpl.java
@@ -0,0 +1,208 @@
+package org.openecomp.sdc.model.impl.zusammen;
+
+import com.amdocs.zusammen.adaptor.inbound.api.types.item.Element;
+import com.amdocs.zusammen.adaptor.inbound.api.types.item.ElementInfo;
+import com.amdocs.zusammen.adaptor.inbound.api.types.item.ZusammenElement;
+import com.amdocs.zusammen.datatypes.Id;
+import com.amdocs.zusammen.datatypes.SessionContext;
+import com.amdocs.zusammen.datatypes.item.Action;
+import com.amdocs.zusammen.datatypes.item.ElementContext;
+import com.amdocs.zusammen.datatypes.item.Info;
+import com.amdocs.zusammen.datatypes.item.ItemVersion;
+import org.openecomp.core.model.dao.ServiceModelDao;
+import org.openecomp.core.model.types.ServiceElement;
+import org.openecomp.core.utilities.file.FileContentHandler;
+import org.openecomp.core.zusammen.api.ZusammenAdaptor;
+import org.openecomp.core.zusammen.api.ZusammenUtil;
+import org.openecomp.sdc.logging.api.Logger;
+import org.openecomp.sdc.logging.api.LoggerFactory;
+import org.openecomp.sdc.tosca.datatypes.ToscaServiceModel;
+import org.openecomp.sdc.tosca.datatypes.model.ServiceTemplate;
+import org.openecomp.sdc.tosca.services.yamlutil.ToscaExtensionYamlUtil;
+import org.openecomp.sdc.versioning.dao.types.Version;
+import org.openecomp.sdc.versioning.dao.types.VersionStatus;
+
+import java.io.ByteArrayInputStream;
+import java.util.Collection;
+import java.util.Map;
+import java.util.Optional;
+import java.util.stream.Collectors;
+
+public class ServiceModelDaoZusammenImpl
+    implements ServiceModelDao<ToscaServiceModel, ServiceElement> {
+  private static final Logger logger = LoggerFactory.getLogger(ServiceModelDaoZusammenImpl.class);
+
+  protected ZusammenAdaptor zusammenAdaptor;
+  protected String name;
+
+  public ServiceModelDaoZusammenImpl(ZusammenAdaptor zusammenAdaptor) {
+    this.zusammenAdaptor = zusammenAdaptor;
+    this.name = StructureElement.ServiceModel.name();
+  }
+
+  @Override
+  public void registerVersioning(String versionableEntityType) {
+
+  }
+
+  @Override
+  public ToscaServiceModel getServiceModel(String vspId, Version version) {
+    SessionContext context = ZusammenUtil.createSessionContext();
+    Id itemId = new Id(vspId);
+    ElementContext elementContext = new ElementContext(itemId, getFirstVersionId(context, itemId),
+        version.getStatus() == VersionStatus.Locked ? null : version.toString());
+
+    Optional<ElementInfo> serviceModelElement = zusammenAdaptor
+        .getElementInfoByName(context, elementContext, null, name);
+
+    if (serviceModelElement.isPresent()) {
+      String entryDefinitionServiceTemplate =
+          serviceModelElement.get().getInfo().getProperty("base");
+      Id serviceModelElementId = serviceModelElement.get().getId();
+      Map<String, ServiceTemplate> serviceTemplates =
+          getTemplates(context, elementContext, serviceModelElementId);
+      if (serviceTemplates == null) {
+        return null;
+      }
+      FileContentHandler artifacts = getArtifacts(context, elementContext, serviceModelElementId);
+
+
+      return new ToscaServiceModel(
+          artifacts, serviceTemplates, entryDefinitionServiceTemplate);
+    } else {
+      return null;
+    }
+  }
+
+  protected Map<String, ServiceTemplate> getTemplates(SessionContext context,
+                                                      ElementContext elementContext,
+                                                      Id serviceModelElementId) {
+    Optional<ElementInfo> templatesElementInfo = zusammenAdaptor.getElementInfoByName(
+        context, elementContext, serviceModelElementId, StructureElement.Templates.name());
+
+    if (templatesElementInfo.isPresent()) {
+
+      //Map<String, ServiceTemplate> serviceTemplateMap = new HashMap<>();
+      Collection<Element> elements = zusammenAdaptor.listElementData(context, elementContext,
+          templatesElementInfo.get().getId());
+
+      return elements.stream().collect(Collectors.toMap(
+          element -> element.getInfo().getName(),
+          this::elementToServiceTemplate));
+
+    /*elements.stream().forEach(element ->serviceTemplateMap.put(element.getInfo().getName(),
+        elementToServiceTemplate(element)));
+    return serviceTemplateMap;*/
+    }
+    return null;
+  }
+
+  protected FileContentHandler getArtifacts(SessionContext context, ElementContext elementContext,
+                                            Id serviceModelElementId) {
+    Optional<ElementInfo> artifactsElement = zusammenAdaptor.getElementInfoByName(
+        context, elementContext, serviceModelElementId, StructureElement.Artifacts.name());
+
+    if (artifactsElement.isPresent()) {
+
+      Collection<Element> elements = zusammenAdaptor.listElementData(context, elementContext,
+          artifactsElement.get().getId());
+      FileContentHandler fileContentHandler = new FileContentHandler();
+      elements.forEach(element -> fileContentHandler.addFile(element.getInfo().getName(),
+          element.getData()));
+      return fileContentHandler;
+    }
+
+    return null;
+  }
+
+  @Override
+  public void storeServiceModel(String vspId, Version version, ToscaServiceModel serviceModel) {
+    logger.info("Storing service model for vsp id -> " + vspId);
+
+    ZusammenElement templatesElement =
+        buildStructuralElement(StructureElement.Templates.name(), null);
+    serviceModel.getServiceTemplates().entrySet().forEach(entry -> templatesElement.addSubElement(
+        buildServiceTemplateElement(entry.getKey(), entry.getValue(),
+            serviceModel.getEntryDefinitionServiceTemplate(), Action.CREATE)));
+
+    ZusammenElement artifactsElement =
+        buildStructuralElement(StructureElement.Artifacts.name(), Action.UPDATE);
+    serviceModel.getArtifactFiles().getFiles().entrySet().forEach(entry -> artifactsElement
+        .addSubElement(buildArtifactElement(entry.getKey(), entry.getValue(), Action.CREATE)));
+
+    ZusammenElement serviceModelElement = buildStructuralElement(name, Action.UPDATE);
+    serviceModelElement.getInfo()
+        .addProperty("base", serviceModel.getEntryDefinitionServiceTemplate());
+
+    serviceModelElement.addSubElement(templatesElement);
+    serviceModelElement.addSubElement(artifactsElement);
+
+    SessionContext context = ZusammenUtil.createSessionContext();
+    Id itemId = new Id(vspId);
+    ElementContext elementContext = new ElementContext(itemId, getFirstVersionId(context, itemId));
+    zusammenAdaptor
+        .saveElement(context, elementContext, serviceModelElement, "Store service model");
+
+    logger.info("Finished storing service model for vsp id -> " + vspId);
+  }
+
+  @Override
+  public ServiceElement getServiceModelInfo(String vspId, Version version, String name) {
+    return null;
+  }
+
+  @Override
+  public void deleteAll(String vspId, Version version) {
+    logger.info("started deleting service model for vsp id -> " + vspId);
+    SessionContext context = ZusammenUtil.createSessionContext();
+    Id itemId = new Id(vspId);
+    ElementContext elementContext = new ElementContext(itemId, getFirstVersionId(context, itemId));
+
+    ZusammenElement zusammenElement = ZusammenUtil.buildStructuralElement(name, Action.DELETE);
+    zusammenAdaptor.saveElement(context, elementContext, zusammenElement, "delete:" + name + ".");
+    logger.info("Finished deleting service model for vsp id -> " + vspId);
+  }
+
+  protected ZusammenElement buildArtifactElement(String name, byte[] artifact, Action action) {
+    ZusammenElement artifactElement = new ZusammenElement();
+    artifactElement.setAction(action);
+    Info info = new Info();
+    info.setName(name);
+    info.addProperty("type", ElementType.Artifact.name());
+    artifactElement.setInfo(info);
+    artifactElement.setData(new ByteArrayInputStream(artifact));
+
+    return artifactElement;
+  }
+
+  private ServiceTemplate elementToServiceTemplate(Element element) {
+    return new ToscaExtensionYamlUtil().yamlToObject(element.getData(), ServiceTemplate.class);
+  }
+
+  private Element buildServiceTemplateElement(String name, ServiceTemplate serviceTemplate,
+                                              String entryDefinitionServiceTemplate,
+                                              Action action) {
+    ZusammenElement zusammenElement = new ZusammenElement();
+    zusammenElement.setAction(action);
+    Info info = new Info();
+    info.setName(name);
+    info.setDescription(serviceTemplate.getDescription());
+    info.addProperty("type", ElementType.Servicetemplate.name());
+    info.addProperty("base", entryDefinitionServiceTemplate);
+    String yaml = new ToscaExtensionYamlUtil().objectToYaml(serviceTemplate);
+    zusammenElement.setData(new ByteArrayInputStream(yaml.getBytes()));
+    zusammenElement.setInfo(info);
+    return zusammenElement;
+  }
+
+  protected Id getFirstVersionId(SessionContext context, Id vspId) {
+    Optional<ItemVersion> itemVersionOptional = zusammenAdaptor.getFirstVersion(context, vspId);
+    ItemVersion itemVersion = itemVersionOptional.orElseThrow(() ->
+        new RuntimeException(String.format("Vsp %s does not contain any version.", vspId))); //todo
+    return itemVersion.getId();
+  }
+
+  protected ZusammenElement buildStructuralElement(String structureElement, Action action) {
+    return ZusammenUtil.buildStructuralElement(structureElement, action);
+  }
+}