Fix formatter range tracking through iterative edits

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

Author: Copilot

Engine/ScriptAnalyzer.cs

@@ -1556,7 +1556,6 @@ public EditableText Fix(EditableText text, Range range, bool skipParsing, out Ra
             }
 
             range = isRangeNull ? null : SnapToEdges(text, range);
-            var previousLineCount = text.LineCount;
             var previousUnusedCorrections = 0;
             do
             {
@@ -1578,7 +1577,7 @@ public EditableText Fix(EditableText text, Range range, bool skipParsing, out Ra
 
                 this.outputWriter.WriteVerbose($"Found {corrections.Count} violations.");
                 int unusedCorrections;
-                Fix(text, corrections, out unusedCorrections);
+                Fix(text, corrections, out unusedCorrections, ref range);
                 var numberOfFixedViolatons = corrections.Count - unusedCorrections;
                 fixesWereApplied = numberOfFixedViolatons > 0;
                 this.outputWriter.WriteVerbose($"Fixed {numberOfFixedViolatons} violations.");
@@ -1595,16 +1594,6 @@ public EditableText Fix(EditableText text, Range range, bool skipParsing, out Ra
                 }
 
                 previousUnusedCorrections = unusedCorrections;
-                var lineCount = text.LineCount;
-                if (!isRangeNull && lineCount != previousLineCount)
-                {
-                    range = new Range(
-                        range.Start,
-                        range.End.Shift(lineCount - previousLineCount, 0));
-                    range = SnapToEdges(text, range);
-                }
-
-                previousLineCount = lineCount;
             } while (previousUnusedCorrections > 0);
 
             updatedRange = range;
@@ -1728,14 +1717,29 @@ private static IEnumerable<CorrectionExtent> GetCorrectionExtentsForFix(
         private static EditableText Fix(
             EditableText text,
             List<CorrectionExtent> correctionExtents,
-            out int unusedCorrections)
+            out int unusedCorrections,
+            ref Range range)
         {
             var correctionsToUse = GetCorrectionExtentsForFix(correctionExtents);
             var count = 0;
             foreach (var correction in correctionsToUse)
             {
                 count++;
                 text.ApplyEdit(correction);
+                if (range != null)
+                {
+                    var lineDelta = correction.Lines.Length - 1
+                        - (correction.EndLineNumber - correction.StartLineNumber);
+                    var columnDelta = 0;
+                    if (correction.EndLineNumber == range.End.Line)
+                    {
+                        var replacementEndColumn = correction.Lines.Last().TrimEnd('\r').Length
+                            + (correction.Lines.Length == 1 ? correction.StartColumnNumber : 1);
+                        columnDelta = replacementEndColumn - correction.EndColumnNumber;
+                    }
+
+                    range = new Range(range.Start, range.End.Shift(lineDelta, columnDelta));
+                }
             }
 
             unusedCorrections = correctionExtents.Count - count;

Tests/Engine/InvokeFormatter.tests.ps1

@@ -81,6 +81,51 @@ function foo {
 
             Invoke-Formatter -ScriptDefinition $def -Range @(3, 1, 4, 1) | Should -Be $expected
         }
+
+        It "Should apply all corrections when earlier fixes expand the selected line" {
+            $def = "if(`$a){'x'}`nif(`$b){'y'}"
+            $expected = "if(`$a){'x'}`nif (`$b) { 'y' }"
+
+            Invoke-Formatter -ScriptDefinition $def -Range @(2, 1, 2, 12) | Should -Be $expected
+        }
+
+        It "Should track the selected lines when fixes <Name>" -TestCases @(
+            @{
+                Name = 'expand the last line'
+                Selected = "if(`$b){'y'}"
+                Expected = "if (`$b) { 'y' }"
+            }
+            @{
+                Name = 'shorten the last line'
+                Selected = "if  (`$b)  {  'y'  }"
+                Expected = "if (`$b) { 'y' }"
+            }
+            @{
+                Name = 'expand multiple lines'
+                Selected = "if(`$b){'y'}`nif(`$c){'z'}"
+                Expected = "if (`$b) { 'y' }`nif (`$c) { 'z' }"
+            }
+            @{
+                Name = 'remove a line'
+                Selected = "if(`$b)`n{`n'y'`n}"
+                Expected = "if (`$b) {`n    'y'`n}"
+            }
+            @{
+                Name = 'add a line'
+                Selected = "if(`$b) {`n'y'`n}"
+                Expected = "if (`$b)`n{`n    'y'`n}"
+                Settings = 'CodeFormattingAllman'
+            }
+        ) {
+            param($Selected, $Expected, $Settings = 'CodeFormatting')
+
+            $def = "if(`$a){'x'}`n$Selected`nif(`$d){'w'}"
+            $lines = $Selected -split "`n"
+            $range = @(2, 1, (1 + $lines.Count), (1 + $lines[-1].Length))
+
+            Invoke-Formatter -ScriptDefinition $def -Range $range -Settings $Settings |
+                Should -Be "if(`$a){'x'}`n$Expected`nif(`$d){'w'}"
+        }
     }
 
     Context "When no settings are given" {