From bacc781ea755d3771e1c8d09e1fbfd873872f750 Mon Sep 17 00:00:00 2001
From: Michael Williamson <mike@zwobble.org>
Date: Sun, 15 Nov 2015 15:31:09 +0000
Subject: Allow descendants of nodes to be traversed

---
 src/main/java/org/zwobble/couscous/ast/ClassNode.java | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

(limited to 'src/main/java/org/zwobble/couscous/ast/ClassNode.java')

diff --git a/src/main/java/org/zwobble/couscous/ast/ClassNode.java b/src/main/java/org/zwobble/couscous/ast/ClassNode.java
index d2c64f5..c252bb3 100644
--- a/src/main/java/org/zwobble/couscous/ast/ClassNode.java
+++ b/src/main/java/org/zwobble/couscous/ast/ClassNode.java
@@ -1,8 +1,8 @@
 package org.zwobble.couscous.ast;
 
-import java.util.List;
+import org.zwobble.couscous.ast.visitors.NodeMapper;
 
-import org.zwobble.couscous.ast.visitors.NodeVisitor;
+import java.util.List;
 
 public class ClassNode implements Node {
     public static ClassNodeBuilder builder(String name) {
@@ -54,8 +54,8 @@ public class ClassNode implements Node {
     }
 
     @Override
-    public void accept(NodeVisitor visitor) {
-        visitor.visit(this);
+    public <T> T accept(NodeMapper<T> visitor) {
+        return visitor.visit(this);
     }
 
     @Override
-- 
cgit v1.2.3