Adding collector to ActionMailer

1 files changed, 32 insertions, 0 deletions

diff --git a/actionmailer/lib/action_mailer/collector.rb b/actionmailer/lib/action_mailer/collector.rb
new file mode 100644
index 0000000000..49c3f04bad
--- /dev/null
+++ b/actionmailer/lib/action_mailer/collector.rb
@@ -0,0 +1,32 @@
+require 'abstract_controller/collector'
+
+module ActionMailer #:nodoc:
+
+  class Collector
+  
+    include AbstractController::Collector
+  
+    attr_accessor :responses
+
+    def initialize(context, options, &block)
+      @default_options = options
+      @default_render = block
+      @default_formats = context.formats
+      @context = context
+      @responses = []
+    end
+
+    def custom(mime, options={}, &block)
+      options = @default_options.merge(:content_type => mime.to_s).merge(options)
+      @context.formats = [mime.to_sym]
+      options[:body] = if block
+        block.call
+      else
+        @default_render.call
+      end
+      @responses << options
+      @context.formats = @default_formats
+    end
+
+  end
+end
\ No newline at end of file