lib/webrick/cookie.rb

   1 #
   2 # cookie.rb -- Cookie class
   3 #
   4 # Author: IPR -- Internet Programming with Ruby -- writers
   5 # Copyright (c) 2000, 2001 TAKAHASHI Masayoshi, GOTOU Yuuzou
   6 # Copyright (c) 2002 Internet Programming with Ruby writers. All rights
   7 # reserved.
   8 #
   9 # $IPR: cookie.rb,v 1.16 2002/09/21 12:23:35 gotoyuzo Exp $
  10
  11 require 'time'
  12 require 'webrick/httputils'
  13
  14 module WEBrick
  15   class Cookie
  16
  17     attr_reader :name
  18     attr_accessor :value, :version
  19     attr_accessor :domain, :path, :secure
  20     attr_accessor :comment, :max_age
  21     #attr_accessor :comment_url, :discard, :port
  22
  23     def initialize(name, value)
  24       @name = name
  25       @value = value
  26       @version = 0     # Netscape Cookie
  27
  28       @domain = @path = @secure = @comment = @max_age =
  29       @expires = @comment_url = @discard = @port = nil
  30     end
  31
  32     def expires=(t)
  33       @expires = t && (t.is_a?(Time) ? t.httpdate : t.to_s)
  34     end
  35
  36     def expires
  37       @expires && Time.parse(@expires)
  38     end
  39
  40     def to_s
  41       ret = ""
  42       ret << @name << "=" << @value
  43       ret << "; " << "Version=" << @version.to_s if @version > 0
  44       ret << "; " << "Domain="  << @domain  if @domain
  45       ret << "; " << "Expires=" << @expires if @expires
  46       ret << "; " << "Max-Age=" << @max_age.to_s if @max_age
  47       ret << "; " << "Comment=" << @comment if @comment
  48       ret << "; " << "Path="    << @path if @path
  49       ret << "; " << "Secure"   if @secure
  50       ret
  51     end
  52
  53     # Cookie::parse()
  54     #   It parses Cookie field sent from the user agent.
  55     def self.parse(str)
  56       if str
  57         ret = []
  58         cookie = nil
  59         ver = 0
  60         str.split(/[;,]\s+/).each{|x|
  61           key, val = x.split(/=/,2)
  62           val = val ? HTTPUtils::dequote(val) : ""
  63           case key
  64           when "$Version"; ver = val.to_i
  65           when "$Path";    cookie.path = val
  66           when "$Domain";  cookie.domain = val
  67           when "$Port";    cookie.port = val
  68           else
  69             ret << cookie if cookie
  70             cookie = self.new(key, val)
  71             cookie.version = ver
  72           end
  73         }
  74         ret << cookie if cookie
  75         ret
  76       end
  77     end
  78
  79     def self.parse_set_cookie(str)
  80       cookie_elem = str.split(/;/)
  81       first_elem = cookie_elem.shift
  82       first_elem.strip!
  83       key, value = first_elem.split(/=/, 2)
  84       cookie = new(key, HTTPUtils.dequote(value))
  85       cookie_elem.each{|pair|
  86         pair.strip!
  87         key, value = pair.split(/=/, 2)
  88         if value
  89           value = HTTPUtils.dequote(value.strip)
  90         end
  91         case key.downcase
  92         when "domain"  then cookie.domain  = value
  93         when "path"    then cookie.path    = value
  94         when "expires" then cookie.expires = value
  95         when "max-age" then cookie.max_age = Integer(value)
  96         when "comment" then cookie.comment = value
  97         when "version" then cookie.version = Integer(value)
  98         when "secure"  then cookie.secure = true
  99         end
 100       }
 101       return cookie
 102     end
 103
 104     def self.parse_set_cookies(str)
 105       return str.split(/,(?=[^;,]*=)|,$/).collect{|c|
 106         parse_set_cookie(c)
 107       }
 108     end
 109   end
 110 end