FileDownloadServlet.java 3.8 KB
package fi.insomnia.bortal.servlet;

import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.util.ArrayList;
import java.util.List;
import java.util.regex.Matcher;
import java.util.regex.Pattern;

import javax.ejb.EJB;
import javax.servlet.annotation.WebServlet;
import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

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

import fi.insomnia.bortal.beans.CardTemplateBeanLocal;
import fi.insomnia.bortal.beans.EventBeanLocal;
import fi.insomnia.bortal.beans.PermissionBeanLocal;
import fi.insomnia.bortal.beans.UserBeanLocal;
import fi.insomnia.bortal.model.CardTemplate;
import fi.insomnia.bortal.model.LanEventProperty;
import fi.insomnia.bortal.model.LanEventPropertyKey;
import fi.insomnia.bortal.model.UserImage;

/**
 * Servlet implementation class UploadServlet
 */
@WebServlet(urlPatterns = { "/dydata/*" })
public class FileDownloadServlet extends GenericImageServlet {

	/**
     * 
     */
	private static final long serialVersionUID = -3359999630873773508L;
	@EJB
	private transient CardTemplateBeanLocal ctbean;

	@EJB
	private transient UserBeanLocal userbean;

	@EJB
	private transient PermissionBeanLocal permbean;

	@EJB
	private transient EventBeanLocal orgbean;

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

	/**
	 * @see HttpServlet#doPost(HttpServletRequest request, HttpServletResponse
	 *      response)
	 */

	private static final Pattern URLPATTERN = Pattern.compile("([^./]+)");

	@Override
	protected ImageMover getImagedata() {
		ImageMover ret = new ImageMover();
		Matcher matcher = URLPATTERN.matcher(super.request.getPathInfo());
		List<String> urlparts = new ArrayList<String>();

		while (matcher.find()) {
			urlparts.add(matcher.group());
		}
		logger.debug("filedownload urlparts {}", urlparts);

		if (urlparts.isEmpty()) {
			ret.setResponse(HttpServletResponse.SC_NOT_FOUND);
		} else if (urlparts.get(0).equals("logo")) {
			LanEventProperty logo = orgbean.getProperty(LanEventPropertyKey.EVENT_LOGO);
			if (logo != null) {
				ret.setData(logo.getByteValue());
				ret.setImagetype(logo.getByteMime());
			}
		} else if (urlparts.get(0).equals("cardtemplate") && urlparts.size() > 2)
		{
			int imageid = Integer.parseInt(urlparts.get(1));
			CardTemplate templ = ctbean.find(imageid);
			logger.info("Cardtemplate {}, {}", imageid, templ);
			if (templ != null && templ.getImage().length > 0) {
				ret.setData(templ.getImage());
				ret.setImagetype("image/jpeg");
			}
		}
		else if (urlparts.get(0).equals("userimage") && urlparts.size() > 2) {
			int imageid = Integer.parseInt(urlparts.get(1));
			UserImage image = userbean.findUserimageFORCE(imageid);
			if (image != null)
			{
				ret.setData(image.getImageData());
				ret.setImagetype(image.getMimeType());

				// Create image on file for cropper to work...
				// Check
				// http://code.google.com/p/primefaces/issues/detail?id=3751
				String imagefilePath = this.getServletContext().getRealPath("/dydata/" + super.request.getPathInfo());
				File imagefile = new File(imagefilePath);
				if (!imagefile.exists())
				{
					File parentPath = new File(imagefile.getParent());
					if (!parentPath.isDirectory() && !parentPath.mkdirs()) {
						logger.warn("Error creating directory {}", parentPath.getAbsolutePath());
					}
					try {
						FileOutputStream writer = new FileOutputStream(imagefile);
						writer.write(image.getImageData());
						writer.close();
						logger.info("Created image on file: {}", imagefile.getAbsolutePath());
					} catch (IOException e) {
						logger.warn("error creating image on file {}", e);
					}
				}
			}

		}

		if (ret.getImagetype() == null) {
			ret.setResponse(HttpServletResponse.SC_NOT_FOUND);
		}
		return ret;

	}
}