asdctool/src/main/java/org/openecomp/sdc/asdctool/impl/internal/tool/CsarGenerator.java

   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 package org.openecomp.sdc.asdctool.impl.internal.tool;
  21
  22 import org.openecomp.sdc.asdctool.utils.ConsoleWriter;
  23 import org.openecomp.sdc.be.dao.cassandra.ArtifactCassandraDao;
  24 import org.openecomp.sdc.be.dao.janusgraph.JanusGraphOperationStatus;
  25 import org.openecomp.sdc.be.dao.jsongraph.GraphVertex;
  26 import org.openecomp.sdc.be.dao.jsongraph.JanusGraphDao;
  27 import org.openecomp.sdc.be.dao.jsongraph.types.EdgeLabelEnum;
  28 import org.openecomp.sdc.be.dao.jsongraph.types.JsonParseFlagEnum;
  29 import org.openecomp.sdc.be.dao.jsongraph.types.VertexTypeEnum;
  30 import org.openecomp.sdc.be.datatypes.elements.ArtifactDataDefinition;
  31 import org.openecomp.sdc.be.datatypes.enums.ComponentTypeEnum;
  32 import org.openecomp.sdc.be.datatypes.enums.GraphPropertyEnum;
  33 import org.openecomp.sdc.be.model.ArtifactDefinition;
  34 import org.openecomp.sdc.be.model.Component;
  35 import org.openecomp.sdc.be.model.LifecycleStateEnum;
  36 import org.openecomp.sdc.be.model.jsonjanusgraph.operations.ToscaOperationFacade;
  37 import org.openecomp.sdc.be.model.operations.impl.UniqueIdBuilder;
  38 import org.openecomp.sdc.be.resources.data.DAOArtifactData;
  39 import org.openecomp.sdc.be.tosca.CsarUtils;
  40 import org.openecomp.sdc.be.tosca.ToscaExportHandler;
  41 import org.openecomp.sdc.common.api.ArtifactTypeEnum;
  42 import org.openecomp.sdc.common.log.wrappers.Logger;
  43 import org.openecomp.sdc.common.util.GeneralUtility;
  44 import org.springframework.beans.factory.annotation.Autowired;
  45
  46 import java.io.IOException;
  47 import java.util.EnumMap;
  48 import java.util.HashMap;
  49 import java.util.List;
  50 import java.util.Map;
  51 import java.util.Optional;
  52 import java.util.Scanner;
  53 import java.util.function.Supplier;
  54 import java.util.stream.Collectors;
  55
  56 @org.springframework.stereotype.Component("csarGenerator")
  57 public class CsarGenerator extends CommonInternalTool {
  58
  59     private JanusGraphDao janusGraphDao;
  60     private CsarUtils csarUtils;
  61     private ToscaOperationFacade toscaOperationFacade;
  62     private ArtifactCassandraDao artifactCassandraDao;
  63     private ToscaExportHandler toscaExportHandler;
  64
  65     @Autowired
  66     public CsarGenerator(JanusGraphDao janusGraphDao, CsarUtils csarUtils,
  67         ToscaOperationFacade toscaOperationFacade,
  68         ArtifactCassandraDao artifactCassandraDao, ToscaExportHandler toscaExportHandler) {
  69         super("generate");
  70         this.janusGraphDao = janusGraphDao;
  71         this.csarUtils = csarUtils;
  72         this.toscaOperationFacade = toscaOperationFacade;
  73         this.artifactCassandraDao = artifactCassandraDao;
  74         this.toscaExportHandler = toscaExportHandler;
  75     }
  76
  77     private static Logger log = Logger.getLogger(CsarGenerator.class.getName());
  78
  79     public void generateCsar(String uuid, Scanner scanner) {
  80         JanusGraphOperationStatus status = JanusGraphOperationStatus.OK;
  81
  82         Map<GraphPropertyEnum, Object> props = new EnumMap<>(GraphPropertyEnum.class);
  83         props.put(GraphPropertyEnum.UUID, uuid);
  84         props.put(GraphPropertyEnum.STATE, LifecycleStateEnum.CERTIFIED.name());
  85         props.put(GraphPropertyEnum.COMPONENT_TYPE, ComponentTypeEnum.SERVICE.name());
  86
  87         List<GraphVertex> byCriteria = janusGraphDao
  88             .getByCriteria(VertexTypeEnum.TOPOLOGY_TEMPLATE, props).either(l -> l, r -> null);
  89         if (byCriteria != null && !byCriteria.isEmpty()) {
  90             if (byCriteria.size() > 1) {
  91                 ConsoleWriter.dataLine("Warning ! More that 1 certified service with uuid", uuid);
  92                 // TBD
  93             } else {
  94                 GraphVertex metadataV = byCriteria.get(0);
  95
  96                 printComponentInfo(metadataV.getMetadataProperties());
  97                 ConsoleWriter.dataLine("\nGenerate CSAR (yes/no)?");
  98                 String input = scanner.nextLine();
  99                 if (input.equalsIgnoreCase("yes")) {
 100
 101                     status = handleService(metadataV, uuid);
 102                 }
 103             }
 104         } else {
 105             ConsoleWriter.dataLine("No certified service with UUID ", uuid);
 106         }
 107         if (status == JanusGraphOperationStatus.OK) {
 108             janusGraphDao.commit();
 109         } else {
 110             janusGraphDao.rollback();
 111         }
 112     }
 113
 114     private JanusGraphOperationStatus handleService(GraphVertex metadataV, String uuid) {
 115         JanusGraphOperationStatus status = JanusGraphOperationStatus.OK;
 116         org.openecomp.sdc.be.model.Component component = toscaOperationFacade.getToscaFullElement(metadataV.getUniqueId()).either(l -> l, r -> null);
 117         if (component != null) {
 118
 119             Supplier<byte[]> supplier = () -> generateToscaPayload(component);
 120             generateArtifact(component, ArtifactTypeEnum.TOSCA_TEMPLATE, supplier);
 121
 122             supplier = () -> generateCsarPayload(component);
 123             generateArtifact(component, ArtifactTypeEnum.TOSCA_CSAR, supplier);
 124
 125             GraphVertex toscaArtifactV = janusGraphDao
 126                 .getChildVertex(metadataV, EdgeLabelEnum.TOSCA_ARTIFACTS, JsonParseFlagEnum.ParseJson).either(l->l, r->null);
 127             if ( toscaArtifactV != null ){
 128                 Map<String, ArtifactDataDefinition> copy = component.getToscaArtifacts().entrySet().stream().collect(Collectors.toMap(Map.Entry::getKey, e -> new ArtifactDataDefinition(e.getValue())));
 129                 toscaArtifactV.setJson(copy);
 130                 janusGraphDao.updateVertex(toscaArtifactV);
 131             }
 132
 133         } else {
 134             ConsoleWriter.dataLine("Failed to fetch certified service with UUID ", uuid);
 135         }
 136         return status;
 137     }
 138
 139     private JanusGraphOperationStatus generateArtifact(Component component, ArtifactTypeEnum artifactType, Supplier<byte[]> supplier){
 140         JanusGraphOperationStatus status = JanusGraphOperationStatus.GENERAL_ERROR;
 141         ArtifactDefinition csarArtifact;
 142         Optional<ArtifactDefinition> op = component.getToscaArtifacts().values().stream().filter(p -> p.getArtifactType().equals(artifactType.getType())).findAny();
 143         if (op.isPresent()) {
 144             csarArtifact = op.get();
 145
 146             status = savePayload(component, csarArtifact, supplier);
 147         }
 148         return status;
 149     }
 150
 151     private byte[] generateCsarPayload(org.openecomp.sdc.be.model.Component component) {
 152         return csarUtils.createCsar(component, true, true).either( l -> l, r -> null);
 153     }
 154     private byte[] generateToscaPayload(Component component){
 155        return toscaExportHandler.exportComponent(component).either(l -> l.getMainYaml().getBytes(), r -> null);
 156     }
 157
 158     private JanusGraphOperationStatus savePayload(org.openecomp.sdc.be.model.Component component, ArtifactDefinition csarArtifact, Supplier<byte[]> supplier) {
 159         byte[] payload = supplier.get();
 160
 161         if ( payload == null ) {
 162             ConsoleWriter.dataLine("create artifact failed ", csarArtifact.getArtifactLabel());
 163             return JanusGraphOperationStatus.GENERAL_ERROR;
 164         }
 165         ConsoleWriter.dataLine("create artifact  success ", csarArtifact.getArtifactLabel());
 166         csarArtifact.setPayload(payload);
 167         byte[] decodedPayload = csarArtifact.getPayloadData();
 168
 169         String uniqueId = UniqueIdBuilder.buildPropertyUniqueId(component.getUniqueId(), csarArtifact.getArtifactLabel());
 170         csarArtifact.setUniqueId(uniqueId);
 171         csarArtifact.setEsId(csarArtifact.getUniqueId());
 172
 173         ConsoleWriter.dataLine("create artifact unique id ", uniqueId);
 174
 175
 176         csarArtifact.setArtifactChecksum(GeneralUtility.calculateMD5Base64EncodedByByteArray(decodedPayload));
 177         DAOArtifactData artifactData = new DAOArtifactData(csarArtifact.getEsId(), decodedPayload);
 178         artifactCassandraDao.saveArtifact(artifactData);
 179         ConsoleWriter.dataLine("Artifact generated and saved into Cassandra ", csarArtifact.getArtifactLabel());
 180         report(component, csarArtifact);
 181
 182         return JanusGraphOperationStatus.OK;
 183     }
 184
 185     private void report(org.openecomp.sdc.be.model.Component component, ArtifactDefinition csarArtifact) {
 186         Map<String, Object> dataToPrint = new HashMap<>();
 187         dataToPrint.put("name", component.getName());
 188         dataToPrint.put("type", component.getComponentType());
 189         dataToPrint.put("version", component.getVersion());
 190         dataToPrint.put("UUID", component.getUUID());
 191         dataToPrint.put("state", component.getLifecycleState());
 192         dataToPrint.put("archive", component.isArchived());
 193         dataToPrint.put("id", component.getUniqueId());
 194         dataToPrint.put("artifact name", csarArtifact.getArtifactLabel());
 195         dataToPrint.put("artifact id", csarArtifact.getUniqueId());
 196         dataToPrint.put("csar es id", csarArtifact.getEsId());
 197         dataToPrint.put("artifact checksum", csarArtifact.getArtifactChecksum());
 198
 199         try {
 200             getReportWriter().report(dataToPrint);
 201         } catch (IOException e) {
 202             ConsoleWriter.dataLine("\nFailed to created report file.");
 203         }
 204     }
 205 }