在 InvocatioExpression 中插入新的方法调用

Question

我正在编写一个带有提供程序的 Visual Studio 代码修复程序，它将在这种语句中插入 AsEnumerable 调用：

model.clients.Take(10).Select(c => new Client(c));

我能够检测到此类语句，但在 Select 方法之前插入 AsEnumerable 时遇到问题。这是我尝试过的：

        var expressionSyntax = SyntaxFactory.ParseExpression(".AsEnumerable()");
        var memberAccessExpressionSyntax = invocationExpr.Expression as MemberAccessExpressionSyntax;

        var root = await document.GetSyntaxRootAsync(cancellationToken);

        var newRoot = root.InsertNodesAfter(memberAccessExpressionSyntax.Expression, new[] {expressionSyntax});

        return document.WithSyntaxRoot(newRoot);

但我明白了

"System.InvalidOperationException: 指定的项目不是 列表的元素。”

查看堆栈跟踪：https://gist.github.com/anonymous/02c41f232829163b25ae

调用InsertNodesAfter方法的正确方法是什么？

Answer 1

我就是这样解决的：

var memberAccessExpressionSyntax = invocationExpr.Expression as MemberAccessExpressionSyntax;

var root = await document.GetSyntaxRootAsync(cancellationToken);

var accessExpression = SyntaxFactory.MemberAccessExpression(SyntaxKind.SimpleMemberAccessExpression, memberAccessExpressionSyntax.Expression, SyntaxFactory.IdentifierName("AsEnumerable"));

var invocationExpression = SyntaxFactory.InvocationExpression(accessExpression);
var enumerableMemberAccessExpression = memberAccessExpressionSyntax.WithExpression(invocationExpression);

root = root.ReplaceNode(invocationExpr, invocationExpr.WithExpression(enumerableMemberAccessExpression));

Answer 2

InsertNodesAfter 仅适用于列表中的节点。方法调用不是一个列表，而是一个调用表达式树。

您需要将XXX.Select(...) 调用中的表达式替换为XXX.AsEnumerable()（重复使用相同的XXX 节点）。

要了解这些表达式树的工作原理，请使用 Roslyn 语法可视化工具。