Enforce .rtex extension requirement [#6 state:open]

bruce · Sep 20, 2008 · 2d2890a · 2d2890a
1 parent 07278e3
commit 2d2890a
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/lib/rtex/framework/rails.rb b/lib/rtex/framework/rails.rb
@@ -25,6 +25,9 @@ def initialize(*args)
           if defined?(@view)
             @view.template_format = :pdf
           end
+          def @view.rendered_with_rtex
+            true
+          end
         end
       end
 
@@ -35,7 +38,7 @@ def self.included(base)
 
         def render_with_rtex(options=nil, *args, &block)
           result = render_without_rtex(options, *args, &block)
-          if result.is_a?(String) && @template.template_format == :pdf
+          if result.is_a?(String) && @template.respond_to?(:rendered_with_rtex)
             options ||= {}
             ::RTeX::Document.new(result, options.merge(:processed => true)).to_pdf do |filename|
               serve_file = Tempfile.new('rtex-pdf')