package org.webslinger.servlet;

 import java.io.BufferedReader;

 import java.io.IOException;

 import java.io.InputStream;

 import java.io.InputStreamReader;

 import java.io.UnsupportedEncodingException;

 import java.util.Enumeration;

 import java.util.HashMap;

 import java.util.Locale;

 import java.util.HashSet;

 import java.util.Map;

 import java.util.Set;

 import javax.servlet.ServletInputStream;

 import javax.servlet.ServletRequest;

 import javax.servlet.RequestDispatcher;

 import org.apache.commons.collections.iterators.IteratorEnumeration;

 public class FakeServletRequest implements ServletRequest {

     protected final FakeServletContext context;

     protected final FakeResource resource;

     protected final Map<String, Object> attributes = new HashMap<String, Object>();

     protected final Map<String, String[]> parameters = new HashMap<String, String[]>();

     protected boolean getInputStreamCalled;

     protected ServletInputStream servletInputStream;

     protected boolean getReaderCalled;

     protected BufferedReader bufferedReader;

     protected String encoding;

     protected final ServletRequestConfig config;

     public FakeServletRequest(FakeServletContext context, FakeResource resource) {

         this(context, resource, new ServletRequestConfig());

     }

     public FakeServletRequest(FakeServletContext context, FakeResource resource, ServletRequestConfig config) {

         this.context = context;

         this.resource = resource;

         this.config = config;

     }

     public ServletRequestConfig getConfig() {

         return config;

     }

     public Object getAttribute(String name) {

         Object value = attributes.get(name);

         //System.err.println(this + ".getAttribute(" + name + ")=[" + value + "]");

         return value;

     }

     public Enumeration getAttributeNames() {

         return new IteratorEnumeration(attributes.keySet().iterator());

     }

     public void setAttribute(String name, Object value) {

         //System.err.println(this + ".setAttribute(" + name + ", " + value + ")");

         attributes.put(name, value);

     }

     public void removeAttribute(String name) {

         attributes.remove(name);

     }

     public String getCharacterEncoding() {

         return encoding;

     }

     public int getContentLength() {

         return -1;

     }

     public String getContentType() {

         return null;

     }

     public ServletInputStream getInputStream() throws IOException {

         if (getReaderCalled) throw new IllegalStateException();

         if (servletInputStream == null) {

             getInputStreamCalled = true;

             servletInputStream = getRealServletInputStream();

         }

         return servletInputStream;

     }

     public String getLocalAddr() {

         return config.getLocalAddr();

     }

     public Locale getLocale() {

         return config.getLocale();

     }

     public Enumeration getLocales() {

         return new IteratorEnumeration(config.getLocales().iterator());

     }

     public String getLocalName() {

         return config.getLocalName();

     }

     public int getLocalPort() {

         return config.getLocalPort();

     }

     public String getParameter(String name) {

         String[] values = getParameterValues(name);

         return values == null ? null : values[0];

     }

     public Map getParameterMap() {

         return parameters;

     }

     public Enumeration<String> getParameterNames() {

         return new IteratorEnumeration(parameters.keySet().iterator());

     }

     public String[] getParameterValues(String name) {

         return parameters.get(name);

     }

     public String getProtocol() {

         return config.getProtocol();

     }

     public BufferedReader getReader() throws IOException {

         if (getInputStreamCalled) throw new IllegalStateException();

         if (bufferedReader == null) {

             getReaderCalled = true;

             bufferedReader = getRealReader();

         }

         return bufferedReader;

     }

     /** @deprecated */

     public String getRealPath(String path) {

         return context.getRealPath(path);

     }

     public String getRemoteAddr() {

         return config.getRemoteAddr();

     }

     public String getRemoteHost() {

         return config.getRemoteHost();

     }

     public int getRemotePort() {

         return config.getRemotePort();

     }

     public RequestDispatcher getRequestDispatcher(String path) {

         try {

             return context.getRequestDispatcher(context.resolveResource(resource, path));

         } catch (IOException e){

             context.log(resource + ":getRequestDispatcher(" + path + ")", e);

             return null;

         }

     }

     public String getScheme() {

         return config.getScheme();

     }

     public String getServerName() {

         return config.getServerName();

     }

     public int getServerPort() {

         return config.getServerPort();

     }

     public boolean isSecure() {

         return config.isSecure();

     }

     public void setCharacterEncoding(String encoding) throws UnsupportedEncodingException {

         this.encoding = encoding;

     }

     /* local methods */

     public void setLocalAddr(String localAddr) {

         config.setLocalAddr(localAddr);

     }

     public void setLocalName(String localName) {

         config.setLocalName(localName);

     }

     public void setLocalPort(int localPort) {

         config.setLocalPort(localPort);

     }

     public void setRemoteAddr(String remoteAddr) {

         config.setRemoteAddr(remoteAddr);

     }

     public void setRemoteHost(String remoteHost) {

         config.setRemoteHost(remoteHost);

     }

     public void setRemotePort(int remotePort) {

         config.setRemotePort(remotePort);

     }

     public void setServerName(String serverName) {

         config.setServerName(serverName);

     }

     public void setServerPort(int serverPort) {

         config.setServerPort(serverPort);

     }

     public void setScheme(String scheme) {

         config.setScheme(scheme);

     }

     public void setIsSecure(boolean isSecure) {

         config.setIsSecure(isSecure);

     }

     public void setLocale(Locale locale) {

         config.setLocale(locale);

     }

     public void setProtocol(String protocol) {

         config.setProtocol(protocol);

     }

     public void addLocale(Locale locale) {

         config.addLocale(locale);

     }

     public void removeLocale(Locale locale) {

         config.removeLocale(locale);

     }

     public Map getAttributes() {

         return attributes;

     }

     public void setParameter(String name, String value) {

         parameters.put(name, new String[] {value});

     }

     public void addParameter(String name, String value) {

         String[] values = parameters.get(name);

         if (values == null) {

             values = new String[] {value};

         } else {

             String[] newValues = new String[values.length + 1];

             System.arraycopy(values, 0, newValues, 0, values.length);

             newValues[values.length] = value;

             values = newValues;

         }

         parameters.put(name, values);

     }

     protected ServletInputStream getRealServletInputStream() throws IOException {

         return new ServletInputStream() {

             public int available() throws IOException {

                 return 0;

             }

             public int read() throws IOException {

                 return -1;

             }

             public int read(byte[] b, int off, int len) throws IOException {

                 return -1;

             }

         };

     }

     protected BufferedReader getRealReader() throws IOException {

         return new BufferedReader(new InputStreamReader(getRealServletInputStream()));

     }

 }