org.megatome.frame2.front.FileUploadSupport.java Source code

Introduction

Here is the source code for org.megatome.frame2.front.FileUploadSupport.java

Source

/*
 * ====================================================================
 *
 * Frame2 Open Source License
 *
 * Copyright (c) 2004-2007 Megatome Technologies.  All rights
 * reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 *
 * 1. Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 *
 * 2. Redistributions in binary form must reproduce the above copyright
 *    notice, this list of conditions and the following disclaimer in
 *    the documentation and/or other materials provided with the
 *    distribution.
 *
 * 3. The end-user documentation included with the redistribution, if
 *    any, must include the following acknowlegement:
 *       "This product includes software developed by
 *        Megatome Technologies."
 *    Alternately, this acknowlegement may appear in the software itself,
 *    if and wherever such third-party acknowlegements normally appear.
 *
 * 4. The names "The Frame2 Project", and "Frame2", 
 *    must not be used to endorse or promote products derived
 *    from this software without prior written permission. For written
 *    permission, please contact iamthechad@sourceforge.net.
 *
 * 5. Products derived from this software may not be called "Frame2"
 *    nor may "Frame2" appear in their names without prior written
 *    permission of Megatome Technologies.
 *
 * THIS SOFTWARE IS PROVIDED ``AS IS'' AND ANY EXPRESSED OR IMPLIED
 * WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
 * DISCLAIMED.  IN NO EVENT SHALL MEGATOME TECHNOLOGIES OR
 * ITS CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
 * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF
 * USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
 * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
 * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT
 * OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
 * SUCH DAMAGE.
 * ====================================================================
 */
package org.megatome.frame2.front;

import java.io.File;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

import javax.servlet.http.HttpServletRequest;

import org.apache.commons.fileupload.FileItem;
import org.apache.commons.fileupload.FileUploadException;
import org.apache.commons.fileupload.disk.DiskFileItemFactory;
import org.apache.commons.fileupload.servlet.ServletFileUpload;
import org.megatome.frame2.Frame2Exception;
import org.megatome.frame2.log.Logger;
import org.megatome.frame2.log.LoggerFactory;

public final class FileUploadSupport {
    private static final Logger LOGGER = LoggerFactory.instance(FileUploadSupport.class.getName());

    private FileUploadSupport() { // Non-public ctor
    }

    @SuppressWarnings("unchecked")
    public static Map<String, Object> processMultipartRequest(final HttpServletRequest request)
            throws Frame2Exception {
        Map<String, Object> parameters = new HashMap<String, Object>();

        // Create a factory for disk-based file items
        DiskFileItemFactory factory = new DiskFileItemFactory();

        // Set factory constraints
        factory.setSizeThreshold(FileUploadConfig.getBufferSize());
        factory.setRepository(new File(FileUploadConfig.getFileTempDir()));

        // Create a new file upload handler
        ServletFileUpload upload = new ServletFileUpload(factory);

        // Set overall request size constraint
        upload.setSizeMax(FileUploadConfig.getMaxFileSize());

        List<FileItem> fileItems = null;
        try {
            fileItems = upload.parseRequest(request);
        } catch (FileUploadException fue) {
            LOGGER.severe("File Upload Error", fue); //$NON-NLS-1$
            throw new Frame2Exception("File Upload Exception", fue); //$NON-NLS-1$
        }

        for (FileItem fi : fileItems) {
            String fieldName = fi.getFieldName();

            if (fi.isFormField()) {
                if (parameters.containsKey(fieldName)) {
                    List<Object> tmpArray = new ArrayList<Object>();
                    if (parameters.get(fieldName) instanceof String[]) {
                        String[] origValues = (String[]) parameters.get(fieldName);
                        for (int idx = 0; idx < origValues.length; idx++) {
                            tmpArray.add(origValues[idx]);
                        }
                        tmpArray.add(fi.getString());
                    } else {
                        tmpArray.add(parameters.get(fieldName));
                        tmpArray.add(fi.getString());
                    }
                    String[] newValues = new String[tmpArray.size()];
                    newValues = tmpArray.toArray(newValues);
                    parameters.put(fieldName, newValues);
                } else {
                    parameters.put(fieldName, fi.getString());
                }
            } else {
                if (parameters.containsKey(fieldName)) {
                    List<Object> tmpArray = new ArrayList<Object>();
                    if (parameters.get(fieldName) instanceof FileItem[]) {
                        FileItem[] origValues = (FileItem[]) parameters.get(fieldName);
                        for (int idx = 0; idx < origValues.length; idx++) {
                            tmpArray.add(origValues[idx]);
                        }
                        tmpArray.add(fi);
                    } else {
                        tmpArray.add(parameters.get(fieldName));
                        tmpArray.add(fi);
                    }
                    FileItem[] newValues = new FileItem[tmpArray.size()];
                    newValues = tmpArray.toArray(newValues);
                    parameters.put(fieldName, newValues);
                } else {
                    parameters.put(fieldName, fi);
                }
            }
        }

        return parameters;
    }

}