Generating constants classes for enum definitions.

Author: gericlockify

annotation-processor/pom.xml

@@ -6,7 +6,7 @@
 
     <groupId>com.cake.clockify</groupId>
     <artifactId>addon-sdk-annotation-processor</artifactId>
-    <version>1.0.6</version>
+    <version>1.0.7</version>
 
     <properties>
         <maven.compiler.source>17</maven.compiler.source>

annotation-processor/src/main/java/com/cake/clockify/annotationprocessor/clockify/ClockifyManifestProcessor.java

@@ -4,6 +4,7 @@
 import com.cake.clockify.annotationprocessor.Utils;
 import com.fasterxml.jackson.databind.JsonNode;
 import com.fasterxml.jackson.databind.ObjectMapper;
+import com.fasterxml.jackson.databind.node.ArrayNode;
 import com.squareup.javapoet.JavaFile;
 
 import javax.lang.model.type.DeclaredType;
@@ -44,6 +45,14 @@ private List<JavaFile> getObjectDefinitions() {
                         definition
                 );
 
+                javaFiles.addAll(processor.process());
+            } else if (isEnumDefinition(node)) {
+                EnumConstantsProcessor processor = new EnumConstantsProcessor(
+                        manifest,
+                        packageName,
+                        definition
+                );
+
                 javaFiles.addAll(processor.process());
             }
         });
@@ -58,4 +67,9 @@ private List<JavaFile> getManifestDefinition() {
     private boolean isObjectDefinition(JsonNode node) {
         return NodeConstants.OBJECT.equals(Utils.getNodeType(node, manifest.get(NodeConstants.DEFINITIONS)));
     }
+
+    private boolean isEnumDefinition(JsonNode node) {
+        String type = Utils.getNodeType(node, manifest.get(NodeConstants.DEFINITIONS));
+        return NodeConstants.STRING.equals(type) && node.get(NodeConstants.ENUM) instanceof ArrayNode;
+    }
 }

annotation-processor/src/main/java/com/cake/clockify/annotationprocessor/clockify/EnumConstantsProcessor.java

@@ -0,0 +1,60 @@
+package com.cake.clockify.annotationprocessor.clockify;
+
+import com.cake.clockify.annotationprocessor.NodeConstants;
+import com.cake.clockify.annotationprocessor.Utils;
+import com.fasterxml.jackson.databind.JsonNode;
+import com.squareup.javapoet.ClassName;
+import com.squareup.javapoet.FieldSpec;
+import com.squareup.javapoet.JavaFile;
+import com.squareup.javapoet.TypeSpec;
+
+import javax.lang.model.element.Modifier;
+import java.util.LinkedList;
+import java.util.List;
+import java.util.Objects;
+
+class EnumConstantsProcessor {
+
+    private final String packageName;
+    private final String definition;
+
+    private final List<String> values = new LinkedList<>();
+
+    public EnumConstantsProcessor(JsonNode manifest, String packageName, String definition) {
+        this.packageName = packageName;
+        this.definition = definition;
+
+        JsonNode enumNode = manifest.get(NodeConstants.DEFINITIONS).get(definition).get(NodeConstants.ENUM);
+
+        Objects.requireNonNull(enumNode);
+        if (!enumNode.isArray()) {
+            throw new IllegalArgumentException("The provided definition is not of enum type.");
+        }
+
+        enumNode.forEach(node -> values.add(node.asText()));
+    }
+
+    public List<JavaFile> process() {
+        return List.of(
+                JavaFile.builder(packageName, TypeSpec.interfaceBuilder(getInterfaceClassName(definition))
+                                .addModifiers(Modifier.PUBLIC)
+                                .addFields(values.stream()
+                                        .map(enumValue -> FieldSpec.builder(
+                                                        String.class,
+                                                        enumValue.replace(' ', '_'),
+                                                        Modifier.PUBLIC,
+                                                        Modifier.STATIC,
+                                                        Modifier.FINAL
+                                                )
+                                                .initializer("\"" + enumValue + "\"")
+                                                .build()).toList()
+                                )
+                                .build())
+                        .build()
+        );
+    }
+
+    private ClassName getInterfaceClassName(String step) {
+        return ClassName.get(packageName, Utils.getDefinitionSimpleClassName(step));
+    }
+}