add options #18

Author: wcoder

README.md

@@ -1,4 +1,4 @@
-# highlightjs-line-numbers.js [![version](http://img.shields.io/badge/release-v2.0.0-brightgreen.svg?style=flat)](https://github.com/wcoder/highlightjs-line-numbers.js/archive/master.zip)
+# highlightjs-line-numbers.js [![version](http://img.shields.io/badge/release-v2.1.0-brightgreen.svg?style=flat)](https://github.com/wcoder/highlightjs-line-numbers.js/archive/master.zip)
 
 Highlight.js line numbers plugin.
 
@@ -67,5 +67,25 @@ td.hljs-ln-code {
 }
 ```
 
+## Options
+
+After version 2.1 plugin has optional parameter `options` - for custom setup.
+
+name | type | default value | description
+-----|------|---------------|------------
+singleLine | boolean | false | enable plugin for code block with one line
+
+#### Examples of using
+
+```js
+hljs.initLineNumbersOnLoad({
+    singleLine: true
+});
+```
+
+```js
+hljs.lineNumbersBlock(myCodeBlock, myOptions);
+```
+
 ---
 © 2017 Yauheni Pakala | MIT License

dist/highlightjs-line-numbers.min.js

@@ -6,7 +6,7 @@
 	    CODE_BLOCK_NAME = 'hljs-ln-code',
 	    NUMBERS_BLOCK_NAME = 'hljs-ln-numbers',
 	    NUMBER_LINE_NAME = 'hljs-ln-n',
-	    DATA_ATTR_NAME = 'data-line-number';
+		DATA_ATTR_NAME = 'data-line-number';
 
 	// https://wcoder.github.io/notes/string-format-for-string-formating-in-javascript
 	String.prototype.format = String.prototype.f = function () {
@@ -34,34 +34,44 @@
 		document.getElementsByTagName('head')[0].appendChild(css);
 	}
 
-	function initLineNumbersOnLoad () {
+	function initLineNumbersOnLoad (options) {
 		if (document.readyState === 'complete') {
-			documentReady();
+			documentReady(options);
 		} else {
-			w.addEventListener('DOMContentLoaded', documentReady);
+			w.addEventListener('DOMContentLoaded', function () {
+				documentReady(options);
+			});
 		}
 	}
 
-	function documentReady () {
+	function documentReady (options) {
 		try {
 			var blocks = document.querySelectorAll('code.hljs');
 
 			for (var i in blocks) {
 				if (blocks.hasOwnProperty(i)) {
-					lineNumbersBlock(blocks[i]);
+					lineNumbersBlock(blocks[i], options);
 				}
 			}
 		} catch (e) {
 			console.error('LineNumbers error: ', e);
 		}
 	}
 
-	function lineNumbersBlock (element) {
+	function lineNumbersBlock (element, options) {
 		if (typeof element !== 'object') return;
 
+		// define options or set default
+		options = options || {
+			singleLine: false
+		};
+
+		// convert options
+		var firstLineIndex = !!options.singleLine ? 0 : 1;
+
 		var lines = getLines(element.innerHTML);
 
-		if (lines.length > 1) {
+		if (lines.length > firstLineIndex) {
 			var html = '';
 
 			for (var i = 0; i < lines.length; i++) {