haml--haml/lib/haml/html.rb

require File.dirname(__FILE__) + '/../haml'

require 'haml/engine'
require 'rubygems'
require 'hpricot'

module Haml
  # This class contains the functionality used in the +html2haml+ utility,
  # namely converting HTML documents to Haml templates.
  # It depends on Hpricot for HTML parsing (http://code.whytheluckystiff.net/hpricot/).
  class HTML
    # Creates a new instance of Haml::HTML that will compile the given template,
    # which can either be a string containing HTML or an Hpricot node,
    # to a Haml string when +render+ is called.
    def initialize(template)
      if template.is_a? Hpricot::Node
        @template = template
      else
        @template = Hpricot(template)
      end
    end

    # Processes the document and returns the result as a string
    # containing the Haml template.
    def render
      @template.to_haml(0)
    end
    alias_method :to_haml, :render

    module ::Hpricot::Node
      # Returns the Haml representation of the given node,
      # at the given tabulation.
      def to_haml(tabs = 0)
        parse_text(self.to_s, tabs)
      end

      private

      def tabulate(tabs)
        '  ' * tabs
      end

      def parse_text(text, tabs)
        text.strip!
        if text.empty?
          String.new
        else
          lines = text.split("\n")
          
          lines.map do |line|
            line.strip!
            "#{tabulate(tabs)}#{'\\' if Haml::Engine::SPECIAL_CHARACTERS.include?(line[0])}#{line}\n"
          end.join
        end
      end
    end

    # :stopdoc:
    
    TEXT_REGEXP = /^(\s*).*$/

    class ::Hpricot::Doc
      def to_haml(tabs = 0)
        output = ''
        children.each { |child| output += child.to_haml(0) }
        output
      end
    end

    class ::Hpricot::XMLDecl
      def to_haml(tabs = 0)
        "#{tabulate(tabs)}!!! XML\n"
      end
    end

    # TODO: Understand STRICT, etc. doctypes
    class ::Hpricot::DocType
      def to_haml(tabs = 0)
        "#{tabulate(tabs)}!!!\n"
      end
    end

    class ::Hpricot::Comment
      def to_haml(tabs = 0)
        "#{tabulate(tabs)}/\n#{parse_text(self.content, tabs + 1)}"
      end
    end

    class ::Hpricot::Elem
      def to_haml(tabs = 0)
        output = "#{tabulate(tabs)}"
        output += "%#{name}" unless name == 'div' && (attributes.include?('id') || attributes.include?('class'))
        
        if attributes
          output += "##{attributes['id']}" if attributes['id']
          attributes['class'].split(' ').each { |c| output += ".#{c}" } if attributes['class']
          attributes.delete("id")
          attributes.delete("class")
          output += attributes.inspect if attributes.length > 0
        end
        
        output += "/" if children.length == 0
        output += "\n"
        
        self.children.each do |child|
          output += child.to_haml(tabs + 1)
        end

        output
      end
    end
    # :startdoc:
  end
end