Add rudimentary type checking

1 files changed, 7 insertions, 0 deletions

diff --git a/src/viz.c b/src/viz.c
index d409472..8b5d4cf 100644
--- a/src/viz.c
+++ b/src/viz.c
@@ -19,6 +19,10 @@ viz_node(Node *node) {
     }
     printf("%zu [width=2.5,shape=Mrecord,label=\"", node->id);
     printf("<top> %s -- [%4ld:%-4ld] ", node_str[node->type], node->line, node->col);
+    if (node->expr_type != NULL) {
+        printf("| T: ");
+        sv_write(&node->expr_type->name);
+    }
     switch (node->type) {
         case NODE_NUMBER: {
             printf("| Value: ");
@@ -136,6 +140,9 @@ viz_node(Node *node) {
 
 void
 viz_ast(ParseTree *parse_tree) {
+    if (parse_tree == NULL) {
+        return;
+    }
     printf("digraph ast {\n");
     printf("rankdir=LR;\n");
     printf("ranksep=\"0.95 equally\";\n");