[SDC-29] Amdocs OnBoard 1707 initial commit.

Change-Id: Ie4d12a3f574008b792899b368a0902a8b46b5370
Signed-off-by: AviZi <avi.ziv@amdocs.com>

2 files changed, 180 insertions, 0 deletions

diff --git a/openecomp-be/lib/openecomp-core-lib/openecomp-zusammen-lib/openecomp-zusammen-plugin/src/main/java/org/openecomp/core/zusammen/plugin/collaboration/ElementCollaborationStore.java b/openecomp-be/lib/openecomp-core-lib/openecomp-zusammen-lib/openecomp-zusammen-plugin/src/main/java/org/openecomp/core/zusammen/plugin/collaboration/ElementCollaborationStore.java
new file mode 100644
index 0000000000..93ee2d058b
--- /dev/null
+++ b/openecomp-be/lib/openecomp-core-lib/openecomp-zusammen-lib/openecomp-zusammen-plugin/src/main/java/org/openecomp/core/zusammen/plugin/collaboration/ElementCollaborationStore.java
@@ -0,0 +1,101 @@
+package org.openecomp.core.zusammen.plugin.collaboration;
+
+import com.amdocs.zusammen.datatypes.Id;
+import com.amdocs.zusammen.datatypes.SessionContext;
+import com.amdocs.zusammen.datatypes.item.ElementContext;
+import com.amdocs.zusammen.plugin.statestore.cassandra.dao.types.ElementEntityContext;
+import com.amdocs.zusammen.sdk.collaboration.types.CollaborationElement;
+import org.openecomp.core.zusammen.plugin.ZusammenPluginConstants;
+import org.openecomp.core.zusammen.plugin.ZusammenPluginUtil;
+import org.openecomp.core.zusammen.plugin.dao.ElementRepository;
+import org.openecomp.core.zusammen.plugin.dao.ElementRepositoryFactory;
+import org.openecomp.core.zusammen.plugin.dao.types.ElementEntity;
+
+import java.util.Collection;
+import java.util.HashSet;
+import java.util.Objects;
+import java.util.Optional;
+import java.util.stream.Collectors;
+
+import static org.openecomp.core.zusammen.plugin.ZusammenPluginUtil.getSpaceName;
+
+
+public class ElementCollaborationStore {
+
+  public Collection<CollaborationElement> listElements(SessionContext context,
+                                                       ElementContext elementContext,
+                                                       Id elementId) {
+    ElementEntityContext elementEntityContext =
+        new ElementEntityContext(ZusammenPluginUtil.getPrivateSpaceName(context), elementContext);
+
+    if (elementId == null) {
+      elementId = ZusammenPluginConstants.ROOT_ELEMENTS_PARENT_ID;
+    }
+
+    ElementRepository elementRepository = getElementRepository(context);
+    return elementRepository.get(context, elementEntityContext, new ElementEntity(elementId))
+        .map(ElementEntity::getSubElementIds).orElse(new HashSet<>()).stream()
+        .map(subElementId -> elementRepository
+            .get(context, elementEntityContext, new ElementEntity(subElementId)).get())
+        .filter(Objects::nonNull)
+        .map(subElement -> ZusammenPluginUtil
+            .getCollaborationElement(elementEntityContext, subElement))
+        .collect(Collectors.toList());
+  }
+
+  public CollaborationElement getElement(SessionContext context, ElementContext elementContext,
+                                         Id elementId) {
+    ElementEntityContext elementEntityContext =
+        new ElementEntityContext(ZusammenPluginUtil.getPrivateSpaceName(context), elementContext);
+    return getElementRepository(context)
+        .get(context, elementEntityContext, new ElementEntity(elementId))
+        .map(elementEntity -> ZusammenPluginUtil
+            .getCollaborationElement(elementEntityContext, elementEntity))
+        .orElse(null);
+  }
+
+  public void createElement(SessionContext context, CollaborationElement element) {
+    getElementRepository(context)
+        .create(context,
+            new ElementEntityContext(getSpaceName(context, element.getSpace()),
+                element.getItemId(), element.getVersionId()),
+            ZusammenPluginUtil.getElementEntity(element));
+  }
+
+  public void updateElement(SessionContext context, CollaborationElement element) {
+    getElementRepository(context)
+        .update(context,
+            new ElementEntityContext(getSpaceName(context, element.getSpace()),
+                element.getItemId(), element.getVersionId()),
+            ZusammenPluginUtil.getElementEntity(element));
+  }
+
+  public void deleteElement(SessionContext context, CollaborationElement element) {
+    deleteElementHierarchy(getElementRepository(context),
+        context,
+        new ElementEntityContext(getSpaceName(context, element.getSpace()),
+            element.getItemId(), element.getVersionId()),
+        ZusammenPluginUtil.getElementEntity(element));
+  }
+
+  private void deleteElementHierarchy(ElementRepository elementRepository, SessionContext context,
+                                      ElementEntityContext elementEntityContext,
+                                      ElementEntity elementEntity) {
+    Optional<ElementEntity> retrieved =
+        elementRepository.get(context, elementEntityContext, elementEntity);
+    if (!retrieved.isPresent()) {
+      return;
+    }
+    retrieved.get().getSubElementIds().stream()
+        .map(ElementEntity::new)
+        .forEach(subElementEntity -> deleteElementHierarchy(
+            elementRepository, context, elementEntityContext, subElementEntity));
+
+    // only for the first one the parentId will populated (so it'll be removed from its parent)
+    elementRepository.delete(context, elementEntityContext, elementEntity);
+  }
+
+  protected ElementRepository getElementRepository(SessionContext context) {
+    return ElementRepositoryFactory.getInstance().createInterface(context);
+  }
+}
diff --git a/openecomp-be/lib/openecomp-core-lib/openecomp-zusammen-lib/openecomp-zusammen-plugin/src/main/java/org/openecomp/core/zusammen/plugin/collaboration/VersionCollaborationStore.java b/openecomp-be/lib/openecomp-core-lib/openecomp-zusammen-lib/openecomp-zusammen-plugin/src/main/java/org/openecomp/core/zusammen/plugin/collaboration/VersionCollaborationStore.java
new file mode 100644
index 0000000000..ae23b6e8a9
--- /dev/null
+++ b/openecomp-be/lib/openecomp-core-lib/openecomp-zusammen-lib/openecomp-zusammen-plugin/src/main/java/org/openecomp/core/zusammen/plugin/collaboration/VersionCollaborationStore.java
@@ -0,0 +1,79 @@
+package org.openecomp.core.zusammen.plugin.collaboration;
+
+import com.amdocs.zusammen.datatypes.Id;
+import com.amdocs.zusammen.datatypes.SessionContext;
+import com.amdocs.zusammen.datatypes.Space;
+import com.amdocs.zusammen.datatypes.item.Action;
+import com.amdocs.zusammen.datatypes.itemversion.Tag;
+import com.amdocs.zusammen.plugin.statestore.cassandra.dao.types.ElementEntityContext;
+import com.amdocs.zusammen.sdk.collaboration.types.CollaborationElementChange;
+import com.amdocs.zusammen.sdk.collaboration.types.CollaborationMergeChange;
+import org.openecomp.core.zusammen.plugin.ZusammenPluginUtil;
+import org.openecomp.core.zusammen.plugin.dao.ElementRepository;
+import org.openecomp.core.zusammen.plugin.dao.ElementRepositoryFactory;
+import org.openecomp.core.zusammen.plugin.dao.types.ElementEntity;
+
+import java.util.Collection;
+
+import static org.openecomp.core.zusammen.plugin.ZusammenPluginUtil.getSpaceName;
+
+public class VersionCollaborationStore {
+
+  public void tagItemVersion(SessionContext context, Id itemId, Id versionId, Id changeId,
+                             Tag tag) {
+    if (changeId != null) {
+      throw new UnsupportedOperationException(
+          "In this plugin implementation tag is supported only on versionId");
+    }
+    copyElements(context, getSpaceName(context, Space.PRIVATE), itemId, versionId, tag.getName());
+  }
+
+  public CollaborationMergeChange resetItemVersionHistory(SessionContext context, Id itemId,
+                                                          Id versionId, String changeRef) {
+    ElementRepository elementRepository = getElementRepository(context);
+    ElementEntityContext elementContext =
+        new ElementEntityContext(getSpaceName(context, Space.PRIVATE), itemId, versionId);
+
+    CollaborationMergeChange resetChange = new CollaborationMergeChange();
+
+    Collection<ElementEntity> versionElements = elementRepository.list(context, elementContext);
+    versionElements.stream()
+        .map(elementEntity ->
+            convertElementEntityToElementChange(elementEntity, elementContext, Action.DELETE))
+        .forEach(resetChange.getChangedElements()::add);
+
+    elementContext.setChangeRef(changeRef);
+    Collection<ElementEntity> changeRefElements = elementRepository.list(context, elementContext);
+    changeRefElements.stream()
+        .map(elementEntity ->
+            convertElementEntityToElementChange(elementEntity, elementContext, Action.CREATE))
+        .forEach(resetChange.getChangedElements()::add);
+
+    return resetChange; // TODO: 4/19/2017 version change...
+  }
+
+  private void copyElements(SessionContext context, String space, Id itemId, Id sourceVersionId,
+                            String targetTag) {
+    ElementRepository elementRepository = getElementRepository(context);
+    ElementEntityContext elementContext = new ElementEntityContext(space, itemId, sourceVersionId);
+
+    Collection<ElementEntity> versionElements = elementRepository.list(context, elementContext);
+
+    elementContext.setChangeRef(targetTag);
+    versionElements
+        .forEach(elementEntity -> elementRepository.create(context, elementContext, elementEntity));
+  }
+
+  private CollaborationElementChange convertElementEntityToElementChange(
+      ElementEntity elementEntity, ElementEntityContext elementContext, Action action) {
+    CollaborationElementChange elementChange = new CollaborationElementChange();
+    elementChange
+        .setElement(ZusammenPluginUtil.getCollaborationElement(elementContext, elementEntity));
+    elementChange.setAction(action);
+    return elementChange;
+  }
+
+  protected ElementRepository getElementRepository(SessionContext context) {
+    return ElementRepositoryFactory.getInstance().createInterface(context);
+  }
+}