package org.webslinger.servlet;

 import java.io.Serializable;

 import java.util.ArrayList;

 import java.util.Collections;

 import java.util.Date;

 import java.util.Enumeration;

 import java.util.HashMap;

 import java.util.Iterator;

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

 import org.webslinger.io.HttpUtil;

 public class HttpHeadersImpl implements HttpHeaders, Serializable {

     protected final HashMap<String, HeaderValue> headers = new HashMap<String, HeaderValue>();

     protected String getFirst(String name) {

         HeaderValue hv = headers.get(name.toLowerCase());

         return hv != null ? (String) hv.values.get(0) : null;

     }

     public void addDateHeader(String name, long date) {

         addHeader(name, HttpUtil.formatDate(date));

     }

     public void addHeader(String name, String value) {

         HeaderValue hv = headers.get(name.toLowerCase());

         if (hv == null) {

             headers.put(name.toLowerCase(), new HeaderValue(name, value));

         } else {

             hv.values.add(value);

         }

     }

     public void addIntHeader(String name, int value) {

         addHeader(name, Integer.toString(value));

     }

     public boolean containsHeader(String name) {

         return headers.containsKey(name);

     }

     public long getDateHeader(String name) {

         String value = getFirst(name);

         if (value == null) return -1;

         Date date = HttpUtil.parseDate(value);

         if (date != null) return date.getTime();

         throw new IllegalArgumentException("not a date(" + name + ", " + value + ")");

     }

     public String getHeader(String name) {

         return getFirst(name);

     }

     public Enumeration<String> getHeaderNames() {

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

     }

     public Enumeration<String> getHeaders(String name) {

         HeaderValue hv = headers.get(name.toLowerCase());

         if (hv == null) return new IteratorEnumeration(Collections.EMPTY_LIST.iterator());

         return new IteratorEnumeration(hv.values.iterator());

     }

     public int getIntHeader(String name) {

         String value = getFirst(name);

         if (value == null) return -1;

         return Integer.parseInt(value);

     }

     public void setDateHeader(String name, long date) {

         setHeader(name, HttpUtil.formatDate(date));

     }

     public void setHeader(String name, String value) {

         headers.put(name.toLowerCase(), new HeaderValue(name, value));

     }

     public void setIntHeader(String name, int value) {

         setHeader(name, Integer.toString(value));

     }

     protected static final class HeaderValue implements Serializable {

         private final String name;

         private final ArrayList<String> values = new ArrayList<String>();

         protected HeaderValue(String name, String firstValue) {

             this.name = name;

             values.add(firstValue);

         }

         public boolean equals(Object o) {

             if (!(o instanceof HeaderValue)) return false;

             HeaderValue other = (HeaderValue) o;

             return name.toLowerCase().equals(other.name.toLowerCase())

                     && values.equals(other.values);

         }

         public int hashCode() {

             return HeaderValue.class.hashCode() ^ name.hashCode();

         }

         public String toString() {

             return "HV(" + name + ':' + values + ')';

         }

     }

     public String toString() {

         return headers.values().toString();

     }

 }