diff --git a/src/LineNumbers.js b/src/LineNumbers.js new file mode 100644 index 0000000..6ca86a7 --- /dev/null +++ b/src/LineNumbers.js @@ -0,0 +1,15 @@ +function LineNumbers(lineCount) { + let lineNumbersWrapper = ""; + if (!lineCount) { + return lineNumbersWrapper; + } + + const linesNum = lineCount; + + const lines = new Array(linesNum + 1).join(""); + lineNumbersWrapper = ``; + + return lineNumbersWrapper; +} + +module.exports = LineNumbers; diff --git a/src/markdownSyntaxHighlight.js b/src/markdownSyntaxHighlight.js index 531928d..1f9bdcf 100644 --- a/src/markdownSyntaxHighlight.js +++ b/src/markdownSyntaxHighlight.js @@ -1,6 +1,7 @@ const Prism = require("prismjs"); const PrismLoader = require("prismjs/components/index.js"); const HighlightLinesGroup = require("./HighlightLinesGroup"); +const LineNumbers = require("./LineNumbers"); module.exports = function(str, language) { if(!language) { @@ -31,5 +32,15 @@ module.exports = function(str, language) { return highlights.getLineMarkup(j, line); }); - return `
${highlightedLines.join("
")}
`; + + // Not sure how to configure this on/off + const numberLines = LineNumbers(highlightedLines.length); + const preClasses = [ + `language-${language}`, + ]; + if (numberLines !== "") { + preClasses.push("line-numbers"); + } + + return `
${highlightedLines.join("
")}${numberLines}
`; };