使用C# Roslyn为.cs文件内自定义方法及调用统一添加前缀

C# 方法名批量加前缀重构解决方案 #

你现有代码的核心问题是缺少两个关键逻辑：一是没有预先收集当前类下需要加前缀的原始方法名，无法区分要修改的本类方法和不需要修改的系统/第三方方法；二是VisitInvocationExpression方法为空，没有实现调用点位的修改逻辑。

以下是可直接运行的修改后完整代码：

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

namespace CodeScanner { 
  internal sealed class Fixer : CSharpSyntaxRewriter {
    // 存储当前类下所有需要加前缀的原始方法名
    private readonly HashSet<string> _currentClassMethodNames = new HashSet<string>();

    public override SyntaxNode VisitClassDeclaration(ClassDeclarationSyntax node)
    {
        // 进入类时先收集所有本类定义的方法名
        _currentClassMethodNames.Clear();
        foreach (var member in node.Members)
        {
            if (member is MethodDeclarationSyntax method)
            {
                _currentClassMethodNames.Add(method.Identifier.ValueText);
            }
        }
        // 再遍历类内部的所有节点
        return base.VisitClassDeclaration(node);
    }

    public override SyntaxNode VisitInvocationExpression(InvocationExpressionSyntax node) {
        // 只处理直接调用本类方法的场景（即没有前缀的方法调用，如test1()，而非xxx.test1()）
        if (node.Expression is IdentifierNameSyntax identifierNode 
            && _currentClassMethodNames.Contains(identifierNode.Identifier.ValueText))
        {
            // 生成新的方法名标识符，保留原有格式信息
            var newIdentifier = SyntaxFactory.Identifier(
                identifierNode.Identifier.LeadingTrivia,
                "prefix_" + identifierNode.Identifier.ValueText,
                identifierNode.Identifier.TrailingTrivia);
            // 替换标识符后返回新的调用表达式
            var newIdentifierNode = identifierNode.WithIdentifier(newIdentifier);
            return node.WithExpression(newIdentifierNode);
        }
        return base.VisitInvocationExpression(node);
    }

    public override SyntaxNode VisitMethodDeclaration(MethodDeclarationSyntax node) {
        // 生成带前缀的新方法名，保留原有格式信息
        var newIdentifier = SyntaxFactory.Identifier(
            node.Identifier.LeadingTrivia,
            "prefix_" + node.Identifier.ValueText,
            node.Identifier.TrailingTrivia);
        return node.WithIdentifier(newIdentifier);
    }
  }

  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());
    }
  }
}

改动说明

• 新增方法名收集逻辑：进入类声明时先收集该类下所有原始方法名，避免误改Console.WriteLine这类系统方法的调用
• 完善调用点位修改逻辑：匹配到本类的无前缀直接调用时，同步修改方法名
• 保留原代码格式：所有修改都保留了原有节点的前后空白、缩进等格式信息，不会破坏原代码的排版
  运行上述代码即可得到你期望的重构结果。

内容的提问来源于stack exchange，提问作者James W.