d87c9c8dbb5517975c2c93dde84de2085b1914c5
[sdc.git] /
1 /*-
2  * ============LICENSE_START=======================================================
3  * SDC
4  * ================================================================================
5  * Copyright (C) 2019 AT&T Intellectual Property. All rights reserved.
6  * ================================================================================
7  * Licensed under the Apache License, Version 2.0 (the "License");
8  * you may not use this file except in compliance with the License.
9  * You may obtain a copy of the License at
10  * 
11  *      http://www.apache.org/licenses/LICENSE-2.0
12  * 
13  * Unless required by applicable law or agreed to in writing, software
14  * distributed under the License is distributed on an "AS IS" BASIS,
15  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
16  * See the License for the specific language governing permissions and
17  * limitations under the License.
18  * ============LICENSE_END=========================================================
19  */
20
21 package org.openecomp.sdc.vendorsoftwareproduct.impl.orchestration;
22
23 import static org.openecomp.core.validation.errors.ErrorMessagesFormatBuilder.getErrorWithParameters;
24
25 import java.io.ByteArrayInputStream;
26 import java.util.Optional;
27 import org.apache.commons.collections4.MapUtils;
28 import org.openecomp.core.utilities.file.FileContentHandler;
29 import org.openecomp.core.utilities.orchestration.OnboardingTypesEnum;
30 import org.openecomp.sdc.common.errors.Messages;
31 import org.openecomp.sdc.common.utils.SdcCommon;
32 import org.openecomp.sdc.datatypes.error.ErrorLevel;
33 import org.openecomp.sdc.datatypes.error.ErrorMessage;
34 import org.openecomp.sdc.logging.api.Logger;
35 import org.openecomp.sdc.logging.api.LoggerFactory;
36 import org.openecomp.sdc.vendorsoftwareproduct.dao.type.VspDetails;
37 import org.openecomp.sdc.vendorsoftwareproduct.services.filedatastructuremodule.CandidateService;
38 import org.openecomp.sdc.vendorsoftwareproduct.types.OnboardPackage;
39 import org.openecomp.sdc.vendorsoftwareproduct.types.OnboardPackageInfo;
40 import org.openecomp.sdc.vendorsoftwareproduct.types.UploadFileResponse;
41
42 public abstract class BaseOrchestrationTemplateHandler implements OrchestrationTemplateFileHandler {
43   protected static final Logger logger = LoggerFactory.getLogger(BaseOrchestrationTemplateHandler.class);
44
45   @Override
46   public UploadFileResponse upload(final VspDetails vspDetails,
47                                    final OnboardPackageInfo onboardPackageInfo,
48                                    final CandidateService candidateService) {
49     final OnboardPackage onboardPackage = onboardPackageInfo.getOnboardPackage();
50     final UploadFileResponse uploadFileResponse = new UploadFileResponse();
51     uploadFileResponse.setOnboardingType(getHandlerType());
52     if (isFileFileToUploadEmpty(onboardPackage, uploadFileResponse, candidateService)) {
53       return uploadFileResponse;
54     }
55
56     final byte[] fileContentByteArray = onboardPackage.getFileContent().array();
57     if (isInvalidRawZipData(onboardPackage.getFileExtension(),
58         uploadFileResponse, fileContentByteArray, candidateService)) {
59       return uploadFileResponse;
60     }
61
62     final Optional<FileContentHandler> optionalContentMap =
63         getFileContentMap(uploadFileResponse, fileContentByteArray);
64     if (!optionalContentMap.isPresent()) {
65       logger.error(getErrorWithParameters(Messages.FILE_CONTENT_MAP.getErrorMessage(),
66           getHandlerType().toString()));
67       uploadFileResponse.addStructureError(SdcCommon.UPLOAD_FILE, new ErrorMessage(ErrorLevel.ERROR,
68           getErrorWithParameters(Messages.FILE_CONTENT_MAP.getErrorMessage(),
69               getHandlerType().toString())));
70       return uploadFileResponse;
71     }
72
73     if (!MapUtils.isEmpty(uploadFileResponse.getErrors())) {
74       return uploadFileResponse;
75     }
76     if (updateCandidateData(vspDetails, onboardPackageInfo, candidateService, uploadFileResponse,
77         optionalContentMap.get())) {
78       return uploadFileResponse;
79     }
80     return uploadFileResponse;
81
82   }
83
84   protected abstract boolean updateCandidateData(final VspDetails vspDetails,
85                                                  final OnboardPackageInfo onboardPackageInfo,
86                                                  final CandidateService candidateService,
87                                                  final UploadFileResponse uploadFileResponse,
88                                                  final FileContentHandler contentMap);
89
90   private boolean isFileFileToUploadEmpty(final OnboardPackage onboardPackage,
91                                           final UploadFileResponse uploadFileResponse,
92                                           final CandidateService candidateService) {
93     final ByteArrayInputStream fileToUpload = new ByteArrayInputStream(
94         onboardPackage.getFileContent().array());
95     Optional<ErrorMessage> errorMessage =
96         candidateService.validateNonEmptyFileToUpload(fileToUpload, onboardPackage.getFileExtension());
97     if (errorMessage.isPresent()) {
98       uploadFileResponse.addStructureError(SdcCommon.UPLOAD_FILE, errorMessage.get());
99       return true;
100     }
101     return false;
102   }
103
104   protected boolean isInvalidRawZipData(String fileSuffix,
105                                         UploadFileResponse uploadFileResponse,
106                                         byte[] uploadedFileData,
107                                         CandidateService candidateService) {
108     Optional<ErrorMessage> errorMessage;
109     errorMessage = candidateService.validateRawZipData(fileSuffix, uploadedFileData);
110     if (errorMessage.isPresent()) {
111       uploadFileResponse.addStructureError(SdcCommon.UPLOAD_FILE, errorMessage.get());
112       return true;
113     }
114     return false;
115   }
116
117   public abstract Optional<FileContentHandler> getFileContentMap(
118       UploadFileResponse uploadFileResponse,
119       byte[] uploadedFileData);
120
121   protected abstract OnboardingTypesEnum getHandlerType();
122 }