Add parsing for function definitions

This commits also changes the structure of some existing functions.
Namely, parse_* functions other than parse_next check that the type of
the token to parse is correct. This allow us to use them directly in the
rest of the code to consume tokens and properly produce an error if the
token type is not the expected one.

In the same fashion, two new functions consume_lparen and consume_rparen
are implemented. They only report true/false and report errors if
something went wrong.

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;
     }
 }