From c56ac5ed2c45e1a1a955667d4fcc3646baf8bb3b Mon Sep 17 00:00:00 2001
From: Benny Wong <benny@bwong.net>
Date: Fri, 28 Oct 2011 13:16:25 -0400
Subject: [PATCH] Include Draper::ViewContextFilter in ActionMailer::Base

* Only send the :before_filter method if it exists
* This will allow decorators to be used when rendering email
---
 Readme.markdown            | 15 +++++++++++++++
 lib/draper/system.rb       |  5 +++--
 lib/draper/view_context.rb |  8 ++++----
 3 files changed, 22 insertions(+), 6 deletions(-)

diff --git a/Readme.markdown b/Readme.markdown
index 56baff9..a57d7b3 100644
--- a/Readme.markdown
+++ b/Readme.markdown
@@ -229,6 +229,21 @@ Use the new methods in your views like any other model method (ex: `@article.pub
 <h1><%= @article.title %> <%= @article.published_at %></h1>
 ```
 
+### Using in Mailers
+
+To use decorators in mailers that use helpers, you have to call `set_current_view_context` in your
+mailer method:
+
+```ruby
+class ActicleMailer < ActionMailer::Base
+  def new_article(article)
+    set_current_view_context
+    @article_decorator = ArticleDecorator.decorate(article)
+    mail(:to => 'come@me.bro', :subject => "New Article: #{@article_decorator.title}")
+  end
+end
+```
+
 ## Possible Decoration Methods
 
 Here are some ideas of what you might do in decorator methods:
diff --git a/lib/draper/system.rb b/lib/draper/system.rb
index 29cf957..3cf420c 100644
--- a/lib/draper/system.rb
+++ b/lib/draper/system.rb
@@ -1,7 +1,8 @@
 module Draper
-  class System    
+  class System
     def self.setup
       ActionController::Base.send(:include, Draper::ViewContextFilter) if defined?(ActionController::Base)
+      ActionMailer::Base.send(:include, Draper::ViewContextFilter) if defined?(ActionMailer::Base)
     end
   end
-end
\ No newline at end of file
+end
diff --git a/lib/draper/view_context.rb b/lib/draper/view_context.rb
index 6b5ab21..8a0bb13 100644
--- a/lib/draper/view_context.rb
+++ b/lib/draper/view_context.rb
@@ -8,14 +8,14 @@ module Draper
       Thread.current[:current_view_context] = input
     end
   end
-  
+
   module ViewContextFilter
     def set_current_view_context
       Draper::ViewContext.current = self.view_context
     end
-    
+
     def self.included(source)
-      source.send(:before_filter, :set_current_view_context)
+      source.send(:before_filter, :set_current_view_context) if source.respond_to?(:before_filter)
     end
   end
-end
\ No newline at end of file
+end