52 lines
1.4 KiB
JavaScript
52 lines
1.4 KiB
JavaScript
// Modified from https://github.com/egoist/markdown-it-highlight-lines
|
|
|
|
const RE = /{([\d,-]+)}/
|
|
const wrapperRE = /^<pre .*?><code>/
|
|
|
|
module.exports = md => {
|
|
const fence = md.renderer.rules.fence
|
|
md.renderer.rules.fence = (...args) => {
|
|
const [tokens, idx, options] = args
|
|
const token = tokens[idx]
|
|
|
|
if (!token.lineNumbers) {
|
|
const rawInfo = token.info
|
|
if (!rawInfo || !RE.test(rawInfo)) {
|
|
return fence(...args)
|
|
}
|
|
|
|
const langName = rawInfo.replace(RE, '').trim()
|
|
// ensure the next plugin get the correct lang.
|
|
token.info = langName
|
|
|
|
token.lineNumbers = RE.exec(rawInfo)[1]
|
|
.split(',')
|
|
.map(v => v.split('-').map(v => parseInt(v, 10)))
|
|
}
|
|
|
|
const code = options.highlight
|
|
? options.highlight(token.content, token.info)
|
|
: token.content
|
|
|
|
const rawCode = code.replace(wrapperRE, '')
|
|
const highlightLinesCode = rawCode.split('\n').map((split, index) => {
|
|
const lineNumber = index + 1
|
|
const inRange = token.lineNumbers.some(([start, end]) => {
|
|
if (start && end) {
|
|
return lineNumber >= start && lineNumber <= end
|
|
}
|
|
return lineNumber === start
|
|
})
|
|
if (inRange) {
|
|
return `<div class="highlighted"> </div>`
|
|
}
|
|
return '<br>'
|
|
}).join('')
|
|
|
|
const highlightLinesWrapperCode
|
|
= `<div class="highlight-lines">${highlightLinesCode}</div>`
|
|
|
|
return highlightLinesWrapperCode + code
|
|
}
|
|
}
|