diff options
Diffstat (limited to 'trunk/infrastructure/net.appjet.common/util/HttpServletRequestFactory.java')
| -rw-r--r-- | trunk/infrastructure/net.appjet.common/util/HttpServletRequestFactory.java | 306 |
1 files changed, 306 insertions, 0 deletions
diff --git a/trunk/infrastructure/net.appjet.common/util/HttpServletRequestFactory.java b/trunk/infrastructure/net.appjet.common/util/HttpServletRequestFactory.java new file mode 100644 index 0000000..4d7826a --- /dev/null +++ b/trunk/infrastructure/net.appjet.common/util/HttpServletRequestFactory.java @@ -0,0 +1,306 @@ +/** + * Copyright 2009 Google 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 net.appjet.common.util; + +import javax.servlet.http.HttpServletRequest; +import javax.servlet.http.HttpServletResponse; +import java.net.URL; +import java.net.URI; +import java.net.URLDecoder; +import java.util.ArrayList; +import java.util.HashMap; +import java.util.List; +import java.util.Map; +import java.util.Enumeration; +import java.util.Collections; +import java.util.Locale; +import java.util.Iterator; +import java.util.Vector; + +public class HttpServletRequestFactory { + public static class RequestResponse { + public final HttpServletRequest request; + public final HttpServletResponse response; + + private RequestResponse(HttpServletRequest req, HttpServletResponse res) { + request = req; + response = res; + } + } + + public static HttpServletRequest createRequest(String uri, Map<String, String> headers, + String method, String body) + throws java.net.URISyntaxException { + return new InnerHttpServletRequest(new URI(uri), headers, method, body); + } + + public static HttpServletRequest createRequest(HttpServletRequest req) + throws java.net.URISyntaxException { + Map<String, String> headers = new java.util.HashMap<String, String>(); + Enumeration<String> headerNames = (Enumeration<String>) req.getHeaderNames(); + while (headerNames.hasMoreElements()) { + String e = headerNames.nextElement(); + headers.put(e, req.getHeader(e)); + } + return createRequest( + req.getRequestURL() + + (req.getQueryString() != null ? "?"+req.getQueryString() : ""), + headers, req.getMethod(), null); + } + + public static HttpServletResponse createResponse() { + return new InnerHttpServletResponse(); + } + + public static RequestResponse createPair(String uri, Map<String, String> headers, + String method, String body) + throws java.net.URISyntaxException { + return new RequestResponse(createRequest(uri, headers, method, body), createResponse()); + } + + public static interface ServletAccessor { + int getStatusCode(); + String getOutput(); + } + + private static class InnerHttpServletRequest implements HttpServletRequest { + private String method; + private String host; + private String scheme; + private int port; + private String path; + private String queryString; + private Map<String, String[]> parameters; + private Map<String, String> headers; + private final String body; + + public InnerHttpServletRequest(URI uri, Map<String, String> headers, String method, + String body) + throws java.net.URISyntaxException { + this.method = method; + this.host = uri.getHost(); + this.scheme = uri.getScheme(); + this.port = uri.getPort(); + this.path = uri.getRawPath(); + this.queryString = uri.getRawQuery(); + extractParameters(); + extractHeaders(headers); + this.headers.put("host", host); + if (body != null) + this.headers.put("content-length", Integer.toString(body.length())); + this.body = body; + } + + private void extractHeaders(Map<String, String> headers) { + this.headers = new HashMap<String, String>(); + for (Map.Entry<String, String> kv : headers.entrySet()) { + this.headers.put(kv.getKey().toLowerCase(), kv.getValue()); + } + } + + private String decodeUTF8(String s) { + try { + return URLDecoder.decode(s, "UTF-8"); + } catch (java.io.UnsupportedEncodingException e) { + System.err.println("Unsupported character encoding! UTF-8"); + return s; + } + } + + private void extractParameters() { + parameters = new HashMap<String, String[]>(); + if (queryString == null) + return; + + Map<String, List<String> > params = new HashMap<String, List<String> >(); + String[] pairs = queryString.split("&"); + for (String s : pairs) { + String[] kv = s.split("=", 2); + if (! params.containsKey(kv[0])) { + params.put(decodeUTF8(kv[0]), new ArrayList<String>()); + } + params.get(decodeUTF8(kv[0])).add(decodeUTF8(kv[1])); + } + String[] stringArray = new String[0]; + + for (Map.Entry<String, List<String> > e : params.entrySet()) { + parameters.put(e.getKey(), e.getValue().toArray(stringArray)); + } + } + + // HttpServletRequest methods + public String getAuthType() { return null; } + public String getContextPath() { return ""; } + public javax.servlet.http.Cookie[] getCookies() { return new javax.servlet.http.Cookie[0]; } + @SuppressWarnings({"deprecation"}) + public long getDateHeader(String name) { return java.util.Date.parse(getHeader(name)); } + public String getHeader(String name) { return headers.get(name.toLowerCase()); } + public Enumeration<String> getHeaders(String name) { + Vector<String> v = new Vector<String>(); + v.add(getHeader(name)); + return v.elements(); + } + public Enumeration<String> getHeaderNames() { + return Collections.enumeration(headers.keySet()); + } + public int getIntHeader(String name) { return Integer.parseInt(getHeader(name)); } + public String getMethod() { return method.toUpperCase(); } + public String getPathInfo() { return null; } + public String getPathTranslated() { return null; } + public String getQueryString() { return queryString; } + public String getRemoteUser() { return null; } + public boolean isUserInRole(String role) { return false; } + public java.security.Principal getUserPrincipal() { return null; } + public String getRequestedSessionId() { return null; } + public String getRequestURI() { return path; } + public StringBuffer getRequestURL() { + return new StringBuffer(scheme+"://"+host+(port==-1?"":":"+port)+path); + } + public String getServletPath() { return ""; } + public javax.servlet.http.HttpSession getSession(boolean create) { return null; } + public javax.servlet.http.HttpSession getSession() { return null; } + public boolean isRequestedSessionIdValid() { return false; } + public boolean isRequestedSessionIdFromCookie() { return false; } + public boolean isRequestedSessionIdFromURL() { return false; } + public boolean isRequestedSessionIdFromUrl() { return isRequestedSessionIdFromURL(); } + + // ServletRequest methods + public Object getAttribute(String name) { return null; } + public Enumeration<String> getAttributeNames() { + return Collections.enumeration(new ArrayList<String>()); + } + public String getCharacterEncoding() { return null; } + public void setCharacterEncoding(String env) { } + public int getContentLength() { + return ((getHeader("Content-Length") == null) + ? (body == null ? 0 : body.length()) + : getIntHeader("Content-Length")); + } + public String getContentType() { return getHeader("Content-Type"); } + public javax.servlet.ServletInputStream getInputStream() throws java.io.IOException{ + return new javax.servlet.ServletInputStream() { + private java.io.InputStream istream = + new java.io.ByteArrayInputStream(body.getBytes()); + public int read() throws java.io.IOException { + return istream.read(); + } + }; + } + public String getParameter(String name) { + String[] vals = getParameterValues(name); + if (vals == null) return null; + if (vals.length < 1) return null; + return vals[0]; + } + public Enumeration<String> getParameterNames() { + return Collections.enumeration(parameters.keySet()); + } + public String[] getParameterValues(String name) { return parameters.get(name); } + public Map getParameterMap() { return Collections.unmodifiableMap(parameters); } + public String getProtocol() { return "HTTP/1.1"; } + public String getScheme() { return scheme; } + public String getServerName() { return host; } + public int getServerPort() { return port; } + public java.io.BufferedReader getReader() { + return new java.io.BufferedReader(new java.io.StringReader(body)); + } + public String getRemoteAddr() { return "127.0.0.1"; } + public String getRemoteHost() { return "localhost"; } + public void setAttribute(String name, Object o) { } + public void removeAttribute(String name) { } + public java.util.Locale getLocale() { return java.util.Locale.US; } + public Enumeration<java.util.Locale> getLocales() { + Vector<java.util.Locale> v = new Vector<java.util.Locale>(); + v.add(java.util.Locale.US); + return v.elements(); + } + public boolean isSecure() { return false; } + public javax.servlet.RequestDispatcher getRequestDispatcher(String path) { return null; } + public String getRealPath(String path) { return null; } + public int getRemotePort() { return -1; } + public String getLocalName() { return "localhost"; } + public String getLocalAddr() { return "127.0.0.1"; } + public int getLocalPort() { return 80; } + } + + private static class InnerHttpServletResponse implements HttpServletResponse, ServletAccessor { + private InnerHttpServletResponse() { } + + // ServletAccessor methods + public int getStatusCode() { return e_code; } + public String getOutput() { + try { + writer.flush(); + ostream.flush(); + } catch (java.io.IOException e) { + return "(An IOException occurred while getting output: "+e.getMessage()+")"; + } + return ostream.toString(); + } + + // HttpServletResponse methods + private int e_code = 200; + private String e_msg = ""; + + public void addCookie(javax.servlet.http.Cookie cookie) { } + public void addDateHeader(String name, long date) { } + public void addHeader(String name, String value) { } + public void addIntHeader(String name, int value) { } + public boolean containsHeader(String name) { return true; } + public String encodeRedirectUrl(String url) { return encodeRedirectURL(url); } + public String encodeRedirectURL(String url) { return url; } + public String encodeUrl(String url) { return encodeURL(url); } + public String encodeURL(String url) { return url; } + public void sendError(int sc) { e_code = sc; } + public void sendError(int sc, String msg) { e_code = sc; e_msg = msg;} + public void sendRedirect(String location) { } + public void setDateHeader(String name, long date) { } + public void setHeader(String name, String value) { } + public void setIntHeader(String name, int value) { } + public void setStatus(int sc) { e_code = sc; } + public void setStatus(int sc, String sm) { e_code = sc; e_msg = sm; } + + // ServletResponse methods + private String c_enc = ""; + private String c_type = ""; + private java.util.Locale locale = java.util.Locale.US; + private final java.io.OutputStream ostream = new java.io.ByteArrayOutputStream(); + private final javax.servlet.ServletOutputStream sostream = + new javax.servlet.ServletOutputStream() { + public void write(int b) throws java.io.IOException { + ostream.write(b); + } + }; + private final java.io.PrintWriter writer = new java.io.PrintWriter(ostream); + + public void flushBuffer() { } + public int getBufferSize() { return 0; } + public String getCharacterEncoding() { return c_enc; } + public String getContentType() { return c_type; } + public java.util.Locale getLocale() { return locale; } + public javax.servlet.ServletOutputStream getOutputStream() { return sostream; } + public java.io.PrintWriter getWriter() { return writer; } + public boolean isCommitted() { return false; } + public void reset() { } + public void resetBuffer() { } + public void setBufferSize(int size) { } + public void setCharacterEncoding(String charset) { c_enc = charset; } + public void setContentLength(int len) { } + public void setContentType(String type) { c_type = type; } + public void setLocale(java.util.Locale loc) { locale = loc; } + } +}
\ No newline at end of file |