Detter symbol detection was timing out on larger files

Author: Razmo99

src/PowerShellEditorServices/Services/PowerShell/Handlers/PrepareRenameSymbol.cs

@@ -54,15 +54,17 @@ public async Task<PrepareRenameSymbolResult> Handle(PrepareRenameSymbolParams re
                 {
                     message = ""
                 };
-
-                IEnumerable<Ast> tokens = scriptFile.ScriptAst.FindAll(ast =>
+                // ast is FunctionDefinitionAst or CommandAst or VariableExpressionAst or StringConstantExpressionAst &&
+                Ast token = scriptFile.ScriptAst.Find(ast =>
                 {
                     return request.Line + 1 == ast.Extent.StartLineNumber &&
                            request.Column + 1 >= ast.Extent.StartColumnNumber;
-                }, false);
-
-                Ast token = tokens.LastOrDefault();
-
+                }, true);
+                IEnumerable<Ast> tokens = token.FindAll(ast =>{
+                    return ast.Extent.StartColumnNumber <= request.Column &&
+                    ast.Extent.EndColumnNumber >= request.Column;
+                },true);
+                token = tokens.LastOrDefault();
                 if (token == null)
                 {
                     result.message = "Unable to find symbol";

src/PowerShellEditorServices/Services/PowerShell/Handlers/RenameSymbol.cs

@@ -12,7 +12,6 @@
 using Microsoft.PowerShell.EditorServices.Services.TextDocument;
 using Microsoft.PowerShell.EditorServices.Refactoring;
 using System.Linq;
-
 namespace Microsoft.PowerShell.EditorServices.Handlers
 {
     [Serial, Method("powerShell/renameSymbol")]
@@ -123,13 +122,16 @@ public async Task<RenameSymbolResult> Handle(RenameSymbolParams request, Cancell
             return await Task.Run(() =>
             {
 
-                IEnumerable<Ast> tokens = scriptFile.ScriptAst.FindAll(ast =>
+                Ast token = scriptFile.ScriptAst.Find(ast =>
                 {
-                    return request.Line+1 == ast.Extent.StartLineNumber &&
-                           request.Column+1 >= ast.Extent.StartColumnNumber;
-                }, false);
-
-                Ast token = tokens.LastOrDefault();
+                    return request.Line + 1 == ast.Extent.StartLineNumber &&
+                           request.Column + 1 >= ast.Extent.StartColumnNumber;
+                }, true);
+                IEnumerable<Ast> tokens = token.FindAll(ast =>{
+                    return ast.Extent.StartColumnNumber <= request.Column &&
+                    ast.Extent.EndColumnNumber >= request.Column;
+                },true);
+                token = tokens.LastOrDefault();
 
                 if (token == null) { return null; }