loop support

1 files changed, 29 insertions, 0 deletions

diff --git a/ast.c b/ast.c
index 69173bd..7c7432b 100644
--- a/ast.c
+++ b/ast.c
@@ -116,12 +116,38 @@ static void expr_destroy(struct expr_node* node) {
 static void if_destroy(struct if_node* node) {
     expr_destroy(node->cond);
     free(node->cond);
+
     stmt_destroy(node->true_branch);
     free(node->true_branch);
+
     if (node->false_branch != NULL) {
         stmt_destroy(node->false_branch);
         free(node->false_branch);
     }
+
+    scope_destroy(node->scope);
+    free(node->scope);
+}
+
+static void loop_destroy(struct loop_node* node) {
+    if (node->init != NULL) {
+        expr_destroy(node->init);
+        free(node->init);
+    }
+    if (node->cond != NULL) {
+        expr_destroy(node->cond);
+        free(node->cond);
+    }
+    if (node->incr != NULL) {
+        expr_destroy(node->incr);
+        free(node->incr);
+    }
+
+    stmt_destroy(node->body);
+    free(node->body);
+
+    scope_destroy(node->scope);
+    free(node->scope);
 }
 
 static void stmt_destroy(struct stmt_node* node) {
@@ -142,6 +168,9 @@ static void stmt_destroy(struct stmt_node* node) {
         case STMT_IF:
             if_destroy(&node->inner.if_);
             break;
+        case STMT_LOOP:
+            loop_destroy(&node->inner.loop);
+            break;
     }
 }