org.threadly.litesockets.protocols.http.response.HTTPResponseBuilder Maven / Gradle / Ivy

Go to download
package org.threadly.litesockets.protocols.http.response;

import java.util.Map;
import java.util.Map.Entry;
import java.util.TreeMap;

import org.threadly.litesockets.protocols.http.shared.HTTPConstants;
import org.threadly.litesockets.protocols.http.shared.HTTPHeaders;
import org.threadly.litesockets.protocols.http.shared.HTTPResponseCode;

/**
 * A builder to help create {@link HTTPResponse} objects.
 * 
 * @author lwahlmeier
 *
 */
public class HTTPResponseBuilder {
  private final Map headers = new TreeMap(String.CASE_INSENSITIVE_ORDER);
  private HTTPResponseHeader rHeader = HTTPConstants.OK_RESPONSE_HEADER;
  
  public HTTPResponseBuilder() {
    headers.putAll(HTTPConstants.DEFAULT_HEADERS.getHeadersMap());
    headers.put(HTTPConstants.HTTP_KEY_CONTENT_LENGTH, "0");
  }
  
  public HTTPResponseBuilder setHTTPVersion(String version) {
    rHeader = new HTTPResponseHeader(rHeader.getResponseCode(), version);
    return this;
  }
  
  
  public HTTPResponseBuilder setHeader(String key, String value) {
    if(value == null) {
      headers.remove(key);
    } else {
      headers.put(key, value);
    }
    return this;
  }
  
  public HTTPResponseBuilder setHeaders(HTTPHeaders hh) {
    this.headers.clear();
    for(Entry head: hh.getHeadersMap().entrySet()) {
      setHeader(head.getKey(), head.getValue());
    }
    return this;
  }
  
  public HTTPResponseBuilder setResponseHeader(HTTPResponseHeader hrh) {
    rHeader = hrh;
    return this;
  }
  
  public HTTPResponseBuilder setResponseCode(HTTPResponseCode rc) {
    rHeader = new HTTPResponseHeader(rc, rHeader.getHTTPVersion());
    return this;
  }
  
  public HTTPResponse build() {
    return new HTTPResponse(rHeader, new HTTPHeaders(headers));
  }
}