From 4603d2525ebdfa57522ec60db4a86cbc99251ee5 Mon Sep 17 00:00:00 2001
From: Ginger Bill <bill@gingerbill.org>
Date: Thu, 19 Jan 2017 11:29:15 +0000
Subject: Closed range `...` (both inclusive); Type comparisons with == and !=

---
 src/parser.c | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

(limited to 'src/parser.c')

diff --git a/src/parser.c b/src/parser.c
index 53173eb4b..f083c2c8f 100644
--- a/src/parser.c
+++ b/src/parser.c
@@ -2848,10 +2848,14 @@ AstNode *parse_for_stmt(AstFile *f) {
 	isize prev_level = f->expr_level;
 	f->expr_level = -1;
 	AstNode *expr = parse_expr(f, false);
-	if (f->curr_token.kind == Token_Interval) {
-		Token op = expect_token(f, Token_Interval);
+	switch (f->curr_token.kind) {
+	case Token_HalfOpenRange:
+	case Token_Ellipsis: {
+		Token op = f->curr_token;
+		next_token(f);
 		AstNode *right = parse_expr(f, false);
 		expr = make_interval_expr(f, op, expr, right);
+	} break;
 	}
 	f->expr_level = prev_level;
 
-- 
cgit v1.2.3