Skip to content

R
Roslyn

dotNET / Roslyn

通知 1
Star 0
Fork 0
R
Roslyn

体验新版 GitCode，发现更多精彩内容 >>

提交 9b8d0b9a 编写于 作者: C CyrusNajmabadi
浏览文件
操作

Break types into their own files.

上级 290be822
展开全部 隐藏空白更改
内联 并排
Showing with 1461 addition and 1420 deletion
src/Features/CSharp/Portable/CSharpFeatures.csproj
浏览文件 @ 9b8d0b9a
......@@ -59,6 +59,7 @@
<Compile Include="..\..\..\Compilers\CSharp\Portable\Syntax\LambdaUtilities.cs">
<Link>InternalUtilities\LambdaUtilities.cs</Link>
</Compile>
<Compile Include="AddImport\CSharpAddImportFeatureService.cs" />
<Compile Include="AddPackage\CSharpAddSpecificPackageCodeFixProvider.cs" />
<Compile Include="AddParameter\CSharpAddParameterCodeFixProvider.cs" />
<Compile Include="CodeFixes\RemoveUnusedVariable\CSharpRemoveUnusedVariableCodeFixProvider.cs" />
......
src/Features/Core/Portable/AddImport/IAddImportFeatureService.cs 0 → 100644
浏览文件 @ 9b8d0b9a
// Copyright (c) Microsoft. All Rights Reserved. Licensed under the Apache License, Version 2.0. See License.txt in the project root for license information.
using System.Collections.Immutable;
using System.Threading;
using System.Threading.Tasks;
using Microsoft.CodeAnalysis.Host;
using Microsoft.CodeAnalysis.Packaging;
using Microsoft.CodeAnalysis.SymbolSearch;
using Microsoft.CodeAnalysis.Text;
namespace Microsoft.CodeAnalysis.AddImport
{
internal interface IAddImportFeatureService : ILanguageService
{
Task<ImmutableArray<AddImportFixData>> GetFixesAsync(
Document document, TextSpan span, string diagnosticId, ISymbolSearchService symbolSearchService,
bool searchReferenceAssemblies, ImmutableArray<PackageSource> packageSources, CancellationToken cancellationToken);
}
}
\ No newline at end of file
src/Features/Core/Portable/Features.csproj
浏览文件 @ 9b8d0b9a
......@@ -99,12 +99,14 @@
<Compile Include="..\..\..\Compilers\Shared\DesktopShim.cs">
<Link>Shared\Utilities\DesktopShim.cs</Link>
</Compile>
<Compile Include="AddImport\AbstractAddImportFeatureService.cs" />
<Compile Include="AddImport\AddImportFixData.cs" />
<Compile Include="AddImport\AddImportFixKind.cs" />
<Compile Include="AddImport\CodeActions\AddImportCodeAction.cs" />
<Compile Include="AddImport\CodeActions\AssemblyReferenceCodeAction.cs" />
<Compile Include="AddImport\CodeActions\MetadataSymbolReferenceCodeAction.cs" />
<Compile Include="AddImport\CodeActions\ProjectSymbolReferenceCodeAction.cs" />
<Compile Include="AddImport\IAddImportFeatureService.cs" />
<Compile Include="AddImport\SearchScopes\AllSymbolsProjectSearchScope.cs" />
<Compile Include="AddImport\SearchScopes\MetadataSymbolsSearchScope.cs" />
<Compile Include="AddImport\SearchScopes\ProjectSearchScope.cs" />
......
src/Features/VisualBasic/Portable/AddImport/VisualBasicAddImportCodeFixProvider.vb
浏览文件 @ 9b8d0b9a
......@@ -2,18 +2,10 @@
Imports System.Collections.Immutable
Imports System.Composition
Imports System.Threading
Imports Microsoft.CodeAnalysis.AddImport
Imports Microsoft.CodeAnalysis.AddImports
Imports Microsoft.CodeAnalysis.CaseCorrection
Imports Microsoft.CodeAnalysis.CodeFixes
Imports Microsoft.CodeAnalysis.Formatting
Imports Microsoft.CodeAnalysis.Host.Mef
Imports Microsoft.CodeAnalysis.LanguageServices
Imports Microsoft.CodeAnalysis.Packaging
Imports Microsoft.CodeAnalysis.Simplification
Imports Microsoft.CodeAnalysis.SymbolSearch
Imports Microsoft.CodeAnalysis.VisualBasic.Syntax
Namespace Microsoft.CodeAnalysis.VisualBasic.AddImport
<ExportCodeFixProvider(LanguageNames.VisualBasic, Name:=PredefinedCodeFixProviderNames.AddImport), [Shared]>
......@@ -112,341 +104,4 @@ Namespace Microsoft.CodeAnalysis.VisualBasic.AddImport
End Get
End Property
End Class
<ExportLanguageService(GetType(IAddImportFeatureService), LanguageNames.VisualBasic), [Shared]>
Friend Class VisualBasicAddImportFeatureService
Inherits AbstractAddImportFeatureService(Of SimpleNameSyntax)
Protected Overrides Function CanAddImport(node As SyntaxNode, cancellationToken As CancellationToken) As Boolean
If node.GetAncestor(Of ImportsStatementSyntax)() IsNot Nothing Then
Return False
End If
Return node.CanAddImportsStatements(cancellationToken)
End Function
Protected Overrides Function CanAddImportForMethod(
diagnosticId As String,
syntaxFacts As ISyntaxFactsService,
node As SyntaxNode,
ByRef nameNode As SimpleNameSyntax) As Boolean
Select Case diagnosticId
Case VisualBasicAddImportCodeFixProvider.BC30456,
VisualBasicAddImportCodeFixProvider.BC30390,
VisualBasicAddImportCodeFixProvider.BC42309,
VisualBasicAddImportCodeFixProvider.BC30451
Exit Select
Case VisualBasicAddImportCodeFixProvider.BC30512
' look up its corresponding method name
Dim parent = node.GetAncestor(Of InvocationExpressionSyntax)()
If parent Is Nothing Then
Return False
End If
Dim method = TryCast(parent.Expression, MemberAccessExpressionSyntax)
If method IsNot Nothing Then
node = method.Name
Else
node = parent.Expression
End If
Exit Select
Case VisualBasicAddImportCodeFixProvider.BC36719
If node.IsKind(SyntaxKind.ObjectCollectionInitializer) Then
Return True
End If
Return False
Case VisualBasicAddImportCodeFixProvider.BC32016
Dim memberAccessName = TryCast(node, MemberAccessExpressionSyntax)?.Name
Dim conditionalAccessName = TryCast(TryCast(TryCast(node, ConditionalAccessExpressionSyntax)?.WhenNotNull, InvocationExpressionSyntax)?.Expression, MemberAccessExpressionSyntax)?.Name
If memberAccessName Is Nothing AndAlso conditionalAccessName Is Nothing Then
Return False
End If
node = If(memberAccessName Is Nothing, conditionalAccessName, memberAccessName)
Exit Select
Case Else
Return False
End Select
Dim memberAccess = TryCast(node, MemberAccessExpressionSyntax)
If memberAccess IsNot Nothing Then
node = memberAccess.Name
End If
If memberAccess.IsParentKind(SyntaxKind.SimpleMemberAccessExpression) Then
Return False
End If
nameNode = TryCast(node, SimpleNameSyntax)
If nameNode Is Nothing Then
Return False
End If
Return True
End Function
Protected Overrides Function CanAddImportForNamespace(diagnosticId As String, node As SyntaxNode, ByRef nameNode As SimpleNameSyntax) As Boolean
Select Case diagnosticId
Case VisualBasicAddImportCodeFixProvider.BC30002,
VisualBasicAddImportCodeFixProvider.BC30451
Exit Select
Case Else
Return False
End Select
Return CanAddImportForTypeOrNamespaceCore(node, nameNode)
End Function
Protected Overrides Function CanAddImportForDeconstruct(diagnosticId As String, node As SyntaxNode) As Boolean
' Not supported yet.
Return False
End Function
Protected Overrides Function CanAddImportForQuery(diagnosticId As String, node As SyntaxNode) As Boolean
If diagnosticId <> VisualBasicAddImportCodeFixProvider.BC36593 Then
Return False
End If
Dim queryClause = node.GetAncestor(Of QueryExpressionSyntax)()
Return queryClause IsNot Nothing
End Function
Private Function IsOutermostQueryExpression(node As SyntaxNode) As Boolean
' TODO(cyrusn): Figure out how to implement this.
Return True
End Function
Protected Overrides Function CanAddImportForType(
diagnosticId As String, node As SyntaxNode, ByRef nameNode As SimpleNameSyntax) As Boolean
Select Case diagnosticId
Case VisualBasicAddImportCodeFixProvider.BC30002,
VisualBasicAddImportCodeFixProvider.BC30451,
VisualBasicAddImportCodeFixProvider.BC32042,
VisualBasicAddImportCodeFixProvider.BC32045,
VisualBasicAddImportCodeFixProvider.BC30389,
VisualBasicAddImportCodeFixProvider.BC31504,
VisualBasicAddImportCodeFixProvider.BC36610,
VisualBasicAddImportCodeFixProvider.BC30182
Exit Select
Case VisualBasicAddImportCodeFixProvider.BC42309
Select Case node.Kind
Case SyntaxKind.XmlCrefAttribute
node = CType(node, XmlCrefAttributeSyntax).Reference.DescendantNodes().OfType(Of IdentifierNameSyntax).FirstOrDefault()
Case SyntaxKind.CrefReference
node = CType(node, CrefReferenceSyntax).DescendantNodes().OfType(Of IdentifierNameSyntax).FirstOrDefault()
End Select
Case Else
Return False
End Select
Return CanAddImportForTypeOrNamespaceCore(node, nameNode)
End Function
Private Shared Function CanAddImportForTypeOrNamespaceCore(node As SyntaxNode, ByRef nameNode As SimpleNameSyntax) As Boolean
Dim qn = TryCast(node, QualifiedNameSyntax)
If qn IsNot Nothing Then
node = GetLeftMostSimpleName(qn)
End If
nameNode = TryCast(node, SimpleNameSyntax)
Return nameNode.LooksLikeStandaloneTypeName()
End Function
Private Shared Function GetLeftMostSimpleName(qn As QualifiedNameSyntax) As SimpleNameSyntax
While (qn IsNot Nothing)
Dim left = qn.Left
Dim simpleName = TryCast(left, SimpleNameSyntax)
If simpleName IsNot Nothing Then
Return simpleName
End If
qn = TryCast(left, QualifiedNameSyntax)
End While
Return Nothing
End Function
Protected Overrides Function GetDescription(nameParts As IReadOnlyList(Of String)) As String
Return $"Imports { String.Join(".", nameParts) }"
End Function
Protected Overrides Function GetDescription(
document As Document,
symbol As INamespaceOrTypeSymbol,
semanticModel As SemanticModel,
root As SyntaxNode,
cancellationToken As CancellationToken) As (description As String, hasExistingImport As Boolean)
Dim importsStatement = GetImportsStatement(symbol)
Dim addImportService = document.GetLanguageService(Of IAddImportsService)
Return ($"Imports {symbol.ToDisplayString()}",
addImportService.HasExistingImport(semanticModel.Compilation, root, root, importsStatement))
End Function
Private Function GetImportsStatement(symbol As INamespaceOrTypeSymbol) As ImportsStatementSyntax
Dim nameSyntax = DirectCast(symbol.GenerateTypeSyntax(addGlobal:=False), NameSyntax)
Return GetImportsStatement(nameSyntax)
End Function
Private Function GetImportsStatement(nameSyntax As NameSyntax) As ImportsStatementSyntax
nameSyntax = nameSyntax.WithAdditionalAnnotations(Simplifier.Annotation)
Dim memberImportsClause = SyntaxFactory.SimpleImportsClause(nameSyntax)
Dim newImport = SyntaxFactory.ImportsStatement(
importsClauses:=SyntaxFactory.SingletonSeparatedList(Of ImportsClauseSyntax)(memberImportsClause))
Return newImport
End Function
Protected Overrides Function GetImportNamespacesInScope(semanticModel As SemanticModel, node As SyntaxNode, cancellationToken As CancellationToken) As ISet(Of INamespaceSymbol)
Return semanticModel.GetImportNamespacesInScope(node)
End Function
Protected Overrides Function GetDeconstructInfo(semanticModel As SemanticModel, node As SyntaxNode, cancellationToken As CancellationToken) As ITypeSymbol
Return Nothing
End Function
Protected Overrides Function GetQueryClauseInfo(
model As SemanticModel,
node As SyntaxNode,
cancellationToken As CancellationToken) As ITypeSymbol
Dim query = TryCast(node, QueryExpressionSyntax)
If query Is Nothing Then
query = node.GetAncestor(Of QueryExpressionSyntax)()
End If
Dim semanticModel = DirectCast(model, SemanticModel)
For Each clause In query.Clauses
If TypeOf clause Is AggregateClauseSyntax Then
Dim aggregateClause = DirectCast(clause, AggregateClauseSyntax)
Dim aggregateInfo = semanticModel.GetAggregateClauseSymbolInfo(aggregateClause, cancellationToken)
If IsValid(aggregateInfo.Select1) OrElse IsValid(aggregateInfo.Select2) Then
Return Nothing
End If
For Each variable In aggregateClause.AggregationVariables
Dim info = semanticModel.GetSymbolInfo(variable.Aggregation, cancellationToken)
If IsValid(info) Then
Return Nothing
End If
Next
Else
Dim symbolInfo = semanticModel.GetSymbolInfo(clause, cancellationToken)
If IsValid(symbolInfo) Then
Return Nothing
End If
End If
Next
Dim type As ITypeSymbol
Dim fromOrAggregateClause = query.Clauses.First()
If TypeOf fromOrAggregateClause Is FromClauseSyntax Then
Dim fromClause = DirectCast(fromOrAggregateClause, FromClauseSyntax)
type = semanticModel.GetTypeInfo(fromClause.Variables.First().Expression, cancellationToken).Type
Else
Dim aggregateClause = DirectCast(fromOrAggregateClause, AggregateClauseSyntax)
type = semanticModel.GetTypeInfo(aggregateClause.Variables.First().Expression, cancellationToken).Type
End If
Return type
End Function
Private Function IsValid(info As SymbolInfo) As Boolean
Dim symbol = info.Symbol.GetOriginalUnreducedDefinition()
Return symbol IsNot Nothing AndAlso symbol.Locations.Length > 0
End Function
Protected Overloads Overrides Async Function AddImportAsync(
contextNode As SyntaxNode,
symbol As INamespaceOrTypeSymbol,
document As Document,
placeSystemNamespaceFirst As Boolean,
cancellationToken As CancellationToken) As Task(Of Document)
Dim importsStatement = GetImportsStatement(symbol)
Return Await AddImportAsync(
contextNode, document, placeSystemNamespaceFirst,
importsStatement, cancellationToken).ConfigureAwait(False)
End Function
Private Overloads Shared Async Function AddImportAsync(
contextNode As SyntaxNode, document As Document, placeSystemNamespaceFirst As Boolean,
importsStatement As ImportsStatementSyntax, cancellationToken As CancellationToken) As Task(Of Document)
Dim compilation = Await document.Project.GetCompilationAsync(cancellationToken).ConfigureAwait(False)
Dim importService = document.GetLanguageService(Of IAddImportsService)
Dim root = Await document.GetSyntaxRootAsync(cancellationToken).ConfigureAwait(False)
Dim newRoot = importService.AddImport(compilation, root, contextNode, importsStatement, placeSystemNamespaceFirst)
newRoot = newRoot.WithAdditionalAnnotations(CaseCorrector.Annotation, Formatter.Annotation)
Dim newDocument = document.WithSyntaxRoot(newRoot)
Return newDocument
End Function
Protected Overrides Function AddImportAsync(
contextNode As SyntaxNode,
nameSpaceParts As IReadOnlyList(Of String),
Document As Document,
placeSystemNamespaceFirst As Boolean,
cancellationToken As CancellationToken) As Task(Of Document)
Dim nameSyntax = CreateNameSyntax(nameSpaceParts, nameSpaceParts.Count - 1)
Dim importsStatement = GetImportsStatement(nameSyntax)
Return AddImportAsync(
contextNode, Document, placeSystemNamespaceFirst,
importsStatement, cancellationToken)
End Function
Private Function CreateNameSyntax(nameSpaceParts As IReadOnlyList(Of String), index As Integer) As NameSyntax
Dim namePiece = SyntaxFactory.IdentifierName(nameSpaceParts(index))
Return If(index = 0,
DirectCast(namePiece, NameSyntax),
SyntaxFactory.QualifiedName(CreateNameSyntax(nameSpaceParts, index - 1), namePiece))
End Function
Protected Overrides Function IsViableExtensionMethod(method As IMethodSymbol,
expression As SyntaxNode,
semanticModel As SemanticModel,
syntaxFacts As ISyntaxFactsService,
cancellationToken As CancellationToken) As Boolean
Dim leftExpressionType As ITypeSymbol = Nothing
If syntaxFacts.IsInvocationExpression(expression) Then
leftExpressionType = semanticModel.GetEnclosingNamedType(expression.SpanStart, cancellationToken)
Else
Dim leftExpression As SyntaxNode
If TypeOf expression Is ObjectCreationExpressionSyntax Then
leftExpression = expression
Else
leftExpression = syntaxFacts.GetExpressionOfMemberAccessExpression(
expression, allowImplicitTarget:=True)
If leftExpression Is Nothing Then
Return False
End If
End If
Dim semanticInfo = semanticModel.GetTypeInfo(leftExpression, cancellationToken)
leftExpressionType = semanticInfo.Type
End If
Return IsViableExtensionMethod(method, leftExpressionType)
End Function
Friend Overrides Function IsAddMethodContext(node As SyntaxNode, semanticModel As SemanticModel) As Boolean
If node.IsKind(SyntaxKind.ObjectCollectionInitializer) Then
Dim objectCreateExpression = node.GetAncestor(Of ObjectCreationExpressionSyntax)
If objectCreateExpression Is Nothing Then
Return False
End If
Return True
End If
Return False
End Function
End Class
End Namespace
\ No newline at end of file
src/Features/VisualBasic/Portable/AddImport/VisualBasicAddImportFeatureService.vb 0 → 100644
浏览文件 @ 9b8d0b9a
' Copyright (c) Microsoft. All Rights Reserved. Licensed under the Apache License, Version 2.0. See License.txt in the project root for license information.
Imports System.Composition
Imports System.Threading
Imports Microsoft.CodeAnalysis.AddImport
Imports Microsoft.CodeAnalysis.AddImports
Imports Microsoft.CodeAnalysis.CaseCorrection
Imports Microsoft.CodeAnalysis.Formatting
Imports Microsoft.CodeAnalysis.Host.Mef
Imports Microsoft.CodeAnalysis.LanguageServices
Imports Microsoft.CodeAnalysis.Simplification
Imports Microsoft.CodeAnalysis.VisualBasic.Syntax
Namespace Microsoft.CodeAnalysis.VisualBasic.AddImport
<ExportLanguageService(GetType(IAddImportFeatureService), LanguageNames.VisualBasic), [Shared]>
Friend Class VisualBasicAddImportFeatureService
Inherits AbstractAddImportFeatureService(Of SimpleNameSyntax)
Protected Overrides Function CanAddImport(node As SyntaxNode, cancellationToken As CancellationToken) As Boolean
If node.GetAncestor(Of ImportsStatementSyntax)() IsNot Nothing Then
Return False
End If
Return node.CanAddImportsStatements(cancellationToken)
End Function
Protected Overrides Function CanAddImportForMethod(
diagnosticId As String,
syntaxFacts As ISyntaxFactsService,
node As SyntaxNode,
ByRef nameNode As SimpleNameSyntax) As Boolean
Select Case diagnosticId
Case VisualBasicAddImportCodeFixProvider.BC30456,
VisualBasicAddImportCodeFixProvider.BC30390,
VisualBasicAddImportCodeFixProvider.BC42309,
VisualBasicAddImportCodeFixProvider.BC30451
Exit Select
Case VisualBasicAddImportCodeFixProvider.BC30512
' look up its corresponding method name
Dim parent = node.GetAncestor(Of InvocationExpressionSyntax)()
If parent Is Nothing Then
Return False
End If
Dim method = TryCast(parent.Expression, MemberAccessExpressionSyntax)
If method IsNot Nothing Then
node = method.Name
Else
node = parent.Expression
End If
Exit Select
Case VisualBasicAddImportCodeFixProvider.BC36719
If node.IsKind(SyntaxKind.ObjectCollectionInitializer) Then
Return True
End If
Return False
Case VisualBasicAddImportCodeFixProvider.BC32016
Dim memberAccessName = TryCast(node, MemberAccessExpressionSyntax)?.Name
Dim conditionalAccessName = TryCast(TryCast(TryCast(node, ConditionalAccessExpressionSyntax)?.WhenNotNull, InvocationExpressionSyntax)?.Expression, MemberAccessExpressionSyntax)?.Name
If memberAccessName Is Nothing AndAlso conditionalAccessName Is Nothing Then
Return False
End If
node = If(memberAccessName Is Nothing, conditionalAccessName, memberAccessName)
Exit Select
Case Else
Return False
End Select
Dim memberAccess = TryCast(node, MemberAccessExpressionSyntax)
If memberAccess IsNot Nothing Then
node = memberAccess.Name
End If
If memberAccess.IsParentKind(SyntaxKind.SimpleMemberAccessExpression) Then
Return False
End If
nameNode = TryCast(node, SimpleNameSyntax)
If nameNode Is Nothing Then
Return False
End If
Return True
End Function
Protected Overrides Function CanAddImportForNamespace(diagnosticId As String, node As SyntaxNode, ByRef nameNode As SimpleNameSyntax) As Boolean
Select Case diagnosticId
Case VisualBasicAddImportCodeFixProvider.BC30002,
VisualBasicAddImportCodeFixProvider.BC30451
Exit Select
Case Else
Return False
End Select
Return CanAddImportForTypeOrNamespaceCore(node, nameNode)
End Function
Protected Overrides Function CanAddImportForDeconstruct(diagnosticId As String, node As SyntaxNode) As Boolean
' Not supported yet.
Return False
End Function
Protected Overrides Function CanAddImportForQuery(diagnosticId As String, node As SyntaxNode) As Boolean
If diagnosticId <> VisualBasicAddImportCodeFixProvider.BC36593 Then
Return False
End If
Dim queryClause = node.GetAncestor(Of QueryExpressionSyntax)()
Return queryClause IsNot Nothing
End Function
Private Function IsOutermostQueryExpression(node As SyntaxNode) As Boolean
' TODO(cyrusn): Figure out how to implement this.
Return True
End Function
Protected Overrides Function CanAddImportForType(
diagnosticId As String, node As SyntaxNode, ByRef nameNode As SimpleNameSyntax) As Boolean
Select Case diagnosticId
Case VisualBasicAddImportCodeFixProvider.BC30002,
VisualBasicAddImportCodeFixProvider.BC30451,
VisualBasicAddImportCodeFixProvider.BC32042,
VisualBasicAddImportCodeFixProvider.BC32045,
VisualBasicAddImportCodeFixProvider.BC30389,
VisualBasicAddImportCodeFixProvider.BC31504,
VisualBasicAddImportCodeFixProvider.BC36610,
VisualBasicAddImportCodeFixProvider.BC30182
Exit Select
Case VisualBasicAddImportCodeFixProvider.BC42309
Select Case node.Kind
Case SyntaxKind.XmlCrefAttribute
node = CType(node, XmlCrefAttributeSyntax).Reference.DescendantNodes().OfType(Of IdentifierNameSyntax).FirstOrDefault()
Case SyntaxKind.CrefReference
node = CType(node, CrefReferenceSyntax).DescendantNodes().OfType(Of IdentifierNameSyntax).FirstOrDefault()
End Select
Case Else
Return False
End Select
Return CanAddImportForTypeOrNamespaceCore(node, nameNode)
End Function
Private Shared Function CanAddImportForTypeOrNamespaceCore(node As SyntaxNode, ByRef nameNode As SimpleNameSyntax) As Boolean
Dim qn = TryCast(node, QualifiedNameSyntax)
If qn IsNot Nothing Then
node = GetLeftMostSimpleName(qn)
End If
nameNode = TryCast(node, SimpleNameSyntax)
Return nameNode.LooksLikeStandaloneTypeName()
End Function
Private Shared Function GetLeftMostSimpleName(qn As QualifiedNameSyntax) As SimpleNameSyntax
While (qn IsNot Nothing)
Dim left = qn.Left
Dim simpleName = TryCast(left, SimpleNameSyntax)
If simpleName IsNot Nothing Then
Return simpleName
End If
qn = TryCast(left, QualifiedNameSyntax)
End While
Return Nothing
End Function
Protected Overrides Function GetDescription(nameParts As IReadOnlyList(Of String)) As String
Return $"Imports { String.Join(".", nameParts) }"
End Function
Protected Overrides Function GetDescription(
document As Document,
symbol As INamespaceOrTypeSymbol,
semanticModel As SemanticModel,
root As SyntaxNode,
cancellationToken As CancellationToken) As (description As String, hasExistingImport As Boolean)
Dim importsStatement = GetImportsStatement(symbol)
Dim addImportService = document.GetLanguageService(Of IAddImportsService)
Return ($"Imports {symbol.ToDisplayString()}",
addImportService.HasExistingImport(semanticModel.Compilation, root, root, importsStatement))
End Function
Private Function GetImportsStatement(symbol As INamespaceOrTypeSymbol) As ImportsStatementSyntax
Dim nameSyntax = DirectCast(symbol.GenerateTypeSyntax(addGlobal:=False), NameSyntax)
Return GetImportsStatement(nameSyntax)
End Function
Private Function GetImportsStatement(nameSyntax As NameSyntax) As ImportsStatementSyntax
nameSyntax = nameSyntax.WithAdditionalAnnotations(Simplifier.Annotation)
Dim memberImportsClause = SyntaxFactory.SimpleImportsClause(nameSyntax)
Dim newImport = SyntaxFactory.ImportsStatement(
importsClauses:=SyntaxFactory.SingletonSeparatedList(Of ImportsClauseSyntax)(memberImportsClause))
Return newImport
End Function
Protected Overrides Function GetImportNamespacesInScope(semanticModel As SemanticModel, node As SyntaxNode, cancellationToken As CancellationToken) As ISet(Of INamespaceSymbol)
Return semanticModel.GetImportNamespacesInScope(node)
End Function
Protected Overrides Function GetDeconstructInfo(semanticModel As SemanticModel, node As SyntaxNode, cancellationToken As CancellationToken) As ITypeSymbol
Return Nothing
End Function
Protected Overrides Function GetQueryClauseInfo(
model As SemanticModel,
node As SyntaxNode,
cancellationToken As CancellationToken) As ITypeSymbol
Dim query = TryCast(node, QueryExpressionSyntax)
If query Is Nothing Then
query = node.GetAncestor(Of QueryExpressionSyntax)()
End If
Dim semanticModel = DirectCast(model, SemanticModel)
For Each clause In query.Clauses
If TypeOf clause Is AggregateClauseSyntax Then
Dim aggregateClause = DirectCast(clause, AggregateClauseSyntax)
Dim aggregateInfo = semanticModel.GetAggregateClauseSymbolInfo(aggregateClause, cancellationToken)
If IsValid(aggregateInfo.Select1) OrElse IsValid(aggregateInfo.Select2) Then
Return Nothing
End If
For Each variable In aggregateClause.AggregationVariables
Dim info = semanticModel.GetSymbolInfo(variable.Aggregation, cancellationToken)
If IsValid(info) Then
Return Nothing
End If
Next
Else
Dim symbolInfo = semanticModel.GetSymbolInfo(clause, cancellationToken)
If IsValid(symbolInfo) Then
Return Nothing
End If
End If
Next
Dim type As ITypeSymbol
Dim fromOrAggregateClause = query.Clauses.First()
If TypeOf fromOrAggregateClause Is FromClauseSyntax Then
Dim fromClause = DirectCast(fromOrAggregateClause, FromClauseSyntax)
type = semanticModel.GetTypeInfo(fromClause.Variables.First().Expression, cancellationToken).Type
Else
Dim aggregateClause = DirectCast(fromOrAggregateClause, AggregateClauseSyntax)
type = semanticModel.GetTypeInfo(aggregateClause.Variables.First().Expression, cancellationToken).Type
End If
Return type
End Function
Private Function IsValid(info As SymbolInfo) As Boolean
Dim symbol = info.Symbol.GetOriginalUnreducedDefinition()
Return symbol IsNot Nothing AndAlso symbol.Locations.Length > 0
End Function
Protected Overloads Overrides Async Function AddImportAsync(
contextNode As SyntaxNode,
symbol As INamespaceOrTypeSymbol,
document As Document,
placeSystemNamespaceFirst As Boolean,
cancellationToken As CancellationToken) As Task(Of Document)
Dim importsStatement = GetImportsStatement(symbol)
Return Await AddImportAsync(
contextNode, document, placeSystemNamespaceFirst,
importsStatement, cancellationToken).ConfigureAwait(False)
End Function
Private Overloads Shared Async Function AddImportAsync(
contextNode As SyntaxNode, document As Document, placeSystemNamespaceFirst As Boolean,
importsStatement As ImportsStatementSyntax, cancellationToken As CancellationToken) As Task(Of Document)
Dim compilation = Await document.Project.GetCompilationAsync(cancellationToken).ConfigureAwait(False)
Dim importService = document.GetLanguageService(Of IAddImportsService)
Dim root = Await document.GetSyntaxRootAsync(cancellationToken).ConfigureAwait(False)
Dim newRoot = importService.AddImport(compilation, root, contextNode, importsStatement, placeSystemNamespaceFirst)
newRoot = newRoot.WithAdditionalAnnotations(CaseCorrector.Annotation, Formatter.Annotation)
Dim newDocument = document.WithSyntaxRoot(newRoot)
Return newDocument
End Function
Protected Overrides Function AddImportAsync(
contextNode As SyntaxNode,
nameSpaceParts As IReadOnlyList(Of String),
Document As Document,
placeSystemNamespaceFirst As Boolean,
cancellationToken As CancellationToken) As Task(Of Document)
Dim nameSyntax = CreateNameSyntax(nameSpaceParts, nameSpaceParts.Count - 1)
Dim importsStatement = GetImportsStatement(nameSyntax)
Return AddImportAsync(
contextNode, Document, placeSystemNamespaceFirst,
importsStatement, cancellationToken)
End Function
Private Function CreateNameSyntax(nameSpaceParts As IReadOnlyList(Of String), index As Integer) As NameSyntax
Dim namePiece = SyntaxFactory.IdentifierName(nameSpaceParts(index))
Return If(index = 0,
DirectCast(namePiece, NameSyntax),
SyntaxFactory.QualifiedName(CreateNameSyntax(nameSpaceParts, index - 1), namePiece))
End Function
Protected Overrides Function IsViableExtensionMethod(method As IMethodSymbol,
expression As SyntaxNode,
semanticModel As SemanticModel,
syntaxFacts As ISyntaxFactsService,
cancellationToken As CancellationToken) As Boolean
Dim leftExpressionType As ITypeSymbol = Nothing
If syntaxFacts.IsInvocationExpression(expression) Then
leftExpressionType = semanticModel.GetEnclosingNamedType(expression.SpanStart, cancellationToken)
Else
Dim leftExpression As SyntaxNode
If TypeOf expression Is ObjectCreationExpressionSyntax Then
leftExpression = expression
Else
leftExpression = syntaxFacts.GetExpressionOfMemberAccessExpression(
expression, allowImplicitTarget:=True)
If leftExpression Is Nothing Then
Return False
End If
End If
Dim semanticInfo = semanticModel.GetTypeInfo(leftExpression, cancellationToken)
leftExpressionType = semanticInfo.Type
End If
Return IsViableExtensionMethod(method, leftExpressionType)
End Function
Friend Overrides Function IsAddMethodContext(node As SyntaxNode, semanticModel As SemanticModel) As Boolean
If node.IsKind(SyntaxKind.ObjectCollectionInitializer) Then
Dim objectCreateExpression = node.GetAncestor(Of ObjectCreationExpressionSyntax)
If objectCreateExpression Is Nothing Then
Return False
End If
Return True
End If
Return False
End Function
End Class
End Namespace
\ No newline at end of file
src/Features/VisualBasic/Portable/BasicFeatures.vbproj
浏览文件 @ 9b8d0b9a
......@@ -64,6 +64,7 @@
<Compile Include="..\..\..\Compilers\VisualBasic\Portable\Syntax\LambdaUtilities.vb">
<Link>InternalUtilities\LambdaUtilities.vb</Link>
</Compile>
<Compile Include="AddImport\VisualBasicAddImportFeatureService.vb" />
<Compile Include="AddPackage\VisualBasicAddSpecificPackageCodeFixProvider.vb" />
<Compile Include="DocumentHighlighting\VisualBasicDocumentHighlightsService.vb" />
<Compile Include="InitializeParameter\InitializeParameterHelpers.vb" />
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或