1 files changed, 284 insertions, 0 deletions
diff --git a/bpmn/MSOCommonBPMN/src/main/java/org/onap/so/client/cds/NssiCDSRequestProvider.java b/bpmn/MSOCommonBPMN/src/main/java/org/onap/so/client/cds/NssiCDSRequestProvider.java
new file mode 100644
index 0000000000..d283092e0a
--- /dev/null
+++ b/bpmn/MSOCommonBPMN/src/main/java/org/onap/so/client/cds/NssiCDSRequestProvider.java
@@ -0,0 +1,284 @@
+/*-
+ * ============LICENSE_START=======================================================
+ * ONAP - SO
+ * ================================================================================
+ * Copyright (c) 2022 Deutsche telekom
+ * ================================================================================
+ * 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.onap.so.client.cds;
+
+import com.fasterxml.jackson.core.JsonProcessingException;
+import com.fasterxml.jackson.databind.ObjectMapper;
+import com.google.gson.JsonObject;
+import java.util.ArrayList;
+import org.onap.aai.domain.yang.ServiceInstances;
+import org.onap.aaiclient.client.aai.entities.CustomQuery;
+import org.onap.aaiclient.client.aai.entities.uri.AAIPluralResourceUri;
+import org.onap.aaiclient.client.aai.entities.uri.AAIResourceUri;
+import org.onap.aaiclient.client.aai.entities.uri.AAIUriFactory;
+import org.onap.aaiclient.client.aai.entities.uri.NodesSingleUri;
+import org.onap.aaiclient.client.generated.fluentbuilders.AAIFluentTypeBuilder;
+import org.onap.aaiclient.client.graphinventory.Format;
+import org.onap.so.bpmn.common.BuildingBlockExecution;
+import org.onap.so.bpmn.common.InjectionHelper;
+import org.onap.so.bpmn.core.json.JsonUtils;
+import org.onap.so.bpmn.servicedecomposition.bbobjects.ServiceInstance;
+import org.onap.so.bpmn.servicedecomposition.entities.GeneralBuildingBlock;
+import org.onap.so.bpmn.servicedecomposition.entities.ResourceKey;
+import org.onap.so.bpmn.servicedecomposition.tasks.ExtractPojosForBB;
+import org.onap.so.client.exception.BBObjectNotFoundException;
+import org.onap.so.client.exception.PayloadGenerationException;
+import org.onap.so.moi.Attributes;
+import org.onap.so.moi.Snssai;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.beans.factory.config.ConfigurableBeanFactory;
+import org.springframework.context.annotation.Scope;
+import org.springframework.stereotype.Component;
+import java.util.List;
+import java.util.Map;
+import java.util.Optional;
+import static org.onap.so.client.cds.PayloadConstants.PROPERTIES;
+import static org.onap.so.client.cds.PayloadConstants.SEPARATOR;
+
+@Component
+@Scope(value = ConfigurableBeanFactory.SCOPE_PROTOTYPE)
+public class NssiCDSRequestProvider implements CDSRequestProvider {
+
+    private static final Logger LOGGER = LoggerFactory.getLogger(NssiCDSRequestProvider.class);
+
+    private String blueprintName;
+    private String blueprintVersion;
+    private BuildingBlockExecution execution;
+
+    private static final String DELETE_SLICE_PROFILE_ACTION = "delete-sliceprofile";
+    private static final String CREATE_SLICE_PROFILE_ACTION = "create-sliceprofile";
+    private static final String MODIFY_SLICE_PROFILE_ACTION = "modify-sliceprofile";
+    private static final String SERVICE_INSTANCE_KEY = "service-instance";
+    private static final String SERVICE_INSTANCE_ID_KEY = "service-instance-id";
+    private static final String NSSI = "nssi";
+    private static final String NSSI_ID = "nssiId";
+    private static final String NSSI_NAME = "nssiName";
+    private static final String NSI_NAME = "nsiName";
+    private static final String NSI_ID = "nsiId";
+    private static final String SLICE_PROFILE_INSTANCE_ID = "sliceProfileInstanceId";
+    private static final String SLICE_INSTANCE_FROM_PROFILE_ID_CUSTOM_QUERY =
+            "related-to?startingNodeType=slice-profile&relatedToNodeType=service-instance";
+    private static final String ENVIRONMENT_CONTEXT_QUERY_PARAM = "environment-context";
+    private static final String AAI_SUPPORTED_SLICE_PROFILE =
+            "latency|maxNumberofUEs|coverageAreaTAList|areaTrafficCapDL|resourceSharingLevel|serviceType|uEMobilityLevel|expDataRateUL|expDataRateDL";
+    private static final ObjectMapper mapper = new ObjectMapper();
+
+    JsonUtils jsonUtil = new JsonUtils();
+
+    @Autowired
+    private ExtractPojosForBB extractPojosForBB;
+
+    @Autowired
+    private InjectionHelper injectionHelper;
+
+    @Override
+    public String getBlueprintName() {
+        return blueprintName;
+    }
+
+    @Override
+    public String getBlueprintVersion() {
+        return blueprintVersion;
+    }
+
+    @Override
+    public <T> void setExecutionObject(T executionObject) {
+        execution = (BuildingBlockExecution) executionObject;
+    }
+
+    @Override
+    public Optional<String> buildRequestPayload(String action) throws PayloadGenerationException {
+        JsonObject cdsPropertyObject = null;
+
+        final GeneralBuildingBlock buildingBlock = execution.getGeneralBuildingBlock();
+        List<Map<String, Object>> userParamsFromRequest =
+                buildingBlock.getRequestContext().getRequestParameters().getUserParams();
+
+        try {
+            ServiceInstance serviceInstance =
+                    extractPojosForBB.extractByKey(execution, ResourceKey.SERVICE_INSTANCE_ID);
+            // get values from service Instance
+            blueprintName = serviceInstance.getModelInfoServiceInstance().getBlueprintName();
+            blueprintVersion = serviceInstance.getModelInfoServiceInstance().getBlueprintVersion();
+
+        } catch (Exception e) {
+            throw new PayloadGenerationException("Failed to buildPropertyObjectForNssi", e);
+        }
+
+        cdsPropertyObject = setCreateSliceProfileRequestValues(action, buildingBlock);
+
+        return Optional.of(buildRequestJsonObject(cdsPropertyObject, action));
+    }
+
+    private JsonObject setCreateSliceProfileRequestValues(String action, GeneralBuildingBlock buildingBlock) {
+
+        JsonObject cdsRequestObject = new JsonObject();
+
+        ServiceInstance serviceInstance = null;
+        try {
+            serviceInstance = extractPojosForBB.extractByKey(execution, ResourceKey.SERVICE_INSTANCE_ID);
+        } catch (BBObjectNotFoundException e) {
+            e.printStackTrace();
+        }
+
+        List<Map<String, Object>> userParamsFromRequest =
+                buildingBlock.getRequestContext().getRequestParameters().getUserParams();
+
+        Attributes attributes = null;
+
+        for (Map<String, Object> userParamData : userParamsFromRequest) {
+            if (userParamData.get(NSSI) != null) {
+                attributes = mapper.convertValue(userParamData.get(NSSI), Attributes.class);
+            }
+        }
+        // Value to come as a request == Hardcoded for now
+        String nssiId = serviceInstance.getServiceInstanceId();
+        String nssiName = getNssiName(nssiId);
+        String nsiId = "nssi-id";
+        String nsiName = "nssi-name";
+
+        cdsRequestObject.addProperty(NSSI_ID, nssiId);
+
+        if (!DELETE_SLICE_PROFILE_ACTION.equals(action)) {
+            cdsRequestObject.addProperty(NSSI_NAME, nssiName);
+            cdsRequestObject.addProperty(NSI_NAME, nsiName);
+        }
+
+        String sliceProfileInstanceId = null;
+        if (CREATE_SLICE_PROFILE_ACTION.equalsIgnoreCase(action)) {
+            Snssai snssai = attributes.getSliceProfileList().get(0).getPlmnInfoList().get(0).getSnssai();
+            String sNssaiString = snssai.getSst() + SEPARATOR + snssai.getSd();
+            sliceProfileInstanceId = getSliceProfileInstanceIdForNssi(sNssaiString);
+        } else {
+            String sliceProfileId = attributes.getSliceProfileList().get(0).getSliceProfileId();
+            sliceProfileInstanceId = getSliceProfileInstanceFromSliceProfileId(sliceProfileId);
+        }
+
+        cdsRequestObject.addProperty(SLICE_PROFILE_INSTANCE_ID, sliceProfileInstanceId);
+
+        cdsRequestObject.addProperty(NSI_ID, nsiId);
+        /*
+         * JsonObject nssiPropertyObject = setSliceProfileProperties(getSliceProfilesFromUserParams(buildingBlock));
+         * cdsRequestObject.add(action + SEPARATOR + PROPERTIES, nssiPropertyObject);
+         */
+        return cdsRequestObject;
+    }
+
+    private Map<String, Object> getSliceProfilesFromUserParams(GeneralBuildingBlock gBB) {
+
+
+        List<Map<String, Object>> mapUserParams = gBB.getRequestContext().getRequestParameters().getUserParams();
+
+        Map<String, Object> sliceProfileMap = null;
+        try {
+            String userParamsJson = mapper.writeValueAsString(mapUserParams.get(0));
+            String rANSliceSubnetProfile =
+                    jsonUtil.getJsonParamValue(userParamsJson, "nssi.sliceProfileList", "RANSliceSubnetProfile");
+
+            if (rANSliceSubnetProfile != null) {
+                sliceProfileMap = mapper.readValue(rANSliceSubnetProfile, Map.class);
+            }
+        } catch (JsonProcessingException e) {
+            e.printStackTrace();
+        }
+        return sliceProfileMap;
+    }
+
+    // May be needed later
+
+    /*
+     * private JsonObject setSliceProfileProperties(Map<String, Object> userParamsMap) { JsonObject
+     * sliceProfilePropertiesNotPresentInAai = new JsonObject();
+     * 
+     * if (userParamsMap != null) { userParamsMap.forEach((k, v) -> { if (!AAI_SUPPORTED_SLICE_PROFILE.contains((String)
+     * k)) { sliceProfilePropertiesNotPresentInAai.addProperty(k, v.toString()); } }); }
+     * 
+     * return sliceProfilePropertiesNotPresentInAai; }
+     */
+
+    private String getSliceProfileInstanceFromSliceProfileId(String sliceProfileId) {
+
+        List<AAIResourceUri> startNodes = new ArrayList<>();
+        startNodes.add(
+                AAIUriFactory.createNodesUri(AAIFluentTypeBuilder.Types.SLICE_PROFILE.getFragment(sliceProfileId)));
+
+        CustomQuery customQuery = new CustomQuery(startNodes, SLICE_INSTANCE_FROM_PROFILE_ID_CUSTOM_QUERY);
+
+        String results = injectionHelper.getAaiQueryClient().query(Format.RESOURCE, customQuery);
+
+        if (results == null || results.isEmpty()) {
+            throw new RuntimeException("Slice Profile Instance Not found");
+        }
+
+        Map<String, List<Map<String, Object>>> serviceInstancesMap = null;
+        try {
+            serviceInstancesMap = mapper.readValue(results, Map.class);
+        } catch (JsonProcessingException e) {
+            e.printStackTrace();
+        }
+        Map<String, Object> serviceInstance =
+                (Map<String, Object>) serviceInstancesMap.get("results").get(0).get(SERVICE_INSTANCE_KEY);
+
+        return (String) serviceInstance.get(SERVICE_INSTANCE_ID_KEY);
+    }
+
+    private String getSliceProfileInstanceIdForNssi(String sNssai) {
+
+        String sliceProfileInstanceId = null;
+
+        try {
+            AAIPluralResourceUri uriSI =
+                    AAIUriFactory.createNodesUri(AAIFluentTypeBuilder.Types.SERVICE_INSTANCES.getFragment())
+                            .queryParam(ENVIRONMENT_CONTEXT_QUERY_PARAM, sNssai);
+            Optional<ServiceInstances> sliceProfileInstancesOptional =
+                    injectionHelper.getAaiClient().get(ServiceInstances.class, uriSI);
+
+            if (sliceProfileInstancesOptional.isPresent()) {
+                sliceProfileInstanceId =
+                        sliceProfileInstancesOptional.get().getServiceInstance().get(0).getServiceInstanceId();
+            }
+        } catch (Exception e) {
+            LOGGER.error("Error in getting sliceProfile Instance" + e.getMessage());
+        }
+        return sliceProfileInstanceId;
+    }
+
+    private String getNssiName(String nssiId) {
+
+        String nssiName = null;
+
+        try {
+            NodesSingleUri uriSI =
+                    AAIUriFactory.createNodesUri(AAIFluentTypeBuilder.Types.SERVICE_INSTANCE.getFragment(nssiId));
+            Optional<org.onap.aai.domain.yang.ServiceInstance> sliceProfileInstancesOptional =
+                    injectionHelper.getAaiClient().get(org.onap.aai.domain.yang.ServiceInstance.class, uriSI);
+
+            if (sliceProfileInstancesOptional.isPresent()) {
+                nssiName = sliceProfileInstancesOptional.get().getServiceInstanceName();
+            }
+        } catch (Exception e) {
+            LOGGER.error("Error in getting Nssi Instance" + e.getMessage());
+        }
+        return nssiName;
+    }
+}