1 files changed, 169 insertions, 0 deletions
diff --git a/openecomp-be/api/openecomp-sdc-rest-webapp/vendor-license-rest/vendor-license-rest-services/src/main/java/org/openecomp/sdcrests/vendorlicense/rest/services/FeatureGroupsImpl.java b/openecomp-be/api/openecomp-sdc-rest-webapp/vendor-license-rest/vendor-license-rest-services/src/main/java/org/openecomp/sdcrests/vendorlicense/rest/services/FeatureGroupsImpl.java
new file mode 100644
index 0000000000..daf35f9ee6
--- /dev/null
+++ b/openecomp-be/api/openecomp-sdc-rest-webapp/vendor-license-rest/vendor-license-rest-services/src/main/java/org/openecomp/sdcrests/vendorlicense/rest/services/FeatureGroupsImpl.java
@@ -0,0 +1,169 @@
+/*-
+ * ============LICENSE_START=======================================================
+ * SDC
+ * ================================================================================
+ * Copyright (C) 2017 AT&T Intellectual Property. All rights reserved.
+ * ================================================================================
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ * 
+ *      http://www.apache.org/licenses/LICENSE-2.0
+ * 
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ * ============LICENSE_END=========================================================
+ */
+
+package org.openecomp.sdcrests.vendorlicense.rest.services;
+
+import org.openecomp.core.utilities.CommonMethods;
+import org.openecomp.sdc.vendorlicense.VendorLicenseManager;
+import org.openecomp.sdc.vendorlicense.dao.types.EntitlementPoolEntity;
+import org.openecomp.sdc.vendorlicense.dao.types.FeatureGroupEntity;
+import org.openecomp.sdc.vendorlicense.dao.types.FeatureGroupModel;
+import org.openecomp.sdc.vendorlicense.dao.types.LicenseKeyGroupEntity;
+import org.openecomp.sdc.versioning.dao.types.Version;
+import org.openecomp.sdcrests.vendorlicense.rest.FeatureGroups;
+import org.openecomp.sdcrests.vendorlicense.rest.mapping.MapEntitlementPoolEntityToEntitlementPoolEntityDto;
+import org.openecomp.sdcrests.vendorlicense.rest.mapping.MapFeatureGroupDescriptorDtoToFeatureGroupEntity;
+import org.openecomp.sdcrests.vendorlicense.rest.mapping.MapFeatureGroupEntityToFeatureGroupDescriptorDto;
+import org.openecomp.sdcrests.vendorlicense.rest.mapping.MapLicenseKeyGroupEntityToLicenseKeyGroupEntityDto;
+import org.openecomp.sdcrests.vendorlicense.types.EntitlementPoolEntityDto;
+import org.openecomp.sdcrests.vendorlicense.types.FeatureGroupEntityDto;
+import org.openecomp.sdcrests.vendorlicense.types.FeatureGroupModelDto;
+import org.openecomp.sdcrests.vendorlicense.types.FeatureGroupRequestDto;
+import org.openecomp.sdcrests.vendorlicense.types.FeatureGroupUpdateRequestDto;
+import org.openecomp.sdcrests.vendorlicense.types.LicenseKeyGroupEntityDto;
+import org.openecomp.sdcrests.wrappers.GenericCollectionWrapper;
+import org.openecomp.sdcrests.wrappers.StringWrapperResponse;
+
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.context.annotation.Scope;
+import org.springframework.stereotype.Service;
+
+import java.util.Collection;
+import java.util.HashSet;
+import javax.inject.Named;
+import javax.ws.rs.core.Response;
+
+
+@Named
+@Service("featureGroups")
+@Scope(value = "prototype")
+public class FeatureGroupsImpl implements FeatureGroups {
+
+  @Autowired
+  private VendorLicenseManager vendorLicenseManager;
+
+  @Override
+  public Response listFeatureGroups(String vlmId, String version, String user) {
+    Collection<FeatureGroupEntity> featureGroupEntities =
+        vendorLicenseManager.listFeatureGroups(vlmId, Version.valueOf(version), user);
+
+    MapFeatureGroupEntityToFeatureGroupDescriptorDto outputMapper =
+        new MapFeatureGroupEntityToFeatureGroupDescriptorDto();
+    GenericCollectionWrapper<FeatureGroupEntityDto> results = new GenericCollectionWrapper<>();
+
+    for (FeatureGroupEntity fg : featureGroupEntities) {
+      FeatureGroupEntityDto fgDto = new FeatureGroupEntityDto();
+      fgDto.setId(fg.getId());
+      fgDto.setLicenseKeyGroupsIds(fg.getLicenseKeyGroupIds());
+      fgDto.setEntitlementPoolsIds(fg.getEntitlementPoolIds());
+      fgDto.setReferencingLicenseAgreements(fg.getReferencingLicenseAgreements());
+      outputMapper.doMapping(fg, fgDto);
+      results.add(fgDto);
+    }
+
+    return Response.ok(results).build();
+  }
+
+  @Override
+  public Response createFeatureGroup(FeatureGroupRequestDto request, String vlmId, String user) {
+    FeatureGroupEntity featureGroupEntity = new MapFeatureGroupDescriptorDtoToFeatureGroupEntity()
+        .applyMapping(request, FeatureGroupEntity.class);
+    featureGroupEntity.setVendorLicenseModelId(vlmId);
+    featureGroupEntity.setLicenseKeyGroupIds(request.getAddedLicenseKeyGroupsIds());
+    featureGroupEntity.setEntitlementPoolIds(request.getAddedEntitlementPoolsIds());
+
+    FeatureGroupEntity createdFeatureGroup =
+        vendorLicenseManager.createFeatureGroup(featureGroupEntity, user);
+
+    StringWrapperResponse result =
+        createdFeatureGroup != null ? new StringWrapperResponse(createdFeatureGroup.getId()) : null;
+    return Response.ok(result).build();
+  }
+
+  @Override
+  public Response updateFeatureGroup(FeatureGroupUpdateRequestDto request, String vlmId,
+                                     String featureGroupId, String user) {
+    FeatureGroupEntity featureGroupEntity = new MapFeatureGroupDescriptorDtoToFeatureGroupEntity()
+        .applyMapping(request, FeatureGroupEntity.class);
+    featureGroupEntity.setVendorLicenseModelId(vlmId);
+    featureGroupEntity.setId(featureGroupId);
+
+    vendorLicenseManager
+        .updateFeatureGroup(featureGroupEntity, request.getAddedLicenseKeyGroupsIds(),
+            request.getRemovedLicenseKeyGroupsIds(), request.getAddedEntitlementPoolsIds(),
+            request.getRemovedEntitlementPoolsIds(), user);
+    return Response.ok().build();
+  }
+
+  @Override
+  public Response getFeatureGroup(String vlmId, String version, String featureGroupId,
+                                  String user) {
+    FeatureGroupEntity fgInput = new FeatureGroupEntity();
+    fgInput.setVendorLicenseModelId(vlmId);
+    fgInput.setVersion(Version.valueOf(version));
+    fgInput.setId(featureGroupId);
+    FeatureGroupModel featureGroupModel = vendorLicenseManager.getFeatureGroupModel(fgInput, user);
+
+    if (featureGroupModel == null) {
+      return Response.ok().build();
+    }
+
+    FeatureGroupModelDto fgmDto = new FeatureGroupModelDto();
+    fgmDto.setId(featureGroupModel.getFeatureGroup().getId());
+    fgmDto.setReferencingLicenseAgreements(
+        featureGroupModel.getFeatureGroup().getReferencingLicenseAgreements());
+    new MapFeatureGroupEntityToFeatureGroupDescriptorDto()
+        .doMapping(featureGroupModel.getFeatureGroup(), fgmDto);
+
+    if (!CommonMethods.isEmpty(featureGroupModel.getLicenseKeyGroups())) {
+      fgmDto.setLicenseKeyGroups(new HashSet<>());
+
+      MapLicenseKeyGroupEntityToLicenseKeyGroupEntityDto lkgMapper =
+          new MapLicenseKeyGroupEntityToLicenseKeyGroupEntityDto();
+      for (LicenseKeyGroupEntity lkg : featureGroupModel.getLicenseKeyGroups()) {
+        fgmDto.getLicenseKeyGroups()
+            .add(lkgMapper.applyMapping(lkg, LicenseKeyGroupEntityDto.class));
+      }
+    }
+
+    if (!CommonMethods.isEmpty(featureGroupModel.getEntitlementPools())) {
+      fgmDto.setEntitlementPools(new HashSet<>());
+
+      MapEntitlementPoolEntityToEntitlementPoolEntityDto epMapper =
+          new MapEntitlementPoolEntityToEntitlementPoolEntityDto();
+      for (EntitlementPoolEntity ep : featureGroupModel.getEntitlementPools()) {
+        fgmDto.getEntitlementPools().add(epMapper.applyMapping(ep, EntitlementPoolEntityDto.class));
+
+      }
+    }
+
+    return Response.ok(fgmDto).build();
+  }
+
+  @Override
+  public Response deleteFeatureGroup(String vlmId, String featureGroupId, String user) {
+    FeatureGroupEntity fgInput = new FeatureGroupEntity();
+    fgInput.setVendorLicenseModelId(vlmId);
+    fgInput.setId(featureGroupId);
+    vendorLicenseManager.deleteFeatureGroup(fgInput, user);
+    return Response.ok().build();
+  }
+
+}