Add while typechecking

2 files changed, 22 insertions, 4 deletions

diff --git a/src/main.c b/src/main.c
index df2f687..ffe9e10 100644
--- a/src/main.c
+++ b/src/main.c
@@ -254,8 +254,8 @@ type_inference(Analyzer *a, Node *node, TypeScope *scope) {
             return node->type;
         } break;
         case NODE_IF: {
-            Type type = type_inference(a, node->cond_if, scope);
-            if (!str_eq(type, cstr("bool"))) {
+            Type cond_type = type_inference(a, node->cond_if, scope);
+            if (!str_eq(cond_type, cstr("bool"))) {
                 emit_semantic_error(
                     a, node->cond_if,
                     cstr("non boolean expression on if condition"));
@@ -277,12 +277,26 @@ type_inference(Analyzer *a, Node *node, TypeScope *scope) {
                 }
                 if (!str_eq(node->type, else_type)) {
                     emit_semantic_error(
-                        a, node,
-                        cstr("mismatch types for if/else branches"));
+                        a, node, cstr("mismatch types for if/else branches"));
                     return cstr("");
                 }
             }
         } break;
+        case NODE_WHILE: {
+            Type cond_type = type_inference(a, node->while_cond, scope);
+            if (!str_eq(cond_type, cstr("bool"))) {
+                emit_semantic_error(
+                    a, node->cond_if,
+                    cstr("non boolean expression on while condition"));
+                return cstr("");
+            }
+            if (node->while_expr->kind != NODE_BLOCK) {
+                scope = typescope_alloc(a, scope);
+            }
+            type_inference(a, node->while_expr, scope);
+            node->type = cstr("nil");
+            return node->type;
+        } break;
         case NODE_TRUE:
         case NODE_FALSE: {
             node->type = cstr("bool");
diff --git a/tests/semantics.bad b/tests/semantics.bad
index c316a3c..a66c8ba 100644
--- a/tests/semantics.bad
+++ b/tests/semantics.bad
@@ -22,6 +22,10 @@ let single = if (true) {
     123
 }
 
+while (!true) {
+    "asjdflasdjf"
+}
+
 ; enum test {
 ;     a = 1
 ;     b