1 files changed, 39 insertions, 0 deletions
diff --git a/lib/attachment_to_html/view.rb b/lib/attachment_to_html/view.rb
new file mode 100644
index 000000000..e6991d44e
--- /dev/null
+++ b/lib/attachment_to_html/view.rb
@@ -0,0 +1,39 @@
+module AttachmentToHTML
+    class View < ERB
+
+        def self.template
+            @template || "#{ File.dirname(__FILE__) }/template.html.erb"
+        end
+
+        def self.template=(path)
+            @template = path
+        end
+
+        attr_accessor :title, :body, :template, :wrapper
+
+        def initialize(adapter, opts = {})
+            self.title    = adapter.title
+            self.body     = adapter.body
+            self.template = opts.fetch(:template, self.class.template)
+            self.wrapper  = opts.fetch(:wrapper, 'wrapper')
+            super(File.read(template))
+        end
+
+        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