Skip to content

Commit c5bebe3

Browse files
committed
additional checks in getast for namedblock detection
1 parent 271b67f commit c5bebe3

File tree

1 file changed

+15
-0
lines changed
  • src/PowerShellEditorServices/Services/PowerShell/Refactoring

1 file changed

+15
-0
lines changed

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

Lines changed: 15 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -117,6 +117,21 @@ public static Ast GetAst(int StartLineNumber, int StartColumnNumber, Ast Ast)
117117
StartColumnNumber >= ast.Extent.StartColumnNumber;
118118
}, true);
119119

120+
if (token is NamedBlockAst)
121+
{
122+
return token.Parent;
123+
}
124+
125+
if (null == token)
126+
{
127+
IEnumerable<Ast> LineT = Ast.FindAll(ast =>
128+
{
129+
return StartLineNumber == ast.Extent.StartLineNumber &&
130+
StartColumnNumber >= ast.Extent.StartColumnNumber;
131+
}, true);
132+
return LineT.OfType<FunctionDefinitionAst>()?.LastOrDefault();
133+
}
134+
120135
IEnumerable<Ast> tokens = token.FindAll(ast =>
121136
{
122137
return ast.Extent.EndColumnNumber >= StartColumnNumber

0 commit comments

Comments
 (0)