求助：用Java AST批量将项目字符串字面量重构为常量的实现问题

先看你的需求：遍历src目录下的Java文件，提取所有字符串字面量替换为常量引用，同时把这些常量统一添加到指定的常量类中，还没法用Eclipse的外部化字符串功能。下面分两部分给你解决——先搞定AST卡住的第2、4步，再聊聊除AST外的替代方案。

一、用AST完成第2步和第4步的具体实现 #

第2步：检查字符串字面量并自动导入常量类 #

要完成这件事，你需要在遍历AST时做两个判断：

1. 标记当前文件是否包含StringLiteral节点（也就是有没有字符串字面量）；
2. 如果有，就给当前编译单元（CompilationUnit）添加常量类的导入语句。

具体代码实现片段：

// 在你的ASTVisitor类里加个标记字段
private boolean hasStringLiterals = false;

@Override
public boolean visit(StringLiteral node) {
    hasStringLiterals = true; // 标记当前文件有字符串
    // 你的其他字符串处理逻辑...
    return super.visit(node);
}

// 遍历完AST后执行导入逻辑（放在parse方法里，ASTVisitor遍历完成后）
if (visitor.hasStringLiterals) {
    AST ast = cu.getAST();
    // 创建导入声明，替换成你的常量类全限定名
    ImportDeclaration importDecl = ast.newImportDeclaration();
    importDecl.setName(ast.newName("com.yourproject.constants.GlobalConstants"));
    // 添加到编译单元的import列表
    cu.imports().add(importDecl);
    
    // 将修改应用到原文件
    IDocument doc = new Document(str);
    TextEdit edits = cu.rewrite(doc, null);
    edits.apply(doc);
    // 这里要把修改后的内容写回原文件（你之前的代码漏掉了这一步）
    Files.write(Paths.get(filePath), doc.get().getBytes(StandardCharsets.UTF_8));
}

第4步：将新常量添加到常量类中 #

这需要单独解析常量类的AST，然后动态添加public static final String类型的字段：

1. 先读取常量类的内容，解析为CompilationUnit；
2. 遍历你收集到的所有字符串常量，生成对应的字段声明；
3. 将字段添加到常量类的类体中；
4. 把修改后的内容写回常量类文件。

代码示例片段：

// 假设你已经收集了需要添加的常量：key是常量名，value是字符串值
Map<String, String> constantsToAdd = new HashMap<>();
constantsToAdd.put("USER_WELCOME_MSG", "Hello, welcome to our system!");
constantsToAdd.put("ERROR_INVALID_INPUT", "Invalid input, please try again.");

// 读取并解析常量类
String constantsContent = readFileToString("src/com/yourproject/constants/GlobalConstants.java");
ASTParser parser = ASTParser.newParser(AST.JLS3);
parser.setSource(constantsContent.toCharArray());
CompilationUnit constantsCu = (CompilationUnit) parser.createAST(null);

// 获取常量类的TypeDeclaration节点
TypeDeclaration constantsClass = (TypeDeclaration) constantsCu.types().get(0);
AST ast = constantsCu.getAST();

for (Map.Entry<String, String> entry : constantsToAdd.entrySet()) {
    // 创建字段声明：public static final String CONST_NAME = "value";
    FieldDeclaration fieldDecl = ast.newFieldDeclaration();
    // 设置修饰符
    fieldDecl.modifiers().add(ast.newModifier(ModifierKeyword.PUBLIC_KEYWORD));
    fieldDecl.modifiers().add(ast.newModifier(ModifierKeyword.STATIC_KEYWORD));
    fieldDecl.modifiers().add(ast.newModifier(ModifierKeyword.FINAL_KEYWORD));
    // 设置类型为String
    fieldDecl.setType(ast.newSimpleType(ast.newSimpleName("String")));
    // 设置变量名和初始化值
    VariableDeclarationFragment frag = ast.newVariableDeclarationFragment();
    frag.setName(ast.newSimpleName(entry.getKey()));
    frag.setInitializer(ast.newStringLiteral(entry.getValue()));
    fieldDecl.fragments().add(frag);
    
    // 添加到常量类的类体中
    constantsClass.bodyDeclarations().add(fieldDecl);
}

// 写回常量类文件
IDocument doc = new Document(constantsContent);
TextEdit edits = constantsCu.rewrite(doc, null);
edits.apply(doc);
Files.write(Paths.get("src/com/yourproject/constants/GlobalConstants.java"), 
            doc.get().getBytes(StandardCharsets.UTF_8));

二、除AST外的替代方案 #

如果觉得Eclipse AST学习成本太高，还有两个更易用的选择：

1. 正则表达式（适合简单场景） #

• 用正则匹配字符串字面量："([^"\\]|\\.)*"（注意处理转义字符）；
• 缺点：无法区分注释里的字符串、字符串拼接片段，也没法处理复杂语法场景（比如嵌套在表达式里的字符串）；
• 适合代码风格统一、没有复杂字符串用法的小型项目。

2. JavaParser库（推荐） #

JavaParser比Eclipse AST的API更直观，学习成本低很多，专门用于Java代码的解析和修改。示例流程：

// 解析源文件
CompilationUnit cu = JavaParser.parse(new File("src/com/yourproject/MyService.java"));
// 收集所有字符串字面量
List<StringLiteralExpr> literals = cu.findAll(StringLiteralExpr.class);

if (!literals.isEmpty()) {
    // 添加常量类导入
    cu.addImport("com.yourproject.constants.GlobalConstants");
    // 替换每个字符串为常量引用
    for (StringLiteralExpr literal : literals) {
        // 自己实现常量名生成逻辑，比如把字符串转成大写下划线格式
        String constantName = generateConstantName(literal.getValue());
        literal.replace(new FieldAccessExpr(null, constantName));
    }
    // 写回修改后的源文件
    Files.write(new File("src/com/yourproject/MyService.java").toPath(), 
                cu.toString().getBytes(StandardCharsets.UTF_8));
}

// 常量名生成示例方法
private static String generateConstantName(String value) {
    return value.trim().toUpperCase().replaceAll("[^A-Z0-9]", "_");
}

三、修正你现有代码的关键问题 #

你的代码里有几处需要调整的地方：

1. 要把文件路径传入parse方法，方便修改后写回原文件；
2. 遍历文件时要过滤.java后缀，避免解析非Java文件；
3. ASTRewrite的使用要更规范，确保修改能正确应用。

修正后的核心片段：

// 修改parse方法，添加filePath参数
public static void parse(final String str, final String filePath) {
    // ... 原有解析逻辑 ...
    cu.accept(new ASTVisitor() {
        // ... 其他visit方法 ...
        public boolean visit (StringLiteral node){
            System.out.println(node.getLiteralValue()+" string literal ");
            AST ast = node.getAST();
            ASTRewrite rewrite = ASTRewrite.create(ast);
            
            // 替换为常量引用，这里的常量名要和你添加到常量类的一致
            SimpleName constantName = ast.newSimpleName(generateConstantName(node.getLiteralValue()));
            rewrite.replace(node, constantName, null);
            
            // 应用修改并写回文件
            IDocument doc = new Document(str);
            TextEdit edits = rewrite.rewriteAST(doc, null);
            try {
                edits.apply(doc);
                Files.write(Paths.get(filePath), doc.get().getBytes(StandardCharsets.UTF_8));
            } catch (Exception e) {
                e.printStackTrace();
            }
            return false;
        }
    });
}

// 修改遍历文件的方法
public static void ParseFilesInDir() throws IOException{
    File dirs = new File(".");
    String dirPath = dirs.getCanonicalPath() + File.separator+"src"+File.separator;
    File root = new File(dirPath);
    File[] files = root.listFiles();
    String filePath = null;
    for (File f : files ) {
        filePath = f.getAbsolutePath();
        // 只处理Java文件
        if(f.isFile() && filePath.endsWith(".java")){
            parse(readFileToString(filePath), filePath);
        }
    }
}

内容的提问来源于stack exchange，提问作者Sergio Navarro Carrero