Added a new container for cookies that makes them more intuative to use. The old methods of cookie and @cookies have been deprecated.

git-svn-id: http://svn-commit.rubyonrails.org/rails/trunk@20 5ecf4fe2-1ee6-0310-87b1-e25e094e27de

3 files changed, 68 insertions, 27 deletions

diff --git a/actionpack/lib/action_controller.rb b/actionpack/lib/action_controller.rb
index 6445940d78..9cc31cd90c 100755
--- a/actionpack/lib/action_controller.rb
+++ b/actionpack/lib/action_controller.rb
@@ -34,6 +34,7 @@ require 'action_controller/flash'
 require 'action_controller/scaffolding'
 require 'action_controller/helpers'
 require 'action_controller/dependencies'
+require 'action_controller/cookies'
 require 'action_controller/cgi_process'
 
 ActionController::Base.class_eval do
@@ -45,6 +46,7 @@ ActionController::Base.class_eval do
   include ActionController::Scaffolding
   include ActionController::Helpers
   include ActionController::Dependencies
+  include ActionController::Cookies
 end
 
 require 'action_view'
diff --git a/actionpack/lib/action_controller/base.rb b/actionpack/lib/action_controller/base.rb
index 5c386ac05a..a794272a9a 100755
--- a/actionpack/lib/action_controller/base.rb
+++ b/actionpack/lib/action_controller/base.rb
@@ -539,33 +539,6 @@ module ActionController #:nodoc:
         @performed_redirect = true
       end
 
-      # Creates a new cookie that is sent along-side the next render or redirect command. API is the same as for CGI::Cookie.
-      # Examples:
-      #
-      #   cookie("name", "value1", "value2", ...)
-      #   cookie("name" => "name", "value" => "value")
-      #   cookie('name'    => 'name',
-      #          'value'   => ['value1', 'value2', ...],
-      #          'path'    => 'path',   # optional
-      #          'domain'  => 'domain', # optional
-      #          'expires' => Time.now, # optional
-      #          'secure'  => true      # optional
-      #   )
-      def cookie(*options) #:doc:
-        @response.headers["cookie"] << CGI::Cookie.new(*options)
-      end
-
-      # Alias for cookie "name", "value"
-      def cookies[]=(name, value)
-        cookie(name, value)
-      end
-
-      # Returns the value of the cookie by +name+ -- or nil if no such cookie exist. You set new cookies using either the cookie method
-      # or cookies[]= (for simple name/value cookies without options).
-      def cookies[](name)
-        @cookies[name].value if @cookies[name]
-      end
-      
       # Resets the session by clearsing out all the objects stored within and initializing a new session object.
       def reset_session #:doc:
         @request.reset_session
@@ -573,6 +546,11 @@ module ActionController #:nodoc:
         @response.session = @session
       end
     
+      # Deprecated cookie writer method
+      def cookie(*options)
+        @response.headers["cookie"] << CGI::Cookie.new(*options)
+      end
+
     private
       def initialize_template_class(response)
         begin
diff --git a/actionpack/lib/action_controller/cookies.rb b/actionpack/lib/action_controller/cookies.rb
new file mode 100644
index 0000000000..78815b7bc4
--- /dev/null
+++ b/actionpack/lib/action_controller/cookies.rb
@@ -0,0 +1,61 @@
+module ActionController #:nodoc:
+  # Cookies are read and written through ActionController#cookies. The cookies being read is what was received along with the request,
+  # the cookies being written is what will be sent out will the response. Cookies are read by value (so you won't get the cookie object
+  # itself back -- just the value it holds). Examples for writting:
+  #
+  #   cookies["user_name"] = "david" # => Will set a simple session cookie
+  #   cookies["login"] = { "value" => "XJ-122", "expires" => Time.now + 360} # => Will set a cookie that expires in 1 hour
+  #   
+  # Examples for reading:
+  #
+  #   cookies["user_name"] # => "david"
+  #   cookies.size         # => 2
+  #
+  # All the options for setting cookies are:
+  #
+  # value:: the cookie's value or list of values (as an array).
+  # path:: the path for which this cookie applies.  Defaults to the root of the application.
+  # domain:: the domain for which this cookie applies.
+  # expires:: the time at which this cookie expires, as a +Time+ object.
+  # secure:: whether this cookie is a secure cookie or not (default to false).
+  #          Secure cookies are only transmitted to HTTPS servers.
+  module Cookies
+    # Returns the cookie container, which operates as described above.
+    def cookies
+      CookieContainer.new(self)
+    end
+  end
+  
+  class CookieContainer < Hash #:nodoc:
+    def initialize(controller)
+      @controller, @cookies = controller, controller.instance_variable_get("@cookies")
+      super()
+      update(@cookies)
+    end
+
+    # Returns the value of the cookie by +name+ -- or nil if no such cookie exist. You set new cookies using either the cookie method
+    # or cookies[]= (for simple name/value cookies without options).
+    def [](name)
+      @cookies[name].value if @cookies[name]
+    end
+    
+    def []=(name, options)
+      if options.is_a?(Hash)
+        options["name"] = name
+      else
+        options = [ name, options ]
+      end
+      
+      set_cookie(name, options)
+    end
+
+    private
+      def set_cookie(name, options) #:doc:
+        if options.is_a?(Array)
+          @controller.response.headers["cookie"] << CGI::Cookie.new(*options)
+        else
+          @controller.response.headers["cookie"] << CGI::Cookie.new(options)
+        end
+      end
+  end
+end