使用Roslyn更改语法标记

时间:2014-06-26 14:36:26

标签: c# roslyn

我正在尝试用Roslyn重写代码。 我想将GreaterThanToken更改为EqualsEqualsToken。到目前为止,这是我的代码:

ToParse.cs:

public class ToParse
{
    public bool MethodToConvert(int param)
    {
        return (1 > param);
    }
}

的Program.cs:

class Rewriter : SyntaxRewriter
{
    private readonly SyntaxKind _replace;
    private readonly SyntaxKind _replacewith;
    public Rewriter(SyntaxKind replace, SyntaxKind replacewith)
    {
        _replace = replace;
        _replacewith = replacewith;
    }

    public override SyntaxToken VisitToken(SyntaxToken token)
    {
        if (token.Kind != _replace)
            return token;

        return Syntax.Token(_replacewith);
    }
}

用法:

var code = new StreamReader("ToParse.cs").ReadToEnd();
var tree = SyntaxTree.ParseText(code);
var root = tree.GetRoot();
var rewriter = new Rewriter(SyntaxKind.GreaterThanToken, SyntaxKind.EqualsEqualsToken);
var newRoot = rewriter.Visit(root);

var newTree = SyntaxTree.Create((CompilationUnitSyntax)newRoot);
var compilation = Compilation.Create("TestAssembly.dll", 
      new CompilationOptions(OutputKind.DynamicallyLinkedLibrary),
      references: new[]{ new MetadataFileReference(typeof(object).Assembly.Location)},
      syntaxTrees: new[] { newTree });
Console.WriteLine(newTree);
EmitResult res;

using (var file = new FileStream("e:\\TestAssembly.dll", FileMode.Create))
   res = compilation.Emit(file);

执行后,Console.WriteLine打印更改的标记return (1 == param); 但是当我用ilspy打开testassembly.dll时,我仍然看到return 1 > param; 有什么建议吗?

1 个答案:

答案 0 :(得分:6)

[注意:您使用的是较旧版本的Roslyn。这个答案也适用于该版本,但我可以通过更新的名称引用类和成员,以便它们与CodePlex上可用的源匹配。]

您已解析的原始树包含一个BinaryExpressionSyntax节点SyntaxKind GreaterThanExpression。当您使用GreaterThanToken换出EqualsEqualsToken时 在此BinaryExpressionSyntax内,会自动将包含SyntaxNode种类调整为EqualsExpression

因此,您的GreaterThanExpression结果为EqualsEqualsToken。由于这不是可能由编译器本身合法生成的语法树,因此您可能会看到类似的意外行为。

要在这种情况下生成正确的树,我建议通过覆盖CSharpSyntaxRewriter.VisitBinaryExpression并执行以下操作来重写节点本身而不是令牌:

public override SyntaxNode VisitBinaryExpression(BinaryExpressionSyntax node)
{
    if (node.CSharpKind() == SyntaxKind.GreaterThanExpression)
    {
        return SyntaxFactory.BinaryExpression(SyntaxKind.EqualsExpression, node.Left, node.Right);
    }

    return node;
}