From 3954e78a4d330e3e9d82be0d239d5bf25d2f6a04 Mon Sep 17 00:00:00 2001
From: Gareth Rees <gareth@mysociety.org>
Date: Wed, 2 Apr 2014 17:15:41 +0100
Subject: Dynamically inject AttachmentToHTML::View content

---
 lib/attachment_to_html/view.rb | 14 +++++++++++++-
 1 file changed, 13 insertions(+), 1 deletion(-)

(limited to 'lib/attachment_to_html/view.rb')

diff --git a/lib/attachment_to_html/view.rb b/lib/attachment_to_html/view.rb
index 5cdd3823b..e6991d44e 100644
--- a/lib/attachment_to_html/view.rb
+++ b/lib/attachment_to_html/view.rb
@@ -19,9 +19,21 @@ module AttachmentToHTML
             super(File.read(template))
         end
 
-        def render
+        def render(&block)
+            instance_eval(&block) if block_given?
             result(binding)
         end
 
+        def content_for(area)
+            send(area) if respond_to?(area)
+        end
+
+        private
+
+        def inject_content(area, &block)
+          instance_variable_set("@#{ area }".to_sym, block.call)
+          self.class.send(:attr_accessor, area)
+        end
+
     end
 end
-- 
cgit v1.2.3