lib/rouge/formatters/html.rb in rouge-0.2.7 vs lib/rouge/formatters/html.rb in rouge-0.2.8

@@ -38,9 +38,12 @@
         tokens.each do |tok, val|
           num_lines += val.scan(/\n/).size
           span(tok, val) { |str| code << str }
         end
 
+        # add an extra line number for non-newline-terminated strings
+        num_lines += 1 if code[-1] != "\n"
+
         # generate a string of newline-separated line numbers for the gutter
         numbers = num_lines.times.map do |x|
           %<<div class="lineno">#{x+1}</div>>
         end.join