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}
`;
};