File tree Expand file tree Collapse file tree 3 files changed +9
-17
lines changed Expand file tree Collapse file tree 3 files changed +9
-17
lines changed Original file line number Diff line number Diff line change @@ -173,15 +173,12 @@ await semanticDocument.WithSyntaxRootAsync(result.Root, cancellationToken).Confi
173173 protected override AbstractFormattingRule GetCustomFormattingRule ( Document document )
174174 => FormattingRule . Instance ;
175175
176- protected override SyntaxToken ? GetInvocationNameToken ( IEnumerable < SyntaxToken > methodNames )
177- => methodNames . FirstOrNull ( t => ! t . Parent . IsKind ( SyntaxKind . MethodDeclaration ) ) ;
178-
179176 protected override SyntaxNode ParseTypeName ( string name )
180177 => SyntaxFactory . ParseTypeName ( name ) ;
181178
182- protected override async Task < ( Document document , SyntaxToken ? invocationNameToken ) > InsertNewLineBeforeLocalFunctionIfNecessaryAsync (
179+ protected override async Task < ( Document document , SyntaxToken invocationNameToken ) > InsertNewLineBeforeLocalFunctionIfNecessaryAsync (
183180 Document document ,
184- SyntaxToken ? invocationNameToken ,
181+ SyntaxToken invocationNameToken ,
185182 SyntaxNode methodDefinition ,
186183 CancellationToken cancellationToken )
187184 {
@@ -206,8 +203,7 @@ protected override SyntaxNode ParseTypeName(string name)
206203
207204 var newRoot = await document . GetSyntaxRootAsync ( cancellationToken ) . ConfigureAwait ( false ) ;
208205
209- if ( invocationNameToken != null )
210- invocationNameToken = newRoot . FindToken ( invocationNameToken . Value . SpanStart ) ;
206+ invocationNameToken = newRoot . FindToken ( invocationNameToken . SpanStart ) ;
211207 }
212208
213209 return ( document , invocationNameToken ) ;
Original file line number Diff line number Diff line change @@ -42,11 +42,10 @@ internal abstract partial class MethodExtractor(
4242
4343 protected abstract CodeGenerator CreateCodeGenerator ( SelectionResult selectionResult , AnalyzerResult analyzerResult ) ;
4444
45- protected abstract SyntaxToken ? GetInvocationNameToken ( IEnumerable < SyntaxToken > tokens ) ;
4645 protected abstract AbstractFormattingRule GetCustomFormattingRule ( Document document ) ;
4746
48- protected abstract Task < ( Document document , SyntaxToken ? invocationNameToken ) > InsertNewLineBeforeLocalFunctionIfNecessaryAsync (
49- Document document , SyntaxToken ? invocationNameToken , SyntaxNode methodDefinition , CancellationToken cancellationToken ) ;
47+ protected abstract Task < ( Document document , SyntaxToken invocationNameToken ) > InsertNewLineBeforeLocalFunctionIfNecessaryAsync (
48+ Document document , SyntaxToken invocationNameToken , SyntaxNode methodDefinition , CancellationToken cancellationToken ) ;
5049
5150 public ExtractMethodResult ExtractMethod ( OperationStatus initialStatus , CancellationToken cancellationToken )
5251 {
@@ -92,7 +91,8 @@ public ExtractMethodResult ExtractMethod(OperationStatus initialStatus, Cancella
9291 cancellationToken . ThrowIfCancellationRequested ( ) ;
9392
9493 var newRoot = afterTriviaRestored . Root ;
95- var invocationNameToken = GetInvocationNameToken ( newRoot . GetAnnotatedTokens ( MethodNameAnnotation ) ) ;
94+
95+ var invocationNameToken = newRoot . GetAnnotatedTokens ( MethodNameAnnotation ) . Single ( ) ;
9696
9797 // Do some final patchups of whitespace when inserting a local function.
9898 if ( LocalFunction )
Original file line number Diff line number Diff line change @@ -61,10 +61,6 @@ Namespace Microsoft.CodeAnalysis.VisualBasic.ExtractMethod
6161 Return FormattingRule.Instance
6262 End Function
6363
64- Protected Overrides Function GetInvocationNameToken(methodNames As IEnumerable( Of SyntaxToken)) As SyntaxToken?
65- Return methodNames.FirstOrNull( Function (t) t.Parent.Kind <> SyntaxKind.SubStatement AndAlso t.Parent.Kind <> SyntaxKind.FunctionStatement)
66- End Function
67-
6864 Protected Overrides Function ParseTypeName(name As String ) As SyntaxNode
6965 Return SyntaxFactory.ParseTypeName(name)
7066 End Function
@@ -115,9 +111,9 @@ Namespace Microsoft.CodeAnalysis.VisualBasic.ExtractMethod
115111
116112 Protected Overrides Function InsertNewLineBeforeLocalFunctionIfNecessaryAsync(
117113 document As Document,
118- invocationNameToken? As SyntaxToken,
114+ invocationNameToken As SyntaxToken,
119115 methodDefinition As SyntaxNode,
120- cancellationToken As CancellationToken) As Task( Of (document As Document, invocationNameToken As SyntaxToken? ))
116+ cancellationToken As CancellationToken) As Task( Of (document As Document, invocationNameToken As SyntaxToken))
121117 ' VB doesn't need to do any correction, so we just return the values untouched
122118 Return Task.FromResult((document, invocationNameToken))
123119 End Function
You can’t perform that action at this time.
0 commit comments