Added library functions for setting and parsing cookies.

[jsvc.git] / src / dolda / jsvc / util / Cookie.java

diff --git a/src/dolda/jsvc/util/Cookie.java b/src/dolda/jsvc/util/Cookie.java
new file mode 100644 (file)
index 0000000..8783427
--- /dev/null
+++ b/src/dolda/jsvc/util/Cookie.java
@@ -0,0 +1,98 @@
+package dolda.jsvc.util;
+
+import dolda.jsvc.*;
+import java.util.*;
+import java.text.*;
+import java.io.*;
+
+public class Cookie {
+    public final static DateFormat datefmt;
+    static {
+       datefmt = new SimpleDateFormat("EEE, dd-MMM-yyyy HH:mm:ss z", Locale.ENGLISH);
+       datefmt.setCalendar(Calendar.getInstance(TimeZone.getTimeZone("UTC")));
+    }
+    public final String name;
+    public String value;
+    public Date expires;
+    public String domain, path;
+    public boolean secure;
+    
+    public Cookie(String name, String value, Date expires, String domain, String path, boolean secure) {
+       if(!Http.istoken(name))
+           throw(new RuntimeException("Invalid cookie name: `" + name + "'"));
+       this.name = name;
+       this.value = value;
+       this.expires = expires;
+       this.domain = domain;
+       this.path = path;
+       this.secure = secure;
+    }
+    
+    public Cookie(String name) {
+       this(name, null, null, null, null, false);
+    }
+    
+    public Cookie(String name, String value) {
+       this(name, value, null, null, null, false);
+    }
+    
+    public String format() {
+       StringBuilder buf = new StringBuilder();
+       buf.append(Http.tokenquote(name));
+       buf.append('=');
+       buf.append(Http.tokenquote(value));
+       if(domain != null)
+           buf.append("; Domain=" + Http.tokenquote(domain));
+       if(path != null)
+           buf.append("; Path=" + Http.tokenquote(path));
+       if(expires != null)
+           buf.append("; Expires=" + Http.tokenquote(datefmt.format(expires)));
+       if(secure)
+           buf.append("; Secure");
+       return(buf.toString());
+    }
+
+    public void addto(Request req) {
+       req.outheaders().add("Set-Cookie", format());
+    }
+    
+    public static MultiMap<String, Cookie> parse(Request req) {
+       MultiMap<String, Cookie> ret = new WrappedMultiMap<String, Cookie>(new TreeMap<String, Collection<Cookie>>());
+       for(String in : req.inheaders().values("Cookie")) {
+           try {
+               StringReader r = new StringReader(in);
+               Cookie c = null;
+               while(true) {
+                   String k = Http.tokenunquote(r);
+                   String v = Http.tokenunquote(r);
+                   if(k == null)
+                       break;
+                   if(k.equals("$Version")) {
+                       if(Integer.parseInt(v) != 1)
+                           throw(new Http.EncodingException("Unknown cookie format version"));
+                   } else if(k.equals("$Path")) {
+                       if(c != null)
+                           c.path = v;
+                   } else if(k.equals("$Domain")) {
+                       if(c != null)
+                           c.domain = v;
+                   } else {
+                       c = new Cookie(k, v);
+                       ret.add(k, c);
+                   }
+               }
+           } catch(IOException e) {
+               throw(new Error(e));
+           }
+       }
+       return(ret);
+    }
+    
+    public String toString() {
+       StringBuilder buf = new StringBuilder();
+       buf.append("Cookie(");
+       buf.append(format());
+       buf.append(")");
+       return(buf.toString());
+    }
+}