Add parsing of if statements

1 files changed, 14 insertions, 0 deletions

diff --git a/src/nodes.c b/src/nodes.c
index 3a63246..af3b772 100644
--- a/src/nodes.c
+++ b/src/nodes.c
@@ -11,6 +11,9 @@ alloc_node(NodeType type) {
 
 void
 print_node(Node *node) {
+    if (node == NULL) {
+        return;
+    }
     switch (node->type) {
         case NODE_NUMBER: {
             if (node->number.negative) {
@@ -73,6 +76,17 @@ print_node(Node *node) {
             }
             printf(" }");
         } break;
+        case NODE_IF: {
+            printf("(if ");
+            print_node(node->ifexpr.cond);
+            printf(" ");
+            print_node(node->ifexpr.expr_true);
+            if (node->ifexpr.expr_false != NULL) {
+                printf(" ");
+                print_node(node->ifexpr.expr_false);
+            }
+            printf(")");
+        } break;
         case NODE_FUN: {
             printf("(fun ");
             print_node(node->fun.name);