Allow descendants of nodes to be traversed

1 files changed, 8 insertions, 1 deletions

diff --git a/src/main/java/org/zwobble/couscous/ast/AnnotationNode.java b/src/main/java/org/zwobble/couscous/ast/AnnotationNode.java
index f7a646b..0d135ba 100644
--- a/src/main/java/org/zwobble/couscous/ast/AnnotationNode.java
+++ b/src/main/java/org/zwobble/couscous/ast/AnnotationNode.java
@@ -1,6 +1,8 @@
 package org.zwobble.couscous.ast;
 
-public class AnnotationNode {
+import org.zwobble.couscous.ast.visitors.NodeMapper;
+
+public class AnnotationNode implements Node {
     public static AnnotationNode annotation(TypeName type) {
         return new AnnotationNode(type);
     }
@@ -16,6 +18,11 @@ public class AnnotationNode {
     }
 
     @Override
+    public <T> T accept(NodeMapper<T> visitor) {
+        return visitor.visit(this);
+    }
+
+    @Override
     public String toString() {
         return "AnnotationNode(type=" + type + ")";
     }