HttpResponse.java

import java.util.HashMap;
import java.util.Map;

/**
 * Represents an HTTP response
 */
public class HttpResponse {
    private final int statusCode;
    private final String statusMessage;
    private final String contentType;
    private final Map<String, String> headers;
    private byte[] content;
    private String textContent;
    
    public HttpResponse(int statusCode, String statusMessage, String content, String contentType) {
        this.statusCode = statusCode;
        this.statusMessage = statusMessage;
        this.contentType = contentType;
        this.headers = new HashMap<>();
        this.textContent = content;
    }
    
    public HttpResponse(int statusCode, String statusMessage, byte[] content, String contentType) {
        this.statusCode = statusCode;
        this.statusMessage = statusMessage;
        this.contentType = contentType;
        this.headers = new HashMap<>();
        this.content = content;
    }
    
    public int getStatusCode() {
        return statusCode;
    }
    
    public String getStatusMessage() {
        return statusMessage;
    }
    
    public String getContentType() {
        return contentType;
    }
    
    public Map<String, String> getHeaders() {
        return headers;
    }
    
    public void addHeader(String name, String value) {
        headers.put(name, value);
    }
    
    public byte[] getContent() {
        if (content != null) {
            return content;
        } else if (textContent != null) {
            return textContent.getBytes();
        }
        return new byte[0];
    }
    
    public int getContentLength() {
        return getContent().length;
    }
}