Subversion Repositories SmartDukaan

package com.spice.profitmandi.common.web.util;

import javax.servlet.http.HttpServletRequest;

import javax.servlet.http.HttpServletResponse;

import org.springframework.http.HttpStatus;

import org.springframework.http.MediaType;

import org.springframework.http.ResponseEntity;

import org.springframework.stereotype.Component;

import com.fasterxml.jackson.databind.ObjectMapper;

import com.spice.profitmandi.common.exception.ProfitMandiBusinessException;

import com.spice.profitmandi.common.model.ProfitMandiResponse;

import com.spice.profitmandi.common.model.Response;

import com.spice.profitmandi.common.model.ResponseStatus;

@Component

public class ResponseSender<T> {

	private final ObjectMapper objectMapper = new ObjectMapper();

	public ResponseEntity<?> ok(Object response){

		final ProfitMandiResponse<?> profitMandiResponse=new ProfitMandiResponse<>(HttpStatus.OK.toString(), HttpStatus.OK, ResponseStatus.SUCCESS, response);

		return new ResponseEntity<>(profitMandiResponse, HttpStatus.OK);

	}

	public ResponseEntity<?> badRequest(ProfitMandiBusinessException profitMandiBusinessException){

		return this.genericError(profitMandiBusinessException, HttpStatus.BAD_REQUEST);

	}

	public ResponseEntity<?> unauthorized(ProfitMandiBusinessException profitMandiBusinessException){

		return this.genericError(profitMandiBusinessException, HttpStatus.UNAUTHORIZED);

	}

	public ResponseEntity<?> forbidden(ProfitMandiBusinessException profitMandiBusinessException){

		return this.genericError(profitMandiBusinessException, HttpStatus.FORBIDDEN);

	}

	public ResponseEntity<?> notFound(ProfitMandiBusinessException profitMandiBusinessException){

		return this.genericError(profitMandiBusinessException, HttpStatus.NOT_FOUND);

	}

	public ResponseEntity<?> internalServerError(Throwable exception){

		final Response response=new Response("", "", "errorCode", exception.getMessage());

		return this.generic(response, HttpStatus.INTERNAL_SERVER_ERROR);

	}

	private ResponseEntity<?> genericError(ProfitMandiBusinessException profitMandiBusinessException, HttpStatus status){

		final Response response=new Response(profitMandiBusinessException.getRejectedType(), profitMandiBusinessException.getRejectedValue(),profitMandiBusinessException.getCode(), profitMandiBusinessException.getMessage());

		return this.generic(response, status);

	}

	private ResponseEntity<?> generic(Response response, HttpStatus status){

		final ProfitMandiResponse<Response> profitMandiResponse=new ProfitMandiResponse<Response>(status.toString(), status, ResponseStatus.FAILURE, response);

		return new ResponseEntity<>(profitMandiResponse, status);

	}

	public void writeBadRequest(HttpServletRequest request, HttpServletResponse response, Response failedResponse) throws Exception{

		this.writeGenericRequest(request, response, failedResponse, HttpStatus.BAD_REQUEST);

	}

	public void writeOk(HttpServletRequest request, HttpServletResponse response, Object data) throws Exception{

		response.setStatus(HttpStatus.OK.value());

		response.setContentType(MediaType.APPLICATION_JSON_VALUE);

		response.setCharacterEncoding("UTF-8");

		response.getWriter().println(data);

		response.getWriter().flush();

	}

	public void writeOkFileContentType(HttpServletRequest request, HttpServletResponse response, File file, String contentType) throws Exception{

		response.setStatus(HttpStatus.OK.value());

		response.setContentType(contentType);

		response.getOutputStream().write(Files.readAllBytes(file.toPath()));

		response.getOutputStream().flush();

	}

	public void writeUnauthorized(HttpServletRequest request, HttpServletResponse response, Response failedResponse) throws Exception{

		this.writeGenericRequest(request, response, failedResponse, HttpStatus.UNAUTHORIZED);

	}

	public void writeForbidden(HttpServletRequest request, HttpServletResponse response, Response failedResponse) throws Exception{

		this.writeGenericRequest(request, response, failedResponse, HttpStatus.FORBIDDEN);

	}

	public void writeNotFound(HttpServletRequest request, HttpServletResponse response, Response failedResponse) throws Exception{

		this.writeGenericRequest(request, response, failedResponse, HttpStatus.NOT_FOUND);

	}

	private void writeGenericRequest(HttpServletRequest request, HttpServletResponse response, Response failedResponse, HttpStatus httpStatus) throws Exception{

		response.setStatus(httpStatus.value());

		response.setContentType(MediaType.APPLICATION_JSON_VALUE);

		response.setCharacterEncoding("UTF-8");

		final ProfitMandiResponse<?> profitMandiResponse=new ProfitMandiResponse<>(httpStatus.toString(), httpStatus, ResponseStatus.FAILURE, failedResponse);

		final Object obj = objectMapper.readValue(objectMapper.writeValueAsString(profitMandiResponse), Object.class);

		final String jsonString = objectMapper.writerWithDefaultPrettyPrinter().writeValueAsString(obj);

		response.getWriter().write(jsonString.toString());

		response.getWriter().flush();

	}

}