1 files changed, 187 insertions, 0 deletions
diff --git a/asdctool/src/main/java/org/openecomp/sdc/asdctool/impl/internal/tool/CsarGenerator.java b/asdctool/src/main/java/org/openecomp/sdc/asdctool/impl/internal/tool/CsarGenerator.java
new file mode 100644
index 0000000000..21085942ac
--- /dev/null
+++ b/asdctool/src/main/java/org/openecomp/sdc/asdctool/impl/internal/tool/CsarGenerator.java
@@ -0,0 +1,187 @@
+package org.openecomp.sdc.asdctool.impl.internal.tool;
+
+import java.io.IOException;
+import java.util.EnumMap;
+import java.util.HashMap;
+import java.util.List;
+import java.util.Map;
+import java.util.Optional;
+import java.util.Scanner;
+import java.util.function.Supplier;
+import java.util.stream.Collectors;
+
+import org.openecomp.sdc.asdctool.utils.ConsoleWriter;
+import org.openecomp.sdc.be.dao.cassandra.ArtifactCassandraDao;
+import org.openecomp.sdc.be.dao.jsongraph.GraphVertex;
+import org.openecomp.sdc.be.dao.jsongraph.TitanDao;
+import org.openecomp.sdc.be.dao.jsongraph.types.EdgeLabelEnum;
+import org.openecomp.sdc.be.dao.jsongraph.types.JsonParseFlagEnum;
+import org.openecomp.sdc.be.dao.jsongraph.types.VertexTypeEnum;
+import org.openecomp.sdc.be.dao.titan.TitanOperationStatus;
+import org.openecomp.sdc.be.datatypes.elements.ArtifactDataDefinition;
+import org.openecomp.sdc.be.datatypes.enums.ComponentTypeEnum;
+import org.openecomp.sdc.be.datatypes.enums.GraphPropertyEnum;
+import org.openecomp.sdc.be.datatypes.enums.JsonPresentationFields;
+import org.openecomp.sdc.be.model.ArtifactDefinition;
+import org.openecomp.sdc.be.model.Component;
+import org.openecomp.sdc.be.model.LifecycleStateEnum;
+import org.openecomp.sdc.be.model.jsontitan.operations.ToscaOperationFacade;
+import org.openecomp.sdc.be.model.operations.impl.UniqueIdBuilder;
+import org.openecomp.sdc.be.resources.data.ESArtifactData;
+import org.openecomp.sdc.be.tosca.CsarUtils;
+import org.openecomp.sdc.be.tosca.ToscaError;
+import org.openecomp.sdc.be.tosca.ToscaExportHandler;
+import org.openecomp.sdc.be.tosca.ToscaRepresentation;
+import org.openecomp.sdc.common.api.ArtifactTypeEnum;
+import org.openecomp.sdc.common.log.wrappers.Logger;
+import org.openecomp.sdc.common.util.GeneralUtility;
+import org.openecomp.sdc.exception.ResponseFormat;
+import org.springframework.beans.factory.annotation.Autowired;
+
+import fj.data.Either;
+
+@org.springframework.stereotype.Component("csarGenerator")
+public class CsarGenerator extends CommonInternalTool {
+    public CsarGenerator() {
+        super("generate");
+    }
+
+    @Autowired
+    private TitanDao titanDao;
+    @Autowired
+    private CsarUtils csarUtils;
+    @Autowired
+    private ToscaOperationFacade toscaOperationFacade;
+    @Autowired
+    private ArtifactCassandraDao artifactCassandraDao;
+    @Autowired
+    private ToscaExportHandler toscaExportHandler;
+    
+
+    private static Logger log = Logger.getLogger(CsarGenerator.class.getName());
+
+    public void generateCsar(String uuid, Scanner scanner) {
+        TitanOperationStatus status = TitanOperationStatus.OK;
+
+        Map<GraphPropertyEnum, Object> props = new EnumMap<>(GraphPropertyEnum.class);
+        props.put(GraphPropertyEnum.UUID, uuid);
+        props.put(GraphPropertyEnum.STATE, LifecycleStateEnum.CERTIFIED.name());
+        props.put(GraphPropertyEnum.COMPONENT_TYPE, ComponentTypeEnum.SERVICE.name());
+
+        List<GraphVertex> byCriterria = titanDao.getByCriteria(VertexTypeEnum.TOPOLOGY_TEMPLATE, props).either(l -> l, r -> null);
+        if (byCriterria != null && !byCriterria.isEmpty()) {
+            if (byCriterria.size() > 1) {
+                ConsoleWriter.dataLine("Warning ! More that 1 certified service with uuid", uuid);
+                // TBD
+            } else {
+                GraphVertex metadataV = byCriterria.get(0);
+
+                printComponentInfo(metadataV.getMetadataProperties());
+                ConsoleWriter.dataLine("\nGenerate CSAR (yes/no)?");
+                String input = scanner.nextLine();
+                if (input.equalsIgnoreCase("yes")) {
+                    
+                    status = handleService(metadataV, uuid);
+                }
+            }
+        } else {
+            ConsoleWriter.dataLine("No certified service with UUID", uuid);
+        }
+        if (status == TitanOperationStatus.OK) {
+            titanDao.commit();
+        } else {
+            titanDao.rollback();
+        }
+    }
+
+    private TitanOperationStatus handleService(GraphVertex metadataV, String uuid) {
+        TitanOperationStatus status = TitanOperationStatus.OK;
+        org.openecomp.sdc.be.model.Component component = toscaOperationFacade.getToscaFullElement(metadataV.getUniqueId()).either(l -> l, r -> null);
+        if (component != null) {
+
+            Supplier<byte[]> supplier = () -> generateToscaPayload(component);
+            generateArtifact(component, ArtifactTypeEnum.TOSCA_TEMPLATE, supplier);
+            
+            supplier = () -> generateCsarPayload(component);
+            generateArtifact(component, ArtifactTypeEnum.TOSCA_CSAR, supplier);
+            
+            GraphVertex toscaArtifactV = titanDao.getChildVertex(metadataV, EdgeLabelEnum.TOSCA_ARTIFACTS, JsonParseFlagEnum.ParseJson).either(l->l, r->null);
+            if ( toscaArtifactV != null ){
+                Map<String, ArtifactDataDefinition> copy = component.getToscaArtifacts().entrySet().stream().collect(Collectors.toMap(Map.Entry::getKey, e -> new ArtifactDataDefinition(e.getValue())));
+                toscaArtifactV.setJson(copy);
+                titanDao.updateVertex(toscaArtifactV);
+            }
+           
+        } else {
+            ConsoleWriter.dataLine("Failed to fetch certified service with UUID", uuid);
+        }
+        return status;
+    }
+
+    private TitanOperationStatus generateArtifact(Component component, ArtifactTypeEnum artifactType, Supplier<byte[]> supplier){
+        TitanOperationStatus status = TitanOperationStatus.GENERAL_ERROR;
+        ArtifactDefinition csarArtifact = null;
+        Optional<ArtifactDefinition> op = component.getToscaArtifacts().values().stream().filter(p -> p.getArtifactType().equals(artifactType.getType())).findAny();
+        if (op.isPresent()) {
+            csarArtifact = op.get();
+              
+            status = savePayload(component, csarArtifact, supplier);
+        }
+        return status;
+    }
+    
+    private byte[] generateCsarPayload(org.openecomp.sdc.be.model.Component component) {
+        return csarUtils.createCsar(component, true, true).either( l -> l, r -> null);
+    }
+    private byte[] generateToscaPayload(Component component){
+       return toscaExportHandler.exportComponent(component).either(l -> l.getMainYaml().getBytes(), r -> null);
+    }
+
+    private TitanOperationStatus savePayload(org.openecomp.sdc.be.model.Component component, ArtifactDefinition csarArtifact, Supplier<byte[]> supplier) {
+        byte[] payload = supplier.get();
+
+        if ( payload == null ) {
+            ConsoleWriter.dataLine("create artifact failed ", csarArtifact.getArtifactLabel());
+            return TitanOperationStatus.GENERAL_ERROR;
+        }
+        ConsoleWriter.dataLine("createartifact  success ", csarArtifact.getArtifactLabel());
+        csarArtifact.setPayload(payload);
+        byte[] decodedPayload = csarArtifact.getPayloadData();
+
+        String uniqueId = UniqueIdBuilder.buildPropertyUniqueId(component.getUniqueId(), csarArtifact.getArtifactLabel());
+        csarArtifact.setUniqueId(uniqueId);
+        csarArtifact.setEsId(csarArtifact.getUniqueId());
+        
+        ConsoleWriter.dataLine("create artifact unique id ", uniqueId);
+        
+        
+        csarArtifact.setArtifactChecksum(GeneralUtility.calculateMD5Base64EncodedByByteArray(decodedPayload));
+        ESArtifactData artifactData = new ESArtifactData(csarArtifact.getEsId(), decodedPayload);
+        artifactCassandraDao.saveArtifact(artifactData);
+        ConsoleWriter.dataLine("Artifact generated and saved into Cassandra ", csarArtifact.getArtifactLabel());
+        report(component, csarArtifact);
+
+        return TitanOperationStatus.OK;
+    }
+
+    private void report(org.openecomp.sdc.be.model.Component component, ArtifactDefinition csarArtifact) {
+        Map<String, Object> dataToPrint = new HashMap<>();
+        dataToPrint.put("name", component.getName());
+        dataToPrint.put("type", component.getComponentType());
+        dataToPrint.put("version", component.getVersion());
+        dataToPrint.put("UUID", component.getUUID());
+        dataToPrint.put("state", component.getLifecycleState());
+        dataToPrint.put("archive", component.isArchived());
+        dataToPrint.put("id", component.getUniqueId());
+        dataToPrint.put("artifact name", csarArtifact.getArtifactLabel());
+        dataToPrint.put("artifact id", csarArtifact.getUniqueId());
+        dataToPrint.put("csar es id", csarArtifact.getEsId());
+        dataToPrint.put("artifact checksum", csarArtifact.getArtifactChecksum());
+
+        try {
+            getReportWriter().report(dataToPrint);
+        } catch (IOException e) {
+            ConsoleWriter.dataLine("\nFailed to created report file.");
+        }
+    }
+}