From b224c116fb19374ba9f970eb9bae3d8d95716e26 Mon Sep 17 00:00:00 2001
From: surunzi <surunzi@foxmail.com>
Date: Sun, 1 Jan 2017 21:14:21 +0800
Subject: [PATCH] Fix: Code line number

---
 src/Sources/Sources.es6 | 9 ++++++---
 src/Sources/code.hbs    | 4 ++--
 test/snippets.js        | 6 ++++++
 3 files changed, 14 insertions(+), 5 deletions(-)

diff --git a/src/Sources/Sources.es6 b/src/Sources/Sources.es6
index 557e846..18ef8e0 100644
--- a/src/Sources/Sources.es6
+++ b/src/Sources/Sources.es6
@@ -233,11 +233,14 @@ export default class Sources extends Tool
 
         if (this._showLineNum)
         {
-            code = code.split('\n').map(line =>
+            code = code.split('\n').map((line, idx) =>
             {
-                if (util.trim(line) === '') return '&nbsp;';
+                if (util.trim(line) === '') line = '&nbsp;';
 
-                return line;
+                return {
+                    idx: idx + 1,
+                    val: line
+                };
             });
         }
 
diff --git a/src/Sources/code.hbs b/src/Sources/code.hbs
index 3127aa9..f7a95c4 100644
--- a/src/Sources/code.hbs
+++ b/src/Sources/code.hbs
@@ -5,12 +5,12 @@
                 <tr>
                     <td class="eruda-gutter">
                         {{#each code}}
-                            <div class="eruda-line-num">{{@key}}</div>
+                            <div class="eruda-line-num">{{idx}}</div>
                         {{/each}}
                     </td>
                     <td class="eruda-content">
                         {{#each code}}
-                            <pre class="eruda-code-line">{{{.}}}</pre>
+                            <pre class="eruda-code-line">{{{val}}}</pre>
                         {{/each}}
                     </td>
                 </tr>
diff --git a/test/snippets.js b/test/snippets.js
index 8e54060..5b24cd2 100644
--- a/test/snippets.js
+++ b/test/snippets.js
@@ -7,10 +7,16 @@ describe('default', function ()
     {
         expect($tool.find('.eruda-name').eq(0)).toContainText('Border All');
     });
+
     it('refresh page', function ()
     {
         expect($tool.find('.eruda-name').eq(1)).toContainText('Refresh Page');
     });
+
+    it('search text', function ()
+    {
+        expect($tool.find('.eruda-name').eq(2)).toContainText('Search Text');
+    });
 });
 
 describe('basic', function ()