Allow descendants of nodes to be traversed

1 files changed, 8 insertions, 1 deletions

diff --git a/src/main/java/org/zwobble/couscous/ast/FormalArgumentNode.java b/src/main/java/org/zwobble/couscous/ast/FormalArgumentNode.java
index 80c3273..41fb982 100644
--- a/src/main/java/org/zwobble/couscous/ast/FormalArgumentNode.java
+++ b/src/main/java/org/zwobble/couscous/ast/FormalArgumentNode.java
@@ -1,6 +1,8 @@
 package org.zwobble.couscous.ast;
 
-public class FormalArgumentNode implements VariableNode {
+import org.zwobble.couscous.ast.visitors.NodeMapper;
+
+public class FormalArgumentNode implements VariableNode, Node {
     public static FormalArgumentNode formalArg(VariableDeclaration declaration) {
         return new FormalArgumentNode(declaration);
     }
@@ -24,6 +26,11 @@ public class FormalArgumentNode implements VariableNode {
     }
 
     @Override
+    public <T> T accept(NodeMapper<T> visitor) {
+        return visitor.visit(this);
+    }
+
+    @Override
     public String toString() {
         return "FormalArgumentNode(declaration=" + declaration + ")";
     }