Untitled

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:hprange@gmail.com.br">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;
        }
}