appc-sdc-listener/appc-sdc-listener-bundle/src/main/java/org/onap/appc/sdc/artifacts/impl/ToscaCsarArtifactProcessor.java

   1 /*-
   2  * ============LICENSE_START=======================================================
   3  * ONAP : APPC
   4  * ================================================================================
   5  * Copyright (C) 2017 AT&T Intellectual Property. All rights reserved.
   6  * ================================================================================
   7  * Copyright (C) 2017 Amdocs
   8  * =============================================================================
   9  * Licensed under the Apache License, Version 2.0 (the "License");
  10  * you may not use this file except in compliance with the License.
  11  * You may obtain a copy of the License at
  12  *
  13  *      http://www.apache.org/licenses/LICENSE-2.0
  14  *
  15  * Unless required by applicable law or agreed to in writing, software
  16  * distributed under the License is distributed on an "AS IS" BASIS,
  17  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  18  * See the License for the specific language governing permissions and
  19  * limitations under the License.
  20  *
  21  * ECOMP is a trademark and service mark of AT&T Intellectual Property.
  22  * ============LICENSE_END=========================================================
  23  */
  24
  25 package org.onap.appc.sdc.artifacts.impl;
  26
  27 import com.att.eelf.configuration.EELFLogger;
  28 import com.att.eelf.configuration.EELFManager;
  29 import com.fasterxml.jackson.databind.JsonNode;
  30 import com.fasterxml.jackson.databind.ObjectMapper;
  31 import com.fasterxml.jackson.dataformat.yaml.YAMLFactory;
  32 import org.apache.commons.lang.StringUtils;
  33 import org.onap.appc.adapter.message.EventSender;
  34 import org.onap.appc.exceptions.APPCException;
  35 import org.onap.appc.licmgr.Constants;
  36 import org.onap.appc.sdc.artifacts.helper.DependencyModelGenerator;
  37 import org.onap.appc.sdc.artifacts.object.Resource;
  38 import org.onap.appc.sdc.artifacts.object.SDCArtifact;
  39 import org.onap.appc.sdc.artifacts.object.SDCReference;
  40 import org.openecomp.sdc.api.IDistributionClient;
  41 import org.openecomp.sdc.api.notification.IArtifactInfo;
  42 import org.openecomp.sdc.api.notification.INotificationData;
  43 import org.openecomp.sdc.api.notification.IResourceInstance;
  44 import org.openecomp.sdc.api.results.IDistributionClientDownloadResult;
  45
  46 import javax.json.Json;
  47 import java.io.*;
  48 import java.net.URI;
  49 import java.util.*;
  50 import java.util.zip.ZipEntry;
  51 import java.util.zip.ZipInputStream;
  52
  53 public class ToscaCsarArtifactProcessor extends AbstractArtifactProcessor{
  54
  55     private final EELFLogger logger = EELFManager.getInstance().getLogger(ToscaCsarArtifactProcessor.class);
  56
  57     private DependencyModelGenerator dependencyModelGenerator;
  58
  59     public ToscaCsarArtifactProcessor(IDistributionClient client, EventSender eventSender, INotificationData notification, IResourceInstance resource,
  60                                       IArtifactInfo artifact, URI storeUri){
  61         super(client,eventSender,notification,resource,artifact,storeUri);
  62         dependencyModelGenerator = new DependencyModelGenerator();
  63     }
  64
  65     @Override
  66     public void processArtifact(IDistributionClientDownloadResult download) throws APPCException {
  67         logger.debug("processing artifact " + super.artifact.getArtifactType());
  68         byte[] byteArray = download.getArtifactPayload();
  69         String serviceFileName = "";
  70         String serviceTemplateContent = "";
  71         List<Resource> resources = null;
  72         Map<String,String> csarFiles = new HashMap<>();
  73         try (ZipInputStream inputStream = new ZipInputStream(new ByteArrayInputStream(byteArray))) {
  74             ZipEntry entry = inputStream.getNextEntry();
  75             logger.debug("First Entry = " +entry);
  76             while(entry!= null){
  77                 String filename = entry.getName();
  78                 logger.debug("Next Entry = "+ filename);
  79
  80                 BufferedReader bufferedReader = new BufferedReader(new InputStreamReader(inputStream));
  81                 String str = null;
  82                 StringBuilder sb = new StringBuilder();
  83                 while((str = bufferedReader.readLine()) != null){
  84                     sb.append(new String(str)).append(System.getProperty("line.separator"));
  85                 }
  86                 csarFiles.put(filename,sb.toString());
  87                 entry = inputStream.getNextEntry();
  88             }
  89
  90         } catch (IOException e) {
  91             logger.error("Error Reading TOSCA.meta from CSAR",e);
  92             throw new APPCException(e);
  93         }
  94         serviceFileName = readServiceFileName(csarFiles.get("TOSCA-Metadata/TOSCA.meta"));
  95         logger.debug("Service File Name = " + serviceFileName);
  96         serviceTemplateContent = csarFiles.get(serviceFileName);
  97
  98         try {
  99             resources = readResources (serviceTemplateContent);
 100         } catch (Exception e) {
 101             logger.error("Error reading resources from " + ", serviceFileName = " + serviceFileName
 102                     + ", TOSCA Metadata = " + csarFiles.get("TOSCA-Metadata/TOSCA.meta"),e);
 103             throw new APPCException(e);
 104         }
 105
 106         for(Resource resource:resources){
 107             String resourceTemplate = csarFiles.get("Definitions/resource-" + resource.getFileNameTag() + "-template.yml");
 108             SDCArtifact artifact = this.getArtifactObject(resource,resourceTemplate);
 109             processArtifact(artifact);
 110         }
 111     }
 112
 113     private String readServiceFileName(String toscaMetadata) {
 114         toscaMetadata = toscaMetadata.substring(toscaMetadata.indexOf("Entry-Definitions"), toscaMetadata.indexOf(System.getProperty("line.separator"),toscaMetadata.indexOf("Entry-Definitions")));
 115         toscaMetadata =toscaMetadata.split(":")[1].trim();
 116         return toscaMetadata;
 117     }
 118
 119     protected SDCArtifact getArtifactObject(Resource resource, String data){
 120
 121         SDCArtifact sdcArtifact = new SDCArtifact();
 122
 123         sdcArtifact.setArtifactUUID(this.artifact.getArtifactUUID());
 124         sdcArtifact.setArtifactName(this.artifact.getArtifactName());
 125         sdcArtifact.setArtifactType(this.artifact.getArtifactType());
 126         sdcArtifact.setArtifactVersion(this.artifact.getArtifactVersion());
 127         sdcArtifact.setArtifactDescription(this.artifact.getArtifactDescription());
 128         sdcArtifact.setArtifactContent(data);
 129         sdcArtifact.setCreationDate(super.getCurrentDateTime());
 130
 131         sdcArtifact.setDistributionId(this.notification.getDistributionID());
 132         sdcArtifact.setServiceUUID(this.notification.getServiceUUID());
 133         sdcArtifact.setServiceName(this.notification.getServiceName());
 134         sdcArtifact.setServiceDescription(this.notification.getServiceDescription());
 135
 136         sdcArtifact.setResourceName(resource.getName());
 137         sdcArtifact.setResourceType(resource.getType());
 138         sdcArtifact.setResourceVersion(resource.getVersion());
 139         sdcArtifact.setResourceUUID(resource.getUuid());
 140         sdcArtifact.setResourceInstanceName(resource.getInstanceName());
 141
 142         return sdcArtifact;
 143     }
 144
 145     private List<Resource> readResources(String serviceTemplateContent) throws IOException {
 146         List<Resource> resources = new LinkedList<>();
 147         ObjectMapper mapper = new ObjectMapper(new YAMLFactory());
 148         JsonNode root = mapper.readTree(serviceTemplateContent);
 149         JsonNode topologyTemplate =  root.get("topology_template");
 150         JsonNode nodeTemplates =  topologyTemplate.get("node_templates");
 151         Iterator<Map.Entry<String, JsonNode>> itr = nodeTemplates.fields();
 152         while(itr.hasNext()){
 153             Map.Entry<String, JsonNode> entry = itr.next();
 154             String instanceName = entry.getKey();
 155             JsonNode nodeTemplate = entry.getValue();
 156
 157             String fileNameTag =  nodeTemplate.get("type").asText();
 158             logger.debug("Resource type in Service Template = " + fileNameTag);
 159             fileNameTag = fileNameTag.substring(fileNameTag.lastIndexOf(".")+1,fileNameTag.length());
 160             String version = nodeTemplate.get("metadata").get("version").asText();
 161             String uuid = nodeTemplate.get("metadata").get("UUID").asText();
 162             String name = nodeTemplate.get("metadata").get("name").asText();
 163             String type = nodeTemplate.get("metadata").get("type").asText();
 164
 165             if(!"VF".equalsIgnoreCase(type)){
 166                 continue;
 167             }
 168
 169             Resource resource = new Resource();
 170             resource.setFileNameTag(fileNameTag);
 171             resource.setVersion(version);
 172             resource.setUuid(uuid);
 173             resource.setInstanceName(instanceName);
 174             resource.setName(name);
 175             resource.setType(type);
 176
 177             resources.add(resource);
 178         }
 179         return resources;
 180     }
 181
 182
 183     @Override
 184     protected void processArtifact(SDCArtifact artifact) throws APPCException {
 185         String vnfType = artifact.getResourceName();
 186         String version = artifact.getResourceVersion();
 187         String packageArtifactID = artifact.getArtifactUUID();
 188
 189         if (StringUtils.isEmpty(vnfType) ||
 190                 StringUtils.isEmpty(version) ||
 191                 StringUtils.isEmpty(packageArtifactID)) {
 192             String errStr = String.format("Missing information in SDC request. Details: resource_type='%s', resource_version='%s', artifactID='%s'", vnfType, version, packageArtifactID);
 193             logger.error(errStr);
 194             throw new APPCException(errStr);
 195         }
 196         try {
 197             SDCReference reference = new SDCReference();
 198             reference.setVnfType(vnfType);
 199             reference.setFileCategory("tosca_model");
 200             reference.setArtifactName(artifact.getArtifactName());
 201             logger.debug("Storing TOSCA to SDC Artifact");
 202             artifactStorageService.storeSDCArtifactWithReference(artifact,reference);
 203
 204             SDCArtifact dependencyArtifact = getDependencyArtifact(artifact);
 205             SDCReference dependencyReference = new SDCReference();
 206             dependencyReference.setVnfType(vnfType);
 207             dependencyReference.setFileCategory("tosca_dependency_model");
 208             dependencyReference.setArtifactName(dependencyArtifact.getArtifactName());
 209             logger.debug("Storing Dependency to SDC Artifact");
 210             artifactStorageService.storeSDCArtifactWithReference(dependencyArtifact,dependencyReference);
 211         } catch (Exception e) {
 212             logger.error("Error processing artifact : " + artifact.toString() );
 213             throw new APPCException(e.getMessage(),e);
 214         }
 215     }
 216
 217     private SDCArtifact getDependencyArtifact(SDCArtifact toscaArtifact) throws APPCException {
 218         SDCArtifact artifact = new SDCArtifact();
 219         artifact.setArtifactName("dependency_"+toscaArtifact.getArtifactName());
 220         String dependencyModel = dependencyModelGenerator.getDependencyModel(toscaArtifact.getArtifactContent(),toscaArtifact.getResourceName());
 221         artifact.setArtifactContent(dependencyModel);
 222         artifact.setArtifactType("DEPENDENCY_MODEL");
 223
 224         artifact.setArtifactUUID(toscaArtifact.getArtifactUUID());
 225         artifact.setArtifactVersion(toscaArtifact.getArtifactVersion());
 226         artifact.setArtifactDescription(toscaArtifact.getArtifactDescription());
 227         artifact.setCreationDate(super.getCurrentDateTime());
 228         artifact.setDistributionId(toscaArtifact.getDistributionId());
 229         artifact.setServiceUUID(toscaArtifact.getServiceUUID());
 230         artifact.setServiceName(toscaArtifact.getServiceName());
 231         artifact.setServiceDescription(toscaArtifact.getServiceDescription());
 232         artifact.setResourceName(toscaArtifact.getResourceName());
 233         artifact.setResourceType(toscaArtifact.getResourceType());
 234         artifact.setResourceVersion(toscaArtifact.getResourceVersion());
 235         artifact.setResourceUUID(toscaArtifact.getResourceUUID());
 236         artifact.setResourceInstanceName(toscaArtifact.getResourceInstanceName());
 237         return artifact;
 238     }
 239
 240
 241 }