lib/hamlit/parsers/script.rb in hamlit-0.1.3 vs lib/hamlit/parsers/script.rb in hamlit-0.2.0

@@ -68,10 +68,19 @@
 
           @current_lineno += 1
           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?
         next_indent == @current_indent + 1
       end