FileDownloadServlet.java

package fi.codecrew.moya.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.codecrew.moya.beans.CardPrintBeanLocal;
import fi.codecrew.moya.beans.CardTemplateBeanLocal;
import fi.codecrew.moya.beans.EventBeanLocal;
import fi.codecrew.moya.beans.PermissionBeanLocal;
import fi.codecrew.moya.beans.UserBeanLocal;
import fi.codecrew.moya.enums.apps.UserPermission;
import fi.codecrew.moya.model.CardTemplate;
import fi.codecrew.moya.model.EventUser;
import fi.codecrew.moya.model.LanEventProperty;
import fi.codecrew.moya.model.LanEventPropertyKey;
import fi.codecrew.moya.model.PrintedCard;
import fi.codecrew.moya.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 CardTemplateBeanLocal cardbean;

	@EJB
	private transient CardPrintBeanLocal cardprint;

	@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("([^./]+)");

	/**
	 * Usage MoyaWeb/dydata/format/ possible formats logo
	 * userimage/<eventuserid>.jpg cardtemplate/<cardtemplateid>.png usercard/
	 */
	@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)
			{

				if (!permbean.isCurrentUser(image.getUser())
						&& !permbean.hasPermission(UserPermission.VIEW_ALL))
				{
					ret.setResponse(HttpServletResponse.SC_FORBIDDEN);
					return ret;
				}

				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 dydataRoot = this.getServletContext().getRealPath("/dydata/");

				File imagefile = null;
				if (dydataRoot != null) {
					imagefile = new File(dydataRoot + super.request.getPathInfo());
				}
				if (imagefile != null && !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);
					}

				}

			}

		} else if (urlparts.get(0).equals("usercard") && urlparts.size() > 2) {
			int userid = Integer.parseInt(urlparts.get(1));

			EventUser usr = userbean.findByUserId(userid, false);
			logger.info("Trying to print usercard for user {}", usr);
			if (usr != null) {
				if (!permbean.isCurrentUser(usr.getUser())
						&& !permbean.hasPermission(UserPermission.VIEW_ALL)) {
					ret.setResponse(HttpServletResponse.SC_FORBIDDEN);
					return ret;
				}
				PrintedCard card = cardbean.checkPrintedCard(usr);
				try {
					byte[] img = cardprint.constructPNG(card);
					if (img != null && img.length > 0) {
						ret.setData(img);
						ret.setImagetype("image/png");
					}
				} catch (Exception e) {
					logger.warn("Error generating image", e);
				}

			}
		}

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

	}

}