' 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.Runtime.CompilerServices
Imports Microsoft.CodeAnalysis.Text
Imports Microsoft.CodeAnalysis.VisualBasic.Symbols
Imports Microsoft.CodeAnalysis.VisualBasic.Syntax

Namespace Microsoft.CodeAnalysis.VisualBasic.Symbols

    Friend Module EmbeddedSymbolExtensions

        ''' <summary>
        ''' True if the syntax tree is an embedded syntax tree
        ''' </summary>
        <Extension()>
        Public Function IsEmbeddedSyntaxTree(tree As SyntaxTree) As Boolean
            Return EmbeddedSymbolManager.GetEmbeddedKind(tree) <> EmbeddedSymbolKind.None
        End Function

        <Extension()>
        Public Function GetEmbeddedKind(tree As SyntaxTree) As EmbeddedSymbolKind
            Return EmbeddedSymbolManager.GetEmbeddedKind(tree)
        End Function

        <Extension()>
        Public Function IsEmbeddedOrMyTemplateTree(tree As SyntaxTree) As Boolean
            Dim vbTree = TryCast(tree, VisualBasicSyntaxTree)
            Return vbTree IsNot Nothing AndAlso vbTree.IsMyTemplate OrElse vbTree.IsEmbeddedSyntaxTree
        End Function

        <Extension()>
        Public Function IsEmbeddedOrMyTemplateLocation(location As Location) As Boolean
            Return TypeOf location Is EmbeddedTreeLocation OrElse
                TypeOf location Is MyTemplateLocation
        End Function
    End Module

End Namespace