Create Hamlit::Compiler

lib/hamlit/compiler.rb

@@ -0,0 +1,8 @@
+require 'hamlit/compilers/preserve'
+require 'hamlit/filter'
+
+module Hamlit
+  class Compiler < Hamlit::Filter
+    include Compilers::Preserve
+  end
+end

lib/hamlit/compilers/preserve.rb

@@ -0,0 +1,10 @@
+module Hamlit
+  module Compilers
+    module Preserve
+      def on_haml_preserve(code)
+        code = "Hamlit::Helpers.find_and_preserve(#{code})"
+        [:dynamic, code]
+      end
+    end
+  end
+end

lib/hamlit/engine.rb

@@ -1,6 +1,7 @@
 require 'temple'
 require 'hamlit/attribute_compiler'
 require 'hamlit/attribute_sorter'
+require 'hamlit/compiler'
 require 'hamlit/dynamic_formatter'
 require 'hamlit/doctype_compiler'
 require 'hamlit/filter_compiler'
@@ -24,6 +25,7 @@ module Hamlit
 
     use MultilinePreprocessor
     use Parser
+    use Compiler
     use DoctypeCompiler
     use AttributeCompiler
     use NewAttributeCompier

lib/hamlit/filter.rb

@@ -18,5 +18,9 @@ module Hamlit
     def on_haml_text(exp)
       [:haml, :text, compile(exp)]
     end
+
+    def on_haml_preserve(exp)
+      [:haml, :preserve, exp]
+    end
   end
 end

lib/hamlit/parser.rb

@@ -132,8 +132,7 @@ module Hamlit
       raise SyntaxError unless scanner.scan(/~/)
 
       code = scan_code(scanner)
-      code = "Hamlit::Helpers.find_and_preserve(#{code})"
+      escape_html([:haml, :preserve, code])
-      escape_html([:dynamic, code])
     end
 
     def parse_silent_script(scanner)