From b6a679c0edd74d996bd0c1769cf4b161b4d42c4d Mon Sep 17 00:00:00 2001
From: Andrew Kelley <superjoe30@gmail.com>
Date: Thu, 4 May 2017 12:43:11 -0400
Subject: implement else on loops and break can give an expression

closes #357
---
 src/ast_render.cpp | 18 +++++++++++++-----
 1 file changed, 13 insertions(+), 5 deletions(-)

(limited to 'src/ast_render.cpp')

diff --git a/src/ast_render.cpp b/src/ast_render.cpp
index 348d825332..e1186da2c4 100644
--- a/src/ast_render.cpp
+++ b/src/ast_render.cpp
@@ -485,6 +485,15 @@ static void render_node_extra(AstRender *ar, AstNode *node, bool grouped) {
                 }
                 break;
             }
+        case NodeTypeBreak:
+            {
+                fprintf(ar->f, "break");
+                if (node->data.break_expr.expr) {
+                    fprintf(ar->f, " ");
+                    render_node_grouped(ar, node->data.break_expr.expr);
+                }
+                break;
+            }
         case NodeTypeDefer:
             {
                 const char *defer_str = defer_string(node->data.defer.kind);
@@ -880,11 +889,10 @@ static void render_node_extra(AstRender *ar, AstNode *node, bool grouped) {
                     fprintf(ar->f, "| ");
                 }
                 render_node_grouped(ar, node->data.for_expr.body);
-                break;
-            }
-        case NodeTypeBreak:
-            {
-                fprintf(ar->f, "break");
+                if (node->data.for_expr.else_node) {
+                    fprintf(ar->f, " else");
+                    render_node_grouped(ar, node->data.for_expr.else_node);
+                }
                 break;
             }
         case NodeTypeContinue:
-- 
cgit v1.2.3