cn.dreampie.common.http.HttpRequest Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of resty-common Show documentation
Resty common tools
There is a newer version: 1.3.0.RELEASE
package cn.dreampie.common.http;

import javax.servlet.RequestDispatcher;
import javax.servlet.ServletContext;
import javax.servlet.ServletRequest;
import javax.servlet.http.Cookie;
import javax.servlet.http.HttpServletRequest;
import java.io.IOException;
import java.io.InputStream;
import java.io.UnsupportedEncodingException;
import java.net.URLDecoder;
import java.util.*;

/**
 * Date: 1/22/13
 * Time: 2:52 PM
 */
public class HttpRequest extends AbstractRequest {
  private final HttpServletRequest request;
  private final ServletContext servletContext;
  private Map> queryParams;

  public HttpRequest(HttpServletRequest request, ServletContext servletContext) {
    this.request = request;
    this.servletContext = servletContext;
  }

  private static String getCookieValue(Cookie[] cookies,
                                       String cookieName) {
    if (cookies == null) {
      return null;
    }
    for (Cookie cookie : cookies) {
      if (cookieName.equals(cookie.getName()))
        return cookie.getValue();
    }
    return null;
  }

  static Cookie getCookie(Cookie[] cookies, String cookieName) {
    if (cookies == null) {
      return null;
    }
    for (Cookie cookie : cookies) {
      if (cookieName.equals(cookie.getName()))
        return cookie;
    }
    return null;
  }

  public String getLocalClientAddress() {
    return request.getRemoteAddr();
  }

  protected String getBasePath() {
    return request.getContextPath();
  }

  protected String getLocalScheme() {
    return request.getScheme();
  }

  public String getRestPath() {
    String basepath = getBasePath();
    String requestURI = request.getRequestURI();
    if (basepath.length() > 0) {
      requestURI = request.getRequestURI().substring(basepath.length());
    }
    int index = requestURI.toLowerCase().indexOf(";jsessionid=");
    if (index != -1) {
      requestURI = requestURI.substring(0, index);
    }
    try {
      return URLDecoder.decode(requestURI, getCharacterEncoding());
    } catch (UnsupportedEncodingException e) {
      throw new IllegalArgumentException("Invalid character encoding for '" + getCharacterEncoding() + "'");
    }
  }

  public String getRealPath(String path) {
    return servletContext.getRealPath(path);
  }

  public String getRestUri() {
    String queryString = getQueryString();
    if (queryString == null) {
      return getRestPath();
    } else {
      return getRestPath() + "?" + queryString;
    }
  }

  public String getQueryParam(String param) {
    return request.getParameter(param);
  }

  public List getQueryParams(String param) {
    String[] values = request.getParameterValues(param);
    if (values != null)
      return Arrays.asList(values);
    else
      return null;
  }

  public Map> getQueryParams() {
    if (queryParams == null) {
      Map paramMap = getParamMap();
      Map> params = new HashMap>();
      for (Map.Entry entry : paramMap.entrySet()) {
        params.put(entry.getKey(), Arrays.asList(entry.getValue()));
      }
      queryParams = params;
    }
    return queryParams;
  }

  protected Map getParamMap() {
    return request.getParameterMap();
  }

  public int getContentLength() {
    return request.getContentLength();
  }

  public InputStream getContentStream() throws IOException {
    return request.getInputStream();
  }

  public String getHttpMethod() {
    String httpMethod = request.getMethod();
    if (httpMethod == null) {
      throw new IllegalArgumentException("Invalid HTTP Method for " + getRestPath());
    }
    return httpMethod;
  }

  public Map getCookiesMap() {
    Map cookies = new LinkedHashMap();
    Cookie[] requestCookies = request.getCookies();
    if (requestCookies != null) {
      for (Cookie cookie : requestCookies) {
        cookies.put(cookie.getName(), cookie.getValue());
      }
    }
    return cookies;
  }

  public String getCookieValue(String cookieName) {
    return getCookieValue(request.getCookies(), cookieName);
  }

  public boolean isPersistentCookie(String cookie) {
    Cookie c = getCookie(request.getCookies(), cookie);
    return c != null && c.getMaxAge() > 0;
  }

  public String getQueryString() {
    try {
      String queryString = request.getQueryString();
      if (queryString != null && !"".equals(queryString)) {
        return URLDecoder.decode(request.getQueryString(), getCharacterEncoding());
      } else {
        return queryString;
      }
    } catch (UnsupportedEncodingException e) {
      throw new IllegalArgumentException("Invalid character encoding for '" + getCharacterEncoding() + "'");
    }
  }

  public RequestDispatcher getRequestDispatcher(String url) {
    return request.getRequestDispatcher(url);
  }

  public String getHeader(String headerName) {
    return request.getHeader(headerName);
  }

  public Enumeration getHeaders(String headerName) {
    return request.getHeaders(headerName);
  }

  public Map getHeaders() {
    Map map = new HashMap();
    Enumeration headerNames = request.getHeaderNames();
    while (headerNames.hasMoreElements()) {
      String key = (String) headerNames.nextElement();
      String value = request.getHeader(key);
      map.put(key, value);
    }
    return map;
  }

  public String getContentType() {
    return request.getContentType();
  }


  @SuppressWarnings("unchecked")
  public  T unwrap(Class clazz) {
    if (clazz == HttpServletRequest.class || clazz == ServletRequest.class) {
      return (T) request;
    }
    throw new IllegalArgumentException("underlying implementation is HttpServletRequest, not " + clazz.getName());
  }


  public Locale getLocale() {
    return request.getLocale();
  }

  public List getLocales() {
    return Collections.list((request.getLocales()));
  }

  public String getCharacterEncoding() {
    return request.getCharacterEncoding();
  }

  public void setCharacterEncoding(String encoding) throws UnsupportedEncodingException {
    request.setCharacterEncoding(encoding);
  }
}