org.appcelerator.transport.ProxyTransportServlet.java Source code

Introduction

Here is the source code for org.appcelerator.transport.ProxyTransportServlet.java

Source

/*
 * Copyright 2006-2008 Appcelerator, Inc.
 * 
 * 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. 
 */

package org.appcelerator.transport;

import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.net.URLDecoder;

import javax.servlet.ServletConfig;
import javax.servlet.ServletException;
import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

import org.apache.commons.httpclient.Header;
import org.apache.commons.httpclient.HttpClient;
import org.apache.commons.httpclient.HttpMethodBase;
import org.apache.commons.httpclient.methods.ByteArrayRequestEntity;
import org.apache.commons.httpclient.methods.DeleteMethod;
import org.apache.commons.httpclient.methods.GetMethod;
import org.apache.commons.httpclient.methods.OptionsMethod;
import org.apache.commons.httpclient.methods.PostMethod;
import org.apache.commons.httpclient.methods.PutMethod;
import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import org.appcelerator.annotation.AnnotationHelper;
import org.appcelerator.util.Base64;
import org.appcelerator.util.Util;

/**
 * Cross-domain proxy servlet
 *
 * @author <a href="mailto:jhaynie@appcelerator.com">Jeff Haynie</a>
 */
public class ProxyTransportServlet extends HttpServlet {
    private static final long serialVersionUID = 1L;
    private static final Log LOG = LogFactory.getLog(ProxyTransportServlet.class);

    private boolean embeddedMode;

    /* (non-Javadoc)
     * @see javax.servlet.GenericServlet#init(javax.servlet.ServletConfig)
     */
    @Override
    public void init(ServletConfig config) throws ServletException {
        super.init(config);

        if (this.embeddedMode) {
            AnnotationHelper.initializeAnnotationDBFromClasspath();
        } else {
            AnnotationHelper.initializeAnnotationDBFromServlet(config.getServletContext());
        }
    }

    /**
     * called to indicate that the class path must be used when loading annotations instead
     * of WAR lib
     */
    public void setEmbeddedMode(boolean embed) {
        this.embeddedMode = embed;
    }

    @Override
    protected void service(HttpServletRequest request, HttpServletResponse response)
            throws ServletException, IOException {
        String method = request.getMethod();
        String url = request.getParameter("url");
        if (url == null) {
            response.sendError(HttpServletResponse.SC_BAD_REQUEST);
            return;
        }

        url = URLDecoder.decode(url, "UTF-8");

        if (url.indexOf("://") == -1) {
            url = new String(Base64.decode(url));
        }

        HttpClient client = new HttpClient();
        HttpMethodBase methodBase = null;

        if (method.equalsIgnoreCase("POST")) {
            methodBase = new PostMethod(url);
            ByteArrayOutputStream out = new ByteArrayOutputStream();
            Util.copy(request.getInputStream(), out);
            ByteArrayRequestEntity req = new ByteArrayRequestEntity(out.toByteArray());
            ((PostMethod) methodBase).setRequestEntity(req);
        } else if (method.equalsIgnoreCase("GET")) {
            methodBase = new GetMethod(url);
            methodBase.setFollowRedirects(true);
        } else if (method.equalsIgnoreCase("PUT")) {
            methodBase = new PutMethod(url);
            ByteArrayOutputStream out = new ByteArrayOutputStream();
            Util.copy(request.getInputStream(), out);
            ByteArrayRequestEntity req = new ByteArrayRequestEntity(out.toByteArray());
            ((PutMethod) methodBase).setRequestEntity(req);
        } else if (method.equalsIgnoreCase("DELETE")) {
            methodBase = new DeleteMethod(url);
        } else if (method.equalsIgnoreCase("OPTIONS")) {
            methodBase = new OptionsMethod(url);
        }

        methodBase.setRequestHeader("User-Agent", request.getHeader("user-agent") + " (Appcelerator Proxy)");

        if (LOG.isDebugEnabled()) {
            LOG.debug("Proxying url: " + url + ", method: " + method);
        }

        int status = client.executeMethod(methodBase);

        response.setStatus(status);
        response.setContentLength((int) methodBase.getResponseContentLength());

        for (Header header : methodBase.getResponseHeaders()) {
            String name = header.getName();
            if (name.equalsIgnoreCase("Set-Cookie") == false && name.equals("Transfer-Encoding") == false) {
                response.setHeader(name, header.getValue());
            }
        }
        InputStream in = methodBase.getResponseBodyAsStream();
        Util.copy(in, response.getOutputStream());
    }
}