我已经整理了一个带有诊断功能的代码修复程序。
以下是有趣的部分:
ISyntaxNodeAnalyzer中AnalyzeNode的实现
public void AnalyzeNode(SyntaxNode node, SemanticModel semanticModel, Action<Diagnostic> public void AnalyzeNode(SyntaxNode node, SemanticModel semanticModel, Action<Diagnostic> addDiagnostic, CancellationToken cancellationToken)
{
var localDeclaration = (LocalDeclarationStatementSyntax)node;
if (localDeclaration.Declaration.Type.IsVar) return;
var variable = localDeclaration.Declaration.Variables.First();
var initialiser = variable.Initializer;
if (initialiser == null) return;
var variableTypeName = localDeclaration.Declaration.Type;
var variableType = semanticModel.GetTypeInfo(variableTypeName).ConvertedType;
var initialiserInfo = semanticModel.GetTypeInfo(variable.Initializer.Value);
var typeOfRightHandSideOfDeclaration = initialiserInfo.Type;
if (Equals(variableType, typeOfRightHandSideOfDeclaration))
{
addDiagnostic(Diagnostic.Create(Rule, node.GetLocation(), localDeclaration.Declaration.Variables.First().Identifier.Value));
}
}
这实质上是查看声明两边的类型,如果它们相同(并且 RHS 还不是 var),则添加诊断。
这是代码修复的代码:
public async Task<IEnumerable<CodeAction>> GetFixesAsync(Document document, TextSpan span, IEnumerable<Diagnostic> diagnostics, CancellationToken cancellationToken)
{
var root = await document.GetSyntaxRootAsync(cancellationToken).ConfigureAwait(false);
var diagnosticSpan = diagnostics.First().Location.SourceSpan;
var declaration = root.FindToken(diagnosticSpan.Start).Parent.AncestorsAndSelf().OfType<LocalDeclarationStatementSyntax>().First();
return new[] { CodeAction.Create("Use var", c => ChangeDeclarationToVar(document, declaration, c)) };
}
private async Task<Document> ChangeDeclarationToVar(Document document, LocalDeclarationStatementSyntax localDeclaration, CancellationToken cancellationToken)
{
var root = await document.GetSyntaxRootAsync(cancellationToken);
var variableTypeName = localDeclaration.Declaration.Type;
var varTypeName = SyntaxFactory.IdentifierName("var").WithAdditionalAnnotations(Formatter.Annotation);
var newDeclaration = localDeclaration.ReplaceNode(variableTypeName, varTypeName);
var newRoot = root.ReplaceNode(localDeclaration, newDeclaration);
return document.WithSyntaxRoot(newRoot);
}
这一点很好很简单,只需从语法工厂中获取 var 并将其切换出来。
请注意, var 在 SyntaxFactory 中没有自己的静态方法,因此是按名称引用。