Update AttachmentToHTML for new View class

1 files changed, 7 insertions, 6 deletions

diff --git a/lib/attachment_to_html/attachment_to_html.rb b/lib/attachment_to_html/attachment_to_html.rb
index 104dc13e2..8675329c4 100644
--- a/lib/attachment_to_html/attachment_to_html.rb
+++ b/lib/attachment_to_html/attachment_to_html.rb
@@ -10,14 +10,15 @@ module AttachmentToHTML
 
     def to_html(attachment, opts = {})
         adapter = adapter_for(attachment).new(attachment, opts)
-        html = HTML.new(adapter)
 
-        if html.success?
-            html
-        else
-            fallback = fallback_adapter_for(attachment).new(attachment, opts)
-            HTML.new(fallback)
+        unless adapter.success?
+          adapter = fallback_adapter_for(attachment).new(attachment, opts)
         end
+
+        view = View.new(adapter)
+        view.wrapper = 'wrapper_google_embed' if adapter.is_a?(Adapters::GoogleDocsViewer)
+
+        view.render
     end
 
     private