each loops!

1 files changed, 13 insertions, 0 deletions

diff --git a/typedefs_cgen.c b/typedefs_cgen.c
index 2051acb..937be4a 100644
--- a/typedefs_cgen.c
+++ b/typedefs_cgen.c
@@ -56,6 +56,19 @@ static bool typedefs_expr(CGenerator *g, Expression *e) {
 		if (!typedefs_block(g, &e->while_.body))
 			return false;
 		break;
+	case EXPR_EACH: {
+		EachExpr *ea = &e->each;
+		if (ea->flags & EACH_IS_RANGE) {
+			if (!typedefs_expr(g, ea->range.from))
+				return false;
+			if (ea->range.to && !typedefs_expr(g, ea->range.to))
+				return false;
+			/* step is a value, not an expression */
+		} else {
+			if (!typedefs_expr(g, ea->of))
+				return false;
+		}
+	} break;
 	case EXPR_TUPLE:
 		arr_foreach(e->tuple, Expression, x)
 			if (!typedefs_expr(g, x))