Renderer.java

/*
 * Copyright 2013 David Tinker
 *
 * 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 io.qdb.server.controller;

import org.simpleframework.http.Response;

import javax.inject.Inject;
import javax.inject.Singleton;
import java.io.IOException;

/**
 * Helps render HTTP responses.
 */
@Singleton
public class Renderer {

    protected final JsonService jsonService;

    @Inject
    public Renderer(JsonService jsonService) {
        this.jsonService = jsonService;
    }

    public void json(Response resp, Object o, boolean borg) throws IOException {
        byte[] bytes = jsonService.toJson(o, borg);
        resp.set("Content-Type", "application/json;charset=utf-8");
        resp.setContentLength(bytes.length);
        resp.getOutputStream().write(bytes);
    }

    public void setCode(Response resp, int code, String message) throws IOException {
        resp.setCode(code);
        json(resp, new StatusMsg(code, message == null ? toMessage(code) : message), false);
    }

    public void setCode(Response resp, int code, Object data, boolean borg) throws IOException {
        resp.setCode(code);
        json(resp, data, borg);
    }

    public void setText(Response resp, int code, String message) throws IOException {
        resp.setCode(code);
        byte[] bytes = message.getBytes("UTF8");
        resp.set("Content-Type", "text/plain;charset=utf-8");
        resp.setContentLength(bytes.length);
        resp.getOutputStream().write(bytes);
    }

    private String toMessage(int code) {
        switch (code) {
            case 200:   return "OK";
            case 400:   return "Bad request";
            case 403:   return "Forbidden";
            case 404:   return "Not found";
            case 409:   return "Version mismatch";
            case 500:   return "Internal server error";
        }
        return null;
    }

    public static class StatusMsg {

        public int responseCode;
        public String message;

        public StatusMsg(int responseCode, String message) {
            this.responseCode = responseCode;
            this.message = message;
        }
    }
}