diff --git a/lib/rocco.rb b/lib/rocco.rb index 9e2a0eb..655d6c4 100644 --- a/lib/rocco.rb +++ b/lib/rocco.rb @@ -85,7 +85,7 @@ class Rocco } @options = defaults.merge(options) @sources = sources - @comment_pattern = Regexp.new("^\\s*#{@options[:comment_chars]}") + @comment_pattern = Regexp.new("^\\s*#{@options[:comment_chars]}\s?") @sections = highlight(split(parse(@data))) end @@ -176,7 +176,7 @@ class Rocco # Do some post-processing on the pygments output to split things back # into sections and remove partial `
` blocks.
code_html = code_html.
- split(/\n*#{@options[:comment_chars]} DIVIDER<\/span>\n*/m).
+ split(/\n*#{@options[:comment_chars]} DIVIDER<\/span>\n*/m).
map { |code| code.sub(/\n?