From 10f602cabec44841bb947a185dd196d1107c97d9 Mon Sep 17 00:00:00 2001
From: John Hodge <tpg@mutabah.net>
Date: Sun, 21 Aug 2016 16:37:38 +0800
Subject: HIR Dump - Fix printing of closures (that don't contain a block)

---
 src/hir/dump.cpp | 16 +++++++++++++++-
 1 file changed, 15 insertions(+), 1 deletion(-)

(limited to 'src')

diff --git a/src/hir/dump.cpp b/src/hir/dump.cpp
index f5788ca3..32851fd5 100644
--- a/src/hir/dump.cpp
+++ b/src/hir/dump.cpp
@@ -208,7 +208,21 @@ namespace {
             if( item.m_code )
             {
                 m_os << indent();
-                item.m_code->visit( *this );
+                if( dynamic_cast< ::HIR::ExprNode_Block*>(&*item.m_code) ) {
+                    item.m_code->visit( *this );
+                }
+                else {
+                    m_os << "{\n";
+                    inc_indent();
+                    m_os << indent();
+                    
+                    item.m_code->visit( *this );
+                    
+                    m_os << "\n";
+                    dec_indent();
+                    m_os << indent();
+                    m_os << "}";
+                }
                 m_os << "\n";
             }
             else
-- 
cgit v1.2.3