package org.webslinger.resolver;

 import java.util.HashMap;

 import java.util.HashSet;

 import java.util.Set;

 import java.util.Map;

 import javax.servlet.ServletRequest;

 import org.apache.commons.collections.CollectionUtils;

 public class ServletRequestResolver<T extends ServletRequest> extends BeanResolver<T> {

     protected ServletRequestResolver() {

         super();

         addFetcher("character-encoding", new BeanResolver.BeanGet<T, String>("character-encoding", String.class) {

             public String getValue(ServletRequest request) {

                 return request.getCharacterEncoding();

             }

             public void setValue(ServletRequest request, String value) throws Exception {

                 request.setCharacterEncoding(value);

             }

         });

         addFetcher("content-length", new BeanResolver.BeanGet<T, Integer>("content-length", Integer.class) {

             public Integer getValue(ServletRequest request) {

                 return request.getContentLength();

             }

         });

         addFetcher("content-type", new BeanResolver.BeanGet<T, String>("content-type", String.class) {

             public String getValue(ServletRequest request) {

                 return request.getContentType();

             }

         });

         addFetcher("parameters", new BeanResolver.BeanGet<T, Map>("parameters", Map.class) {

             public Map getValue(ServletRequest request) {

                 return request.getParameterMap();

             }

         });

         addFetcher("protocol", new BeanResolver.BeanGet<T, String>("protocol", String.class) {

             public String getValue(ServletRequest request) {

                 return request.getProtocol();

             }

         });

         addFetcher("remote-address", new BeanResolver.BeanGet<T, String>("remote-address", String.class) {

             public String getValue(ServletRequest request) {

                 return request.getRemoteAddr();

             }

         });

         addFetcher("remote-host", new BeanResolver.BeanGet<T, String>("remote-host", String.class) {

             public String getValue(ServletRequest request) {

                 return request.getRemoteHost();

             }

         });

         addFetcher("scheme", new BeanResolver.BeanGet<T, String>("scheme", String.class) {

             public String getValue(ServletRequest request) {

                 return request.getScheme();

             }

         });

         addFetcher("secure", new BeanResolver.BeanGet<T, Boolean>("secure", Boolean.class) {

             public Boolean getValue(ServletRequest request) {

                 return request.isSecure();

             }

         });

         addFetcher("server-port", new BeanResolver.BeanGet<T, Integer>("server-port", Integer.class) {

             public Integer getValue(ServletRequest request) {

                 return request.getServerPort();

             }

         });

         addFetcher("attributes", new BeanResolver.BeanGet<T, ServletRequestAttributesLookup>("attributes", ServletRequestAttributesLookup.class) {

             public ServletRequestAttributesLookup getValue(ServletRequest request) {

                 return new ServletRequestAttributesLookup(request);

             }

         });

     }

     public static final ServletRequestResolver<ServletRequest> RESOLVER = new ServletRequestResolver<ServletRequest>();

     public static final class ServletRequestAttributesLookup implements ObjectLookup {

         private final ServletRequest request;

         private ServletRequestAttributesLookup(ServletRequest request) {

             this.request = request;

         }

         public boolean objectContains(String name) {

             Set names = getAttributeNames();

             return names.contains(name);

         }

         public Object objectGet(String name) {

             return request.getAttribute(name);

         }

         public void objectSet(String name, Object value) throws Exception {

             request.setAttribute(name, value);

         }

         public String[] objectList() {

             Set<String> names = getAttributeNames();

             return (String[]) names.toArray(new String[names.size()]);

         }

         private Set<String> getAttributeNames() {

             Set<String> names = new HashSet<String>();

             CollectionUtils.addAll(names, request.getAttributeNames());

             return names;

         }

         public boolean objectHasChildren() {

             return true;

         }

         public Class getType(String name) {

             Object value = request.getAttribute(name);

             if (value == null) return Void.TYPE;

             return value.getClass();

         }

     }

     public static class ServletRequestResolverInfo implements ObjectResolverInfo<ServletRequest> {

         public String getType() {

             return "javax.servlet.ServletRequest";

         }

         public ServletRequestResolver<ServletRequest> getResolver() {

             return RESOLVER;

         }

     }

 }