Fixing an odd edge case, of not being to rename a variable directly under a function definition, but selecting one column to the right worked

Author: Razmo99

src/PowerShellEditorServices/Services/PowerShell/Refactoring/Utilities.cs

@@ -123,6 +123,19 @@ public static Ast GetAst(int StartLineNumber, int StartColumnNumber, Ast Ast)
 
             if (token is NamedBlockAst)
             {
+                // NamedBlockAST starts on the same line as potentially another AST,
+                // its likley a user is not after the NamedBlockAst but what it contains
+                IEnumerable<Ast> stacked_tokens = token.FindAll(ast =>
+                {
+                    return StartLineNumber == ast.Extent.StartLineNumber &&
+                    ast.Extent.EndColumnNumber >= StartColumnNumber
+                    && StartColumnNumber >= ast.Extent.StartColumnNumber;
+                }, true);
+
+                if(stacked_tokens.Count() > 1){
+                    return stacked_tokens.LastOrDefault();
+                }
+
                 return token.Parent;
             }
 

test/PowerShellEditorServices.Test.Shared/Refactoring/Utilities/TestDetectionUnderFunctionDef.ps1

@@ -0,0 +1,15 @@
+function Write-Item($itemCount) {
+    $i = 1
+
+    while ($i -le $itemCount) {
+        $str = "Output $i"
+        Write-Output $str
+
+        # In the gutter on the left, right click and select "Add Conditional Breakpoint"
+        # on the next line. Use the condition: $i -eq 25
+        $i = $i + 1
+
+        # Slow down execution a bit so user can test the "Pause debugger" feature.
+        Start-Sleep -Milliseconds $DelayMilliseconds
+    }
+}

test/PowerShellEditorServices.Test/Refactoring/RefactorUtilitiesTests.cs

@@ -142,6 +142,23 @@ public void GetFunctionDefinitionAst()
             Assert.Equal(1,symbol.Extent.StartLineNumber);
             Assert.Equal(1,symbol.Extent.StartColumnNumber);
 
+        }
+        [Fact]
+        public void GetVariableUnderFunctionDef()
+        {
+            RenameSymbolParams request = new(){
+                Column=5,
+                Line=2,
+                RenameTo="Renamed",
+                FileName="TestDetectionUnderFunctionDef.ps1"
+            };
+            ScriptFile scriptFile = GetTestScript(request.FileName);
+
+            Ast symbol = Utilities.GetAst(request.Line,request.Column,scriptFile.ScriptAst);
+            Assert.IsType<VariableExpressionAst>(symbol);
+            Assert.Equal(2,symbol.Extent.StartLineNumber);
+            Assert.Equal(5,symbol.Extent.StartColumnNumber);
+
         }
         [Fact]
         public void AssertContainsDotSourcingTrue()