|
' Licensed to the .NET Foundation under one or more agreements.
' The .NET Foundation licenses this file to you under the MIT license.
' See the LICENSE file in the project root for more information.
Imports System.Composition
Imports System.Threading
Imports Microsoft.CodeAnalysis.Highlighting
Imports Microsoft.CodeAnalysis.Host.Mef
Imports Microsoft.CodeAnalysis.Text
Imports Microsoft.CodeAnalysis.VisualBasic.Syntax
Namespace Microsoft.CodeAnalysis.VisualBasic.KeywordHighlighting
<ExportHighlighter(LanguageNames.VisualBasic), [Shared]>
Friend Class MultiLineLambdaExpressionHighlighter
Inherits AbstractKeywordHighlighter(Of SyntaxNode)
<ImportingConstructor>
<Obsolete(MefConstruction.ImportingConstructorMessage, True)>
Public Sub New()
End Sub
Protected Overloads Overrides Sub AddHighlights(node As SyntaxNode, highlights As List(Of TextSpan), cancellationToken As CancellationToken)
Dim lambdaExpression = node.GetAncestor(Of MultiLineLambdaExpressionSyntax)()
If lambdaExpression Is Nothing Then
Return
End If
With lambdaExpression
Dim isAsync = False
Dim isIterator = False
With .SubOrFunctionHeader
isAsync = .Modifiers.Any(SyntaxKind.AsyncKeyword)
isIterator = .Modifiers.Any(SyntaxKind.IteratorKeyword)
Dim firstKeyword = If(.Modifiers.Count > 0, .Modifiers.First(), .DeclarationKeyword)
highlights.Add(TextSpan.FromBounds(firstKeyword.SpanStart, .DeclarationKeyword.Span.End))
End With
highlights.AddRange(
.GetRelatedStatementHighlights(
blockKind:= .SubOrFunctionHeader.DeclarationKeyword.Kind,
checkReturns:=True))
If isIterator Then
highlights.AddRange(.GetRelatedYieldStatementHighlights())
End If
If isAsync Then
HighlightRelatedAwaits(lambdaExpression, highlights, cancellationToken)
End If
highlights.Add(.EndSubOrFunctionStatement.Span)
End With
End Sub
End Class
End Namespace
|