使用 Roslyn 如何使用指令更新类？

Question

刚刚发现罗斯林，请耐心等待。

我想更新班级顶部的 using 指令以包含附加语句，例如：

using System;

public class Foo {

}

应该变成：

using System;
using Custom.Bar;

public class Foo {

}

我看到我可以覆盖 SyntaxRewriter，并且我已经这样做以解决方法级别代码，但我看不到可能让我访问这些 using 指令的覆盖？

谢谢。

编辑：

我找到了这个属性，但是不知道怎么修改。

var tree = document.GetSyntaxTree().GetRoot() as SyntaxNode;

var compilationUnitSyntax = (CompilationUnitSyntax) (tree);

if (compilationUnitSyntax != null)
      compilationUnitSyntax.Usings.Add();

不幸的是 UsingDirectiveSyntax 是内部的，所以我该如何添加一个！ :D

Answer 1

要创建SyntaxNodes，您必须在您的情况下使用Syntax 类工厂方法Syntax.UsingDirective 方法。

您可以使用 AddUsings 方法添加一个新的方法

if (compilationUnitSyntax != null)
{
    var name = Syntax.QualifiedName(Syntax.IdentifierName("Custom"),
                                    Syntax.IdentifierName("Bar"));
    compilationUnitSyntax = compilationUnitSyntax
        .AddUsings(Syntax.UsingDirective(name).NormalizeWhitespace());
}

注意：由于CompilationUnitSyntax 的不变性，您需要使用AddUsings 调用的结果重新分配compilationUnitSyntax 变量。