hamlit 1.7.2 → 2.0.1

data/lib/hamlit/parser/haml_buffer.rb

@@ -0,0 +1,348 @@
+require 'hamlit/parser/haml_helpers'
+require 'hamlit/parser/haml_util'
+require 'hamlit/parser/haml_compiler'
+
+module Hamlit
+  # This class is used only internally. It holds the buffer of HTML that
+  # is eventually output as the resulting document.
+  # It's called from within the precompiled code,
+  # and helps reduce the amount of processing done within `instance_eval`ed code.
+  class HamlBuffer
+    ID_KEY    = 'id'.freeze
+    CLASS_KEY = 'class'.freeze
+    DATA_KEY  = 'data'.freeze
+
+    include ::Hamlit::HamlHelpers
+    include ::Hamlit::HamlUtil
+
+    # The string that holds the compiled HTML. This is aliased as
+    # `_erbout` for compatibility with ERB-specific code.
+    #
+    # @return [String]
+    attr_accessor :buffer
+
+    # The options hash passed in from {Haml::Engine}.
+    #
+    # @return [{String => Object}]
+    # @see Haml::Options#for_buffer
+    attr_accessor :options
+
+    # The {Buffer} for the enclosing Haml document.
+    # This is set for partials and similar sorts of nested templates.
+    # It's `nil` at the top level (see \{#toplevel?}).
+    #
+    # @return [Buffer]
+    attr_accessor :upper
+
+    # nil if there's no capture_haml block running,
+    # and the position at which it's beginning the capture if there is one.
+    #
+    # @return [Fixnum, nil]
+    attr_accessor :capture_position
+
+    # @return [Boolean]
+    # @see #active?
+    attr_writer :active
+
+    # @return [Boolean] Whether or not the format is XHTML
+    def xhtml?
+      not html?
+    end
+
+    # @return [Boolean] Whether or not the format is any flavor of HTML
+    def html?
+      html4? or html5?
+    end
+
+    # @return [Boolean] Whether or not the format is HTML4
+    def html4?
+      @options[:format] == :html4
+    end
+
+    # @return [Boolean] Whether or not the format is HTML5.
+    def html5?
+      @options[:format] == :html5
+    end
+
+    # @return [Boolean] Whether or not this buffer is a top-level template,
+    #   as opposed to a nested partial
+    def toplevel?
+      upper.nil?
+    end
+
+    # Whether or not this buffer is currently being used to render a Haml template.
+    # Returns `false` if a subtemplate is being rendered,
+    # even if it's a subtemplate of this buffer's template.
+    #
+    # @return [Boolean]
+    def active?
+      @active
+    end
+
+    # @return [Fixnum] The current indentation level of the document
+    def tabulation
+      @real_tabs + @tabulation
+    end
+
+    # Sets the current tabulation of the document.
+    #
+    # @param val [Fixnum] The new tabulation
+    def tabulation=(val)
+      val = val - @real_tabs
+      @tabulation = val > -1 ? val : 0
+    end
+
+    # @param upper [Buffer] The parent buffer
+    # @param options [{Symbol => Object}] An options hash.
+    #   See {Haml::Engine#options\_for\_buffer}
+    def initialize(upper = nil, options = {})
+      @active     = true
+      @upper      = upper
+      @options    = options
+      @buffer     = new_encoded_string
+      @tabulation = 0
+
+      # The number of tabs that Engine thinks we should have
+      # @real_tabs + @tabulation is the number of tabs actually output
+      @real_tabs = 0
+    end
+
+    # Appends text to the buffer, properly tabulated.
+    # Also modifies the document's indentation.
+    #
+    # @param text [String] The text to append
+    # @param tab_change [Fixnum] The number of tabs by which to increase
+    #   or decrease the document's indentation
+    # @param dont_tab_up [Boolean] If true, don't indent the first line of `text`
+    def push_text(text, tab_change, dont_tab_up)
+      if @tabulation > 0
+        # Have to push every line in by the extra user set tabulation.
+        # Don't push lines with just whitespace, though,
+        # because that screws up precompiled indentation.
+        text.gsub!(/^(?!\s+$)/m, tabs)
+        text.sub!(tabs, '') if dont_tab_up
+      end
+
+      @real_tabs += tab_change
+      @buffer << text
+    end
+
+    # Modifies the indentation of the document.
+    #
+    # @param tab_change [Fixnum] The number of tabs by which to increase
+    #   or decrease the document's indentation
+    def adjust_tabs(tab_change)
+      @real_tabs += tab_change
+    end
+
+    # the number of arguments here is insane, but passing in an options hash instead of named arguments
+    # causes a significant performance regression
+    def format_script(result, preserve_script, in_tag, preserve_tag, escape_html, nuke_inner_whitespace, interpolated, ugly)
+      result_name = escape_html ? html_escape(result.to_s) : result.to_s
+
+      if ugly
+        result = nuke_inner_whitespace ? result_name.strip : result_name
+        result = preserve(result, preserve_script, preserve_tag)
+        fix_textareas!(result) if toplevel? && result.include?('<textarea')
+        return result
+      end
+
+      # If we're interpolated,
+      # then the custom tabulation is handled in #push_text.
+      # The easiest way to avoid it here is to reset @tabulation.
+      if interpolated
+        old_tabulation = @tabulation
+        @tabulation = 0
+      end
+
+      in_tag_no_nuke = in_tag && !nuke_inner_whitespace
+      preserved_no_nuke = in_tag_no_nuke && preserve_tag
+      tabulation = !preserved_no_nuke && @real_tabs
+
+      result = nuke_inner_whitespace ? result_name.strip : result_name.rstrip
+      result = preserve(result, preserve_script, preserve_tag)
+
+      has_newline = !preserved_no_nuke && result.include?("\n")
+
+      if in_tag_no_nuke && (preserve_tag || !has_newline)
+        @real_tabs -= 1
+        @tabulation = old_tabulation if interpolated
+        return result
+      end
+
+      unless preserved_no_nuke
+        # Precompiled tabulation may be wrong
+        result = "#{tabs}#{result}" if !interpolated && !in_tag && @tabulation > 0
+
+        if has_newline
+          result.gsub! "\n", "\n#{tabs(tabulation)}"
+
+          # Add tabulation if it wasn't precompiled
+          result = "#{tabs(tabulation)}#{result}" if in_tag_no_nuke
+        end
+
+        fix_textareas!(result) if toplevel? && result.include?('<textarea')
+
+        if in_tag_no_nuke
+          result = "\n#{result}\n#{tabs(tabulation-1)}"
+          @real_tabs -= 1
+        end
+        @tabulation = old_tabulation if interpolated
+        result
+      end
+    end
+
+    def attributes(class_id, obj_ref, *attributes_hashes)
+      attributes = class_id
+      attributes_hashes.each do |old|
+        self.class.merge_attrs(attributes, Hash[old.map {|k, v| [k.to_s, v]}])
+      end
+      self.class.merge_attrs(attributes, parse_object_ref(obj_ref)) if obj_ref
+      ::Hamlit::HamlCompiler.build_attributes(
+        html?, @options[:attr_wrapper], @options[:escape_attrs], @options[:hyphenate_data_attrs], attributes)
+    end
+
+    # Remove the whitespace from the right side of the buffer string.
+    # Doesn't do anything if we're at the beginning of a capture_haml block.
+    def rstrip!
+      if capture_position.nil?
+        buffer.rstrip!
+        return
+      end
+
+      buffer << buffer.slice!(capture_position..-1).rstrip
+    end
+
+    # Merges two attribute hashes.
+    # This is the same as `to.merge!(from)`,
+    # except that it merges id, class, and data attributes.
+    #
+    # ids are concatenated with `"_"`,
+    # and classes are concatenated with `" "`.
+    # data hashes are simply merged.
+    #
+    # Destructively modifies both `to` and `from`.
+    #
+    # @param to [{String => String}] The attribute hash to merge into
+    # @param from [{String => #to_s}] The attribute hash to merge from
+    # @return [{String => String}] `to`, after being merged
+    def self.merge_attrs(to, from)
+      from[ID_KEY] = ::Hamlit::HamlCompiler.filter_and_join(from[ID_KEY], '_') if from[ID_KEY]
+      if to[ID_KEY] && from[ID_KEY]
+        to[ID_KEY] << "_#{from.delete(ID_KEY)}"
+      elsif to[ID_KEY] || from[ID_KEY]
+        from[ID_KEY] ||= to[ID_KEY]
+      end
+
+      from[CLASS_KEY] = ::Hamlit::HamlCompiler.filter_and_join(from[CLASS_KEY], ' ') if from[CLASS_KEY]
+      if to[CLASS_KEY] && from[CLASS_KEY]
+        # Make sure we don't duplicate class names
+        from[CLASS_KEY] = (from[CLASS_KEY].to_s.split(' ') | to[CLASS_KEY].split(' ')).sort.join(' ')
+      elsif to[CLASS_KEY] || from[CLASS_KEY]
+        from[CLASS_KEY] ||= to[CLASS_KEY]
+      end
+
+      from.keys.each do |key|
+        next unless from[key].kind_of?(Hash) || to[key].kind_of?(Hash)
+
+        from_data = from.delete(key)
+        # forces to_data & from_data into a hash
+        from_data = { nil => from_data } if from_data && !from_data.is_a?(Hash)
+        to[key] = { nil => to[key] } if to[key] && !to[key].is_a?(Hash)
+
+        if from_data && !to[key]
+          to[key] = from_data
+        elsif from_data && to[key]
+          to[key].merge! from_data
+        end
+      end
+
+      to.merge!(from)
+    end
+
+    private
+
+    def preserve(result, preserve_script, preserve_tag)
+      return ::Hamlit::HamlHelpers.preserve(result) if preserve_tag
+      return ::Hamlit::HamlHelpers.find_and_preserve(result, options[:preserve]) if preserve_script
+      result
+    end
+
+    # Works like #{find_and_preserve}, but allows the first newline after a
+    # preserved opening tag to remain unencoded, and then outdents the content.
+    # This change was motivated primarily by the change in Rails 3.2.3 to emit
+    # a newline after textarea helpers.
+    #
+    # @param input [String] The text to process
+    # @since Haml 4.0.1
+    # @private
+    def fix_textareas!(input)
+      pattern = /<(textarea)([^>]*)>(\n|&#x000A;)(.*?)<\/textarea>/im
+      input.gsub!(pattern) do |s|
+        match = pattern.match(s)
+        content = match[4]
+        if match[3] == '&#x000A;'
+          content.sub!(/\A /, '&#x0020;')
+        else
+          content.sub!(/\A[ ]*/, '')
+        end
+        "<#{match[1]}#{match[2]}>\n#{content}</#{match[1]}>"
+      end
+    end
+
+    def new_encoded_string
+      "".encode(Encoding.find(options[:encoding]))
+    end
+
+    @@tab_cache = {}
+    # Gets `count` tabs. Mostly for internal use.
+    def tabs(count = 0)
+      tabs = [count + @tabulation, 0].max
+      @@tab_cache[tabs] ||= '  ' * tabs
+    end
+
+    # Takes an array of objects and uses the class and id of the first
+    # one to create an attributes hash.
+    # The second object, if present, is used as a prefix,
+    # just like you can do with `dom_id()` and `dom_class()` in Rails
+    def parse_object_ref(ref)
+      prefix = ref[1]
+      ref = ref[0]
+      # Let's make sure the value isn't nil. If it is, return the default Hash.
+      return {} if ref.nil?
+      class_name =
+        if ref.respond_to?(:haml_object_ref)
+          ref.haml_object_ref
+        else
+          underscore(ref.class)
+        end
+      ref_id =
+        if ref.respond_to?(:to_key)
+          key = ref.to_key
+          key.join('_') unless key.nil?
+        else
+          ref.id
+        end
+      id = "#{class_name}_#{ref_id || 'new'}"
+      if prefix
+        class_name = "#{ prefix }_#{ class_name}"
+        id = "#{ prefix }_#{ id }"
+      end
+
+      {ID_KEY => id, CLASS_KEY => class_name}
+    end
+
+    # Changes a word from camel case to underscores.
+    # Based on the method of the same name in Rails' Inflector,
+    # but copied here so it'll run properly without Rails.
+    def underscore(camel_cased_word)
+      word = camel_cased_word.to_s.dup
+      word.gsub!(/::/, '_')
+      word.gsub!(/([A-Z]+)([A-Z][a-z])/, '\1_\2')
+      word.gsub!(/([a-z\d])([A-Z])/, '\1_\2')
+      word.tr!('-', '_')
+      word.downcase!
+      word
+    end
+  end
+end

data/lib/hamlit/parser/haml_compiler.rb

@@ -0,0 +1,553 @@
+require 'hamlit/parser/haml_util'
+require 'hamlit/parser/haml_parser'
+
+module Hamlit
+  class HamlCompiler
+    include ::Hamlit::HamlUtil
+
+    attr_accessor :options
+
+    def initialize(options)
+      @options     = options
+      @output_tabs = 0
+      @to_merge    = []
+      @precompiled = ''
+      @node        = nil
+    end
+
+    def compile(node)
+      parent, @node = @node, node
+      if node.children.empty?
+        send(:"compile_#{node.type}")
+      else
+        send(:"compile_#{node.type}") {node.children.each {|c| compile c}}
+      end
+    ensure
+      @node = parent
+    end
+
+    # The source code that is evaluated to produce the Haml document.
+    #
+    # This is automatically converted to the correct encoding
+    # (see {file:REFERENCE.md#encodings the `:encoding` option}).
+    #
+    # @return [String]
+    def precompiled
+      encoding = Encoding.find(@options.encoding)
+      return @precompiled.force_encoding(encoding) if encoding == Encoding::ASCII_8BIT
+      return @precompiled.encode(encoding)
+    end
+
+    def precompiled_with_return_value
+      "#{precompiled};#{precompiled_method_return_value}"
+    end
+
+    # Returns the precompiled string with the preamble and postamble.
+    #
+    # Initializes to ActionView::OutputBuffer when available; this is necessary
+    # to avoid ordering issues with partial layouts in Rails. If not available,
+    # initializes to nil.
+    def precompiled_with_ambles(local_names)
+      preamble = <<END.tr!("\n", ';')
+begin
+extend ::Hamlit::HamlHelpers
+_hamlout = @haml_buffer = ::Hamlit::HamlBuffer.new(haml_buffer, #{options.for_buffer.inspect})
+_erbout = _hamlout.buffer
+@output_buffer = output_buffer ||= ActionView::OutputBuffer.new rescue nil
+END
+      postamble = <<END.tr!("\n", ';')
+#{precompiled_method_return_value}
+ensure
+@haml_buffer = @haml_buffer.upper if @haml_buffer
+end
+END
+      "#{preamble}#{locals_code(local_names)}#{precompiled}#{postamble}"
+    end
+
+    private
+
+    # Returns the string used as the return value of the precompiled method.
+    # This method exists so it can be monkeypatched to return modified values.
+    def precompiled_method_return_value
+      "_erbout"
+    end
+
+    def locals_code(names)
+      names = names.keys if Hash === names
+
+      names.each_with_object('') do |name, code|
+        # Can't use || because someone might explicitly pass in false with a symbol
+        sym_local = "_haml_locals[#{inspect_obj(name.to_sym)}]"
+        str_local = "_haml_locals[#{inspect_obj(name.to_s)}]"
+        code << "#{name} = #{sym_local}.nil? ? #{str_local} : #{sym_local};"
+      end
+    end
+
+    def compile_root
+      @dont_indent_next_line = @dont_tab_up_next_text = false
+      @output_line = 1
+      yield if block_given?
+      flush_merged_text
+    end
+
+    def compile_plain
+      push_text @node.value[:text]
+    end
+
+    def nuke_inner_whitespace?(node)
+      if node.value && node.value[:nuke_inner_whitespace]
+        true
+      elsif node.parent
+        nuke_inner_whitespace?(node.parent)
+      else
+        false
+      end
+    end
+
+    def compile_script(&block)
+      push_script(@node.value[:text],
+                  :preserve_script       => @node.value[:preserve],
+                  :escape_html           => @node.value[:escape_html],
+                  :nuke_inner_whitespace => nuke_inner_whitespace?(@node),
+                  &block)
+    end
+
+    def compile_silent_script
+      return if @options.suppress_eval
+      push_silent(@node.value[:text])
+      keyword = @node.value[:keyword]
+
+      if block_given?
+        # Store these values because for conditional statements,
+        # we want to restore them for each branch
+        @node.value[:dont_indent_next_line] = @dont_indent_next_line
+        @node.value[:dont_tab_up_next_text] = @dont_tab_up_next_text
+        yield
+        push_silent("end", :can_suppress) unless @node.value[:dont_push_end]
+      elsif keyword == "end"
+        if @node.parent.children.last.equal?(@node)
+          # Since this "end" is ending the block,
+          # we don't need to generate an additional one
+          @node.parent.value[:dont_push_end] = true
+        end
+        # Don't restore dont_* for end because it isn't a conditional branch.
+      elsif ::Hamlit::HamlParser::MID_BLOCK_KEYWORDS.include?(keyword)
+        # Restore dont_* for this conditional branch
+        @dont_indent_next_line = @node.parent.value[:dont_indent_next_line]
+        @dont_tab_up_next_text = @node.parent.value[:dont_tab_up_next_text]
+      end
+    end
+
+    def compile_haml_comment; end
+
+    def compile_tag
+      t = @node.value
+
+      # Get rid of whitespace outside of the tag if we need to
+      rstrip_buffer! if t[:nuke_outer_whitespace]
+
+      dont_indent_next_line =
+        (t[:nuke_outer_whitespace] && !block_given?) ||
+        (t[:nuke_inner_whitespace] && block_given?)
+
+      if @options.suppress_eval
+        object_ref = :nil
+        parse = false
+        value = t[:parse] ? nil : t[:value]
+        attributes_hashes = {}
+        preserve_script = false
+      else
+        object_ref = t[:object_ref]
+        parse = t[:parse]
+        value = t[:value]
+        attributes_hashes = t[:attributes_hashes]
+        preserve_script = t[:preserve_script]
+      end
+
+      if @options[:trace]
+        t[:attributes].merge!({"data-trace" => @options.filename.split('/views').last << ":" << @node.line.to_s})
+      end
+
+      # Check if we can render the tag directly to text and not process it in the buffer
+      if (object_ref == :nil) && attributes_hashes.empty? && !preserve_script
+        tag_closed = !block_given? && !t[:self_closing] && !parse
+
+        open_tag = prerender_tag(t[:name], t[:self_closing], t[:attributes])
+        if tag_closed
+          open_tag << "#{value}</#{t[:name]}>"
+          open_tag << "\n" unless t[:nuke_outer_whitespace]
+        elsif !(parse || t[:nuke_inner_whitespace] ||
+            (t[:self_closing] && t[:nuke_outer_whitespace]))
+          open_tag << "\n"
+        end
+
+        push_merged_text(open_tag,
+          tag_closed || t[:self_closing] || t[:nuke_inner_whitespace] ? 0 : 1,
+          !t[:nuke_outer_whitespace])
+
+        @dont_indent_next_line = dont_indent_next_line
+        return if tag_closed
+      else
+        if attributes_hashes.empty?
+          attributes_hashes = ''
+        elsif attributes_hashes.size == 1
+          attributes_hashes = ", #{attributes_hashes.first}"
+        else
+          attributes_hashes = ", #{attributes_hashes.join(", ")}"
+        end
+
+        push_merged_text "<#{t[:name]}", 0, !t[:nuke_outer_whitespace]
+        push_generated_script(
+          "_hamlout.attributes(#{inspect_obj(t[:attributes])}, #{object_ref}#{attributes_hashes})")
+        concat_merged_text(
+          if t[:self_closing] && @options.xhtml?
+            " />#{"\n" unless t[:nuke_outer_whitespace]}"
+          else
+            ">#{"\n" unless (t[:self_closing] && @options.html?) ? t[:nuke_outer_whitespace] : (!block_given? || t[:preserve_tag] || t[:nuke_inner_whitespace])}"
+          end)
+
+        if value && !parse
+          concat_merged_text("#{value}</#{t[:name]}>#{"\n" unless t[:nuke_outer_whitespace]}")
+        elsif !t[:nuke_inner_whitespace] && !t[:self_closing]
+          @to_merge << [:text, '', 1]
+        end
+
+        @dont_indent_next_line = dont_indent_next_line
+      end
+
+      return if t[:self_closing]
+
+      if value.nil?
+        @output_tabs += 1 unless t[:nuke_inner_whitespace]
+        yield if block_given?
+        @output_tabs -= 1 unless t[:nuke_inner_whitespace]
+        rstrip_buffer! if t[:nuke_inner_whitespace]
+        push_merged_text("</#{t[:name]}>#{"\n" unless t[:nuke_outer_whitespace]}",
+          t[:nuke_inner_whitespace] ? 0 : -1, !t[:nuke_inner_whitespace])
+        @dont_indent_next_line = t[:nuke_outer_whitespace]
+        return
+      end
+
+      if parse
+        push_script(value, t.merge(:in_tag => true))
+        concat_merged_text("</#{t[:name]}>#{"\n" unless t[:nuke_outer_whitespace]}")
+      end
+    end
+
+    def compile_comment
+      condition = "#{@node.value[:conditional]}>" if @node.value[:conditional]
+      revealed = @node.value[:revealed]
+
+      open = "<!--#{condition}#{'<!-->' if revealed}"
+
+      close = "#{'<!--' if revealed}#{'<![endif]' if condition}-->"
+
+      unless block_given?
+        push_merged_text("#{open} ")
+
+        if @node.value[:parse]
+          push_script(@node.value[:text], :in_tag => true, :nuke_inner_whitespace => true)
+        else
+          push_merged_text(@node.value[:text], 0, false)
+        end
+
+        push_merged_text(" #{close}\n", 0, false)
+        return
+      end
+
+      push_text(open, 1)
+      @output_tabs += 1
+      yield if block_given?
+      @output_tabs -= 1
+      push_text(close, -1)
+    end
+
+    def compile_doctype
+      doctype = text_for_doctype
+      push_text doctype if doctype
+    end
+
+    def compile_filter
+      unless filter = Filters.defined[@node.value[:name]]
+        name = @node.value[:name]
+        if ["maruku", "textile"].include?(name)
+          raise ::Hamlit::HamlError.new(::Hamlit::HamlError.message(:install_haml_contrib, name), @node.line - 1)
+        else
+          raise ::Hamlit::HamlError.new(::Hamlit::HamlError.message(:filter_not_defined, name), @node.line - 1)
+        end
+      end
+      filter.internal_compile(self, @node.value[:text])
+    end
+
+    def text_for_doctype
+      if @node.value[:type] == "xml"
+        return nil if @options.html?
+        wrapper = @options.attr_wrapper
+        return "<?xml version=#{wrapper}1.0#{wrapper} encoding=#{wrapper}#{@node.value[:encoding] || "utf-8"}#{wrapper} ?>"
+      end
+
+      if @options.html5?
+        '<!DOCTYPE html>'
+      else
+        if @options.xhtml?
+          if @node.value[:version] == "1.1"
+            '<!DOCTYPE html PUBLIC "-//W3C//DTD XHTML 1.1//EN" "http://www.w3.org/TR/xhtml11/DTD/xhtml11.dtd">'
+          elsif @node.value[:version] == "5"
+            '<!DOCTYPE html>'
+          else
+            case @node.value[:type]
+            when "strict";   '<!DOCTYPE html PUBLIC "-//W3C//DTD XHTML 1.0 Strict//EN" "http://www.w3.org/TR/xhtml1/DTD/xhtml1-strict.dtd">'
+            when "frameset"; '<!DOCTYPE html PUBLIC "-//W3C//DTD XHTML 1.0 Frameset//EN" "http://www.w3.org/TR/xhtml1/DTD/xhtml1-frameset.dtd">'
+            when "mobile";   '<!DOCTYPE html PUBLIC "-//WAPFORUM//DTD XHTML Mobile 1.2//EN" "http://www.openmobilealliance.org/tech/DTD/xhtml-mobile12.dtd">'
+            when "rdfa";     '<!DOCTYPE html PUBLIC "-//W3C//DTD XHTML+RDFa 1.0//EN" "http://www.w3.org/MarkUp/DTD/xhtml-rdfa-1.dtd">'
+            when "basic";    '<!DOCTYPE html PUBLIC "-//W3C//DTD XHTML Basic 1.1//EN" "http://www.w3.org/TR/xhtml-basic/xhtml-basic11.dtd">'
+            else             '<!DOCTYPE html PUBLIC "-//W3C//DTD XHTML 1.0 Transitional//EN" "http://www.w3.org/TR/xhtml1/DTD/xhtml1-transitional.dtd">'
+            end
+          end
+
+        elsif @options.html4?
+          case @node.value[:type]
+          when "strict";   '<!DOCTYPE html PUBLIC "-//W3C//DTD HTML 4.01//EN" "http://www.w3.org/TR/html4/strict.dtd">'
+          when "frameset"; '<!DOCTYPE html PUBLIC "-//W3C//DTD HTML 4.01 Frameset//EN" "http://www.w3.org/TR/html4/frameset.dtd">'
+          else             '<!DOCTYPE html PUBLIC "-//W3C//DTD HTML 4.01 Transitional//EN" "http://www.w3.org/TR/html4/loose.dtd">'
+          end
+        end
+      end
+    end
+
+    # Evaluates `text` in the context of the scope object, but
+    # does not output the result.
+    def push_silent(text, can_suppress = false)
+      flush_merged_text
+      return if can_suppress && @options.suppress_eval?
+      newline = (text == "end") ? ";" : "\n"
+      @precompiled << "#{resolve_newlines}#{text}#{newline}"
+      @output_line = @output_line + text.count("\n") + newline.count("\n")
+    end
+
+    # Adds `text` to `@buffer` with appropriate tabulation
+    # without parsing it.
+    def push_merged_text(text, tab_change = 0, indent = true)
+      text = !indent || @dont_indent_next_line || @options.ugly ? text : "#{'  ' * @output_tabs}#{text}"
+      @to_merge << [:text, text, tab_change]
+      @dont_indent_next_line = false
+    end
+
+    # Concatenate `text` to `@buffer` without tabulation.
+    def concat_merged_text(text)
+      @to_merge << [:text, text, 0]
+    end
+
+    def push_text(text, tab_change = 0)
+      push_merged_text("#{text}\n", tab_change)
+    end
+
+    def flush_merged_text
+      return if @to_merge.empty?
+
+      mtabs = 0
+      @to_merge.map! do |type, val, tabs|
+        case type
+        when :text
+          mtabs += tabs
+          inspect_obj(val)[1...-1]
+        when :script
+          if mtabs != 0 && !@options.ugly
+            val = "_hamlout.adjust_tabs(#{mtabs}); " + val
+          end
+          mtabs = 0
+          "\#{#{val}}"
+        else
+          raise ::Hamlit::HamlSyntaxError.new("[HAML BUG] Undefined entry in ::Hamlit::HamlCompiler@to_merge.")
+        end
+      end
+      str = @to_merge.join
+
+      unless str.empty?
+        @precompiled <<
+          if @options.ugly
+            "_hamlout.buffer << \"#{str}\";"
+          else
+            "_hamlout.push_text(\"#{str}\", #{mtabs}, #{@dont_tab_up_next_text.inspect});"
+          end
+      end
+      @to_merge = []
+      @dont_tab_up_next_text = false
+    end
+
+    # Causes `text` to be evaluated in the context of
+    # the scope object and the result to be added to `@buffer`.
+    #
+    # If `opts[:preserve_script]` is true, Haml::Helpers#find_and_preserve is run on
+    # the result before it is added to `@buffer`
+    def push_script(text, opts = {})
+      return if @options.suppress_eval?
+
+      args = [:preserve_script, :in_tag, :preserve_tag, :escape_html, :nuke_inner_whitespace]
+      args.map! {|name| !!opts[name]}
+      args << !block_given? << @options.ugly
+
+      no_format = @options.ugly &&
+        !(opts[:preserve_script] || opts[:preserve_tag] || opts[:escape_html])
+
+      # Prerender tabulation unless we're in a tag
+      push_merged_text '' unless opts[:in_tag]
+
+      unless block_given?
+        format_script_method = "_hamlout.format_script((#{text}\n),#{args.join(',')});"
+        push_generated_script(no_format ? "#{text}\n" : format_script_method)
+        concat_merged_text("\n") unless opts[:in_tag] || opts[:nuke_inner_whitespace]
+        return
+      end
+
+      flush_merged_text
+      push_silent "haml_temp = #{text}"
+      yield
+      push_silent('end', :can_suppress) unless @node.value[:dont_push_end]
+      format_script_method = "_hamlout.format_script(haml_temp,#{args.join(',')});"
+      @precompiled << "_hamlout.buffer << #{no_format ? "haml_temp.to_s;" : format_script_method}"
+      concat_merged_text("\n") unless opts[:in_tag] || opts[:nuke_inner_whitespace] || @options.ugly
+    end
+
+    def push_generated_script(text)
+      @to_merge << [:script, resolve_newlines + text]
+      @output_line += text.count("\n")
+    end
+
+    # This is a class method so it can be accessed from Buffer.
+    def self.build_attributes(is_html, attr_wrapper, escape_attrs, hyphenate_data_attrs, attributes = {})
+        # @TODO this is an absolutely ridiculous amount of arguments. At least
+      # some of this needs to be moved into an instance method.
+      quote_escape     = attr_wrapper == '"' ? "&#x0022;" : "&#x0027;"
+      other_quote_char = attr_wrapper == '"' ? "'" : '"'
+      join_char        = hyphenate_data_attrs ? '-' : '_'
+
+      attributes.each do |key, value|
+        if value.is_a?(Hash)
+          data_attributes = attributes.delete(key)
+          data_attributes = flatten_data_attributes(data_attributes, '', join_char)
+          data_attributes = build_data_keys(data_attributes, hyphenate_data_attrs, key)
+          attributes = data_attributes.merge(attributes)
+        end
+      end
+
+      result = attributes.collect do |attr, value|
+        next if value.nil?
+
+        value = filter_and_join(value, ' ') if attr == 'class'
+        value = filter_and_join(value, '_') if attr == 'id'
+
+        if value == true
+          next " #{attr}" if is_html
+          next " #{attr}=#{attr_wrapper}#{attr}#{attr_wrapper}"
+        elsif value == false
+          next
+        end
+
+        escaped =
+          if escape_attrs == :once
+            ::Hamlit::HamlHelpers.escape_once(value.to_s)
+          elsif escape_attrs
+            ::Hamlit::HamlHelpers.html_escape(value.to_s)
+          else
+            value.to_s
+          end
+        value = ::Hamlit::HamlHelpers.preserve(escaped)
+        if escape_attrs
+          # We want to decide whether or not to escape quotes
+          value.gsub!(/&quot;|&#x0022;/, '"')
+          this_attr_wrapper = attr_wrapper
+          if value.include? attr_wrapper
+            if value.include? other_quote_char
+              value.gsub!(attr_wrapper, quote_escape)
+            else
+              this_attr_wrapper = other_quote_char
+            end
+          end
+        else
+          this_attr_wrapper = attr_wrapper
+        end
+        " #{attr}=#{this_attr_wrapper}#{value}#{this_attr_wrapper}"
+      end
+      result.compact!
+      result.sort!
+      result.join
+    end
+
+    def self.filter_and_join(value, separator)
+      return '' if (value.respond_to?(:empty?) && value.empty?)
+
+      if value.is_a?(Array)
+        value.flatten!
+        value.map! {|item| item ? item.to_s : nil}
+        value.compact!
+        value = value.join(separator)
+      else
+        value = value ? value.to_s : nil
+      end
+      !value.nil? && !value.empty? && value
+    end
+
+    def self.build_data_keys(data_hash, hyphenate, attr_name="data")
+      Hash[data_hash.map do |name, value|
+        if name == nil
+          [attr_name, value]
+        elsif hyphenate
+          ["#{attr_name}-#{name.to_s.tr('_', '-')}", value]
+        else
+          ["#{attr_name}-#{name}", value]
+        end
+      end]
+    end
+
+    def self.flatten_data_attributes(data, key, join_char, seen = [])
+      return {key => data} unless data.is_a?(Hash)
+
+      return {key => nil} if seen.include? data.object_id
+      seen << data.object_id
+
+      data.sort {|x, y| x[0].to_s <=> y[0].to_s}.inject({}) do |hash, (k, v)|
+        joined = key == '' ? k : [key, k].join(join_char)
+        hash.merge! flatten_data_attributes(v, joined, join_char, seen)
+      end
+    end
+
+    def prerender_tag(name, self_close, attributes)
+      attributes_string = ::Hamlit::HamlCompiler.build_attributes(
+        @options.html?, @options.attr_wrapper, @options.escape_attrs, @options.hyphenate_data_attrs, attributes)
+      "<#{name}#{attributes_string}#{self_close && @options.xhtml? ? ' /' : ''}>"
+    end
+
+    def resolve_newlines
+      diff = @node.line - @output_line
+      return "" if diff <= 0
+      @output_line = @node.line
+      "\n" * diff
+    end
+
+    # Get rid of and whitespace at the end of the buffer
+    # or the merged text
+    def rstrip_buffer!(index = -1)
+      last = @to_merge[index]
+      if last.nil?
+        push_silent("_hamlout.rstrip!", false)
+        @dont_tab_up_next_text = true
+        return
+      end
+
+      case last.first
+      when :text
+        last[1].rstrip!
+        if last[1].empty?
+          @to_merge.slice! index
+          rstrip_buffer! index
+        end
+      when :script
+        last[1].gsub!(/\(haml_temp, (.*?)\);$/, '(haml_temp.rstrip, \1);')
+        rstrip_buffer! index - 1
+      else
+        raise ::Hamlit::HamlSyntaxError.new("[HAML BUG] Undefined entry in ::Hamlit::HamlCompiler@to_merge.")
+      end
+    end
+  end
+end