1 files changed, 46 insertions, 0 deletions
diff --git a/actionmailer/lib/action_mailer/deliver_later/mail_message_wrapper.rb b/actionmailer/lib/action_mailer/deliver_later/mail_message_wrapper.rb
new file mode 100644
index 0000000000..5c0a5304d4
--- /dev/null
+++ b/actionmailer/lib/action_mailer/deliver_later/mail_message_wrapper.rb
@@ -0,0 +1,46 @@
+module ActionMailer
+  module DeliverLater
+    class MailMessageWrapper < Delegator
+      def initialize(mailer, mail_method, *args)
+        @mailer      = mailer
+        @mail_method = mail_method
+        @args        = args
+        __getobj__
+      end
+
+      def __getobj__
+        @obj ||= @mailer.send(:new, @mail_method, *@args).message
+      end
+
+      def __setobj__(obj)
+        @obj = obj
+      end
+
+      def deliver_later!(options={})
+        enqueue_delivery :deliver!, options
+      end
+
+      def deliver_later(options={})
+        enqueue_delivery :deliver, options
+      end
+
+      def method_missing(m, *args, &block)
+        __getobj__.__send__(m, *args, &block)
+      end
+
+      private
+      def enqueue_delivery(delivery_method, options={})
+        args = @mailer.name, @mail_method.to_s, delivery_method.to_s, *@args
+        enqueue_method = :enqueue
+        if options[:at]
+          enqueue_method = :enqueue_at
+          args.unshift options[:at]
+        elsif options[:in]
+          enqueue_method = :enqueue_in
+          args.unshift options[:in]
+        end
+        ActionMailer::DeliverLater::Job.send enqueue_method, *args
+      end
+    end
+  end
+end