fix(highlight&prism): add 'caption' class

Author: curbengh

lib/highlight.js

@@ -45,7 +45,7 @@ function highlightUtil(str, options = {}) {
   let codeCaption = '';
 
   if (caption) {
-    codeCaption = wrap ? `<figcaption>${caption}</figcaption>` : `<div>${caption}</div>`;
+    codeCaption = wrap ? `<figcaption>${caption}</figcaption>` : `<div class="caption">${caption}</div>`;
   }
 
   if (!wrap) {

lib/prism.js

@@ -98,7 +98,7 @@ function PrismUtil(str, options = {}) {
 
   if (tab) str = replaceTabs(str, tab);
 
-  const codeCaption = caption ? `<div>${caption}</div>` : '';
+  const codeCaption = caption ? `<div class="caption">${caption}</div>` : '';
 
   const startTag = `<pre class="${preTagClassArr.join(' ')}"${preTagAttr}>${codeCaption}<code class="language-${language}">`;
   const endTag = '</code></pre>';

test/highlight.spec.js

@@ -218,7 +218,7 @@ describe('highlight', () => {
 
     result.should.eql([
       '<pre>',
-      `<div>${caption}</div>`,
+      `<div class="caption">${caption}</div>`,
       '<code class="highlight plain">',
       entities.encode(testString),
       '</code></pre>'

test/prism.spec.js

@@ -330,7 +330,7 @@ describe('prismHighlight', () => {
     const caption = 'foo';
     const result = prismHighlight(input, { caption });
 
-    result.should.contains('<div>' + caption + '</div>');
+    result.should.contains('<div class="caption">' + caption + '</div>');
 
     validateHtmlAsync(result, done);
   });