C# Roslyn替换方法

Question

我想修改（在前面添加一个前缀）局部声明方法及其在 .cs 文件中的使用情况。最佳做法是什么？目前我的代码只处理声明。

using System;
using System.IO;
using Microsoft.CodeAnalysis;
using Microsoft.CodeAnalysis.CSharp;
using Microsoft.CodeAnalysis.CSharp.Syntax;

namespace CodeScanner { 
  internal sealed class Fixer : CSharpSyntaxRewriter {
    public override SyntaxNode VisitInvocationExpression(InvocationExpressionSyntax node) {
      base.VisitInvocationExpression(node);
      // replace usages
      return node;
    }
    public override SyntaxNode VisitMethodDeclaration(MethodDeclarationSyntax node) {
      base.VisitMethodDeclaration(node);
      return node.ReplaceNode(node, SyntaxFactory.MethodDeclaration(
          node.AttributeLists,
          node.Modifiers,
          node.ReturnType,
          node.ExplicitInterfaceSpecifier,
          SyntaxFactory.Identifier("prefix_" + node.Identifier.Value),
          node.TypeParameterList,
          node.ParameterList,
          node.ConstraintClauses,
          node.Body,
          node.ExpressionBody));
    }
  }

  class Program {
    static void Main(string[] args) {
      var tree = CSharpSyntaxTree.ParseText(File.ReadAllText("./test.cs"));
      var rewriter = new Fixer();
      var result = rewriter.Visit(tree.GetRoot());
      Console.WriteLine(result.ToFullString());
    }
  }
}

输入文件

using System;

namespace TopLevel
{
  class Bar {
    public void test1(){}

    public void test2(){ Console.WriteLine("str"); }

    public void Fizz() {
      Console.WriteLine(test1());
      Console.WriteLine(test1(test2()));
      test2();
    }
  }
}

输出

using System;

namespace TopLevel
{
  class Bar {
    public void prefix_test1(){}

    public void prefix_test2(){ Console.WriteLine("str"); }

    public void prefix_Fizz() {
      Console.WriteLine(test1());
      Console.WriteLine(test1(test2()));
      test2();
    }
  }
}

期望输出（@ Fizz更改）：

using System;

namespace TopLevel
{
  class Bar {
    public void prefix_test1(){}

    public void prefix_test2(){ Console.WriteLine("str"); }

    public void prefix_Fizz() {
      Console.WriteLine(prefix_test1());
      Console.WriteLine(prefix_test1(prefix_test2()));
      prefix_test2();
    }
  }
}

- whoopdedoo

---

你尝试了什么？问题的具体情况是什么？ - Kris Vandermotten

@KrisVandermotten：我看他的问题是方法定义已经被重命名，但对它的引用没有更改。我认为应该将Console.WriteLine(test1());更改为Console.WriteLine(prefix_test1());。 - Le Vu

@LeVu 我明白你的意思，只是我不确定写代码时出了什么问题。这个问题不够精确。 - Kris Vandermotten

我想重命名所有已声明的方法，这不清楚吗？ - whoopdedoo

@whoopdedoo，您的要求很明确。我不清楚的是，具体有什么问题阻止您编写代码。 - Kris Vandermotten

熟悉并具有Roslyn的知识和经验。 - whoopdedoo

Answer 1

我写了一些代码来满足您设置的要求。

请查看.NET Fiddle。

using Microsoft.CodeAnalysis;
using Microsoft.CodeAnalysis.CSharp;
using Microsoft.CodeAnalysis.CSharp.Syntax;
using System;
using System.Collections.Generic;
using System.Diagnostics.CodeAnalysis;
using System.IO;
using static Globals;

tree = CSharpSyntaxTree.ParseText(File.ReadAllText("Test.cs"));
compilation = CSharpCompilation.Create("HelloWorld").AddSyntaxTrees(tree);
model = compilation.GetSemanticModel(tree);
new Finder().Visit(tree.GetRoot());
Console.WriteLine(new Rewriter().Visit(tree.GetRoot()).NormalizeWhitespace().ToFullString());

public static class Globals
{
    public static SyntaxTree tree;
    public static CompilationUnitSyntax root;
    public static CSharpCompilation compilation;
    public static SemanticModel model;
    public static Dictionary<IMethodSymbol, string> renames = new();
}

public sealed class Finder : CSharpSyntaxWalker
{
    public override void VisitMethodDeclaration(MethodDeclarationSyntax node)
    {
        base.VisitMethodDeclaration(node);
        renames.Add(model.GetDeclaredSymbol(node), "prefix_" + node.Identifier.Value);
    }
}

public sealed class Rewriter : CSharpSyntaxRewriter
{
    public override SyntaxNode VisitMethodDeclaration(MethodDeclarationSyntax mds)
    {
        IMethodSymbol symbol = model.GetDeclaredSymbol(mds);
        mds = (MethodDeclarationSyntax)base.VisitMethodDeclaration(mds);
        if (renames.TryGetValue(symbol, out string newName))
            mds = mds.ReplaceToken(mds.Identifier, SyntaxFactory.Identifier(newName));
        return mds;
    }

    [return: NotNullIfNotNull("node")]
    public override SyntaxNode Visit(SyntaxNode node)
    {
        node = base.Visit(node);
        if (node is SimpleNameSyntax sns &&
            model.GetSymbolInfo(sns) is { Symbol: IMethodSymbol ms } && renames.TryGetValue(ms, out string newName)
        )
            node = sns.ReplaceToken(sns.Identifier, SyntaxFactory.Identifier(newName));
        return node;
    }
}

- trinalbadger587