catalog-be/src/main/java/org/openecomp/sdc/be/servlets/TypesUploadEndpoint.java

   1 package org.openecomp.sdc.be.servlets;
   2
   3 import com.google.common.annotations.VisibleForTesting;
   4 import com.jcabi.aspects.Loggable;
   5 import io.swagger.annotations.*;
   6 import org.apache.commons.lang3.tuple.ImmutablePair;
   7 import org.glassfish.jersey.media.multipart.FormDataParam;
   8 import org.openecomp.sdc.be.components.impl.CommonImportManager;
   9 import org.openecomp.sdc.be.components.validation.AccessValidations;
  10 import org.openecomp.sdc.be.datatypes.tosca.ToscaDataDefinition;
  11 import org.openecomp.sdc.be.model.AnnotationTypeDefinition;
  12 import org.openecomp.sdc.be.model.operations.impl.AnnotationTypeOperations;
  13 import org.openecomp.sdc.be.utils.TypeUtils;
  14 import org.openecomp.sdc.common.datastructure.Wrapper;
  15 import org.springframework.http.HttpStatus;
  16 import org.springframework.stereotype.Controller;
  17
  18 import javax.ws.rs.*;
  19 import javax.ws.rs.core.MediaType;
  20 import javax.ws.rs.core.Response;
  21 import java.io.File;
  22 import java.io.IOException;
  23 import java.util.List;
  24 import java.util.Map;
  25
  26 /**
  27  * Here new APIs for types upload written in an attempt to gradually servlet code
  28  */
  29 @Loggable(prepend = true, value = Loggable.DEBUG, trim = false)
  30 @Path("/v1/catalog/uploadType")
  31 @Consumes(MediaType.MULTIPART_FORM_DATA)
  32 @Produces(MediaType.APPLICATION_JSON)
  33 @Api(value = "Catalog Types Upload")
  34 @Controller
  35 public class TypesUploadEndpoint {
  36
  37     private final CommonImportManager commonImportManager;
  38     private final AnnotationTypeOperations annotationTypeOperations;
  39     private final AccessValidations accessValidations;
  40
  41     public TypesUploadEndpoint(CommonImportManager commonImportManager, AnnotationTypeOperations annotationTypeOperations, AccessValidations accessValidations) {
  42         this.commonImportManager = commonImportManager;
  43         this.annotationTypeOperations = annotationTypeOperations;
  44         this.accessValidations = accessValidations;
  45     }
  46
  47     @POST
  48     @Path("/annotationtypes")
  49     @ApiOperation(value = "Create AnnotationTypes from yaml", httpMethod = "POST", notes = "Returns created annotation types", response = Response.class)
  50     @ApiResponses(value = {
  51             @ApiResponse(code = 201, message = "annotation types created"),
  52             @ApiResponse(code = 403, message = "Restricted operation"),
  53             @ApiResponse(code = 400, message = "Invalid content / Missing content"),
  54             @ApiResponse(code = 409, message = "annotation types already exist")})
  55     public Response uploadAnnotationTypes(
  56             @ApiParam("FileInputStream") @FormDataParam("annotationTypesZip") File file,
  57             @HeaderParam("USER_ID") String userId) throws IOException {
  58         accessValidations.validateUserExists(userId, "Annotation Types Creation");
  59         Wrapper<String> yamlStringWrapper = new Wrapper<>();
  60         AbstractValidationsServlet.extractZipContents(yamlStringWrapper, file);
  61         List<ImmutablePair<AnnotationTypeDefinition, Boolean>> typesResults = commonImportManager.createElementTypes(yamlStringWrapper.getInnerElement(), TypesUploadEndpoint::buildAnnotationFromFieldMap, annotationTypeOperations);
  62         HttpStatus status = getHttpStatus(typesResults);
  63         return Response.status(status.value())
  64                 .entity(typesResults)
  65                 .build();
  66     }
  67
  68     @VisibleForTesting
  69     static <T extends ToscaDataDefinition> HttpStatus getHttpStatus(List<ImmutablePair<T, Boolean>> typesResults) {
  70         boolean typeActionFailed = false;
  71         boolean typeExists = false;
  72         boolean typeActionSucceeded = false;
  73         for (ImmutablePair<T, Boolean> typeResult : typesResults) {
  74             Boolean result = typeResult.getRight();
  75             if (result==null) {
  76                 typeExists = true;
  77             } else if (result) {
  78                 typeActionSucceeded = true;
  79             } else {
  80                 typeActionFailed = true;
  81             }
  82         }
  83         HttpStatus status = HttpStatus.OK;
  84         if (typeActionFailed) {
  85             status =  HttpStatus.BAD_REQUEST;
  86         } else if (typeActionSucceeded) {
  87             status = HttpStatus.CREATED;
  88         } else if (typeExists) {
  89             status = HttpStatus.CONFLICT;
  90         }
  91         return status;
  92     }
  93
  94     private static <T extends ToscaDataDefinition> T buildAnnotationFromFieldMap(String typeName, Map<String, Object> toscaJson) {
  95         AnnotationTypeDefinition annotationType = new AnnotationTypeDefinition();
  96         annotationType.setVersion(TypeUtils.FIRST_CERTIFIED_VERSION_VERSION);
  97         annotationType.setHighestVersion(true);
  98         annotationType.setType(typeName);
  99         TypeUtils.setField(toscaJson, TypeUtils.ToscaTagNamesEnum.DESCRIPTION, annotationType::setDescription);
 100         CommonImportManager.setProperties(toscaJson, annotationType::setProperties);
 101         return (T) annotationType;
 102     }
 103
 104
 105 }