Preserve branch indentation after multiline conditions (#2271)

* Initial plan

* Preserve close brace indentation for branches

Co-authored-by: bergmeister <9250262+bergmeister@users.noreply.github.com>

---------

Co-authored-by: copilot-swe-agent[bot] <198982749+Copilot@users.noreply.github.com>
Co-authored-by: bergmeister <9250262+bergmeister@users.noreply.github.com>

Author: Copilot

Rules/PlaceCloseBrace.cs

@@ -277,6 +277,9 @@ private List<CorrectionExtent> GetCorrectionsForBraceShouldHaveNewLineAfter(
             var corrections = new List<CorrectionExtent>();
             var nextToken = tokens[closeBracePos + 1];
             var closeBraceToken = tokens[closeBracePos];
+            var indentation = tokens[closeBracePos - 1].Kind == TokenKind.NewLine
+                ? new String(' ', closeBraceToken.Extent.StartColumnNumber - 1)
+                : GetIndentation(tokens, closeBracePos, openBracePos);
 
             // Also remove the whitespace between the close brace and the next token
             // and indent the next token to the same level as the close brace
@@ -285,7 +288,7 @@ private List<CorrectionExtent> GetCorrectionsForBraceShouldHaveNewLineAfter(
                 nextToken.Extent.StartLineNumber,
                 closeBraceToken.Extent.StartColumnNumber,
                 nextToken.Extent.StartColumnNumber,
-                closeBraceToken.Text + Environment.NewLine + GetIndentation(tokens, closeBracePos, openBracePos),
+                closeBraceToken.Text + Environment.NewLine + indentation,
                 fileName));
             return corrections;
         }

Tests/Rules/PlaceCloseBrace.tests.ps1

@@ -180,6 +180,52 @@ if (Test-Path "blah") {
             Test-CorrectionExtentFromContent @params
         }
 
+        It "Should preserve close brace indentation after a multiline if condition" {
+            $def = @'
+if (
+    $true) {
+    $true
+} else {
+    $false
+}
+'@
+            $expected = @'
+if (
+    $true) {
+    $true
+}
+else {
+    $false
+}
+'@
+            Invoke-Formatter -ScriptDefinition $def -Settings $settings | Should -Be $expected
+        }
+
+        It "Should preserve nested close brace indentation after a multiline if condition" {
+            $def = @'
+function Test {
+    if (
+        $true) {
+        $true
+    } else {
+        $false
+    }
+}
+'@
+            $expected = @'
+function Test {
+    if (
+        $true) {
+        $true
+    }
+    else {
+        $false
+    }
+}
+'@
+            Invoke-Formatter -ScriptDefinition $def -Settings $settings | Should -Be $expected
+        }
+
         It "Should find a violation for a close brace followed by elseif" {
             $def = @'
 if (Test-Path "blah") {