diff --git a/resources/code-specific-lines.js b/resources/code-specific-lines.js index 9fa6e36..27f13db 100644 --- a/resources/code-specific-lines.js +++ b/resources/code-specific-lines.js @@ -4,7 +4,7 @@ Prism.hooks.add( 'complete', function( env ) { if (showLineRanges) { var showLines = showLines( showLineRanges ); - var tokenList = tokenList( env.element.innerHTML, showLines ); + var tokenList = tokenList( env.element, showLines ); var lineNumbersActive = Prism.util.isActive( env.element, 'line-numbers' ); @@ -23,14 +23,14 @@ Prism.hooks.add( 'complete', function( env ) { showLineRanges.replace( /\s+/g, '' ).split( ',' ).filter( Boolean ).forEach( function( currentRange ) { if (currentRange.includes( '-' )) { - var range = currentRange.split( '-' ); + var range = currentRange.split( '-' ).map( Number ); if ( range[0] > range[1] ) { [range[0], range[1]] = [range[1], range[0]]; } for ( var counter = range[0]; counter <= range[1]; counter++ ) { - showLines.push( Number( counter ) ); + showLines.push( counter ); } } else { @@ -41,14 +41,25 @@ Prism.hooks.add( 'complete', function( env ) { return showLines; } - function tokenList( HTML, showLines ) { - var tokenList = HTML.split( /\n(?!$)/g ); + function tokenList( element, showLines ) { + var tokenList = element.innerHTML.split( /\n(?!$)/g ); var tokenCount = tokenList.length; - const regex = /(