Parse Maybe (?) type

Signed-off-by: Mel <mel@rnrd.eu>

1 files changed, 17 insertions, 2 deletions

diff --git a/boot/parse.c b/boot/parse.c
index ac324f9..976bc4b 100644
--- a/boot/parse.c
+++ b/boot/parse.c
@@ -546,9 +546,8 @@ parser_node_type_reference(struct Parser* p, struct Parser_Error* error)
 }
 
 struct Type_Node*
-parser_node_type(struct Parser* p, struct Parser_Error* error)
+parser_node_type_inner(struct Parser* p, struct Parser_Error* error)
 {
-    // TODO: maybe, variant, class
     struct Token token = parser_peek(p);
     switch (token.kind) {
     case TOKEN_NAME:
@@ -573,6 +572,22 @@ parser_node_type(struct Parser* p, struct Parser_Error* error)
     }
 }
 
+struct Type_Node*
+parser_node_type(struct Parser* p, struct Parser_Error* error)
+{
+    struct Type_Node* type = CHECK(parser_node_type_inner(p, error));
+
+    // check if the type is followed by a `?`, which means it may be nil.
+    if (parser_probe(p, TOKEN_QUESTION)) {
+        parser_next(p); // consume the question mark
+        type = type_node_new(
+            TYPE_NODE_MAYBE, (union Type_Node_Value){ .maybe = { type } }, type->span,
+            type->location);
+    }
+
+    return type;
+}
+
 struct Expression*
 parser_expression_primary_name(struct Parser* p, struct Parser_Error* error)
 {