Support passing arguments to Python runner

1 files changed, 6 insertions, 2 deletions

diff --git a/src/main/java/org/zwobble/couscous/ast/VariableReferenceNode.java b/src/main/java/org/zwobble/couscous/ast/VariableReferenceNode.java
index 2410549..afd3918 100644
--- a/src/main/java/org/zwobble/couscous/ast/VariableReferenceNode.java
+++ b/src/main/java/org/zwobble/couscous/ast/VariableReferenceNode.java
@@ -7,13 +7,17 @@ import lombok.Value;
 @Value
 public class VariableReferenceNode implements ExpressionNode {
     public static VariableReferenceNode reference(VariableNode variable) {
-        return new VariableReferenceNode(variable.getDeclaration().getId());
+        return new VariableReferenceNode(variable.getDeclaration());
     }
     
-    int referentId;
+    VariableDeclaration referent;
 
     @Override
     public <T> T accept(ExpressionNodeVisitor<T> visitor) {
         return visitor.visit(this);
     }
+    
+    public int getReferentId() {
+        return referent.getId();
+    }
 }