de.micromata.genome.gwiki.page.impl.actionbean.CommonMultipartRequest.java Source code

Introduction

Here is the source code for de.micromata.genome.gwiki.page.impl.actionbean.CommonMultipartRequest.java

Source

//
// Copyright (C) 2010-2016 Roger Rene Kommer & Micromata GmbH
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//  http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
//

package de.micromata.genome.gwiki.page.impl.actionbean;

import java.io.UnsupportedEncodingException;
import java.util.Collections;
import java.util.Enumeration;
import java.util.HashMap;
import java.util.HashSet;
import java.util.Map;
import java.util.Set;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletRequestWrapper;

import org.apache.commons.fileupload.FileItem;
import org.apache.commons.lang3.ArrayUtils;

public class CommonMultipartRequest extends HttpServletRequestWrapper {
    private Map<String, String[]> multipartParams;

    private Map<String, FileItem> fileItems;

    public CommonMultipartRequest(HttpServletRequest request) {
        super(request);
        multipartParams = new HashMap<String, String[]>();
        fileItems = new HashMap<String, FileItem>();

    }

    public CommonMultipartRequest(HttpServletRequest request, Map<String, String[]> multipartParams,
            Map<String, FileItem> fileItems) {
        super(request);
        this.multipartParams = multipartParams;
        this.fileItems = fileItems;
    }

    public void addFileItem(FileItem item) {
        fileItems.put(item.getFieldName(), item);
    }

    public void addFormField(FileItem item) {
        final String fn = item.getFieldName();
        final String fv;
        try {
            fv = item.getString("UTF-8");
        } catch (UnsupportedEncodingException ex) {
            throw new RuntimeException(ex);
        }
        if (multipartParams.containsKey(fn) == false) {
            multipartParams.put(fn, new String[] { fv });
        } else {
            String[] ci = multipartParams.get(fn);
            multipartParams.put(fn, (String[]) ArrayUtils.add(ci, fv));
        }
    }

    @SuppressWarnings({ "rawtypes", "unchecked" })
    public Enumeration getParameterNames() {
        Set paramNames = new HashSet();
        Enumeration paramEnum = getRequest().getParameterNames();
        while (paramEnum.hasMoreElements()) {
            paramNames.add(paramEnum.nextElement());
        }
        paramNames.addAll(this.multipartParams.keySet());
        return Collections.enumeration(paramNames);
    }

    public String getParameter(String name) {
        String[] values = (String[]) this.multipartParams.get(name);
        if (values != null) {
            return (values.length > 0 ? values[0] : null);
        }
        return getRequest().getParameter(name);
    }

    public String[] getParameterValues(String name) {
        String[] values = (String[]) this.multipartParams.get(name);
        if (values != null) {
            return values;
        }
        return getRequest().getParameterValues(name);
    }

    @SuppressWarnings({ "rawtypes", "unchecked" })
    public Map getParameterMap() {
        Map paramMap = new HashMap();
        paramMap.putAll(getRequest().getParameterMap());
        paramMap.putAll(this.multipartParams);
        return paramMap;
    }

    public Map<String, String[]> getMultipartParams() {
        return multipartParams;
    }

    public Map<String, FileItem> getFileItems() {
        return fileItems;
    }
}