openecomp-be/backend/openecomp-sdc-vendor-software-product-manager/src/main/java/org/openecomp/sdc/vendorsoftwareproduct/impl/orchestration/process/OrchestrationTemplateProcessCsarHandler.java

   1 package org.openecomp.sdc.vendorsoftwareproduct.impl.orchestration.process;
   2
   3 import org.apache.commons.collections4.MapUtils;
   4 import org.openecomp.core.impl.ToscaConverterImpl;
   5 import org.openecomp.core.utilities.file.FileContentHandler;
   6 import org.openecomp.core.utilities.orchestration.OnboardingTypesEnum;
   7 import org.openecomp.core.validation.util.MessageContainerUtil;
   8 import org.openecomp.sdc.common.errors.CoreException;
   9 import org.openecomp.sdc.common.errors.ErrorCode;
  10 import org.openecomp.sdc.common.errors.GeneralErrorBuilder;
  11 import org.openecomp.sdc.datatypes.error.ErrorLevel;
  12 import org.openecomp.sdc.datatypes.error.ErrorMessage;
  13 import org.openecomp.sdc.heat.datatypes.structure.HeatStructureTree;
  14 import org.openecomp.sdc.heat.services.tree.ToscaTreeManager;
  15 import org.openecomp.sdc.logging.api.Logger;
  16 import org.openecomp.sdc.logging.api.LoggerFactory;
  17 import org.openecomp.sdc.logging.messages.AuditMessages;
  18 import org.openecomp.sdc.tosca.datatypes.ToscaServiceModel;
  19 import org.openecomp.sdc.vendorsoftwareproduct.dao.type.ComponentMonitoringUploadEntity;
  20 import org.openecomp.sdc.vendorsoftwareproduct.dao.type.OrchestrationTemplateCandidateData;
  21 import org.openecomp.sdc.vendorsoftwareproduct.dao.type.ProcessEntity;
  22 import org.openecomp.sdc.vendorsoftwareproduct.dao.type.VspDetails;
  23 import org.openecomp.sdc.vendorsoftwareproduct.factory.CandidateServiceFactory;
  24 import org.openecomp.sdc.vendorsoftwareproduct.impl.orchestration.OrchestrationUtil;
  25 import org.openecomp.sdc.vendorsoftwareproduct.services.filedatastructuremodule.CandidateService;
  26 import org.openecomp.sdc.vendorsoftwareproduct.types.OrchestrationTemplateActionResponse;
  27 import org.openecomp.sdc.vendorsoftwareproduct.types.UploadFileResponse;
  28
  29 import java.io.ByteArrayInputStream;
  30 import java.io.IOException;
  31 import java.util.ArrayList;
  32 import java.util.Collection;
  33 import java.util.HashMap;
  34 import java.util.List;
  35 import java.util.Map;
  36 import java.util.Optional;
  37
  38 public class OrchestrationTemplateProcessCsarHandler
  39     implements OrchestrationTemplateProcessHandler {
  40   private static final Logger LOGGER =
  41       LoggerFactory.getLogger(OrchestrationTemplateProcessCsarHandler.class);
  42   private CandidateService candidateService =
  43       CandidateServiceFactory.getInstance().createInterface();
  44   ToscaTreeManager toscaTreeManager = new ToscaTreeManager();
  45
  46   @Override
  47   public OrchestrationTemplateActionResponse process(VspDetails vspDetails,
  48                                                      OrchestrationTemplateCandidateData candidateData) {
  49     LOGGER.audit(
  50         AuditMessages.AUDIT_MSG + AuditMessages.CSAR_VALIDATION_STARTED + vspDetails.getId());
  51
  52     UploadFileResponse uploadFileResponse = new UploadFileResponse();
  53     Optional<FileContentHandler> fileContent = OrchestrationUtil
  54         .getFileContentMap(OnboardingTypesEnum.CSAR, uploadFileResponse,
  55             candidateData.getContentData().array());
  56
  57     OrchestrationTemplateActionResponse response = new OrchestrationTemplateActionResponse();
  58     if (fileContent.isPresent()) {
  59       try {
  60         FileContentHandler fileContentHandler = fileContent.get();
  61         processCsar(vspDetails, fileContentHandler, candidateData, response);
  62       } catch (CoreException e) {
  63         LOGGER.error(e.getMessage());
  64         response.addErrorMessageToMap(e.code().id(), e.code().message(),ErrorLevel.ERROR);
  65       } catch (IOException ioe) {
  66         LOGGER.error(ioe.getMessage());
  67         ErrorCode errorCode = new GeneralErrorBuilder(ioe.getMessage()).build();
  68         response.addErrorMessageToMap(errorCode.id(), errorCode.message(),ErrorLevel.ERROR);
  69       }
  70     } else {
  71       if (!uploadFileResponse.getErrors().isEmpty()) {
  72         response.addStructureErrors(uploadFileResponse.getErrors());
  73       }
  74     }
  75     return response;
  76   }
  77
  78   private void processCsar(VspDetails vspDetails,
  79                            FileContentHandler fileContentHandler,
  80                            OrchestrationTemplateCandidateData candidateData,
  81                            OrchestrationTemplateActionResponse response) throws IOException {
  82     response.setFileNames(new ArrayList<>(fileContentHandler.getFileList()));
  83     Map<String, List<ErrorMessage>> errors = validateCsar(fileContentHandler);
  84     if (!isValid(errors)) {
  85       return;
  86     }
  87
  88     HeatStructureTree tree = toscaTreeManager.getTree();
  89
  90     Map<String, String> componentsQuestionnaire = new HashMap<>();
  91     Map<String, Map<String, String>> componentNicsQuestionnaire = new HashMap<>();
  92     Map<String, Collection<ComponentMonitoringUploadEntity>> componentMibList = new HashMap<>();
  93     Map<String, Collection<ProcessEntity>> processes = new HashMap<>();
  94     Map<String, ProcessEntity> processArtifact = new HashMap<>();
  95     OrchestrationUtil orchestrationUtil = new OrchestrationUtil();
  96     orchestrationUtil.backupComponentsQuestionnaireBeforeDelete(vspDetails.getId(),
  97         vspDetails.getVersion(), componentsQuestionnaire,
  98         componentNicsQuestionnaire, componentMibList, processes, processArtifact);
  99
 100     Optional<ByteArrayInputStream> zipByteArrayInputStream = candidateService
 101         .fetchZipFileByteArrayInputStream(vspDetails.getId(), candidateData, null,
 102             OnboardingTypesEnum.CSAR, errors);
 103
 104     orchestrationUtil.deleteUploadDataAndContent(vspDetails.getId(), vspDetails.getVersion());
 105     orchestrationUtil.saveUploadData(vspDetails, candidateData, zipByteArrayInputStream.get(),
 106             fileContentHandler, tree);
 107
 108     ToscaServiceModel toscaServiceModel = new ToscaConverterImpl().convert(fileContentHandler);
 109     orchestrationUtil.saveServiceModel(vspDetails.getId(), vspDetails.getVersion(), toscaServiceModel,
 110         toscaServiceModel);
 111
 112   }
 113
 114   private void addFiles(FileContentHandler fileContentHandler) {
 115     for (Map.Entry<String, byte[]> fileEntry : fileContentHandler.getFiles().entrySet()) {
 116       toscaTreeManager.addFile(fileEntry.getKey(), fileEntry.getValue());
 117     }
 118   }
 119
 120   private Map<String, List<ErrorMessage>> validateCsar(FileContentHandler fileContentHandler) {
 121     Map<String, List<ErrorMessage>> errors = new HashMap<>();
 122     addFiles(fileContentHandler);
 123     toscaTreeManager.createTree();
 124     toscaTreeManager.addErrors(errors);
 125     //todo - add tosca validation here to the existing validation framework
 126     return errors;
 127   }
 128
 129   private boolean isValid(Map<String, List<ErrorMessage>> errors) {
 130     return MapUtils.isEmpty(MessageContainerUtil.getMessageByLevel(ErrorLevel.ERROR, errors));
 131   }
 132 }