Add viz for symbol tables

1 files changed, 10 insertions, 1 deletions

diff --git a/src/main.c b/src/main.c
index 56c36d7..fc991d2 100644
--- a/src/main.c
+++ b/src/main.c
@@ -17,6 +17,7 @@ typedef enum ExecMode {
     PRINT_LEX,
     PRINT_PARSE,
     PRINT_SEMANTIC,
+    PRINT_SYMTABLES,
 } ExecMode;
 
 static ExecMode mode = RUN_NORMAL;
@@ -46,12 +47,18 @@ process_source(const StringView *source, const char *file_name) {
     check_errors(file_name);
     if (mode == PRINT_PARSE) {
         viz_ast(roots);
+        return;
     }
 
     // Symbol table generation and type checking.
     ParseTree *parse_tree = semantic_analysis(roots);
     if (mode == PRINT_SEMANTIC) {
         viz_ast(parse_tree->roots);
+        return;
+    }
+    if (mode == PRINT_SYMTABLES) {
+        viz_symtables(parse_tree->scopes);
+        return;
     }
 }
 
@@ -116,7 +123,7 @@ print_usage(void) {
     printf("Usage: %s [options] <filename filename ...>\n", BIN_NAME);
     printf("\n");
     printf("\t-h \t\tShow usage.\n");
-    printf("\t-p [l | p | s]\tPrint mode for [l]exing, [p]arsing or [s]emantic analysis\n");
+    printf("\t-p [l|p|s|t]\tPrint mode for [l]exing, [p]arsing, [s]emantic analysis, symbol [t]ables\n");
     printf("\n");
 }
 
@@ -138,6 +145,8 @@ main(int argc, char *argv[]) {
                     mode = PRINT_PARSE;
                 } else if (optarg[0] == 's' && optarg[1] == '\0') {
                     mode = PRINT_SEMANTIC;
+                } else if (optarg[0] == 't' && optarg[1] == '\0') {
+                    mode = PRINT_SYMTABLES;
                 } else {
                     print_usage();
                     return EXIT_FAILURE;