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

Go to download
package cn.dreampie.common.http;

import cn.dreampie.common.Response;
import cn.dreampie.common.http.result.HttpStatus;

import javax.servlet.ServletException;
import javax.servlet.ServletResponse;
import javax.servlet.http.Cookie;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.io.OutputStream;

/**
 * Date: 2/6/13
 * Time: 9:40 PM
 */
public class HttpResponse extends AbstractResponse {
  private final HttpServletResponse response;
  private final HttpServletRequest request;

  public HttpResponse(HttpServletResponse response, HttpServletRequest request) {
    super(HttpServletResponse.class, response);
    this.response = response;
    this.request = request;
  }


  protected void doSetStatus(HttpStatus httpStatus) {
    response.setStatus(httpStatus.getCode());
  }


  protected OutputStream doGetOutputStream() throws IOException {
    return response.getOutputStream();
  }

  public Response addCookie(String cookie, String value, int expiration) {
    return addCookie(cookie, value, expiration, true);
  }

  public Response addCookie(String cookie, String value, int expiration, boolean httpOnly) {
    Cookie existingCookie = HttpRequest.getCookie(request.getCookies(), cookie);
    if (existingCookie != null) {
      if ("/".equals(existingCookie.getPath())
          || existingCookie.getPath() == null // in some cases cookies set on path '/' are returned with a null path
          ) {
        // update existing cookie
        existingCookie.setPath("/");
        existingCookie.setValue(value);
        existingCookie.setMaxAge(expiration);
//        existingCookie.setHttpOnly(httpOnly);
        response.addCookie(existingCookie);
      } else {
        // we have an existing cookie on another path: clear it, and add a new cookie on root path
        existingCookie.setValue("");
        existingCookie.setMaxAge(0);
        response.addCookie(existingCookie);

        Cookie c = new Cookie(cookie, value);
        c.setPath("/");
        c.setMaxAge(expiration);
//        c.setHttpOnly(httpOnly);
        response.addCookie(c);
      }
    } else {
      Cookie c = new Cookie(cookie, value);
      c.setPath("/");
      c.setMaxAge(expiration);
//      c.setHttpOnly(httpOnly);
      response.addCookie(c);
    }
    return this;
  }


  public Response clearCookie(String cookie) {
    Cookie existingCookie = HttpRequest.getCookie(request.getCookies(), cookie);
    if (existingCookie != null) {
      existingCookie.setPath("/");
      existingCookie.setValue("");
      existingCookie.setMaxAge(0);
      response.addCookie(existingCookie);
    }
    return this;
  }


  public void doSetHeader(String headerName, String header) {
    response.setHeader(headerName, header);
  }

  public void doAddHeader(String headerName, String header) {
    response.addHeader(headerName, header);
  }

  public void forward(String url) throws ServletException, IOException {
    request.getRequestDispatcher(url).forward(request, response);
  }

  public void sendRedirect(String url) throws ServletException, IOException {
    response.sendRedirect(url);
  }

  @SuppressWarnings("unchecked")
  public  T unwrap(Class clazz) {
    if (clazz == HttpServletResponse.class || clazz == ServletResponse.class) {
      return (T) response;
    }
    throw new IllegalArgumentException("Underlying implementation is HttpServletResponse, not " + clazz.getName());
  }

  public void reset() {
    response.reset();
  }

}