lib/lucid/ast/comment.rb in lucid-0.0.6 vs lib/lucid/ast/comment.rb in lucid-0.0.7

@@ -9,11 +9,15 @@
         @value.nil? || @value == ""
       end
 
       def accept(visitor)
         return if Lucid.wants_to_quit
-        @value.strip.split("\n").each do |line|
-          visitor.visit_comment_line(line.strip)
+        return if empty?
+
+        visitor.visit_comment(self) do
+          @value.strip.split("\n").each do |line|
+            visitor.visit_comment_line(line.strip)
+          end
         end
       end
 
       def to_sexp
         (@value.nil? || @value == '') ? nil : [:comment, @value]