AST Dump Source

* fixed incorrect indention implementation for structure, feature and implementation definitions * related to sealangdotorg/sea#35
casm-lang · Jul 31, 2017 · b5feb16 · b5feb16
1 parent 73a3732
commit b5feb16
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/src/transform/AstDumpSourcePass.cpp b/src/transform/AstDumpSourcePass.cpp
@@ -332,9 +332,10 @@ void AstDumpSourceVisitor::visit( StructureDefinition& node )
     m_stream << "structure ";
     node.identifier()->accept( *this );
     m_stream << " =\n{";
+    const auto levelGuard = Indentation::NextLevel( m_indentation );
     for( auto& e : *node.functions() )
     {
-        m_stream << "\n    ";
+        m_stream << "\n" << m_indentation;
         e->accept( *this );
     }
     m_stream << "}";
@@ -345,9 +346,10 @@ void AstDumpSourceVisitor::visit( FeatureDefinition& node )
     m_stream << "feature ";
     node.identifier()->accept( *this );
     m_stream << " =\n{";
+    const auto levelGuard = Indentation::NextLevel( m_indentation );
     for( auto& e : *node.definitions() )
     {
-        m_stream << "\n    ";
+        m_stream << "\n" << m_indentation;
         e->accept( *this );
     }
     m_stream << "}";
@@ -363,9 +365,10 @@ void AstDumpSourceVisitor::visit( ImplementationDefinition& node )
     }
     node.type()->accept( *this );
     m_stream << " =\n{";
+    const auto levelGuard = Indentation::NextLevel( m_indentation );
     for( auto& e : *node.definitions() )
     {
-        m_stream << "\n    ";
+        m_stream << "\n" << m_indentation;
         e->accept( *this );
     }
     m_stream << "}";