Implement visitors on OperationNode instead of desugarring

1 files changed, 5 insertions, 2 deletions

diff --git a/src/main/java/org/zwobble/couscous/ast/OperationNode.java b/src/main/java/org/zwobble/couscous/ast/OperationNode.java
index db341a0..0840044 100644
--- a/src/main/java/org/zwobble/couscous/ast/OperationNode.java
+++ b/src/main/java/org/zwobble/couscous/ast/OperationNode.java
@@ -28,6 +28,10 @@ public class OperationNode implements ExpressionNode {
             type);
     }
 
+    public List<ExpressionNode> getArguments() {
+        return arguments;
+    }
+
     @Override
     public TypeName getType() {
         return type;
@@ -35,8 +39,7 @@ public class OperationNode implements ExpressionNode {
 
     @Override
     public <T> T accept(ExpressionNodeMapper<T> visitor) {
-        // TODO: Don't desugar
-        return desugar().accept(visitor);
+        return visitor.visit(this);
     }
 
     @Override