Indentation fix for multi-line call expression

src/services/formatting/smartIndenter.ts

@@ -26,7 +26,7 @@ module ts.formatting {
                 precedingToken.kind === SyntaxKind.TemplateHead ||
                 precedingToken.kind === SyntaxKind.TemplateMiddle ||
                 precedingToken.kind === SyntaxKind.TemplateTail;
-            if (precedingTokenIsLiteral && precedingToken.getStart(sourceFile) <= position &&  precedingToken.end > position) {
+            if (precedingTokenIsLiteral && precedingToken.getStart(sourceFile) <= position && precedingToken.end > position) {
                 return 0;
             }
 
@@ -66,6 +66,10 @@ module ts.formatting {
                 if (actualIndentation !== Value.Unknown) {
                     return actualIndentation;
                 }
+                actualIndentation = getLineIndentationWhenExpressionIsInMultiLine(current, sourceFile, options);
+                if (actualIndentation !== Value.Unknown) {
+                    return actualIndentation + options.IndentSize;
+                }
 
                 previous = current;
                 current = current.parent;
@@ -122,6 +126,10 @@ module ts.formatting {
                     if (actualIndentation !== Value.Unknown) {
                         return actualIndentation + indentationDelta;
                     }
+                    actualIndentation = getLineIndentationWhenExpressionIsInMultiLine(current, sourceFile, options);
+                    if (actualIndentation !== Value.Unknown) {
+                        return actualIndentation + indentationDelta;
+                    }
                 }
 
                 // increase indentation if parent node wants its content to be indented and parent and child nodes don't start on the same line
@@ -287,6 +295,41 @@ module ts.formatting {
             }
         }
 
+        function getLineIndentationWhenExpressionIsInMultiLine(node: Node, sourceFile: SourceFile, options: EditorOptions): number {
+            // actual indentation should not be used when:
+            // - node is close parenthesis - this is the end of the expression
+            // - node is property access expression
+            if (node.kind !== SyntaxKind.CloseParenToken &&
+                node.kind !== SyntaxKind.PropertyAccessExpression &&
+                node.parent && (
+                node.parent.kind === SyntaxKind.CallExpression ||
+                node.parent.kind === SyntaxKind.NewExpression)) {
+
+                let parentExpression = (<CallExpression | NewExpression>node.parent).expression;
+                let startingExpression = getStartingExpression(<PropertyAccessExpression | CallExpression | ElementAccessExpression>parentExpression);
+
+                if (parentExpression === startingExpression) {
+                    return Value.Unknown;
+                }
+
+                let parentExpressionEnd = sourceFile.getLineAndCharacterOfPosition(parentExpression.end);
+                let startingExpressionEnd = sourceFile.getLineAndCharacterOfPosition(startingExpression.end);
+
+                if (parentExpressionEnd.line === startingExpressionEnd.line) {
+                    return Value.Unknown;
+                }
+
+                return findColumnForFirstNonWhitespaceCharacterInLine(parentExpressionEnd, sourceFile, options);
+            }
+            return Value.Unknown;
+
+            function getStartingExpression(expression: PropertyAccessExpression | CallExpression | ElementAccessExpression) {
+                while (expression.expression)
+                    expression = <PropertyAccessExpression | CallExpression | ElementAccessExpression>expression.expression;
+                return expression;
+            }
+        }
+
         function deriveActualIndentationFromList(list: Node[], index: number, sourceFile: SourceFile, options: EditorOptions): number {
             Debug.assert(index >= 0 && index < list.length);
             let node = list[index];

tests/cases/fourslash/consistenceOnIndentionsOfChainedFunctionCalls.ts

@@ -18,4 +18,4 @@ goTo.marker("1");
 edit.insert("\r\n");
 goTo.marker("0");
 // Won't-fixed: Smart indent during chained function calls
-verify.indentationIs(4);
+verify.indentationIs(8);

tests/cases/fourslash/formattingOnChainedCallbacks.ts

@@ -13,10 +13,20 @@
 ////    })/*b*/
 ////}
 
+////Promise
+////    .then(
+////    /*n1*/
+////        )
+////    /*n2*/
+////    .then();
+
+
 goTo.marker('1');
 edit.insertLine('');
 goTo.marker('2');
 verify.currentLineContentIs('        ""');
+edit.insertLine('');
+verify.indentationIs(8);
 goTo.marker('4');
 edit.insertLine('');
 goTo.marker('3');
@@ -34,4 +44,9 @@ edit.insert(';');
 verify.currentLineContentIs('            "";');
 goTo.marker('b');
 edit.insert(';');
-verify.currentLineContentIs('        });');
+verify.currentLineContentIs('        });');
+
+goTo.marker('n1');
+verify.indentationIs(8);
+goTo.marker('n2');
+verify.indentationIs(4);