gr.forth.ics.isl.x3mlEditor.upload.MultipartUploadParser.java Source code

Introduction

Here is the source code for gr.forth.ics.isl.x3mlEditor.upload.MultipartUploadParser.java

Source

/*
 * Copyright 2014-2017 Institute of Computer Science,
 * Foundation for Research and Technology - Hellas
 *
 * Licensed under the EUPL, Version 1.1 or - as soon they will be approved
 * by the European Commission - subsequent versions of the EUPL (the "Licence");
 * You may not use this work except in compliance with the Licence.
 * You may obtain a copy of the Licence at:
 *
 * http://ec.europa.eu/idabc/eupl
 *
 * Unless required by applicable law or agreed to in writing, software distributed
 * under the Licence is distributed on an "AS IS" basis,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the Licence for the specific language governing permissions and limitations
 * under the Licence.
 *
 * Contact:  POBox 1385, Heraklio Crete, GR-700 13 GREECE
 * Tel:+30-2810-391632
 * Fax: +30-2810-391638
 * E-mail: isl@ics.forth.gr
 * http://www.ics.forth.gr/isl
 *
 * Authors : Georgios Samaritakis, Konstantina Konsolaki.
 *
 * This file is part of the 3MEditor webapp of Mapping Memory Manager project.
 */
package gr.forth.ics.isl.x3mlEditor.upload;

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

import javax.servlet.ServletContext;
import javax.servlet.http.HttpServletRequest;
import java.io.File;
import java.io.IOException;
import java.util.*;

/**
 *
 * @author samarita
 */
public class MultipartUploadParser {
    //final Logger log = LoggerFactory.getLogger(MultipartUploadParser.class);

    private Map<String, String> params = new HashMap<String, String>();

    private List<FileItem> files = new ArrayList<FileItem>();

    // fileItemsFactory is a field (even though it's scoped to the constructor) to prevent the
    // org.apache.commons.fileupload.servlet.FileCleanerCleanup thread from attempting to delete the
    // temp file before while it is still being used.
    //
    // FileCleanerCleanup uses a java.lang.ref.ReferenceQueue to delete the temp file when the FileItemsFactory marker object is GCed
    private DiskFileItemFactory fileItemsFactory;

    /**
     *
     * @param request
     * @param repository
     * @param context
     * @throws Exception
     */
    public MultipartUploadParser(HttpServletRequest request, File repository, ServletContext context)
            throws Exception {
        if (!repository.exists() && !repository.mkdirs()) {
            throw new IOException("Unable to mkdirs to " + repository.getAbsolutePath());
        }

        fileItemsFactory = setupFileItemFactory(repository, context);

        ServletFileUpload upload = new ServletFileUpload(fileItemsFactory);
        List<FileItem> formFileItems = upload.parseRequest(request);

        parseFormFields(formFileItems);

        if (files.isEmpty()) {
            writeDebugInfo(request);
            throw new FileUploadException("No files were found when processing the request.");
        } else {

        }
    }

    private DiskFileItemFactory setupFileItemFactory(File repository, ServletContext context) {
        DiskFileItemFactory factory = new DiskFileItemFactory();
        factory.setSizeThreshold(DiskFileItemFactory.DEFAULT_SIZE_THRESHOLD);
        factory.setRepository(repository);

        FileCleaningTracker pTracker = FileCleanerCleanup.getFileCleaningTracker(context);
        factory.setFileCleaningTracker(pTracker);

        return factory;
    }

    private void writeDebugInfo(HttpServletRequest request) {

    }

    private void parseFormFields(List<FileItem> items) {
        for (FileItem item : items) {
            if (item.isFormField()) {
                String key = item.getFieldName();
                String value = item.getString();
                if (StringUtils.isNotBlank(key)) {
                    params.put(key.toLowerCase(), StringUtils.defaultString(value));
                }
            } else {
                files.add(item);
            }
        }
    }

    /**
     *
     * @return
     */
    public Map<String, String> getParams() {
        return params;
    }

    /**
     *
     * @return
     */
    public List<FileItem> getFiles() {
        if (files.isEmpty()) {
            throw new RuntimeException("No FileItems exist.");
        }

        return files;
    }

    /**
     *
     * @return
     */
    public FileItem getFirstFile() {
        if (files.isEmpty()) {
            throw new RuntimeException("No FileItems exist.");
        }

        return files.iterator().next();
    }
}