AST - Switch traits to contain items

1 files changed, 11 insertions, 6 deletions

diff --git a/src/dump_as_rust.cpp b/src/dump_as_rust.cpp
index 79ca0460..42cbf48e 100644
--- a/src/dump_as_rust.cpp
+++ b/src/dump_as_rust.cpp
@@ -949,13 +949,18 @@ void RustPrinter::handle_trait(const AST::Trait& s)
     m_os << indent() << "{\n";
     inc_indent();
     
-    for( const auto& i : s.types() )
+    for( const auto& i : s.items() )
     {
-        m_os << indent() << "type " << i.name << ";\n";
-    }
-    for( const auto& i : s.functions() )
-    {
-        handle_function(false, i.name, i.data);
+        TU_MATCH_DEF(AST::Item, (i.data), (e),
+        (
+            ),
+        (Type,
+            m_os << indent() << "type " << i.name << ";\n";
+            ),
+        (Function,
+            handle_function(false, i.name, e.e);
+            )
+        )
     }
     
     dec_indent();