HIR Dump - Fix printing of closures (that don't contain a block)

1 files changed, 15 insertions, 1 deletions

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