Allow custom template exclusion from BetterErb processing. Fix #1

Shopify · Mar 23, 2017 · a93ac11 · a93ac11
1 parent 60bbdb1
commit a93ac11
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/lib/better_html.rb b/lib/better_html.rb
@@ -25,6 +25,12 @@ class Config
     # name of all html attributes that may contain javascript
     cattr_accessor :javascript_attribute_names
     self.javascript_attribute_names = [/\Aon/i]
+
+    cattr_accessor :template_exclusion_filter_block
+
+    def self.template_exclusion_filter(&block)
+      self.template_exclusion_filter_block = block
+    end
   end
 
   def self.config

diff --git a/lib/better_html/better_erb.rb b/lib/better_html/better_erb.rb
@@ -40,7 +40,8 @@ def generate(template)
       # Always make sure we return a String in the default_internal
       erb.encode!
 
-      klass = BetterHtml::BetterErb.content_types[exts]
+      excluded_template = !!BetterHtml::Config.template_exclusion_filter_block&.call(template.identifier)
+      klass = BetterHtml::BetterErb.content_types[exts] unless excluded_template
       klass ||= self.class.erb_implementation
 
       generator = klass.new(