Revert "Don't soft-wrap on indentation"

src/tokenized-line.coffee

@@ -11,7 +11,6 @@ module.exports =
 class TokenizedLine
   endOfLineInvisibles: null
   lineIsWhitespaceOnly: false
-  firstNonWhitespaceIndex: 0
   foldable: false
 
   constructor: ({tokens, @lineEnding, @ruleStack, @startBufferColumn, @fold, @tabLength, @indentLevel, @invisibles}) ->
@@ -97,7 +96,6 @@ class TokenizedLine
   # Returns a {Number} representing the `line` position where the wrap would take place.
   # Returns `null` if a wrap wouldn't occur.
   findWrapColumn: (maxColumn) ->
-    return unless maxColumn?
     return unless @text.length > maxColumn
 
     if /\s/.test(@text[maxColumn])
@@ -108,7 +106,7 @@ class TokenizedLine
       return @text.length
     else
       # search backward for the start of the word on the boundary
-      for column in [maxColumn..@firstNonWhitespaceIndex]
+      for column in [maxColumn..0] when @isColumnOutsideSoftWrapIndentation(column)
         return column + 1 if /\s/.test(@text[column])
 
       return maxColumn
@@ -129,13 +127,12 @@ class TokenizedLine
 
     rightTokens = new Array(@tokens...)
     leftTokens = []
-    leftScreenColumn = 0
-
-    while leftScreenColumn < column
-      if leftScreenColumn + rightTokens[0].screenDelta > column
-        rightTokens[0..0] = rightTokens[0].splitAt(column - leftScreenColumn)
+    leftTextLength = 0
+    while leftTextLength < column
+      if leftTextLength + rightTokens[0].value.length > column
+        rightTokens[0..0] = rightTokens[0].splitAt(column - leftTextLength)
       nextToken = rightTokens.shift()
-      leftScreenColumn += nextToken.screenDelta
+      leftTextLength += nextToken.value.length
       leftTokens.push nextToken
 
     indentationTokens = @buildSoftWrapIndentationTokens(leftTokens[0], hangingIndent)
@@ -163,6 +160,11 @@ class TokenizedLine
   isSoftWrapped: ->
     @lineEnding is null
 
+  isColumnOutsideSoftWrapIndentation: (column) ->
+    return true if @softWrapIndentationTokens.length == 0
+
+    column > @softWrapIndentationDelta
+
   isColumnInsideSoftWrapIndentation: (column) ->
     return false if @softWrapIndentationTokens.length == 0
 
@@ -207,15 +209,15 @@ class TokenizedLine
     outputTokens
 
   markLeadingAndTrailingWhitespaceTokens: ->
-    @firstNonWhitespaceIndex = @text.search(NonWhitespaceRegex)
-    if @firstNonWhitespaceIndex > 0 and isPairedCharacter(@text, @firstNonWhitespaceIndex - 1)
-      @firstNonWhitespaceIndex--
+    firstNonWhitespaceIndex = @text.search(NonWhitespaceRegex)
+    if firstNonWhitespaceIndex > 0 and isPairedCharacter(@text, firstNonWhitespaceIndex - 1)
+      firstNonWhitespaceIndex--
     firstTrailingWhitespaceIndex = @text.search(TrailingWhitespaceRegex)
     @lineIsWhitespaceOnly = firstTrailingWhitespaceIndex is 0
     index = 0
     for token in @tokens
-      if index < @firstNonWhitespaceIndex
-        token.firstNonWhitespaceIndex = Math.min(index + token.value.length, @firstNonWhitespaceIndex - index)
+      if index < firstNonWhitespaceIndex
+        token.firstNonWhitespaceIndex = Math.min(index + token.value.length, firstNonWhitespaceIndex - index)
       # Only the *last* segment of a soft-wrapped line can have trailing whitespace
       if @lineEnding? and (index + token.value.length > firstTrailingWhitespaceIndex)
         token.firstTrailingWhitespaceIndex = Math.max(0, firstTrailingWhitespaceIndex - index)