diff --git a/lib/rocco.rb b/lib/rocco.rb index 963bc41..1b49f23 100644 --- a/lib/rocco.rb +++ b/lib/rocco.rb @@ -164,7 +164,7 @@ class Rocco # Combine all code blocks into a single big stream and run through either # `pygmentize(1)` or - code_stream = code_blocks.join("\n\n# DIVIDER\n\n") + code_stream = code_blocks.join("\n\n#{@options[:comment_chars]} DIVIDER\n\n") if (`which pygmentize` == "") code_html = highlight_webservice(code_stream) @@ -175,7 +175,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*# DIVIDER<\/span>\n*/m).
+      split(/\n*#{@options[:comment_chars]} DIVIDER<\/span>\n*/m).
       map { |code| code.sub(/\n?
/m, '') }.
       map { |code| code.sub(/\n?<\/pre><\/div>\n/m, '') }