com.runwaysdk.web.WebFileUploadServlet.java Source code

Introduction

Here is the source code for com.runwaysdk.web.WebFileUploadServlet.java

Source

/**
 * Copyright (c) 2015 TerraFrame, Inc. All rights reserved.
 *
 * This file is part of Runway SDK(tm).
 *
 * Runway SDK(tm) is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Lesser General Public License as
 * published by the Free Software Foundation, either version 3 of the
 * License, or (at your option) any later version.
 *
 * Runway SDK(tm) 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 Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with Runway SDK(tm).  If not, see <http://www.gnu.org/licenses/>.
 */
package com.runwaysdk.web;

import java.io.IOException;
import java.io.InputStream;

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

import org.apache.commons.fileupload.FileItemFactory;
import org.apache.commons.fileupload.FileItemIterator;
import org.apache.commons.fileupload.FileItemStream;
import org.apache.commons.fileupload.FileUploadException;
import org.apache.commons.fileupload.disk.DiskFileItemFactory;
import org.apache.commons.fileupload.servlet.ServletFileUpload;
import org.apache.commons.fileupload.util.Streams;

import com.runwaysdk.constants.ClientConstants;
import com.runwaysdk.constants.ClientRequestIF;
import com.runwaysdk.dataaccess.io.FileWriteExceptionDTO;

public class WebFileUploadServlet extends FileUploadServlet {

    /**
     * 
     */
    private static final long serialVersionUID = -2687759556467616281L;

    /**
     * The name of the input that specifies the upload path of the web file.
     */
    public static final String WEB_FILE_UPLOAD_PATH_FIELD_NAME = "webFileUploadPath";

    protected void doPost(HttpServletRequest req, HttpServletResponse res) throws ServletException, IOException {
        ClientRequestIF clientRequest = (ClientRequestIF) req.getAttribute(ClientConstants.CLIENTREQUEST);

        // capture the session id
        boolean isMultipart = ServletFileUpload.isMultipartContent(req);

        if (!isMultipart) {
            // TODO Change exception type
            String msg = "The HTTP Request must contain multipart content.";
            throw new RuntimeException(msg);
        }

        FileItemFactory factory = new DiskFileItemFactory();
        ServletFileUpload upload = new ServletFileUpload();

        upload.setFileItemFactory(factory);

        try {
            // Parse the request
            FileItemIterator iter = upload.getItemIterator(req);

            String fileName = null;
            String extension = null;
            InputStream stream = null;
            String uploadPath = null;
            while (iter.hasNext()) {
                FileItemStream item = iter.next();
                InputStream input = item.openStream();
                if (item.isFormField() && item.getFieldName().equals(WEB_FILE_UPLOAD_PATH_FIELD_NAME)) {
                    uploadPath = Streams.asString(input);
                } else if (!item.isFormField()) {
                    String fullName = item.getName();
                    int extensionInd = fullName.lastIndexOf(".");
                    fileName = fullName.substring(0, extensionInd);
                    extension = fullName.substring(extensionInd + 1);
                    stream = input;
                }
            }

            if (stream != null) {
                clientRequest.newFile(uploadPath, fileName, extension, stream);
            }
        } catch (FileUploadException e) {
            throw new FileWriteExceptionDTO(e.getLocalizedMessage());
        }
    }

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