001    /*
002     * Licensed to the Apache Software Foundation (ASF) under one or more
003     * contributor license agreements.  See the NOTICE file distributed with
004     * this work for additional information regarding copyright ownership.
005     * The ASF licenses this file to You under the Apache License, Version 2.0
006     * (the "License"); you may not use this file except in compliance with
007     * the License.  You may obtain a copy of the License at
008     *
009     *     http://www.apache.org/licenses/LICENSE-2.0
010     *
011     * Unless required by applicable law or agreed to in writing, software
012     * distributed under the License is distributed on an "AS IS" BASIS,
013     * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
014     * See the License for the specific language governing permissions and
015     * limitations under the License.
016     */
017    package org.apache.commons.jxpath.servlet;
018    
019    import java.util.Enumeration;
020    import java.util.HashSet;
021    import javax.servlet.ServletRequest;
022    
023    /**
024     * Implementation of the {@link org.apache.commons.jxpath.DynamicPropertyHandler}
025     * interface that provides access to attributes and parameters
026     * of a {@link ServletRequest}.
027     *
028     * @author Dmitri Plotnikov
029     * @version $Revision: 652848 $ $Date: 2008-05-02 12:53:50 -0500 (Fri, 02 May 2008) $
030     */
031    public class ServletRequestHandler extends HttpSessionHandler {
032    
033        protected void collectPropertyNames(HashSet set, Object bean) {
034            super.collectPropertyNames(set, bean);
035            ServletRequestAndContext handle = (ServletRequestAndContext) bean;
036            ServletRequest servletRequest = handle.getServletRequest();
037            Enumeration e = servletRequest.getAttributeNames();
038            while (e.hasMoreElements()) {
039                set.add(e.nextElement());
040            }
041            e = servletRequest.getParameterNames();
042            while (e.hasMoreElements()) {
043                set.add(e.nextElement());
044            }
045        }
046    
047        public Object getProperty(Object bean, String property) {
048            ServletRequestAndContext handle = (ServletRequestAndContext) bean;
049            ServletRequest servletRequest = handle.getServletRequest();
050            String[] strings = servletRequest.getParameterValues(property);
051    
052            if (strings != null) {
053                if (strings.length == 0) {
054                    return null;
055                }
056                if (strings.length == 1) {
057                    return strings[0];
058                }
059                return strings;
060            }
061    
062            Object object = servletRequest.getAttribute(property);
063            if (object != null) {
064                return object;
065            }
066    
067            return super.getProperty(bean, property);
068        }
069    
070        public void setProperty(Object request, String property, Object value) {
071            ((ServletRequestAndContext) request).getServletRequest().setAttribute(property, value);
072        }
073    }