package er.extensions.appserver;
import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.net.URLDecoder;
import org.apache.commons.lang.CharEncoding;
import org.apache.log4j.Logger;
import com.webobjects.appserver.WOApplication;
import com.webobjects.appserver.WODynamicURL;
import com.webobjects.appserver.WORequest;
import com.webobjects.appserver.WORequestHandler;
import com.webobjects.appserver.WOResourceManager;
import com.webobjects.appserver.WOResponse;
import com.webobjects.appserver._private.WODeployedBundle;
import com.webobjects.foundation.NSBundle;
import com.webobjects.foundation.NSDictionary;
import com.webobjects.foundation.NSNotificationCenter;
import er.extensions.foundation.ERXDictionaryUtilities;
import er.extensions.foundation.ERXProperties;
/**
* Simple static resource request handler. Allows for better debugging
* and you can set the document root via the system property <code>WODocumentRoot</code>.
* @author ak
*/
public class ERXStaticResourceRequestHandler extends WORequestHandler {
private static Logger log = Logger.getLogger(ERXStaticResourceRequestHandler.class);
private static WOApplication application = WOApplication.application();
private String _documentRoot;
private boolean _useRequestHandlerPath;
public ERXStaticResourceRequestHandler() {
_documentRoot = null;
}
/**
* Creates a static resource handler for the given framework, which gives you
* nicer relative URLs to work with. For instance, you could register a request
* handler "aj" that maps to the "Ajax" framework, which would make URLs of the
* form "/aj/wonder.js" map onto Ajax's WebServerResources/wonder.js folder.
*
* @param frameworkName the name of the framework to map to (or null/"app" for the application)
*/
public ERXStaticResourceRequestHandler(String frameworkName) {
if ("app".equals(frameworkName)) {
frameworkName = null;
}
WODeployedBundle bundle = WOApplication.application().resourceManager()._cachedBundleForFrameworkNamed(frameworkName);
File bundleFile = new File(bundle.bundlePath());
if (bundle.isFramework()) {
bundleFile = new File(bundleFile, "WebServerResources");
}
else {
bundleFile = new File(new File(bundleFile, "Contents"), "WebServerResources");
}
_documentRoot = bundleFile.getAbsolutePath();
_useRequestHandlerPath = true;
}
protected WOResponse _generateResponseForInputStream(InputStream is, long length, String type) {
WOResponse response = application.createResponseInContext(null);
if (is != null) {
if (length != 0) {
response.setContentStream(is, 50*1024, length);
}
} else {
response.setStatus(404);
}
if (type != null) {
response.setHeader(type, "content-type");
}
if(length != 0) {
response.setHeader("" + length, "content-length");
}
return response;
}
private String documentRoot() {
if (_documentRoot == null) {
_documentRoot = ERXProperties.stringForKey("WODocumentRoot");
if(_documentRoot == null) {
NSBundle bundle = NSBundle.bundleForName("JavaWebObjects");
NSDictionary dict = ERXDictionaryUtilities.dictionaryFromPropertyList("WebServerConfig", bundle);
_documentRoot = (String) dict.objectForKey("DocumentRoot");
}
}
return _documentRoot;
}
@Override
public WOResponse handleRequest(WORequest request) {
WOResponse response = null;
FileInputStream is = null;
long length = 0;
String contentType = null;
String uri = request.uri();
if (uri.charAt(0) == '/') {
WOResourceManager rm = application.resourceManager();
String documentRoot = documentRoot();
File file = null;
StringBuilder sb = new StringBuilder(documentRoot.length() + uri.length());
String wodataKey = request.stringFormValueForKey("wodata");
if(uri.startsWith("/cgi-bin") && wodataKey != null) {
uri = wodataKey;
if(uri.startsWith("file:")) {
// remove file:/
uri = uri.substring(5);
} else {
}
} else {
int index = uri.indexOf("/wodata=");
if(index >= 0) {
uri = uri.substring(index+"/wodata=".length());
} else {
sb.append(documentRoot);
}
}
if (_useRequestHandlerPath) {
try {
WODynamicURL dynamicURL = new WODynamicURL(uri);
String requestHandlerPath = dynamicURL.requestHandlerPath();
if (requestHandlerPath == null || requestHandlerPath.length() == 0) {
sb.append(uri);
} else {
sb.append('/');
sb.append(requestHandlerPath);
}
}
catch (Exception e) {
throw new RuntimeException("Failed to parse URL '" + uri + "'.", e);
}
}
else {
sb.append(uri);
}
String path = sb.toString();
try {
path = path.replaceAll("\\?.*", "");
if (request.userInfo() != null && !request.userInfo().containsKey("HttpServletRequest")) {
/* PATH_INFO is already decoded by the servlet container */
path = path.replace('+', ' ');
path = URLDecoder.decode(path, CharEncoding.UTF_8);
}
file = new File(path);
length = file.length();
is = new FileInputStream(file);
contentType = rm.contentTypeForResourceNamed(path);
log.debug("Reading file '" + file + "' for uri: " + uri);
} catch (IOException ex) {
if (!uri.toLowerCase().endsWith("/favicon.ico")) {
log.info("Unable to get contents of file '" + file + "' for uri: " + uri);
}
}
} else {
log.error("Can't fetch relative path: " + uri);
}
response = _generateResponseForInputStream(is, length, contentType);
NSNotificationCenter.defaultCenter().postNotification(WORequestHandler.DidHandleRequestNotification, response);
response._finalizeInContext(null);
return response;
}
}