UploadServlet.java 3.9 KB
package fi.insomnia.bortal.servlet;

import java.io.IOException;
import java.io.PrintWriter;

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

import org.apache.commons.fileupload.FileItem;
import org.apache.commons.fileupload.FileItemFactory;
import org.apache.commons.fileupload.FileUploadException;
import org.apache.commons.fileupload.disk.DiskFileItemFactory;
import org.apache.commons.fileupload.servlet.ServletFileUpload;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import fi.insomnia.bortal.beans.EventMapBeanLocal;
import fi.insomnia.bortal.beans.PermissionDeniedException;
import fi.insomnia.bortal.beans.UserBeanLocal;
import fi.insomnia.bortal.beans.UtilBeanLocal;
import fi.insomnia.bortal.model.UserImage;

/**
 * Servlet implementation class UploadServlet
 */
public class UploadServlet extends HttpServlet {
	private static final long serialVersionUID = 1L;

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

	private static final Logger logger = LoggerFactory.getLogger(UploadServlet.class);

	/**
	 * @see HttpServlet#doGet(HttpServletRequest request, HttpServletResponse
	 *      response)
	 */
	@Override
	protected void doGet(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException {
		doPost(request, response);
	}

	@EJB
	private transient UserBeanLocal userbean;
	@EJB
	private transient EventMapBeanLocal eventmapbean;

	@EJB
	private transient UtilBeanLocal utilbean;

	/**
	 * @see HttpServlet#doPost(HttpServletRequest request, HttpServletResponse
	 *      response)
	 */
	@Override
	protected void doPost(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException {
		// boolean isMultipart = ServletFileUpload.isMultipartContent(request);
		FileItemFactory factory = new DiskFileItemFactory();
		ServletFileUpload upload = new ServletFileUpload(factory);

		// Parse the request
		CharSequence retmsg = "Tiedosto tallennettu onnistuneesti.";
		try {

			String contenttype = "";
			byte[] imagedata = null;

			Integer destId = 0;

			String description = "";
			String filename = "";
			String idstr = request.getParameter("id");
			if (idstr != null) {
				try {
					destId = Integer.parseInt(idstr);
				} catch (NumberFormatException e)
				{
					logger.warn("Could not parse ID from string: {}", idstr);
					destId = null;
				}
			}
			for (Object ti : upload.parseRequest(request)) {
				if (ti instanceof FileItem) {

					FileItem fi = (FileItem) ti;

					if (fi.getFieldName().equals("id") && !fi.getString().isEmpty()) {
						logger.debug("Parsing int {}", fi.getString());
						destId = Integer.parseInt(fi.getString());

					} else if (fi.getFieldName().equals("description")) {
						description = fi.getString();

					} else if (fi.getFieldName().equals("file") || fi.getFieldName().equals("Filedata")) {
						imagedata = fi.get();
						filename = fi.getName();
						contenttype = fi.getContentType();
					}

				}
			}
			String type = request.getParameter("type");
			if (type.equals("userimage")) {
				logger.debug("Uploading userimage for user {}");
				UserImage userimage = userbean.uploadImage(destId, contenttype, imagedata, filename, description);
				utilbean.convertImage(userimage.getUser());
			} else if (type.equals("mapimage")) {
				eventmapbean.sendImage(destId, imagedata);
			} else {
				retmsg = "FATAL: servlet type not defined!!";
			}

		} catch (FileUploadException e) {
			logger.warn("Error uploading image", e);
			retmsg = "Tiedoston tallennuksessa tapahtui virhe!";
		} catch (PermissionDeniedException e) {
			logger.warn("Error uploading image", e);
			retmsg = "Tiedoston tallennuksessa tapahtui virhe!";

		}

		PrintWriter writer = response.getWriter();
		writer.append(retmsg);

	}
}