include indentation and find elements more nicely

Author: justinbmeyer

index.js

@@ -85,14 +85,23 @@ var getConfig = function(lineString, lineCount) {
 	};
 };
 
-function findPreviousSibling(el, tag) {
+function findPreviousSibling(start, tag) {
 	tag = tag.toUpperCase();
 
-	while (el = el.previousSibling) {
-		if (el.tagName && el.tagName.toUpperCase() === tag) {
-			return el;
-		}
-	}
+	while(start) {
+        if(start.nodeName === tag) {
+            return start;
+        }
+        if(start.querySelector) {
+            var pre = start.querySelector(tag);
+            if(pre) {
+                return pre;
+            }
+        }
+
+        // needs to be previousSibling for zombie
+        start = start.previousSibling;
+    }
 }
 
 module.exports = function() {

tags.js

@@ -1,16 +1,16 @@
 /**
  * @parent bit-docs-html-highlight-line/tags
  * @module {bit-docs-process-tags/types/tag} bit-docs-html-highlight-line/tags/highlight @highlight
- * 
+ *
  * Highlight the specified code lines.
- * 
+ *
  * @signature `@highlight LINES[,ONLY]`
- * 
+ *
  * @param {String} LINES The lines to highlight like `2-4`.
- * 
+ *
  * @param {String} [,ONLY] Collapse non-highlighted lines greater than three
  * lines away (useful for long code snippets).
- * 
+ *
  * @codestart javascript
  * /**
  *  * ```js
@@ -24,27 +24,28 @@
  *  *   "season": "Summer",
  *  *   "awesome": "yes"
  *  * }
- *  * ``` 
+ *  * ```
  *  *
  *  * @highlight 2-4,only
  *  *|
  * @codeend
- * 
+ *
  * Injects a `<span line-highlight="2-4,only"></span>` element to the page that
  * will be picked up and used by the static front-end script
  * [bit-docs-html-highlight-line/highlight-line.js].
  */
 exports.highlight = {
     add: function(line, curData) {
+        var space = line.substr(0, line.indexOf("@highlight"));
         var lines = line.replace("@highlight","").trim();
-        var html = "<span line-highlight='"+lines+"'></span>";
+        var html = space+"<span line-highlight='"+lines+"'></span>";
         var validCurData =  (curData && curData.length !== 2);
         var useCurData = validCurData && (typeof curData.description === "string") && !curData.body;
 
         if(useCurData) {
-            curData.description += html;
+            curData.description += "\n"+html+"\n";
         } else {
-            this.body += html;
+            this.body += html+"\n";
         }
     }
 };