FileSaveServlet.java

/*******************************************************************************
 * Copyright (c) 2014, 2015 Scott Clarke (scott@dawg6.com).
 *
 * This file is part of Dawg6's Demon Hunter DPS Calculator.
 *
 * Dawg6's Demon Hunter DPS Calculator 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 3 of the License, or
 * (at your option) any later version.
 *
 * Dawg6's Demon Hunter DPS Calculator 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, see <http://www.gnu.org/licenses/>.
 *******************************************************************************/
package com.dawg6.web.dhcalc.server;

import java.io.IOException;
import java.util.logging.Level;
import java.util.logging.Logger;

import javax.servlet.ServletException;
import javax.servlet.ServletOutputStream;
import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

import com.oreilly.servlet.multipart.MultipartParser;
import com.oreilly.servlet.multipart.ParamPart;
import com.oreilly.servlet.multipart.Part;

public class FileSaveServlet extends HttpServlet {

	private static final Logger log = Logger.getLogger(FileSaveServlet.class.getName());
	
	private static final long serialVersionUID = 1L;

	@Override
	protected void doGet(HttpServletRequest req, HttpServletResponse resp)
			throws ServletException, IOException {
		doPost(req, resp);
	}

	@Override
	protected void doPost(HttpServletRequest req, HttpServletResponse resp)
			throws ServletException, IOException {
		try {
			
			MultipartParser parser = new MultipartParser(req, Integer.MAX_VALUE);
			boolean isFile = false;
			
			Part part = parser.readNextPart();

			if (part == null) {
				log.severe("No First Part");
			} else {

				String key = null;
				String filename = null;
				
				while (part != null) {
//					log.info("Found part " + part.getName());
					
					if (part instanceof ParamPart) {
						
						if (part.getName().equals("key"))	 
							key = ((ParamPart)part).getStringValue();
						else if (part.getName().equals("filename"))
							filename = ((ParamPart)part).getStringValue();
						else if (part.getName().equals("isFile"))
							isFile = Boolean.valueOf(((ParamPart)part).getStringValue());
					} else {
						log.info("Skipping part of shooter "
								+ part.getClass().getName());
					}

					part = parser.readNextPart();
				}

				if (key != null) {
					
					if (filename == null)
						filename = "filename";

					Object data = null;
					
					if (isFile) 
						data = ClientBuffer.getInstance().get(key);

					if (!isFile || (data != null)) {
						
						resp.setContentType("application/octet-stream");
						resp.setHeader("Content-Disposition",
								"attachment;filename=" + filename);
						
						ServletOutputStream out = resp.getOutputStream();

						if (isFile)
							out.write((byte[])data);
						else
							out.write(key.getBytes());
						
						out.flush();
						out.close();
					} else if (isFile) {
						log.warning("Unable to find client data: " + key);
					}
					
				} else {
					log.warning("No Key");
				}
			}

		} catch (Exception e) {
			log.log(Level.SEVERE, e.getMessage(), e);
		}
	}

}