Examples with InputPart org.jboss.resteasy.plugins.providers.multipart.InputPart used on opensource projects

Example 11 with InputPart

use of org.jboss.resteasy.plugins.providers.multipart.InputPart in project candlepin by candlepin.

the class OwnerResource method getArchiveFromResponse.

private UploadMetadata getArchiveFromResponse(MultipartInput input) throws IOException {
    String filename = "";
    InputPart part = input.getParts().get(0);
    MultivaluedMap<String, String> headers = part.getHeaders();
    String contDis = headers.getFirst("Content-Disposition");
    StringTokenizer st = new StringTokenizer(contDis, ";");
    while (st.hasMoreTokens()) {
        String entry = st.nextToken().trim();
        if (entry.startsWith("filename")) {
            filename = entry.substring(entry.indexOf("=") + 2, entry.length() - 1);
            break;
        }
    }
    return new UploadMetadata(part.getBody(new GenericType<File>() {
    }), filename);
}

Example 12 with InputPart

use of org.jboss.resteasy.plugins.providers.multipart.InputPart in project syndesis by syndesisio.

the class DriverUploadTest method upload.

@Test
public void upload() throws IOException, URISyntaxException {
    URL driverClasspath = DriverUploadTest.class.getClassLoader().getResource(FILE_CLASSPATH);
    assertThat(driverClasspath).isNotNull();
    MultipartFormDataInput multiPart = mock(MultipartFormDataInput.class);
    InputPart inputPartFileContent = mock(InputPart.class);
    InputPart inputPartFileName = mock(InputPart.class);
    List<InputPart> parts = new ArrayList<>();
    parts.add(inputPartFileContent);
    parts.add(inputPartFileName);
    when(multiPart.getParts()).thenReturn(parts);
    when(multiPart.getFormDataPart("file", InputStream.class, null)).thenReturn(driverClasspath.openStream());
    when(multiPart.getFormDataPart("fileName", String.class, null)).thenReturn(FILE_NAME);
    String tempDir = System.getProperty("java.io.tmpdir");
    System.setProperty("LOADER_HOME", tempDir);
    DriverUploadEndpoint endpoint = new DriverUploadEndpoint();
    Boolean reply = endpoint.upload(multiPart);
    assertThat(reply).isTrue();
// check equality with the original file
// File originalFile = new File(driverClasspath.toURI());
// File uploadedFile = new File(tempDir + File.separator + FILE_NAME);
// Boolean isEquals = FileUtils.contentEquals(originalFile, uploadedFile);
// assertThat(isEquals).isTrue();
}

Example 13 with InputPart

use of org.jboss.resteasy.plugins.providers.multipart.InputPart in project syndesis by syndesisio.

the class ConnectorIconHandler method create.

@POST
@ApiOperation("Updates the connector icon for the specified connector and returns the updated connector")
@ApiResponses(@ApiResponse(code = 200, response = Connector.class, message = "Updated Connector icon"))
@Produces(MediaType.APPLICATION_JSON)
@Consumes(MediaType.MULTIPART_FORM_DATA)
@SuppressWarnings("PMD.CyclomaticComplexity")
public Connector create(MultipartFormDataInput dataInput) {
    if (dataInput == null || dataInput.getParts() == null || dataInput.getParts().isEmpty()) {
        throw new IllegalArgumentException("Multipart request is empty");
    }
    if (dataInput.getParts().size() != 1) {
        throw new IllegalArgumentException("Wrong number of parts in multipart request");
    }
    try {
        InputPart filePart = dataInput.getParts().iterator().next();
        InputStream result = filePart.getBody(InputStream.class, null);
        if (result == null) {
            throw new IllegalArgumentException("Can't find a valid 'icon' part in the multipart request");
        }
        try (BufferedInputStream iconStream = new BufferedInputStream(result)) {
            MediaType mediaType = filePart.getMediaType();
            if (!mediaType.getType().equals("image")) {
                // URLConnection.guessContentTypeFromStream resets the stream after inspecting the media type so
                // can continue to be used, rather than being consumed.
                String guessedMediaType = URLConnection.guessContentTypeFromStream(iconStream);
                if (!guessedMediaType.startsWith("image/")) {
                    throw new IllegalArgumentException("Invalid file contents for an image");
                }
                mediaType = MediaType.valueOf(guessedMediaType);
            }
            Icon.Builder iconBuilder = new Icon.Builder().mediaType(mediaType.toString());
            Icon icon;
            String connectorIcon = connector.getIcon();
            if (connectorIcon != null && connectorIcon.startsWith("db:")) {
                String connectorIconId = connectorIcon.substring(3);
                iconBuilder.id(connectorIconId);
                icon = iconBuilder.build();
                getDataManager().update(icon);
            } else {
                icon = getDataManager().create(iconBuilder.build());
            }
            iconDao.write(icon.getId().get(), iconStream);
            Connector updatedConnector = new Connector.Builder().createFrom(connector).icon("db:" + icon.getId().get()).build();
            getDataManager().update(updatedConnector);
            return updatedConnector;
        }
    } catch (IOException e) {
        throw new IllegalArgumentException("Error while reading multipart request", e);
    }
}

Example 14 with InputPart

use of org.jboss.resteasy.plugins.providers.multipart.InputPart in project syndesis by syndesisio.

the class ExtensionHandler method getBinaryArtifact.

@Nonnull
@SuppressWarnings("PMD.CyclomaticComplexity")
private InputStream getBinaryArtifact(MultipartFormDataInput input) {
    if (input == null || input.getParts() == null || input.getParts().isEmpty()) {
        throw new IllegalArgumentException("Multipart request is empty");
    }
    try {
        InputStream result;
        if (input.getParts().size() == 1) {
            InputPart filePart = input.getParts().iterator().next();
            result = filePart.getBody(InputStream.class, null);
        } else {
            result = input.getFormDataPart("file", InputStream.class, null);
        }
        if (result == null) {
            throw new IllegalArgumentException("Can't find a valid 'file' part in the multipart request");
        }
        return result;
    } catch (IOException e) {
        throw new IllegalArgumentException("Error while reading multipart request", e);
    }
}

Example 15 with InputPart

use of org.jboss.resteasy.plugins.providers.multipart.InputPart in project scheduling by ow2-proactive.

the class SchedulerStateRest method pushFile.

@Override
public boolean pushFile(String sessionId, String spaceName, String filePath, MultipartFormDataInput multipart) throws IOException, NotConnectedRestException, PermissionRestException {
    try {
        checkAccess(sessionId, "pushFile");
        Session session = dataspaceRestApi.checkSessionValidity(sessionId);
        Map<String, List<InputPart>> formDataMap = multipart.getFormDataMap();
        List<InputPart> fNL = formDataMap.get("fileName");
        if ((fNL == null) || (fNL.isEmpty())) {
            throw new IllegalArgumentException("Illegal multipart argument definition (fileName), received " + fNL);
        }
        String fileName = fNL.get(0).getBody(String.class, null);
        List<InputPart> fCL = formDataMap.get("fileContent");
        if ((fCL == null) || (fCL.isEmpty())) {
            throw new IllegalArgumentException("Illegal multipart argument definition (fileContent), received " + fCL);
        }
        InputStream fileContent = fCL.get(0).getBody(InputStream.class, null);
        if (fileName == null) {
            throw new IllegalArgumentException("Wrong file name : " + fileName);
        }
        filePath = normalizeFilePath(filePath, fileName);
        FileObject destfo = dataspaceRestApi.resolveFile(session, spaceName, filePath);
        URL targetUrl = destfo.getURL();
        logger.info("[pushFile] pushing file to " + targetUrl);
        if (!destfo.isWriteable()) {
            RuntimeException ex = new IllegalArgumentException("File " + filePath + " is not writable in space " + spaceName);
            logger.error(ex);
            throw ex;
        }
        if (destfo.exists()) {
            destfo.delete();
        }
        // used to create the necessary directories if needed
        destfo.createFile();
        dataspaceRestApi.writeFile(fileContent, destfo, null);
        return true;
    } finally {
        if (multipart != null) {
            multipart.close();
        }
    }
}