Fix range

johnmai-dev · Dec 12, 2024 · b69ecec · b69ecec
1 parent b11a959
commit b69ecec
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/Sources/Environment.swift b/Sources/Environment.swift
@@ -21,10 +21,14 @@ class Environment {
             return args[0]
         }),
         "range": FunctionValue(value: { args, _ in
+            if args.isEmpty {
+                throw JinjaError.runtime("range() takes at least 1 argument")
+            }
             let start = args[0].value as! Int
             let stop = args.count > 1 ? args[1].value as? Int : nil
             let step = args.count > 2 ? args[2].value as! Int : 1
-            return ArrayValue(value: range(start: start, stop: stop, step: step).map { NumericValue(value: $0) })
+            let result = range(start: start, stop: stop, step: step)
+            return ArrayValue(value: result.map { NumericValue(value: $0) })
         })
     ]
 

diff --git a/Sources/Template.swift b/Sources/Template.swift
@@ -27,7 +27,6 @@ public struct Template {
                 throw JinjaError.runtime("\(args)")
             }
         )
-        try env.set(name: "range", value: range)
 
         for (key, value) in items {
             try env.set(name: key, value: value)