1 files changed, 27 insertions, 1 deletions

diff --git a/src/nodes.c b/src/nodes.c
index 855d186..fc39d8e 100644
--- a/src/nodes.c
+++ b/src/nodes.c
@@ -23,6 +23,7 @@ print_node(Node *node) {
             }
         } break;
         case NODE_SYMBOL:
+        case NODE_TYPE:
         case NODE_STRING: {
             sv_write(&node->string);
         } break;
@@ -51,7 +52,7 @@ print_node(Node *node) {
             printf("(def ");
             print_node(node->def.symbol);
             printf(":");
-            sv_write(&node->def.type);
+            print_node(node->def.type);
             printf(" ");
             print_node(node->def.value);
             printf(")");
@@ -63,6 +64,31 @@ print_node(Node *node) {
             print_node(node->def.value);
             printf(")");
         } break;
+        case NODE_FUN: {
+            printf("(fun ");
+            print_node(node->fun.name);
+            printf(" (");
+            size_t n_params = array_size(node->fun.param_names);
+            for (size_t i = 0; i < n_params; ++i) {
+                print_node(node->fun.param_names[i]);
+                printf(":");
+                print_node(node->fun.param_types[i]);
+                if (i < n_params - 1) {
+                    printf(" ");
+                }
+            }
+            printf(")");
+            printf(": ");
+            print_node(node->fun.return_type);
+
+            size_t n_expr = array_size(node->fun.body);
+            for (size_t i = 0; i < n_expr; ++i) {
+                printf(" ");
+                print_node(node->fun.body[i]);
+            }
+
+            printf(")");
+        } break;
         default: { printf("{#unknown#}"); } break;
     }
 }