From c62b73a2477b9fa625285e9c2ca1e51c2bcb8024 Mon Sep 17 00:00:00 2001
From: Tom Willemse
Date: Thu, 14 Jan 2021 23:53:47 -0800
Subject: Add branching primitives

---
 src/com/craftinginterpreters/lox/AstPrinter.java | 23 +++++++++++++++++++++++
 1 file changed, 23 insertions(+)

(limited to 'src/com/craftinginterpreters/lox/AstPrinter.java')

diff --git a/src/com/craftinginterpreters/lox/AstPrinter.java b/src/com/craftinginterpreters/lox/AstPrinter.java
index d5163df..8f7be78 100644
--- a/src/com/craftinginterpreters/lox/AstPrinter.java
+++ b/src/com/craftinginterpreters/lox/AstPrinter.java
@@ -27,6 +27,11 @@ class AstPrinter implements Expr.Visitor<String>, Stmt.Visitor<String> {
         return expr.value.toString();
     }
 
+    @Override
+    public String visitLogicalExpr(Expr.Logical expr) {
+        return parenthesize(expr.operator.type.toString(), expr.left, expr.right);
+    }
+
     @Override
     public String visitUnaryExpr(Expr.Unary expr) {
         return parenthesize(expr.operator.lexeme, expr.right);
@@ -57,6 +62,20 @@ class AstPrinter implements Expr.Visitor<String>, Stmt.Visitor<String> {
         return expression.expression.accept(this);
     }
 
+    @Override
+    public String visitIfStmt(Stmt.If stmt) {
+        StringBuilder builder = new StringBuilder();
+
+        builder.append("(");
+        builder.append(parenthesize("if", stmt.condition));
+        builder.append(parenthesize("then", stmt.thenBranch));
+
+        if (stmt.elseBranch != null) {
+            builder.append(parenthesize("else", stmt.elseBranch));
+        }
+        return builder.toString();
+    }
+
     @Override
     public String visitBlockStmt(Stmt.Block block) {
         return parenthesize("", block.statements);
@@ -88,6 +107,10 @@ class AstPrinter implements Expr.Visitor<String>, Stmt.Visitor<String> {
         return builder.toString();
     }
 
+    private String parenthesize(String name, Stmt... statements) {
+        return parenthesize(name, statements);
+    }
+
     public static void main(String[] args) {
         Expr expression = new Expr.Binary(
                 new Expr.Unary(new Token(TokenType.MINUS, "-", null, 1), new Expr.Literal(123)),
-- 
cgit v1.2.3-54-g00ecf