' Copyright (c) Microsoft.  All Rights Reserved.  Licensed under the MIT license.  See License.txt in the project root for license information.

Imports System.Collections.Immutable
Imports Microsoft.CodeAnalysis
Imports Microsoft.CodeAnalysis.Diagnostics
Imports Microsoft.CodeAnalysis.Operations
Imports Microsoft.CodeAnalysis.VisualBasic.Syntax
Imports Microsoft.NetCore.Analyzers.Runtime

Namespace Microsoft.NetCore.VisualBasic.Analyzers.Runtime

    <DiagnosticAnalyzer(LanguageNames.VisualBasic)>
    Public NotInheritable Class BasicForwardCancellationTokenToInvocationsAnalyzer

        Inherits ForwardCancellationTokenToInvocationsAnalyzer

        Protected Overrides Function GetInvocationMethodNameNode(invocationNode As SyntaxNode) As SyntaxNode

            Dim invocationExpression = TryCast(invocationNode, InvocationExpressionSyntax)

            If invocationExpression IsNot Nothing Then

                Dim memberBindingExpression = TryCast(invocationExpression.Expression, MemberAccessExpressionSyntax)

                If memberBindingExpression IsNot Nothing Then

                    Return memberBindingExpression.Name

                End If

                Return invocationExpression.Expression

            End If

            Return Nothing

        End Function

        Protected Overrides Function ArgumentsImplicitOrNamed(cancellationTokenType As INamedTypeSymbol, arguments As ImmutableArray(Of IArgumentOperation)) As Boolean

            Return arguments.Any(Function(a)

                                     If a.IsImplicit AndAlso Not a.Parameter.Type.Equals(cancellationTokenType) Then
                                         Return True
                                     End If

                                     Dim argumentNode As ArgumentSyntax = TryCast(a.Syntax, ArgumentSyntax)

                                     Return argumentNode IsNot Nothing AndAlso argumentNode.IsNamed

                                 End Function)

        End Function

    End Class

End Namespace