Add getEnumConstants() method

lukehutch · lukehutch · commit ce52201e3897 · 2021-12-06T11:40:23.000-07:00
diff --git a/src/main/java/io/github/classgraph/ClassInfo.java b/src/main/java/io/github/classgraph/ClassInfo.java
@@ -51,6 +51,7 @@
 
 import io.github.classgraph.Classfile.ClassContainment;
 import io.github.classgraph.Classfile.ClassTypeAnnotationDecorator;
+import io.github.classgraph.FieldInfoList.FieldInfoFilter;
 import nonapi.io.github.classgraph.json.Id;
 import nonapi.io.github.classgraph.scanspec.ScanSpec;
 import nonapi.io.github.classgraph.types.ParseException;
@@ -2666,6 +2667,19 @@ public FieldInfoList getFieldInfo() {
         return fieldInfoList;
     }
 
+    /** Get all enum constants of an enum class (as a list of {@link FieldInfo} objects). */
+    public FieldInfoList getEnumConstants() {
+        if (!isEnum()) {
+            throw new IllegalArgumentException("Class " + getName() + " is not an enum");
+        }
+        return getFieldInfo().filter(new FieldInfoFilter() {
+            @Override
+            public boolean accept(FieldInfo fieldInfo) {
+                return fieldInfo.isEnum();
+            }
+        })
+    }
+    
     /**
      * Returns information on the named field declared by the class, but not by its superclasses. See also:
      * 
diff --git a/src/test/java/io/github/classgraph/features/EnumTest.java b/src/test/java/io/github/classgraph/features/EnumTest.java
@@ -0,0 +1,39 @@
+package io.github.classgraph.features;
+
+import static org.assertj.core.api.Assertions.assertThat;
+
+import org.junit.jupiter.api.Test;
+
+import io.github.classgraph.ClassGraph;
+import io.github.classgraph.ClassInfo;
+import io.github.classgraph.FieldInfo;
+import io.github.classgraph.ScanResult;
+
+/**
+ * Test.
+ */
+public class EnumTest {
+    /** Enum */
+    private static enum MyEnum {
+        A, B, C;
+    }
+
+    /**
+     * Test records (JDK 14+).
+     *
+     * @throws Exception
+     *             the exception
+     */
+    @Test
+    public void recordJar() throws Exception {
+        try (ScanResult scanResult = new ClassGraph().acceptClasses(MyEnum.class.getName()).enableAllInfo()
+                .scan()) {
+            assertThat(scanResult.getAllEnums().size() == 1);
+            final ClassInfo myEnum = scanResult.getAllEnums().get(0);
+            assertThat(myEnum.getName().equals(MyEnum.class.getName()));
+            for (FieldInfo fi : myEnum.getFieldInfo()) {
+                System.out.println(fi);
+            }
+        }
+    }
+}
