Added some functions to increase usability

toolisticon · Oct 27, 2023 · 554fad9 · 554fad9
1 parent b2fbf8c
commit 554fad9
Show file tree

Hide file tree

Showing 4 changed files with 45 additions and 0 deletions.
diff --git a/tools/src/main/java/io/toolisticon/aptk/tools/TypeMirrorWrapper.java b/tools/src/main/java/io/toolisticon/aptk/tools/TypeMirrorWrapper.java
@@ -898,6 +898,16 @@ public static Optional<TypeElementWrapper> getTypeElement(TypeMirror typeMirror)
         return Optional.empty();
     }
 
+    /**
+     * Gets the TypeElements wrapped TypeMirror.
+     *
+     * @param typeElement the TypeElement to get the TypeMirror for
+     * @return The wrapped TypeMirror of the TypeElement
+     */
+    public static TypeMirrorWrapper wrap(TypeElement typeElement) {
+        return TypeElementWrapper.wrap(typeElement).asType();
+    }
+
     /**
      * Wraps a TypeMirror instance to provide some convenience methods
      *

diff --git a/tools/src/main/java/io/toolisticon/aptk/tools/wrapper/AnnotationMirrorWrapper.java b/tools/src/main/java/io/toolisticon/aptk/tools/wrapper/AnnotationMirrorWrapper.java
@@ -7,6 +7,7 @@
 import javax.lang.model.element.AnnotationValue;
 import javax.lang.model.element.Element;
 import javax.lang.model.element.ElementKind;
+import javax.lang.model.type.DeclaredType;
 import java.lang.annotation.Annotation;
 import java.util.Arrays;
 import java.util.Optional;
@@ -38,6 +39,24 @@ public AnnotationMirror unwrap() {
         return annotationMirror;
     }
 
+    /**
+     * Get DeclaredType of wrapped annotation.
+     * @return the annotations declared type
+     */
+    DeclaredType getAnnotationType() {
+        return this.annotationMirror.getAnnotationType();
+    }
+
+    /**
+     * Get wrapped DeclaredType of wrapped annotation.
+     * @return the annotations declared type
+     */
+    TypeMirrorWrapper getWrappedAnnotationType() {
+        return TypeMirrorWrapper.wrap(getAnnotationType());
+    }
+
+
+
     /**
      * Returns the "value" attribute
      *

diff --git a/tools/src/main/java/io/toolisticon/aptk/tools/wrapper/ElementWrapper.java b/tools/src/main/java/io/toolisticon/aptk/tools/wrapper/ElementWrapper.java
@@ -173,6 +173,14 @@ public FluentElementFilter<Element> filterFlattenedEnclosedElementTree(boolean i
         return FluentElementFilter.createFluentElementFilter(getFlattenedEnclosedElementTree(includeSelf, maxDepth).stream().map(ElementWrapper::unwrap).collect(Collectors.toList()));
     }
 
+    /**
+     * Gets all wrapped annotations of element.
+     * @return a list containing all annotations of the element
+     */
+    public List<AnnotationMirrorWrapper> getAnnotations() {
+        return this.element.getAnnotationMirrors().stream().map(AnnotationMirrorWrapper::wrap).collect(Collectors.toList());
+    }
+
     /**
      * Gets the annotation
      *

diff --git a/tools/src/test/java/io/toolisticon/aptk/tools/wrapper/ElementWrapperTest.java b/tools/src/test/java/io/toolisticon/aptk/tools/wrapper/ElementWrapperTest.java
@@ -185,7 +185,15 @@ public void test_validate() {
 
     }
 
+    @Test
+    public void test_getAnnotations() {
+
+        CompileTestBuilder.unitTest().<TypeElement>defineTestWithPassedInElement(TestClass.class, (processingEnvironment, element) -> {
+            ElementWrapper<TypeElement> unit = ElementWrapper.wrap(element);
+            MatcherAssert.assertThat("Should find annotation", unit.getAnnotations(),Matchers.hasSize(1));
+        }).executeTest();
 
+    }
     @Test
     public void test_getAnnotation_byFqnString() {