    001    // Copyright 2007, 2008 The Apache Software Foundation
    002    //
    003    // Licensed under the Apache License, Version 2.0 (the "License");
    004    // you may not use this file except in compliance with the License.
    005    // You may obtain a copy of the License at
    006    //
    007    //     http://www.apache.org/licenses/LICENSE-2.0
    008    //
    009    // Unless required by applicable law or agreed to in writing, software
    010    // distributed under the License is distributed on an "AS IS" BASIS,
    011    // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
    012    // See the License for the specific language governing permissions and
    013    // limitations under the License.
    014
    015    package org.apache.tapestry5.upload.internal.services;
    016
    017    import static org.apache.tapestry5.ioc.internal.util.CollectionFactory.newMap;
    018
    019    import javax.servlet.http.HttpServletRequest;
    020    import javax.servlet.http.HttpServletRequestWrapper;
    021    import java.io.UnsupportedEncodingException;
    022    import java.util.Collections;
    023    import java.util.Enumeration;
    024    import java.util.Map;
    025
    026    /**
    027     * Wrapper for HttpServletRequest that overrides the parameter methods of the wrapped request. i.e. parameters are
    028     * retrieved from the wrapper rather than the real request.
    029     */
    030    public class ParametersServletRequestWrapper extends HttpServletRequestWrapper
    031    {
    032        private final Map<String, ParameterValue> parameters = newMap();
    033
    034        public ParametersServletRequestWrapper(HttpServletRequest httpServletRequest)
    035        {
    036            super(httpServletRequest);
    037        }
    038
    039        @Override
    040        public String getParameter(String name)
    041        {
    042            return getValueFor(name).single();
    043        }
    044
    045        @Override
    046        public Map<String, Object> getParameterMap()
    047        {
    048            Map<String, Object> paramMap = newMap();
    049
    050            for (Map.Entry<String, ParameterValue> e : parameters.entrySet())
    051            {
    052                ParameterValue value = e.getValue();
    053
    054                paramMap.put(e.getKey(), value.isMulti() ? value.multi() : value.single());
    055            }
    056
    057            return paramMap;
    058        }
    059
    060        @Override
    061        public Enumeration getParameterNames()
    062        {
    063            return Collections.enumeration(parameters.keySet());
    064        }
    065
    066        @Override
    067        public String[] getParameterValues(String name)
    068        {
    069            return getValueFor(name).multi();
    070        }
    071
    072        public void addParameter(String name, String value)
    073        {
    074            ParameterValue pv = parameters.get(name);
    075            if (pv == null)
    076            {
    077                pv = new ParameterValue(value);
    078                parameters.put(name, pv);
    079            }
    080            else
    081            {
    082                pv.add(value);
    083            }
    084        }
    085
    086        ParameterValue getValueFor(String name)
    087        {
    088            ParameterValue value = parameters.get(name);
    089
    090            return value == null ? ParameterValue.NULL : value;
    091        }
    092
    093        /**
    094         * Ignores any attempt to set the character encoding, as it already has been set before the request content was
    095         * parsed.
    096         *
    097         * @see org.apache.tapestry5.SymbolConstants#CHARSET
    098         */
    099        @Override
    100        public void setCharacterEncoding(String enc) throws UnsupportedEncodingException
    101        {
    102
    103        }
    104    }