From 5f84ec711b37a1fbe3d6b86f7d01cb2833283c46 Mon Sep 17 00:00:00 2001
From: Kevin Sawicki <kevin@github.com>
Date: Tue, 12 Feb 2013 13:25:20 -0800
Subject: [PATCH] Add class to hard tabs and leading/trailing spaces

This allows whitespace to be still be styled when invisibles
are disabled.
---
 spec/app/editor-spec.coffee | 22 +++++++++++++++++++++-
 src/app/token.coffee        | 11 +++++++++++
 2 files changed, 32 insertions(+), 1 deletion(-)

diff --git a/spec/app/editor-spec.coffee b/spec/app/editor-spec.coffee
index 30c8a0ee2..11f46ed8e 100644
--- a/spec/app/editor-spec.coffee
+++ b/spec/app/editor-spec.coffee
@@ -1270,7 +1270,27 @@ describe "Editor", ->
         expect(span0.children('span:eq(2)').text()).toBe "{"
 
         line12 = editor.renderedLines.find('.line:eq(11)')
-        expect(line12.find('span:eq(1)')).toMatchSelector '.keyword'
+        expect(line12.find('span:eq(2)')).toMatchSelector '.keyword'
+
+      it "wraps hard tabs in a span", ->
+        editor.setText('\t<- hard tab')
+        line0 = editor.renderedLines.find('.line:first')
+        span0_0 = line0.children('span:eq(0)').children('span:eq(0)')
+        expect(span0_0).toMatchSelector '.hard-tab'
+        expect(span0_0.text()).toBe ' '
+
+      it "wraps leading whitespace in a span", ->
+        line1 = editor.renderedLines.find('.line:eq(1)')
+        span0_0 = line1.children('span:eq(0)').children('span:eq(0)')
+        expect(span0_0).toMatchSelector '.leading-whitespace'
+        expect(span0_0.text()).toBe '  '
+
+      it "wraps trailing whitespace in a span", ->
+        editor.setText('trailing whitespace ->   ')
+        line0 = editor.renderedLines.find('.line:first')
+        span0_last = line0.children('span:eq(0)').children('span:last')
+        expect(span0_last).toMatchSelector '.trailing-whitespace'
+        expect(span0_last.text()).toBe '   '
 
       describe "when lines are updated in the buffer", ->
         it "syntax highlights the updated lines", ->
diff --git a/src/app/token.coffee b/src/app/token.coffee
index 657e6658b..5a7c677ed 100644
--- a/src/app/token.coffee
+++ b/src/app/token.coffee
@@ -80,5 +80,16 @@ class Token
         if hasTrailingWhitespace
           html = html.replace /[ ]+$/, (match) ->
             "<span class='invisible trailing-whitespace'>#{match.replace(/./g, invisibles.space)}</span>"
+    else
+      if @isHardTab
+        html = html.replace /^./, (match) ->
+          "<span class='hard-tab'>#{match}</span>"
+      else
+        if hasLeadingWhitespace
+          html = html.replace /^[ ]+/, (match) ->
+            "<span class='leading-whitespace'>#{match}</span>"
+        if hasTrailingWhitespace
+          html = html.replace /[ ]+$/, (match) ->
+            "<span class='trailing-whitespace'>#{match}</span>"
 
     html