handle enum constants

njit-jerse · Aug 14, 2024 · 99515e6 · 99515e6
1 parent 483f32b
commit 99515e6
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/src/main/java/org/checkerframework/specimin/UnusedImportRemoverVisitor.java b/src/main/java/org/checkerframework/specimin/UnusedImportRemoverVisitor.java
@@ -15,6 +15,7 @@
 import com.github.javaparser.ast.visitor.ModifierVisitor;
 import com.github.javaparser.ast.visitor.Visitable;
 import com.github.javaparser.resolution.UnsolvedSymbolException;
+import com.github.javaparser.resolution.declarations.ResolvedEnumConstantDeclaration;
 import com.github.javaparser.resolution.declarations.ResolvedFieldDeclaration;
 import com.github.javaparser.resolution.declarations.ResolvedMethodDeclaration;
 import com.github.javaparser.resolution.declarations.ResolvedValueDeclaration;
@@ -170,6 +171,13 @@ public Visitable visit(NameExpr expr, Void arg) {
       // Check for both cases: java.lang.Math.PI and java.lang.Math.*
       usedImports.add(declaringType + "." + asField.getName());
       usedImports.add(declaringType + ".*");
+    } else if (resolved.isEnumConstant()) {
+      ResolvedEnumConstantDeclaration asEnumConstant = resolved.asEnumConstant();
+      String declaringType = JavaParserUtil.erase(asEnumConstant.getType().describe());
+      // com.example.Enum, com.example.Enum.VALUE, com.example.Enum.*
+      usedImports.add(declaringType);
+      usedImports.add(declaringType + "." + asEnumConstant.getName());
+      usedImports.add(declaringType + ".*");
     }
     return super.visit(expr, arg);
   }