替换Roslyn语法树

时间:2015-07-09 11:01:48

标签: c# roslyn

我正在尝试使用roslyn替换语法树中的几个节点。 但它的不变性似乎妨碍了我。

    public static string Rewrite(string content)
    {
        var tree = CSharpSyntaxTree.ParseText(content);
        var root = tree.GetRoot();

        var methods =root
            .DescendantNodes(node=>true)
            .OfType<MethodDeclarationSyntax>()
            .ToList();

        foreach(var method in methods)
        {
            var returnActions = method
                .DescendantNodes(node => true)
                .OfType<BinaryExpressionSyntax>()
                //Ok this is cheating
                .Where(node => node.OperatorToken.ValueText == "==")
                .Where(node => node.Right.ToString() == "\"#exit#\"" || node.Right.ToString() == "\"#break#\"")
                .Select(node => node.Parent as IfStatementSyntax)
                .ToList();

            var lookup = new Dictionary<StatementSyntax,StatementSyntax>();

            if (returnActions.Count > 0)
            {
                foreach(var ifStatement in returnActions)
                {
                    var mainCall = ifStatement.GetPrevious() as ExpressionStatementSyntax;                        
                    var newIfStatement = ifStatement.WithCondition(mainCall.Expression.WithoutTrivia());

                    lookup[mainCall] = null;
                    lookup[ifStatement] = newIfStatement;
                }

                //this only replace some of the nodes
                root = root.ReplaceNodes(lookup.Keys, (s, d) => lookup[s]);
            }
        }

        return root.ToFullString();
    }

问题在于,当我调用root.ReplaceNodes时,只有部分节点被替换。

我猜替换会更改树,以便其他节点不再与原始树匹配,因此无法替换。

但是解决这个问题的最佳方法是什么?

一遍又一遍地循环过程,直到不再发生变化,感觉很蹩脚:)

这些更改可能会嵌套发生,我认为这就是造成问题的原因。 我可以以某种方式对变更集进行排序以解决这个问题,还是有一种惯用的方式来处理这些事情?

1 个答案:

答案 0 :(得分:17)

  

我猜替换会更改树,以便其他节点不再与原始树匹配,因此无法替换。

你是对的。替换节点会创建全新的语法树。以前语法树中的节点无法与这些新语法树进行比较。

将四种更改应用于语法树有四种方法:

  1. 使用DocumentEditor - 请参阅:https://stackoverflow.com/a/30563669/300908
  2. 使用Annotations(第236和240行)
  3. 使用.TrackNodes()
  4. 创建一个CSharpSyntaxRewriter,以自下而上的方式替换节点。我在blog上写过这篇文章。
  5. 在这些选项中,我认为DocumentEditor具有最容易使用的声誉。它很可能是未来应用多种变化的惯用方法。