ElephantResponse.java

Go to the documentation of this file.

/*
 * Turró i Cutiller Foundation. License notice.
 * Copyright (C) 2015 Lluis Turró Cutiller <http://www.turro.org/>
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Affero General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU Affero General Public License for more details.
 *
 * You should have received a copy of the GNU Affero General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */
 
package org.turro.http;
 
import com.google.gson.Gson;
import com.google.gson.GsonBuilder;
import com.google.gson.JsonSyntaxException;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.Reader;
import java.nio.charset.Charset;
import java.util.logging.Level;
import java.util.logging.Logger;
import javax.servlet.http.HttpServletResponse;
import org.apache.commons.io.Charsets;
import org.apache.hc.core5.http.ContentType;
import org.apache.hc.core5.http.HttpEntity;
import org.turro.elephant.context.ElephantContext;
 
public class ElephantResponse {
 
  public String type;
  public String message;
  public String code;
  
  public static boolean isCorrect(ElephantResponse er) {
    return ElephantResponseType.RESPONSE_CORRECT.equals(ElephantResponse.getType(er));
  }
  
  public static ElephantResponseType getType(ElephantResponse er) {
    try {
      return er != null ? ElephantResponseType.valueOf(er.type) : null;
    } catch(IllegalArgumentException |NullPointerException ex) {
      return null;
    }
  }
  
  public static ElephantResponse getResponse(HttpEntity entity) throws IOException {
    Gson gson = new GsonBuilder().create();
    ContentType contentType = ContentType.create(entity.getContentType(), Charsets.toCharset(entity.getContentEncoding()));
    Charset charset = contentType.getCharset();
    Reader reader = new InputStreamReader(entity.getContent(), charset);
    try {
      return gson.fromJson(reader, ElephantResponse.class);
    } catch(JsonSyntaxException ex) {
      Logger.getLogger(ElephantResponse.class.getName()).log(Level.SEVERE, ElephantContext.logMsg(null), ex);
    } 
    return null;
  }
  
  public static String getResponse(ElephantResponseType type, String message, String code) throws IOException {
    Gson gson = new GsonBuilder().create();
    ElephantResponse er = new ElephantResponse();
    er.type = type.toString();
    er.message = message;
    er.code = code;
    return gson.toJson(er);  
  }
  
  public static void writeToResponse(HttpServletResponse response, ElephantResponseType type, String message, String code) throws IOException {
    response.setContentType("application/json");
    response.setCharacterEncoding(ElephantContext.getEncoding());
    response.getWriter().write(getResponse(type, message, code));
  }
}