Implement dynamic indentation

lib/hamlit/engine.rb

@@ -11,6 +11,7 @@ module Hamlit
       html_type:   nil,
       attr_quote:  "'",
       escape_html: true,
+      buffer:      '_buf',
       autoclose:   %w(area base basefont br col command embed frame
                       hr img input isindex keygen link menuitem meta
                       param source track wbr),

lib/pretty_hamlit/compiler.rb

@@ -25,6 +25,10 @@ module PrettyHamlit
       @indent_level -= 1
     end
 
+    def compile_filter(node)
+      @filter_compiler.compile(node, @indent_level)
+    end
+
     def compile_tag(node)
       @indent_level += 1
       super

lib/pretty_hamlit/dynamic_indentation.rb

@@ -0,0 +1,11 @@
+module PrettyHamlit
+  class DynamicIndentation
+    class << self
+      def indent_with(indent_level)
+        indent = "\n" << '  ' * indent_level
+        text = yield('')
+        text.gsub("\n", indent)
+      end
+    end
+  end
+end

lib/pretty_hamlit/engine.rb

@@ -1,6 +1,7 @@
 require 'temple'
 require 'hamlit/parser'
 require 'pretty_hamlit/compiler'
+require 'pretty_hamlit/dynamic_indentation'
 
 module PrettyHamlit
   class Engine < Temple::Engine
@@ -11,6 +12,7 @@ module PrettyHamlit
       attr_quote:  "'",
       escape_html: true,
       pretty:      true,
+      buffer:      '_buf',
       autoclose:   %w(area base basefont br col command embed frame
                       hr img input isindex keygen link menuitem meta
                       param source track wbr),

lib/pretty_hamlit/filters.rb

@@ -21,8 +21,13 @@ module PrettyHamlit
       @options = options
     end
 
-    def compile(node)
-      find_compiler(node.value[:name]).compile(node)
+    def compile(node, indent_level)
+      content = find_compiler(node.value[:name]).compile(node)
+      [:multi,
+       [:code, "#{@options[:buffer]} << ::PrettyHamlit::DynamicIndentation.indent_with(#{indent_level}) do |#{@options[:buffer]}|"],
+       content,
+       [:code, 'end'],
+      ]
     end
 
     private

test/engine_test.rb

@@ -126,7 +126,7 @@ class EngineTest < Haml::TestCase
     assert_equal "", render("")
   end
 
-  def test_flexible_tabulation; skip # filter
+  def test_flexible_tabulation
     assert_equal("<p>\n  foo\n</p>\n<q>\n  bar\n  <a>\n    baz\n  </a>\n</q>\n",
                  render("%p\n foo\n%q\n bar\n %a\n  baz"))
     assert_equal("<p>\n  foo\n</p>\n<q>\n  bar\n  <a>\n    baz\n  </a>\n</q>\n",