1 files changed, 100 insertions, 0 deletions
diff --git a/openecomp-be/backend/openecomp-sdc-vendor-software-product-manager/src/main/java/org/openecomp/sdc/vendorsoftwareproduct/impl/orchestration/BaseOrchestrationTemplateHandler.java b/openecomp-be/backend/openecomp-sdc-vendor-software-product-manager/src/main/java/org/openecomp/sdc/vendorsoftwareproduct/impl/orchestration/BaseOrchestrationTemplateHandler.java
new file mode 100644
index 0000000000..c7fb43770d
--- /dev/null
+++ b/openecomp-be/backend/openecomp-sdc-vendor-software-product-manager/src/main/java/org/openecomp/sdc/vendorsoftwareproduct/impl/orchestration/BaseOrchestrationTemplateHandler.java
@@ -0,0 +1,100 @@
+package org.openecomp.sdc.vendorsoftwareproduct.impl.orchestration;
+
+import org.apache.commons.collections4.MapUtils;
+import org.openecomp.core.utilities.file.FileContentHandler;
+import org.openecomp.core.utilities.file.FileUtils;
+import org.openecomp.core.utilities.orchestration.OnboardingTypesEnum;
+import org.openecomp.sdc.common.errors.Messages;
+import org.openecomp.sdc.common.utils.SdcCommon;
+import org.openecomp.sdc.datatypes.error.ErrorLevel;
+import org.openecomp.sdc.datatypes.error.ErrorMessage;
+import org.openecomp.sdc.logging.api.Logger;
+import org.openecomp.sdc.logging.api.LoggerFactory;
+import org.openecomp.sdc.logging.context.impl.MdcDataDebugMessage;
+import org.openecomp.sdc.vendorsoftwareproduct.dao.type.VspDetails;
+import org.openecomp.sdc.vendorsoftwareproduct.services.filedatastructuremodule.CandidateService;
+import org.openecomp.sdc.vendorsoftwareproduct.types.UploadFileResponse;
+import org.openecomp.sdc.versioning.dao.types.Version;
+
+import java.io.InputStream;
+import java.util.Optional;
+import static org.openecomp.core.validation.errors.ErrorMessagesFormatBuilder.getErrorWithParameters;
+public abstract class BaseOrchestrationTemplateHandler implements OrchestrationTemplateFileHandler {
+    protected static final Logger logger =
+            LoggerFactory.getLogger(BaseOrchestrationTemplateHandler.class);
+    protected static MdcDataDebugMessage mdcDataDebugMessage = new MdcDataDebugMessage();
+
+    @Override
+    public UploadFileResponse upload(String vspId, Version version, InputStream fileToUpload, String user,
+                                     CandidateService candidateService, VspDetails vspDetails) {
+        UploadFileResponse uploadFileResponse = new UploadFileResponse();
+        uploadFileResponse.setOnboardingType(getHandlerType());
+        if (isNotEmptyFileToUpload(fileToUpload, uploadFileResponse, candidateService)) {
+            return uploadFileResponse;
+        }
+
+        byte[] uploadedFileData = FileUtils.toByteArray(fileToUpload);
+        if (isInvalidRawZipData(uploadFileResponse, uploadedFileData, candidateService)) {
+            return uploadFileResponse;
+        }
+
+        Optional<FileContentHandler> optionalContentMap =
+                getFileContentMap(uploadFileResponse, uploadedFileData);
+        if (!optionalContentMap.isPresent()) {
+            logger.error(getErrorWithParameters(Messages.FILE_CONTENT_MAP.getErrorMessage(), getHandlerType().toString()));
+            uploadFileResponse
+                    .addStructureError(
+                            SdcCommon.UPLOAD_FILE,
+                            new ErrorMessage(ErrorLevel.ERROR,
+                                    getErrorWithParameters(Messages.FILE_CONTENT_MAP.getErrorMessage(), getHandlerType().toString())));
+
+            mdcDataDebugMessage.debugExitMessage("VSP id", vspId);
+            return uploadFileResponse;
+        }
+
+        if (!MapUtils.isEmpty(uploadFileResponse.getErrors())) {
+
+            mdcDataDebugMessage.debugExitMessage("VSP id", vspId);
+            return uploadFileResponse;
+        }
+        if (updateCandidateData(vspId, user, candidateService, vspDetails, uploadFileResponse,
+                uploadedFileData, optionalContentMap))
+            return uploadFileResponse;
+
+        mdcDataDebugMessage.debugExitMessage("VSP id", vspId);
+        return uploadFileResponse;
+
+    }
+
+    protected abstract boolean updateCandidateData(String vspId, String user, CandidateService candidateService,
+                                                   VspDetails vspDetails, UploadFileResponse uploadFileResponse,
+                                                   byte[] uploadedFileData,
+                                                   Optional<FileContentHandler> optionalContentMap);
+
+
+    private boolean isNotEmptyFileToUpload(InputStream fileToUpload,
+                                           UploadFileResponse uploadFileResponse, CandidateService candidateService) {
+        Optional<ErrorMessage> errorMessage =
+                candidateService.validateNonEmptyFileToUpload(fileToUpload);
+        if (errorMessage.isPresent()) {
+            uploadFileResponse.addStructureError(SdcCommon.UPLOAD_FILE, errorMessage.get());
+            return true;
+        }
+        return false;
+    }
+
+    protected boolean isInvalidRawZipData(UploadFileResponse uploadFileResponse,
+                                        byte[] uploadedFileData, CandidateService candidateService) {
+        Optional<ErrorMessage> errorMessage;
+        errorMessage = candidateService.validateRawZipData(uploadedFileData);
+        if (errorMessage.isPresent()) {
+            uploadFileResponse.addStructureError(SdcCommon.UPLOAD_FILE, errorMessage.get());
+            return true;
+        }
+        return false;
+    }
+
+    public abstract Optional<FileContentHandler> getFileContentMap(UploadFileResponse uploadFileResponse,
+                                                                   byte[] uploadedFileData);
+    protected abstract OnboardingTypesEnum getHandlerType();
+}