Remove a comment from script

Fixes #3.
This commit is contained in:
Takashi Kokubun 2015-03-31 03:11:45 +09:00
parent b1458df081
commit 42477c5bdf
2 changed files with 21 additions and 1 deletions

View File

@ -70,7 +70,16 @@ module Hamlit
scanner = StringScanner.new(current_line)
code += ' '
end
code
remove_comment(code)
end
def remove_comment(code)
result = ''
Ripper.lex(code).each do |(row, col), type, str|
next if type == :on_comment
result += str
end
result
end
def has_block?

View File

@ -10,6 +10,17 @@ describe Hamlit::Engine do
HTML
end
it 'renders one-line script with comment' do
assert_render(<<-HAML, <<-HTML)
= '#' + "#" # = 3 #
= ['#',
"#"] # comment
HAML
##
["#", "#"]
HTML
end
it 'renders multi-lines script' do
assert_render(<<-HAML, <<-HTML)
%span