preserve scripts line number

rhyzx · rhyzx · commit 9f702fe192aa · 2015-11-28T14:13:26.000+07:00
diff --git a/lib/parser.js b/lib/parser.js
@@ -70,16 +70,38 @@ module.exports = function (content) {
 
     var start = node.childNodes[0].__location.start
     var end = node.childNodes[node.childNodes.length - 1].__location.end
+
+    var result
+    if (type === 'script') {
+      result = commentScript(content.slice(0, start)) +
+              content.slice(start, end) +
+              commentScript(content.slice(end))
+    } else {
+      result = content.slice(start, end).trim()
+    }
+
     output[type].push({
       lang: lang,
       scoped: scoped,
-      content: content.substring(start, end).trim()
+      content: result
     })
   })
 
   cb(null, 'module.exports = ' + JSON.stringify(output))
 }
 
+function commentScript (content) {
+  return content.split(/\n\r|\n|\r/g)
+  .map(function (line) {
+    if (line.trim() === '') {
+      return line
+    } else {
+      return '// ' + line
+    }
+  })
+  .join('\n')
+}
+
 function getAttribute (node, name) {
   if (node.attrs) {
     var i = node.attrs.length
