From 5de2e1fd9f426348127a66d2c51c300cb90cc3a4 Mon Sep 17 00:00:00 2001
From: Carlos Maniero <carlos@maniero.me>
Date: Wed, 10 May 2023 00:20:00 -0300
Subject: gas: Generate code for if statement

If statements are now working, the only exception is for the comparators
|| and && that will be addressed in a further commit. Checks tested:

  fn main(): i32 {
    let n: i32 = 11;

    if (n == 11) {
      if n != 12 {
        if n < 12 {
          if n <= 11 {
            if n > 10 {
              if n >= 11 {
                return 42;
              }
            }
          }
        }
      }
    }

    return n;
  }

To compile the && and || a precedence issue must be addressed: they must
have the highest precedence, witch is not working now:

  1 == 2 || 3 != 2

The or should be the higher level of the tree in the example above.

Signed-off-by: Carlos Maniero <carlos@maniero.me>
---
 src/ast_pretty_printer.c | 15 +++++++++------
 1 file changed, 9 insertions(+), 6 deletions(-)

(limited to 'src/ast_pretty_printer.c')

diff --git a/src/ast_pretty_printer.c b/src/ast_pretty_printer.c
index e2d007a..b14bf34 100644
--- a/src/ast_pretty_printer.c
+++ b/src/ast_pretty_printer.c
@@ -22,10 +22,12 @@
 #include <stdarg.h>
 #include <stdio.h>
 
-const char operation_kinds[AST_BINOP_N] = { [AST_BINOP_ADITION] = '+',
-                                            [AST_BINOP_SUBTRACTION] = '-',
-                                            [AST_BINOP_MULTIPLICATION] = '*',
-                                            [AST_BINOP_DIVISION] = '/' };
+const char *operation_kinds[AST_BINOP_N] = {
+  [AST_BINOP_ADITION] = "+",  [AST_BINOP_SUBTRACTION] = "-", [AST_BINOP_MULTIPLICATION] = "*",
+  [AST_BINOP_DIVISION] = "/", [AST_BINOP_EQUAL] = "==",      [AST_BINOP_NOT_EQUAL] = "!=",
+  [AST_BINOP_AND] = "&&",     [AST_BINOP_OR] = "||",         [AST_BINOP_GT] = ">",
+  [AST_BINOP_LT] = "<",       [AST_BINOP_LT_EQUAL] = "<=",   [AST_BINOP_GT_EQUAL] = ">=",
+};
 
 void
 ast_pretty_printer_init(ast_pretty_printer_t *printer, FILE *stream);
@@ -76,6 +78,7 @@ ast_pretty_printer_print_ast(ast_pretty_printer_t *printer, ast_node_t *ast)
 
           ast_pretty_printer_rm_indentation(printer);
         }
+        ast_pretty_printer_set_lst_children(printer);
         ast_pretty_printer_printf(printer, "body:\n");
         ast_pretty_printer_add_indentation(printer);
         {
@@ -90,7 +93,7 @@ ast_pretty_printer_print_ast(ast_pretty_printer_t *printer, ast_node_t *ast)
     }
     case AST_BINARY_OPERATION: {
       ast_binary_operation_t binop = ast->data.binary_operation;
-      ast_pretty_printer_printf(printer, "BinaryOperation operation='%c'\n", operation_kinds[binop.kind]);
+      ast_pretty_printer_printf(printer, "BinaryOperation operation='%s'\n", operation_kinds[binop.kind]);
 
       ast_pretty_printer_add_indentation(printer);
       {
@@ -130,6 +133,7 @@ ast_pretty_printer_print_ast(ast_pretty_printer_t *printer, ast_node_t *ast)
 
         ast_pretty_printer_add_indentation(printer);
         {
+          ast_pretty_printer_set_lst_children(printer);
           ast_pretty_printer_print_ast(printer, ast->data.function.body);
           ast_pretty_printer_rm_indentation(printer);
         }
@@ -139,7 +143,6 @@ ast_pretty_printer_print_ast(ast_pretty_printer_t *printer, ast_node_t *ast)
       break;
     }
     case AST_BLOCK: {
-      ast_pretty_printer_set_lst_children(printer);
       ast_pretty_printer_printf(printer, "Block\n");
       ast_pretty_printer_add_indentation(printer);
       {
-- 
cgit v1.2.3