xpressj.Response Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of xpressj Show documentation
A Sinatra/Express.js inspired lightweight Java web framework
There is a newer version: 0.7.2
/*
 * Copyright 2014 - Alexey Kamenskiy
 *
 * 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 xpressj;

import com.google.gson.Gson;

import javax.servlet.http.HttpServletResponse;
import java.util.Collection;
import java.util.HashMap;
import java.util.Map;

/**
 * Created by akamensky on 6/19/14.
 */
public class Response {

    private boolean isConsumed = false;
    private HttpServletResponse httpResponse;
    private static int DEFAULT_STATUS_CODE = 200;
    private Integer statusCode;
    private HashMap cookies;
    private Map headers;
    private Request request;

    public Response(HttpServletResponse httpResponse) {
        this.httpResponse = httpResponse;
        this.cookies = new HashMap<>();
        this.headers = new HashMap<>();
    }
    public Response() {}

    public void setDelegate(Request request){
        this.request = request;
    }

    private void markConsumed(){
        this.isConsumed = true;
    }

    /**
     * True if response was already sent, no changes will affect HTTP response
     * @return boolean
     */
    public boolean isConsumed(){
        return this.isConsumed;
    }

    /**
     * Set status code for response. If not set DEFAULT_STATUS_CODE will be used.
     * @param status int Numeric HTTP response code
     */
    public void setStatusCode(int status){
        this.statusCode = status;
    }

    /**
     * Returns status code for HTTP response
     * @return int
     */
    public int getStatusCode(){
        if(this.statusCode == null){
            return DEFAULT_STATUS_CODE;
        } else {
            return this.statusCode;
        }
    }

    /**
     * Sends plain string response to client
     * @param code int status code
     * @param body string response body
     */
    public void send(int code, String body){
        setStatusCode(code);
        send(body);
    }

    /**
     * Sends plain string response to client
     * @param body String
     */
    public void send(String body) {
        writeResponse(body);
    }

    private void writeResponse(String body){
        try {
            if (this.isConsumed){
                throw new RuntimeException("Request has already been consumed");
            }
            httpResponse.setStatus(getStatusCode());
            httpResponse.setHeader("Content-Type", "text/html; charset=utf-8");

            //Adding cookies to the response
            for (Cookie cookie : this.cookies.values()){
                httpResponse.addCookie(cookie.getServletCookie());
            }

            //Adding headers to the response
            Collection headerNames = this.headers.keySet();
            for (String headerName : headerNames){
                httpResponse.setHeader(headerName, this.headers.get(headerName));
            }

            httpResponse.getOutputStream().write(body.getBytes("utf-8"));
            markConsumed();
        } catch (Exception e){
            e.printStackTrace();
        }
    }

    /**
     * Sends plain string response to client
     * @param code int status code
     * @param obj Object
     */
    public void json(int code, Object obj){
        setStatusCode(code);
        json(obj);
    }

    /**
     * Transforms Object to JSON and sends JSON to client (using Googles GSON library)
     * @param obj Object
     */
    public void json(Object obj){
        Gson gson = new Gson();
        String json = gson.toJson(obj);
        writeResponse(json);
    }

    public void addCookie(Cookie cookie){
        this.cookies.put(cookie.getName(), cookie);
    }

    public void addCookie(String name, String value){
        Cookie cookie = new Cookie(name, value);
        addCookie(cookie);
    }

    public void unsetCookie(String name){
        this.cookies.remove(name);
    }

    public void addHeader(String name, String value){
        this.headers.put(name, value);
    }
}