Range operator

2 files changed, 9 insertions, 0 deletions

diff --git a/src/eval/mod.rs b/src/eval/mod.rs
index 96ac87a9..ba2de8c7 100644
--- a/src/eval/mod.rs
+++ b/src/eval/mod.rs
@@ -383,6 +383,7 @@ impl Eval for BinaryExpr {
             BinOp::SubAssign => self.assign(ctx, ops::sub),
             BinOp::MulAssign => self.assign(ctx, ops::mul),
             BinOp::DivAssign => self.assign(ctx, ops::div),
+            BinOp::Range => self.apply(ctx, ops::range),
         }
     }
 }
diff --git a/src/eval/ops.rs b/src/eval/ops.rs
index 2537f4a5..01044842 100644
--- a/src/eval/ops.rs
+++ b/src/eval/ops.rs
@@ -250,6 +250,14 @@ comparison!(leq, Less | Equal);
 comparison!(gt, Greater);
 comparison!(geq, Greater | Equal);
 
+/// Compute the range from `lhs` to `rhs`.
+pub fn range(lhs: Value, rhs: Value) -> Value {
+    match (lhs, rhs) {
+        (Int(a), Int(b)) => Array((a ..= b).map(Int).collect()),
+        _ => Error,
+    }
+}
+
 /// Concatenate two collections.
 fn concat<T, A>(mut a: T, b: T) -> T
 where