From 4898724edf343650ffb80792caf9e242e5843059 Mon Sep 17 00:00:00 2001
From: Santo Cariotti <santo@dcariotti.me>
Date: Thu, 27 Jun 2024 19:47:05 +0200
Subject: Fix `for` loop for 1+ params

---
 src/ast/nodes/ExprNode.java | 11 ++++++++++-
 1 file changed, 10 insertions(+), 1 deletion(-)

(limited to 'src/ast/nodes/ExprNode.java')

diff --git a/src/ast/nodes/ExprNode.java b/src/ast/nodes/ExprNode.java
index 62b3a94..781cc76 100644
--- a/src/ast/nodes/ExprNode.java
+++ b/src/ast/nodes/ExprNode.java
@@ -26,6 +26,14 @@ public class ExprNode implements Node {
         this.trailers = trailers;
     }
 
+    public Node getExpr(int i) {
+        if (i >= this.exprs.size()) {
+            return null;
+        }
+
+        return this.exprs.get(i);
+    }
+
     public String getId() {
         if (atom != null) {
             return ((AtomNode) this.atom).getId();
@@ -67,7 +75,8 @@ public class ExprNode implements Node {
                         int argNumber = trailer.getArgumentNumber();
 
                         if (paramNumber != argNumber) {
-                            errors.add(new SemanticError(funName + "() takes " + String.valueOf(paramNumber) + " positional arguments but " + String.valueOf(argNumber) + " were given."));
+                            errors.add(new SemanticError(funName + "() takes " + String.valueOf(paramNumber)
+                                    + " positional arguments but " + String.valueOf(argNumber) + " were given."));
                         }
                     }
                 }
-- 
cgit v1.2.3-18-g5258