package com.hoge;

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

import javax.imageio.ImageIO;
import javax.servlet.ServletException;
import javax.servlet.ServletOutputStream;
import javax.servlet.annotation.WebServlet;
import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

import org.apache.tomcat.util.http.fileupload.FileItem;
import org.apache.tomcat.util.http.fileupload.FileUploadException;
import org.apache.tomcat.util.http.fileupload.disk.DiskFileItemFactory;
import org.apache.tomcat.util.http.fileupload.servlet.ServletFileUpload;

/**
 * Servlet implementation class FileUploadServlet
 */
@WebServlet("/FileUploadServlet")
public class FileUploadServlet extends HttpServlet {
    private static final long serialVersionUID = 1L;
	private String uploadDirectoryString = "upload";
	private File storeFile;
	/**
	 * @see HttpServlet#HttpServlet()
	 */
	public FileUploadServlet() {
		super();
		// TODO Auto-generated constructor stub
	}

	/**
	 * @see HttpServlet#doGet(HttpServletRequest request, HttpServletResponse response)
	 */
	protected void doGet(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException {
		// TODO Auto-generated method stub
	}

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

		if(ServletFileUpload.isMultipartContent(request)) {

			DiskFileItemFactory factory = new DiskFileItemFactory();
			factory.setSizeThreshold(1024);
			ServletFileUpload upload = new ServletFileUpload(factory);
			upload.setSizeMax(-1);
			upload.setFileSizeMax(-1);

			List<FileItem> items = null;
			try {
				items = upload.parseRequest(request);
			} catch (FileUploadException e) {
				e.printStackTrace();
			}

			for(FileItem item : items) {
				if(item.isFormField()) {
				} else {
					writeFile(item);
					showThumbnail(request, response);
				}
			}
		}
	}

	private void showThumbnail(HttpServletRequest request, HttpServletResponse response) throws IOException {
		BufferedImage image = ImageIO.read(storeFile);
		response.setContentType("image/jpg");
		ServletOutputStream out = response.getOutputStream();
		ImageIO.write(image, "jpg", out);
		out.close();
	}

	private void writeFile(FileItem item) {
		try {
			String uploadPath = getServletContext().getRealPath("") + File.separator + uploadDirectoryString;
			File uploadDirectory = new File(uploadPath);
			if(!uploadDirectory.exists()) uploadDirectory.mkdir();
			String fileName = new File(item.getName()).getName();
			String filePath = uploadPath + File.separator + fileName;
			storeFile = new File(filePath);
			item.write(storeFile);
		} catch (Exception e) {
			e.printStackTrace();
		}
	}

}