package er.jquery.widgets;
import java.io.File;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import com.webobjects.appserver.WOComponent;
import com.webobjects.appserver.WOContext;
import com.webobjects.appserver.WOMultipartIterator;
import com.webobjects.appserver.WOMultipartIterator.WOFormData;
import com.webobjects.appserver.WORequest;
import com.webobjects.appserver.WOResponse;
import com.webobjects.foundation.NSArray;
import com.webobjects.foundation.NSData;
import com.webobjects.foundation.NSDictionary;
import com.webobjects.foundation.NSMutableArray;
import er.extensions.appserver.ERXResponseRewriter;
import er.extensions.appserver.ajax.ERXAjaxApplication;
import er.extensions.foundation.ERXFileUtilities;
import er.extensions.foundation.ERXProperties;
import er.extensions.foundation.ERXStringUtilities;
/**
* Encapsulation of @see <a href="http://github.com/valums/ajax-upload">Ajax Upload</a>
* @binding name
* @binding onChange
* @binding onSubmit
* @binding onComplete
*
* @property useUnobtrusively For Unobtrusive Javascript programming. Default it is ON.
*
* @author mendis
*
*/
public abstract class AjaxUpload extends WOComponent {
private static boolean useUnobtrusively = ERXProperties.booleanForKeyWithDefault("er.jquery.useUnobtrusively", true);
public AjaxUpload(WOContext aContext) {
super(aContext);
}
/*
* Bindings/API
*/
public static interface Bindings {
public static final String id = "id";
public static final String name = "name";
public static final String onChange = "onChange";
public static final String onSubmit = "onSubmit";
public static final String onComplete = "onComplete";
public static final String filePath = "filePath";
public static final String data = "data";
public static final String mimeType = "mimeType";
public static final String inputStream = "inputStream";
public static final String outputStream = "outputStream";
public static final String finalFilePath = "finalFilePath";
public static final String streamToFilePath = "streamToFilePath";
}
/*
* Headers of multipart form
*/
public static interface Headers {
public static final String filename = "filename";
public static final String contentType = "content-type";
public static final String name = "name";
}
@Override
public boolean synchronizesVariablesWithBindings() {
return false;
}
// accessors
public String id() {
return _id() == null ? "au" + ERXStringUtilities.safeIdentifierName(elementID()) : _id();
}
private String elementID;
private String elementID() {
if (elementID == null) elementID = context().elementID();
return elementID;
}
private String _id() {
return (String) valueForBinding(Bindings.id);
}
private String _script() {
return "new AjaxUpload('" + id() + "', {" + options() + "});";
}
public String script() {
return isAjax() ? _script() : "$(document).ready(function() { " + _script() + " });";
}
private boolean isAjax() {
return ERXAjaxApplication.isAjaxRequest(context().request());
}
/*
* An array of options for Ajax.Updater
*/
protected NSArray<String> _options() {
NSMutableArray _options = new NSMutableArray("action:'" + href() + "'");
// add options
_options.add("name:'" + uploadName() + "'");
if (hasBinding(Bindings.onChange)) _options.add("onChange:" + valueForBinding(Bindings.onChange));
if (hasBinding(Bindings.onComplete)) _options.add("onComplete:" + valueForBinding(Bindings.onComplete));
if (hasBinding(Bindings.onSubmit)) _options.add("onChange:" + valueForBinding(Bindings.onSubmit));
return _options.immutableClone();
}
public String options() {
return _options().componentsJoinedByString(", ");
}
private String _uploadName() {
return (String) valueForBinding(Bindings.name);
}
public String uploadName() {
return _uploadName() != null ? _uploadName() : "userfile"; // Default
}
protected void setFilePath(String aPath) {
setValueForBinding(aPath, Bindings.filePath);
}
protected void setData(NSData data) {
setValueForBinding(data, Bindings.data);
}
protected void setMimeType(String aType) {
setValueForBinding(aType, Bindings.mimeType);
}
/*
* NOTE: this is a standard WO component action url.
* The file upload won't work as an ajax request.
*
*/
private String href() {
return context().componentActionURL();
}
// R&R
@Override
public void appendToResponse(WOResponse response, WOContext context) {
super.appendToResponse(response, context);
if (!useUnobtrusively) {
ERXResponseRewriter.addScriptResourceInHead(response, context, "ERJQuery", "jquery-1.4.2.min.js");
ERXResponseRewriter.addScriptResourceInHead(response, context, "ERJQuery", "ajaxupload.js");
}
}
@Override
public void takeValuesFromRequest(WORequest request, WOContext context) {
super.takeValuesFromRequest(request, context);
if (request.formValueForKey(uploadName()) != null) {
// filepath
if (hasBinding(Bindings.filePath)) {
setFilePath((String) request.formValueForKey(uploadName() + ".filename"));
}
// file data
if (hasBinding(Bindings.data)) {
if (hasBinding(Bindings.filePath)) {
NSArray aValue = request.formValuesForKey(uploadName());
if (aValue != null) {
NSData data = null;
try {
data = (NSData) aValue.objectAtIndex(0);
} catch (ClassCastException e) {
throw new ClassCastException("AjaxUploadButton: Value in request was of type '" + aValue.objectAtIndex(0).getClass().getName() + "' instead of NSData. Verify that the WOForm's 'enctype' binding is set to 'multipart/form-data'");
}
setData(data);
}
// mimetype
if (hasBinding(Bindings.mimeType)) {
setMimeType((String) request.formValueForKey(uploadName() + ".mimetype"));
}
}
} else {
// multipart data
WOMultipartIterator multipartIterator = request.multipartIterator();
WOFormData nextFormData = multipartIterator.nextFormData();
NSDictionary<Object, String> contentDispositionHeaders;
do {
if (nextFormData == null)
break;
contentDispositionHeaders = nextFormData.contentDispositionHeaders();
Object _name = contentDispositionHeaders.objectForKey(Headers.name);
if (uploadName().equals(_name))
break;
nextFormData = multipartIterator.nextFormData();
} while(true);
if(nextFormData == null)
throw new IllegalStateException("AjaxUploadButton: No form data left for WOFileUpload!");
contentDispositionHeaders = nextFormData.contentDispositionHeaders();
String aFileName = null;
if(hasBinding(Bindings.filePath)) {
aFileName = (String) contentDispositionHeaders.valueForKey(Headers.filename);
setFilePath(aFileName);
}
if (hasBinding(Bindings.mimeType)) {
setMimeType((String) contentDispositionHeaders.valueForKey(Headers.contentType));
}
InputStream anInputStream = nextFormData.formDataInputStream();
if(aFileName != null && aFileName.length() > 0) {
if(hasBinding(Bindings.inputStream)) {
setValueForBinding(anInputStream, Bindings.inputStream);
} else {
String localFilePath = null;
File tempFile = null;
if (!hasBinding(Bindings.outputStream)) {
if (hasBinding(Bindings.finalFilePath))
setValueForBinding(null, Bindings.finalFilePath);
try {
tempFile = ERXFileUtilities.writeInputStreamToTempFile(anInputStream, context.session().sessionID(), ".tmp");
} catch (IOException e) {
throw new RuntimeException("Couldn't write input stream to temp file: " + e);
} finally {
try { anInputStream.close(); } catch (IOException e) {}
}
} else {
OutputStream anOutputStream = (OutputStream) valueForBinding(Bindings.outputStream);
try {
ERXFileUtilities.writeInputStreamToOutputStream(anInputStream, anOutputStream);
} catch (IOException e) {
throw new RuntimeException("Couldn't write input stream to output stream: " + e);
} finally {
try { anOutputStream.close(); } catch (IOException e) {}
try { anInputStream.close(); } catch (IOException e) {}
}
}
if (hasBinding(Bindings.streamToFilePath)) {
localFilePath = (String) valueForBinding(Bindings.streamToFilePath);
try {
ERXFileUtilities.renameTo(tempFile, new File(localFilePath));
setValueForBinding(localFilePath, Bindings.finalFilePath);
} catch (Exception e) {
setValueForBinding(tempFile.getPath(), Bindings.finalFilePath);
throw new RuntimeException("Couldn't rename temp file: " + e);
}
}
}
} else {
if (hasBinding(Bindings.inputStream))
setValueForBinding(null, Bindings.inputStream);
if (hasBinding(Bindings.finalFilePath))
setValueForBinding(null, Bindings.finalFilePath);
byte buffer[] = new byte[128];
try {
while (anInputStream.read(buffer) != -1) ;
}
catch (IOException e) {
throw new RuntimeException("Error skipping empty file upload: " + e);
} finally {
try { anInputStream.close(); } catch (IOException e) {}
}
}
}
}
}
}