JarResourceRequestHandler.java

import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.net.MalformedURLException;
import java.net.URL;

import org.apache.commons.lang.StringUtils;
import org.apache.log4j.Logger;

import com.webobjects.appserver.WOApplication;
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.WOResourceRequestHandler;
import com.webobjects.appserver._private.WOShared;
import com.webobjects.appserver._private.WOURLValuedElementData;
import com.webobjects.foundation.NSBundle;
import com.webobjects.foundation.NSDictionary;
import com.webobjects.foundation.NSLog;
import com.webobjects.foundation.NSNotificationCenter;
import com.webobjects.foundation.NSPathUtilities;

import er.extensions.foundation.ERXDictionaryUtilities;
import er.extensions.foundation.ERXProperties;

/** 
 * @author <a href="mailto:[email protected]">Henrique Prange</a>
 */
public class JarResourceRequestHandler extends WOResourceRequestHandler {
	protected static final WOApplication APPLICATION = WOApplication.application();

	private static final Logger LOGGER = Logger.getLogger(JarResourceRequestHandler.class);

	private String _documentRoot;

	public JarResourceRequestHandler() {
		_documentRoot = null;
	}

	protected WOResponse _generateResponseForInputStream(InputStream is, int 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;
	}

	protected WOResponse generateResponseForInputStream(InputStream is, long aContentLength, String aContentType) {
		WOResponse aResponse = APPLICATION.createResponseInContext(null);

		if (aContentType != null) {
			aResponse.setHeader(aContentType, "content-type");
		}

		if (is != null && aContentLength != 0L) {
			aResponse.setHeader(Long.toString(aContentLength), "content-length");
			aResponse.setContentStream(is, 131072, aContentLength);
		} else {
			LOGGER.warn("The resource was not found. Turn log DEBUG on for more details.");

			aResponse.setStatus(404);
			aResponse.setHeader(WOShared.unsignedIntString(0), "content-length");
		}

		return aResponse;
	}

	@Override
	public WOResponse handleRequest(WORequest request) {
		WOResponse response = null;
		FileInputStream is = null;
		int length = 0;
		String contentType = null;
		String uri = request.uri();
		if (uri.charAt(0) == '/') {
			WOResourceManager rm = APPLICATION.resourceManager();
			String documentRoot = documentRoot();
			File file = null;
			StringBuffer sb = new StringBuffer(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 {
				int index = uri.indexOf("/wodata=");

				if (index >= 0) {
					uri = uri.substring(index + "/wodata=".length());
				} else {
					sb.append(documentRoot);
				}
			}
			sb.append(uri);
			String path = sb.toString();
			try {
				path = path.replace('+', ' ');
				path = path.replaceAll("\\?.*", "");
				file = new File(path);
				length = (int) file.length();
				is = new FileInputStream(file);

				contentType = rm.contentTypeForResourceNamed(path);
				LOGGER.debug("Reading file '" + file + "' for uri: " + uri);
			} catch (IOException ex) {
				if (!uri.toLowerCase().endsWith("/favicon.ico")) {
					LOGGER.debug("Unable to get contents of file '" + file + "' for uri: " + uri);

					return handleRequestWithResourceInsideJar(request);
				}
			}
		} else {
			LOGGER.error("Can't fetch relative path: " + uri);
		}
		response = _generateResponseForInputStream(is, length, contentType);
		NSNotificationCenter.defaultCenter().postNotification(WORequestHandler.DidHandleRequestNotification, response);
		response._finalizeInContext(null);
		return response;
	}

	public WOResponse handleRequestWithResourceInsideJar(WORequest request) {
		LOGGER.debug("Handling the request for (entire URI) " + request.uri());

		boolean requestHandlerContainsPath = false;

		String requestHandlerPath = request.requestHandlerPath();
		String resourceDataKey = request.stringFormValueForKey("wodata");

		if (requestHandlerPath != null
				&& (requestHandlerPath.endsWith(".class") || requestHandlerPath.endsWith(".jar") || requestHandlerPath.endsWith(".zip") || requestHandlerPath
						.endsWith(".table")) && requestHandlerPath.indexOf("..") == requestHandlerPath.indexOf('~')) {
			requestHandlerContainsPath = true;
		}

		WOResponse response = null;

		if (requestHandlerContainsPath && APPLICATION.isDirectConnectEnabled()) {
			LOGGER.debug("The path to resources is (based on request handler path) " + requestHandlerPath);

			response = responseForJavaClassAtPath(requestHandlerPath);
		} else if (StringUtils.isNotBlank(requestHandlerPath) && resourceDataKey == null) {
			// A classe ERXResourceManager altera a URL de forma errada. Ao
			// invés de ?wodata=, o Anjo mudou para /wodata=. Isso faz com
			// que seja necessário tratarmos aqui essa key.
			requestHandlerPath = StringUtils.replace(requestHandlerPath, "wodata=", "");

			// Tosca modificacao para carregar recursos no WO541 e Wonder 4 em Windows
			requestHandlerPath = StringUtils.replace(requestHandlerPath, "%3A", ":");

			LOGGER.debug("The path to resources is (based on a corrected path) " + requestHandlerPath);

			URL resourcesUrl = null;

			try {
				resourcesUrl = new URL("file", "", requestHandlerPath);

				response = responseForDataAtURL(resourcesUrl);
			} catch (MalformedURLException exception) {
				LOGGER.error("An error occurred while trying to handle the resource", exception);
			}
		} else if (resourceDataKey != null) {
			LOGGER.debug("The path to resources is (based on wodata key) " + resourceDataKey);

			response = responseForDataCachedWithKey(resourceDataKey);
		}

		if (response == null) {
			LOGGER.warn("THE REQUEST CANNOT BE CORRECTLY HANDLED. GENERATING AN EMPTY RESPONSE.");

			String contentType = request.headerForKey("content-type");

			if (contentType == null) {
				contentType = "text/plain";
			}

			response = generateResponseForInputStream(null, 0L, contentType);
		}

		NSNotificationCenter.defaultCenter().postNotification(WORequestHandler.DidHandleRequestNotification, response);
		response._finalizeInContext(null);

		return response;
	}

	protected WOResponse responseForDataAtURL(URL anURL) {
		InputStream is = null;

		long fileLength = 0L;

		String aResourcePath = anURL.toString();

		String aContentType = APPLICATION.resourceManager().contentTypeForResourceNamed(aResourcePath);

		try {
			fileLength = NSPathUtilities._contentLengthForPathURL(anURL);

			is = anURL.openStream();
		} catch (IOException ioe) {
			NSLog.err.appendln((new StringBuilder()).append("<").append(getClass().getName()).append("> Unable to get contents of file for path '").append(
					aResourcePath).append("': ").append(ioe).toString());

			if (NSLog.debugLoggingAllowedForLevelAndGroups(2, 36L)) {
				NSLog.debug.appendln(ioe);
			}
		}

		WOResponse aResponse = generateResponseForInputStream(is, fileLength, aContentType);

		return aResponse;
	}

	protected WOResponse responseForDataCachedWithKey(String aResourceKey) {
		WOResponse response = APPLICATION.createResponseInContext(null);

		WOResourceManager resourceManager = APPLICATION.resourceManager();

		WOURLValuedElementData aResourceDataObject = resourceManager._cachedDataForKey(aResourceKey);

		if (aResourceDataObject == null) {
			LOGGER.warn("The resource was not found in cache. Turn log DEBUG on for more details.");

			return response;
		}

		aResourceDataObject.appendToResponse(response, null);

		if (aResourceDataObject.isTemporary()) {
			resourceManager.removeDataForKey(aResourceKey, null);
		}

		return response;
	}

	protected WOResponse responseForJavaClassAtPath(String aPath) {
		WOResponse aResponse = null;

		URL anURL = APPLICATION.resourceManager()._pathURLForJavaClass(aPath);

		if (anURL != null) {
			aResponse = responseForDataAtURL(anURL);
		}

		return aResponse;
	}
}