/*
Copyright (C) 2005-2012, by the President and Fellows of Harvard College.
Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at
http://www.apache.org/licenses/LICENSE-2.0
Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.
Dataverse Network - A web application to share, preserve and analyze research data.
Developed at the Institute for Quantitative Social Science, Harvard University.
Version 3.0.
*/
package edu.harvard.iq.dvn.api.entities;
import java.io.IOException;
import java.io.OutputStream;
import java.io.InputStream;
import java.lang.annotation.Annotation;
import java.lang.reflect.Type;
import javax.ejb.Singleton;
import javax.ws.rs.WebApplicationException;
import javax.ws.rs.core.MediaType;
import javax.ws.rs.core.MultivaluedMap;
import javax.ws.rs.ext.MessageBodyWriter;
import javax.ws.rs.ext.Provider;
import javax.ws.rs.core.Response;
import edu.harvard.iq.dvn.core.web.dataaccess.*;
import edu.harvard.iq.dvn.core.study.StudyFile;
/**
*
* @author leonidandreev
*/
@Singleton
@Provider
public class DownloadInstanceWriter implements MessageBodyWriter<DownloadInstance>{
public boolean isWriteable(Class<?> clazz, Type type, Annotation[] annotation, MediaType mediaType) {
return clazz == DownloadInstance.class;
}
public long getSize(DownloadInstance di, Class<?> clazz, Type type, Annotation[] annotation, MediaType mediaType) {
return -1;
}
public void writeTo(DownloadInstance di, Class<?> clazz, Type type, Annotation[] annotation, MediaType mediaType, MultivaluedMap<String, Object> httpHeaders, OutputStream outstream) throws IOException, WebApplicationException {
if (di.getDownloadInfo() != null && di.getDownloadInfo().getStudyFile() != null) {
DataAccessRequest daReq = new DataAccessRequest();
StudyFile sf = di.getDownloadInfo().getStudyFile();
DataAccessObject accessObject = DataAccess.createDataAccessObject(sf, daReq);
if (accessObject != null) {
accessObject.open();
if (di.getConversionParam() != null) {
// Image Thumbnail conversion:
if (di.getConversionParam().equals("imageThumb")) {
accessObject = ImageThumbConverter.getImageThumb(sf, (FileAccessObject)accessObject);
} else if (di.getConversionParam().equals("TermsOfUse")) {
accessObject = ExportTermsOfUse.export(sf.getStudy());
} else if (di.getConversionParam().equals("package")) {
if ("WithTermsOfUse".equals(di.getConversionParamValue())) {
accessObject = PackageWithTermsOfUse.repackage(sf, (FileAccessObject)accessObject);
}
}
// Subsettable:
if (sf.isSubsettable()) {
if (di.getConversionParam().equals("noVarHeader")) {
accessObject.setNoVarHeader(Boolean.TRUE);
accessObject.setVarHeader(null);
} else if (di.getConversionParam().equals("fileFormat")) {
// Saved Original:
if ("original".equals(di.getConversionParamValue())) {
accessObject = StoredOriginalFile.retrieve(sf, (FileAccessObject)accessObject);
} else {
// Other format conversions:
String requestedMimeType = di.getServiceFormatType(di.getConversionParam(), di.getConversionParamValue());
if (requestedMimeType == null) {
// default mime type, in case real type is unknown;
// (this shouldn't happen in real life - but just in case):
requestedMimeType = "application/octet-stream";
}
accessObject =
DataFileConverter.performFormatConversion(
sf,
(FileAccessObject)accessObject,
di.getConversionParamValue(), requestedMimeType);
}
}
}
if (accessObject == null) {
throw new WebApplicationException(Response.Status.SERVICE_UNAVAILABLE);
}
}
InputStream instream = accessObject.getInputStream();
if (instream != null) {
// headers:
String fileName = accessObject.getFileName();
String mimeType = accessObject.getMimeType();
// Provide both the "Content-disposition" and "Content-Type" headers,
// to satisfy the widest selection of browsers out there.
httpHeaders.add("Content-disposition", "attachment; filename=\"" + fileName + "\"");
httpHeaders.add("Content-Type", mimeType + "; name=\"" + fileName);
// (the httpHeaders map must be modified *before* writing any
// data in the output stream!
int bufsize;
byte [] bffr = new byte[4*8192];
// before writing out any bytes from the input stream, flush
// any extra content, such as the variable header for the
// subsettable files:
if (accessObject.getVarHeader() != null) {
outstream.write(accessObject.getVarHeader().getBytes());
}
while ((bufsize = instream.read(bffr)) != -1) {
outstream.write(bffr, 0, bufsize);
}
instream.close();
return;
}
}
}
throw new WebApplicationException(Response.Status.NOT_FOUND);
}
}