Tweak microsoft#1

Author: cspotcode

Diff for: src/services/formatting/formatting.ts

@@ -401,6 +401,7 @@ namespace ts.formatting {
         let previousRange: TextRangeWithKind;
         let previousParent: Node;
         let previousRangeStartLine: number;
+        let previousRangeEndLine: number;
 
         let lastIndentedLine: number;
         let indentationOnLastIndentedLine: number;
@@ -794,11 +795,12 @@ namespace ts.formatting {
                 const isTokenInRange = rangeContainsRange(originalRange, currentTokenInfo.token);
 
                 const tokenStart = sourceFile.getLineAndCharacterOfPosition(currentTokenInfo.token.pos);
+                const tokenEnd = sourceFile.getLineAndCharacterOfPosition(currentTokenInfo.token.end);
                 if (isTokenInRange) {
                     const rangeHasError = rangeContainsError(currentTokenInfo.token);
                     // save previousRange since processRange will overwrite this value with current one
                     const savePreviousRange = previousRange;
-                    lineAdded = processRange(currentTokenInfo.token, tokenStart, parent, childContextNode, dynamicIndentation);
+                    lineAdded = processRange(currentTokenInfo.token, tokenStart, tokenEnd, parent, childContextNode, dynamicIndentation);
                     if (rangeHasError) {
                         // do not indent comments\token if token range overlaps with some error
                         indentToken = false;
@@ -869,13 +871,15 @@ namespace ts.formatting {
             for (const triviaItem of trivia) {
                 if (isComment(triviaItem.kind) && rangeContainsRange(originalRange, triviaItem)) {
                     const triviaItemStart = sourceFile.getLineAndCharacterOfPosition(triviaItem.pos);
-                    processRange(triviaItem, triviaItemStart, parent, contextNode, dynamicIndentation);
+                    const triviaItemEnd = sourceFile.getLineAndCharacterOfPosition(triviaItem.end);
+                    processRange(triviaItem, triviaItemStart, triviaItemEnd, parent, contextNode, dynamicIndentation);
                 }
             }
         }
 
         function processRange(range: TextRangeWithKind,
             rangeStart: LineAndCharacter,
+            rangeEnd: LineAndCharacter,
             parent: Node,
             contextNode: Node,
             dynamicIndentation: DynamicIndentation): boolean {
@@ -890,13 +894,14 @@ namespace ts.formatting {
                 }
                 else {
                     lineAdded =
-                        processPair(range, rangeStart.line, parent, previousRange, previousRangeStartLine, previousParent, contextNode, dynamicIndentation);
+                        processPair(range, rangeStart.line, parent, previousRange, previousRangeStartLine, previousRangeEndLine, previousParent, contextNode, dynamicIndentation);
                 }
             }
 
             previousRange = range;
             previousParent = parent;
             previousRangeStartLine = rangeStart.line;
+            previousRangeEndLine = rangeEnd.line;
 
             return lineAdded;
         }
@@ -906,6 +911,7 @@ namespace ts.formatting {
             currentParent: Node,
             previousItem: TextRangeWithKind,
             previousStartLine: number,
+            previousEndLine: number,
             previousParent: Node,
             contextNode: Node,
             dynamicIndentation: DynamicIndentation): boolean {
@@ -917,7 +923,7 @@ namespace ts.formatting {
             let trimTrailingWhitespaces: boolean;
             let lineAdded: boolean;
             if (rule) {
-                applyRuleEdits(rule, previousItem, previousStartLine, currentItem, currentStartLine);
+                applyRuleEdits(rule, previousItem, previousStartLine, previousEndLine, currentItem, currentStartLine);
 
                 if (rule.Operation.Action & (RuleAction.Space | RuleAction.Delete) && currentStartLine !== previousStartLine) {
                     lineAdded = false;
@@ -1108,6 +1114,7 @@ namespace ts.formatting {
         function applyRuleEdits(rule: Rule,
             previousRange: TextRangeWithKind,
             previousStartLine: number,
+            previousEndLine: number,
             currentRange: TextRangeWithKind,
             currentStartLine: number): void {
 
@@ -1137,7 +1144,7 @@ namespace ts.formatting {
                     break;
                 case RuleAction.Space:
                     // exit early if we on different lines and rule cannot change number of newlines
-                    if (rule.Flag !== RuleFlags.CanDeleteNewLines && previousStartLine !== currentStartLine) {
+                    if (rule.Flag !== RuleFlags.CanDeleteNewLines && previousEndLine !== currentStartLine) {
                         return;
                     }