Carve out DoctypeCompiler

lib/hamlit/compiler.rb

@@ -1,9 +1,11 @@
 require 'hamlit/tag_compiler'
+require 'hamlit/doctype_compiler'
 
 module Hamlit
   class Compiler
     def initialize(options = {})
-      @tag_compiler = TagCompiler.new(options[:attr_quote])
+      @doctype_compiler = DoctypeCompiler.new(options[:format])
+      @tag_compiler     = TagCompiler.new(options[:attr_quote])
     end
 
     def call(ast)
@@ -37,7 +39,7 @@ module Hamlit
     end
 
     def compile_doctype(node)
-      [:html, :doctype, node.value[:type]]
+      @doctype_compiler.compile(node)
     end
 
     def compile_tag(node)

lib/hamlit/doctype_compiler.rb

@@ -0,0 +1,11 @@
+module Hamlit
+  class DoctypeCompiler
+    def initialize(format)
+      @format = format
+    end
+
+    def compile(node)
+      [:html, :doctype, node.value[:type]]
+    end
+  end
+end