From 09df18182dfae22bbc65b2c07c4f610c595aba19 Mon Sep 17 00:00:00 2001
From: mryouse
Date: Sat, 21 May 2022 02:39:43 +0000
Subject: for-count implementation

---
 interpreter.py | 15 +++++++++++++++
 1 file changed, 15 insertions(+)

(limited to 'interpreter.py')

diff --git a/interpreter.py b/interpreter.py
index f252bc9..6a88c57 100644
--- a/interpreter.py
+++ b/interpreter.py
@@ -263,6 +263,21 @@ def interpretToString(expr, env):
 
 GLOBALS.register("->string", Builtin(interpretToString, 1))
 
+def interpretForCount(expr, env):
+    # for-count int exprs
+    num = evaluate(expr.args[0], env)
+    if type(num) is not int:
+        raise Exception("'for-count' count must be an integer")
+    new_env = Environment(env)
+    ret = None
+    for idx in range(0, num):
+        new_env.register("idx", Expr.Literal(idx + 1))
+        for arg in expr.args[1:]:
+            ret = evaluate(arg, new_env)
+    return ret
+
+GLOBALS.register("for-count", Builtin(interpretForCount))
+
 def interpretEnv(expr, env_expr, env):
     ev = evaluate(env_expr, env)
     return ev  # TODO more than this?
-- 
cgit v1.2.3