使用Roslyn将类成员添加到特定位置?

时间:2012-09-12 22:28:16

标签: c# roslyn

我使用ClassDeclarationSyntax.AddMembers方法将私有字段添加到类中。字段出现在类中,但我想知道如何将字段添加到特定位置。截至目前,它们是在#if指令内的类末尾添加的,在运行代码生成时恰好评估为true。

运行代码:

var tree = SyntaxTree.ParseCompilationUnit(@"
namespace Test
{
    public class A
    {
#if !SILVERLIGHT
        public int someField;
#endif
    }
}");
var field =
    Syntax.FieldDeclaration(
        Syntax.VariableDeclaration(
            Syntax.PredefinedType(
                Syntax.Token(
                    SyntaxKind.StringKeyword))))
    .WithModifiers(Syntax.Token(SyntaxKind.PrivateKeyword))
    .AddDeclarationVariables(Syntax.VariableDeclarator("myAddedField"));
var theClass = tree.GetRoot().DescendantNodes()
    .OfType<ClassDeclarationSyntax>().First();
theClass = theClass.AddMembers(field).NormalizeWhitespace();
System.Diagnostics.Debug.Write(theClass.GetFullText());

将导致:

public class A
{
#if !SILVERLIGHT
    public int someField;
    private string myAddedField;
#endif
}

我想得到这个结果:

public class A
{
    private string myAddedField;
#if !SILVERLIGHT
    public int someField;
#endif
}

1 个答案:

答案 0 :(得分:7)

为此,您必须找到您想要放置新成员的确切位置,然后相应地修改类成员列表。类似的东西:

private static SyntaxList<MemberDeclarationSyntax> AddBeforeIfDirective(
    SyntaxList<MemberDeclarationSyntax> oldMembers,
    MemberDeclarationSyntax newMember)
{
    var ifIndex = oldMembers.IndexOf(
        member => member.GetLeadingTrivia()
            .Any(t => t.Kind == SyntaxKind.IfDirective));
    return oldMembers.Insert(ifIndex, newMember);
}


…

var newMembers = AddBeforeIfDirective(theClass.Members, field);

theClass = theClass.WithMembers(newMembers).NormalizeWhitespace();