com.sketchy.server.ImageUploadServlet.java Source code

Introduction

Here is the source code for com.sketchy.server.ImageUploadServlet.java

Source

/*
Sketchy
Copyright (C) 2015 Matthew Havlovick
http://www.quickdrawbot.com
    
This program is free software; you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation; either version 2 of the License, or
(at your option) any later version.
    
This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
GNU General Public License for more details.
    
You should have received a copy of the GNU General Public License along
with this program; if not, write to the Free Software Foundation, Inc.,
51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
    
Contact Info:
    
Matt Havlovick
QuickDraw
470 I St
Washougal, WA 98671
    
matt@quickdrawbot.com
http://www.quickdrawbot.com
    
This General Public License does not permit incorporating your program into
proprietary programs.  If your program is a subroutine library, you may
consider it more useful to permit linking proprietary applications with the
library.  If this is what you want to do, use the GNU Lesser General
Public License instead of this License.
*/

package com.sketchy.server;

import java.awt.image.BufferedImage;
import java.io.File;
import java.io.IOException;
import java.util.List;

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.disk.DiskFileItemFactory;
import org.apache.commons.fileupload.servlet.ServletFileUpload;
import org.apache.commons.io.FileUtils;
import org.apache.commons.lang3.StringUtils;

import com.sketchy.image.SourceImageAttributes;
import com.sketchy.server.JSONServletResult.Status;
import com.sketchy.utils.ImageUtils;

public final class ImageUploadServlet extends HttpServlet {

    private static final int MAX_SIZE = 10000000; // 10Mbytes
    private static final long serialVersionUID = -6172468221875699668L;

    protected void doPost(HttpServletRequest request, HttpServletResponse response)
            throws ServletException, IOException {
        JSONServletResult jsonServletResult = new JSONServletResult(Status.SUCCESS);
        try {

            boolean isMultipart = ServletFileUpload.isMultipartContent(request);
            if (isMultipart) {

                DiskFileItemFactory factory = new DiskFileItemFactory();
                factory.setRepository(FileUtils.getTempDirectory());
                factory.setSizeThreshold(MAX_SIZE);

                ServletFileUpload servletFileUpload = new ServletFileUpload(factory);
                List<FileItem> files = servletFileUpload.parseRequest(request);
                for (FileItem fileItem : files) {
                    String uploadFileName = fileItem.getName();
                    if (StringUtils.isNotBlank(uploadFileName)) {
                        // Don't allow \\ in the filename, assume it's a directory separator and convert to "/"
                        // and take the filename after the last "/"
                        // This will fix the issue of Jetty not reading and serving files
                        // with "\" (%5C) characters 
                        // This also fixes the issue of IE sometimes sending the whole path 
                        // (depending on the security settings)
                        uploadFileName = StringUtils.replaceChars(uploadFileName, "\\", "/");
                        if (StringUtils.contains(uploadFileName, "/")) {
                            uploadFileName = StringUtils.substringAfterLast(uploadFileName, "/");
                        }

                        File uploadFile = HttpServer.getUploadFile(uploadFileName);
                        // make sure filename is actually in the upload directory
                        // we don't want any funny games
                        if (!uploadFile.getParentFile().equals(HttpServer.IMAGE_UPLOAD_DIRECTORY)) {
                            throw new RuntimeException("Can not upload File. Invalid directory!");
                        }

                        // if saved ok, then need to add the data file
                        SourceImageAttributes sourceImageAttributes = new SourceImageAttributes();
                        sourceImageAttributes.setImageName(uploadFileName);

                        File pngFile = HttpServer.getUploadFile(sourceImageAttributes.getImageFilename());
                        if (pngFile.exists()) {
                            throw new Exception(
                                    "Can not Upload file. File '" + uploadFileName + "' already exists!");
                        }
                        File dataFile = HttpServer.getUploadFile(sourceImageAttributes.getDataFilename());

                        // Convert the image to a .PNG file
                        BufferedImage image = ImageUtils.loadImage(fileItem.getInputStream());
                        ImageUtils.saveImage(pngFile, image);

                        sourceImageAttributes.setWidth(image.getWidth());
                        sourceImageAttributes.setHeight(image.getHeight());

                        FileUtils.writeStringToFile(dataFile, sourceImageAttributes.toJson());
                        jsonServletResult.put("imageName", uploadFileName);
                    }
                }
            }
        } catch (Exception e) {
            jsonServletResult = new JSONServletResult(Status.ERROR, e.getMessage());
        }
        response.setContentType("text/html");
        response.setStatus(HttpServletResponse.SC_OK);
        response.getWriter().print(jsonServletResult.toJSONString());
    }

}