/*
* Copyright (c) 2006-2011 Nuxeo SA (http://nuxeo.com/) and others.
*
* All rights reserved. This program and the accompanying materials
* are made available under the terms of the Eclipse Public License v1.0
* which accompanies this distribution, and is available at
* http://www.eclipse.org/legal/epl-v10.html
*
* Contributors:
* bstefanescu
*/
package org.eclipse.ecr.automation.server.jaxrs.io;
import java.io.File;
import java.io.IOException;
import java.io.InputStream;
import java.lang.annotation.Annotation;
import java.lang.reflect.Type;
import java.util.List;
import javax.mail.BodyPart;
import javax.mail.internet.MimeMultipart;
import javax.mail.util.SharedFileInputStream;
import javax.servlet.http.HttpServletRequest;
import javax.ws.rs.Consumes;
import javax.ws.rs.WebApplicationException;
import javax.ws.rs.core.Context;
import javax.ws.rs.core.MediaType;
import javax.ws.rs.core.MultivaluedMap;
import javax.ws.rs.ext.MessageBodyReader;
import javax.ws.rs.ext.Provider;
import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import org.eclipse.ecr.automation.core.util.BlobList;
import org.eclipse.ecr.automation.server.jaxrs.ExceptionHandler;
import org.eclipse.ecr.automation.server.jaxrs.ExecutionRequest;
import org.eclipse.ecr.core.api.Blob;
import org.eclipse.ecr.core.api.impl.blob.FileBlob;
import org.eclipse.ecr.web.jaxrs.context.RequestCleanupHandler;
import org.eclipse.ecr.web.jaxrs.context.RequestContext;
import org.nuxeo.common.utils.FileUtils;
/**
* @author <a href="mailto:bs@nuxeo.com">Bogdan Stefanescu</a>
*/
@Provider
@Consumes("multipart/form-data")
public class MultiPartFormRequestReader implements
MessageBodyReader<ExecutionRequest> {
private static final Log log = LogFactory.getLog(MultiPartFormRequestReader.class);
@Context
protected HttpServletRequest request;
public boolean isReadable(Class<?> arg0, Type arg1, Annotation[] arg2,
MediaType arg3) {
return ExecutionRequest.class.isAssignableFrom(arg0); // TODO check
// media type too
}
public ExecutionRequest readFrom(Class<ExecutionRequest> arg0, Type arg1,
Annotation[] arg2, MediaType arg3,
MultivaluedMap<String, String> headers, InputStream in)
throws IOException, WebApplicationException {
ExecutionRequest req = null;
try {
List<String> ctypes = headers.get("Content-Type");
String ctype = ctypes.get(0);
// we need to copy first the stream into a file otherwise it may
// happen that
// javax.mail fail to receive some parts - I am not sure why -
// perhaps the stream is no more available when javax.mail need it?
File tmp = File.createTempFile("nx-automation-mp-upload-", ".tmp");
FileUtils.copyToFile(in, tmp);
// get the input from the saved file
in = new SharedFileInputStream(tmp);
try {
MimeMultipart mp = new MimeMultipart(new InputStreamDataSource(
in, ctype));
BodyPart part = mp.getBodyPart(0); // use content ids
InputStream pin = part.getInputStream();
req = JsonRequestReader.readRequest(pin, headers);
int cnt = mp.getCount();
if (cnt == 2) { // a blob
req.setInput(readBlob(request, mp.getBodyPart(1)));
} else if (cnt > 2) { // a blob list
BlobList blobs = new BlobList();
for (int i = 1; i < cnt; i++) {
blobs.add(readBlob(request, mp.getBodyPart(i)));
}
req.setInput(blobs);
} else {
log.error("Not all parts received.");
for (int i = 0; i < cnt; i++) {
log.error("Received parts: "
+ mp.getBodyPart(i).getHeader("Content-ID")[0]
+ " -> " + mp.getBodyPart(i).getContentType());
}
throw ExceptionHandler.newException(new IllegalStateException(
"Received only " + cnt
+ " part in a multipart request"));
}
} finally {
try {
in.close();
} catch (IOException e) {
// do nothing
}
tmp.delete();
}
} catch (Throwable e) {
throw ExceptionHandler.newException(
"Failed to parse multipart request", e);
}
return req;
}
public static Blob readBlob(HttpServletRequest request, BodyPart part)
throws Exception {
String ctype = part.getContentType();
String fname = part.getFileName();
InputStream pin = part.getInputStream();
final File tmp = File.createTempFile("nx-automation-upload-", ".tmp");
FileUtils.copyToFile(pin, tmp);
FileBlob blob = new FileBlob(tmp, ctype, null, fname, null);
RequestContext.getActiveContext(request).addRequestCleanupHandler(
new RequestCleanupHandler() {
public void cleanup(HttpServletRequest req) {
tmp.delete();
}
});
return blob;
}
}