From e03777b3044ada72bff1dde2cc42d84060b93cb7 Mon Sep 17 00:00:00 2001
From: Santo Cariotti <santo@dcariotti.me>
Date: Sat, 13 Jul 2024 19:00:22 +0200
Subject: [PATCH] Print AST and fix while

---
 src/Main.java                    | 2 ++
 src/ast/nodes/WhileStmtNode.java | 2 +-
 2 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/src/Main.java b/src/Main.java
index cf8e67e..5970b3e 100644
--- a/src/Main.java
+++ b/src/Main.java
@@ -102,6 +102,8 @@ public static void main(String[] args) {
                     frame.setDefaultCloseOperation(JFrame.EXIT_ON_CLOSE);
                     frame.setSize(800, 600);
                     frame.setVisible(true);
+
+                    System.out.println(ast.printAST(""));
                 }
 
                 if (execute) {
diff --git a/src/ast/nodes/WhileStmtNode.java b/src/ast/nodes/WhileStmtNode.java
index 3e8a36e..394bdd3 100644
--- a/src/ast/nodes/WhileStmtNode.java
+++ b/src/ast/nodes/WhileStmtNode.java
@@ -41,7 +41,7 @@ public String codeGeneration() {
 
     @Override
     public String printAST(String prefix) {
-        return prefix + "While\n" + expr.toPrint(prefix + "  ") + block.toPrint(prefix + "  ");
+        return prefix + "While\n" + expr.printAST(prefix + "  ") + block.printAST(prefix + "  ");
     }
 
     @Override