GenericImageServlet.java

/*
 * Copyright Codecrew Ry
 * 
 * All rights reserved.
 * 
 * This license applies to any software containing a notice placed by the 
 * copyright holder. Such software is herein referred to as the Software. 
 * This license covers modification, distribution and use of the Software. 
 * 
 * Any distribution and use in source and binary forms, with or without 
 * modification is not permitted without explicit written permission from the 
 * copyright owner. 
 * 
 * A non-exclusive royalty-free right is granted to the copyright owner of the 
 * Software to use, modify and distribute all modifications to the Software in 
 * future versions of the Software. 
 * 
 */
package fi.codecrew.moya.servlet;

import java.io.IOException;

import javax.servlet.ServletException;
import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

public abstract class GenericImageServlet extends HttpServlet {
	/**
	 * 
	 */

	protected enum ResponseType {
		OK, FORBIDDEN, NOT_FOUND,

	}

	private static final long serialVersionUID = -2900241580433674121L;

	protected class ImageMover
	{
		private Integer response;
		private byte[] data;
		private String imagetype;

		public byte[] getData() {
			return data;
		}

		public void setData(byte[] data) {
			this.data = data;
		}

		public String getImagetype() {
			return imagetype;
		}

		public void setImagetype(String imagetype) {
			this.imagetype = imagetype;
		}

		public Integer getResponse() {
			return response;
		}

		public void setResponse(Integer response) {
			this.response = response;
		}

	}

	/**
	 * @see HttpServlet#HttpServlet()
	 */
	public GenericImageServlet() {
		super();
		// TODO Auto-generated constructor stub
	}

	@SuppressWarnings("unused")
	private static final Logger logger = LoggerFactory.getLogger(GenericImageServlet.class);

	protected abstract ImageMover getImagedata(HttpServletRequest request);

	protected Integer getIdParam(String name, HttpServletRequest request)
	{
		Integer ret = null;
		String idStr = request.getParameter(name);
		if (idStr != null)
		{
			ret = Integer.parseInt(idStr);
		}
		return ret;

	}

	@Override
	protected void doGet(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException {
		doPost(request, response);
	}

	@Override
	protected void doPost(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException {

		ImageMover data = getImagedata(request);
		if (data == null || data.getData() == null || data.getResponse() != null) {
			if (data != null && data.getData() != null) {
				response.setStatus(data.getResponse());
			} else {
				response.setStatus(HttpServletResponse.SC_NOT_FOUND);
			}
			response.getWriter().append("Error " + (data == null ? "null-data" : data.getResponse()) + " while fetching data");
		} else {
			response.setContentLength(data.getData().length);
			if (request.getParameter("download") != null)
			{
				response.setContentType("application/x-unknown");
			}
			else {
				response.setContentType(data.getImagetype());
			}
			response.getOutputStream().write(data.getData());
		}

		request = null;
	}
}