From f868d02b08688c04509d1abda5af89a182033d88 Mon Sep 17 00:00:00 2001
From: Bruce Hill <bruce@bruce-hill.com>
Date: Thu, 21 Nov 2024 13:00:53 -0500
Subject: Add `NULL` as a syntax for null values.

---
 types.c | 10 +++++++---
 1 file changed, 7 insertions(+), 3 deletions(-)

(limited to 'types.c')

diff --git a/types.c b/types.c
index e03bfb6e..5c26d07a 100644
--- a/types.c
+++ b/types.c
@@ -117,7 +117,7 @@ bool type_eq(type_t *a, type_t *b)
 bool type_is_a(type_t *t, type_t *req)
 {
     if (type_eq(t, req)) return true;
-    if (req->tag == OptionalType)
+    if (req->tag == OptionalType && Match(req, OptionalType)->type)
         return type_is_a(t, Match(req, OptionalType)->type);
     if (t->tag == PointerType && req->tag == PointerType) {
         auto t_ptr = Match(t, PointerType);
@@ -144,10 +144,14 @@ type_t *type_or_type(type_t *a, type_t *b)
 {
     if (!a) return b;
     if (!b) return a;
-    if (type_is_a(b, a)) return a;
-    if (type_is_a(a, b)) return b;
+    if (a->tag == OptionalType && !Match(a, OptionalType)->type)
+        return b->tag == OptionalType ? b : Type(OptionalType, b);
+    if (b->tag == OptionalType && !Match(b, OptionalType)->type)
+        return a->tag == OptionalType ? a : Type(OptionalType, a);
     if (a->tag == ReturnType && b->tag == ReturnType)
         return Type(ReturnType, .ret=type_or_type(Match(a, ReturnType)->ret, Match(b, ReturnType)->ret));
+    if (type_is_a(b, a)) return a;
+    if (type_is_a(a, b)) return b;
     if (a->tag == AbortType || a->tag == ReturnType) return non_optional(b);
     if (b->tag == AbortType || b->tag == ReturnType) return non_optional(a);
     if ((a->tag == IntType || a->tag == NumType) && (b->tag == IntType || b->tag == NumType)) {
-- 
cgit v1.2.3