From bb0b36b86f841ec6340fe6e727ed2ea9db1c2d64 Mon Sep 17 00:00:00 2001
From: Michael Williamson <mike@zwobble.org>
Date: Mon, 4 Jan 2016 18:58:19 +0000
Subject: Capture "this" references

---
 src/main/java/org/zwobble/couscous/ast/VariableReferenceNode.java | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

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

diff --git a/src/main/java/org/zwobble/couscous/ast/VariableReferenceNode.java b/src/main/java/org/zwobble/couscous/ast/VariableReferenceNode.java
index 50462e9..c1b022a 100644
--- a/src/main/java/org/zwobble/couscous/ast/VariableReferenceNode.java
+++ b/src/main/java/org/zwobble/couscous/ast/VariableReferenceNode.java
@@ -6,7 +6,7 @@ import org.zwobble.couscous.ast.visitors.ExpressionNodeMapper;
 
 import java.util.function.Function;
 
-public class VariableReferenceNode implements AssignableExpressionNode {
+public class VariableReferenceNode implements AssignableExpressionNode, ReferenceNode {
     public static VariableReferenceNode reference(VariableNode variable) {
         return new VariableReferenceNode(variable.getDeclaration());
     }
@@ -49,6 +49,11 @@ public class VariableReferenceNode implements AssignableExpressionNode {
         mapper.visit(this);
     }
 
+    @Override
+    public <T> T accept(Visitor<T> visitor) {
+        return visitor.visit(this);
+    }
+
     @Override
     public String toString() {
         return "VariableReferenceNode(referent=" + referent + ")";
-- 
cgit v1.2.3