Support annotate_rendered_view_with_filenames

Fix #180

lib/hamlit/ambles.rb

@@ -0,0 +1,20 @@
+# frozen_string_literal: true
+module Hamlit
+  class Ambles < Temple::Filter
+    define_options :preamble, :postamble
+
+    def initialize(*)
+      super
+      @preamble = options[:preamble]
+      @postamble = options[:postamble]
+    end
+
+    def call(ast)
+      ret = [:multi]
+      ret << [:static, @preamble] if @preamble
+      ret << ast
+      ret << [:static, @postamble] if @postamble
+      ret
+    end
+  end
+end

lib/hamlit/engine.rb

@@ -6,6 +6,7 @@
 require 'hamlit/escapable'
 require 'hamlit/force_escapable'
 require 'hamlit/dynamic_merger'
+require 'hamlit/ambles'
 
 module Hamlit
   class Engine < Temple::Engine
@@ -33,6 +34,7 @@ class Engine < Temple::Engine
     filter :MultiFlattener
     filter :StaticMerger
     use DynamicMerger
+    use Ambles
     use :Generator, -> { options[:generator] }
   end
 end

lib/hamlit/rails_template.rb

@@ -33,6 +33,11 @@ def call(template, source = nil)
         options = options.merge(format: :xhtml)
       end
 
+      if ActionView::Base.try(:annotate_rendered_view_with_filenames) && template.format == :html
+        options[:preamble] = "<!-- BEGIN #{template.short_identifier} -->\n"
+        options[:postamble] = "<!-- END #{template.short_identifier} -->\n"
+      end
+
       Engine.new(options).call(source)
     end