From c4129b1d2468ad824b932cf6f8ec5730b429ba0e Mon Sep 17 00:00:00 2001
From: Recolic Keghart <root@recolic.net>
Date: Sat, 6 Apr 2019 21:13:33 -0700
Subject: [PATCH] if-bug fix

---
 src/main/java/chocopy/pa2/TypeChecker.java | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/src/main/java/chocopy/pa2/TypeChecker.java b/src/main/java/chocopy/pa2/TypeChecker.java
index 6e18fa9..5703e38 100644
--- a/src/main/java/chocopy/pa2/TypeChecker.java
+++ b/src/main/java/chocopy/pa2/TypeChecker.java
@@ -328,8 +328,12 @@ public class TypeChecker extends AbstractNodeAnalyzer<SymbolType> {
 
     @Override
     public SymbolType analyze(IfExpr node) {
+        SymbolType condition_type = node.condition.dispatch(this);
         SymbolType then_type = node.thenExpr.dispatch(this);
         SymbolType else_type = node.elseExpr.dispatch(this);
+        if(!condition_type.equals(BOOL_TYPE)) {
+            err(node, "if_expr should have BOOL_TYPE");
+        }
         if(!then_type.equals(else_type)) {
             err(node, "then_type not equals to else_type. ");
         }
-- 
GitLab