1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155
|
require 'cgi'
require 'erubis'
require 'ruby_parser'
module Html2haml
class HTML
# A class for converting ERB code into a format that's easier
# for the {Html2haml::HTML} Nokogiri-based parser to understand.
#
# Uses [Erubis](http://www.kuwata-lab.com/erubis)'s extensible parsing powers
# to parse the ERB in a reliable way,
# and [ruby_parser](http://parsetree.rubyforge.org/)'s Ruby knowledge
# to figure out whether a given chunk of Ruby code starts a block or not.
#
# The ERB tags are converted to HTML tags in the following way.
# `<% ... %>` is converted into `<haml_silent> ... </haml_silent>`.
# `<%= ... %>` is converted into `<haml_loud> ... </haml_loud>`.
# `<%== ... %>` is converted into `<haml_loud raw=raw> ... </haml_loud>`.
# Finally, if either of these opens a Ruby block,
# `<haml_block> ... </haml_block>` will wrap the entire contents of the block -
# that is, everything that should be indented beneath the previous silent or loud tag.
class ERB < Erubis::Basic::Engine
# Compiles an ERB template into a HTML document containing `haml_*` tags.
#
# @param template [String] The ERB template
# @return [String] The output document
# @see Html2haml::HTML::ERB
def self.compile(template)
new(template).src
end
# The ERB-to-Hamlized-HTML conversion has no preamble.
def add_preamble(src); end
# The ERB-to-Hamlized-HTML conversion has no postamble.
def add_postamble(src); end
# Concatenates the text onto the source buffer.
#
# @param src [String] The source buffer
# @param text [String] The raw text to add to the buffer
def add_text(src, text)
src << text
end
# Concatenates a silent Ruby statement onto the source buffer.
# This uses the `<haml_silent>` tag,
# and may close and/or open a Ruby block with the `<haml_block>` tag.
#
# In particular, a block is closed if this statement is some form of `end`,
# opened if it's a block opener like `do`, `if`, or `begin`,
# and both closed and opened if it's a mid-block keyword
# like `else` or `when`.
#
# @param src [String] The source buffer
# @param code [String] The Ruby statement to add to the buffer
def add_stmt(src, code)
src << '</haml_block>' if block_closer?(code) || mid_block?(code)
src << '<haml_silent>' << h(code) << '</haml_silent>' unless code.strip == "end"
src << '<haml_block>' if block_opener?(code) || mid_block?(code)
end
# Concatenates a Ruby expression that's printed to the document
# onto the source buffer.
# This uses the `<haml:silent>` tag,
# and may open a Ruby block with the `<haml:block>` tag.
# An expression never closes a block.
#
# @param src [String] The source buffer
# @param code [String] The Ruby expression to add to the buffer
def add_expr_literal(src, code)
src << '<haml_loud>' << h(code) << '</haml_loud>'
src << '<haml_block>' if block_opener?(code)
end
def add_expr_escaped(src, code)
src << "<haml_loud raw=raw>" << h(code) << "</haml_loud>"
end
# `html2haml` doesn't support debugging expressions.
def add_expr_debug(src, code)
raise Haml::Error.new("html2haml doesn't support debugging expressions.")
end
private
# HTML-escaped some text (in practice, always Ruby code).
# A utility method.
#
# @param text [String] The text to escape
# @return [String] The escaped text
def h(text)
CGI.escapeHTML(text)
end
# Returns whether the code is valid Ruby code on its own.
#
# @param code [String] Ruby code to check
# @return [Boolean]
def valid_ruby?(code)
RubyParser.for_current_ruby.parse(code)
rescue Racc::ParseError, RubyParser::SyntaxError
false
end
# Returns whether the code has any content
# This is used to test whether lines have been removed by erubis, such as comments
#
# @param code [String] Ruby code to check
# @return [Boolean]
def has_code?(code)
return false if code == "\n"
return false if valid_ruby?(code) == nil
true
end
# Checks if a string of Ruby code opens a block.
# This could either be something like `foo do |a|`
# or a keyword that requires a matching `end`
# like `if`, `begin`, or `case`.
#
# @param code [String] Ruby code to check
# @return [Boolean]
def block_opener?(code)
return unless has_code?(code)
valid_ruby?(code + "\nend") ||
valid_ruby?(code + "\nwhen foo\nend")
end
# Checks if a string of Ruby code closes a block.
# This is always `end` followed optionally by some method calls.
#
# @param code [String] Ruby code to check
# @return [Boolean]
def block_closer?(code)
return unless has_code?(code)
valid_ruby?("begin\n" + code)
end
# Checks if a string of Ruby code comes in the middle of a block.
# This could be a keyword like `else`, `rescue`, or `when`,
# or even `end` with a method call that takes a block.
#
# @param code [String] Ruby code to check
# @return [Boolean]
def mid_block?(code)
return unless has_code?(code)
return if valid_ruby?(code)
valid_ruby?("if foo\n#{code}\nend") || # else, elsif
valid_ruby?("begin\n#{code}\nend") || # rescue, ensure
valid_ruby?("case foo\n#{code}\nend") # when
end
end
end
end
|