ast: Dump: Add Indent class usable with streams

2 files changed, 60 insertions, 0 deletions

diff --git a/gcc/rust/ast/rust-ast-dump.cc b/gcc/rust/ast/rust-ast-dump.cc
index 2d4cb70..99b6e86 100644
--- a/gcc/rust/ast/rust-ast-dump.cc
+++ b/gcc/rust/ast/rust-ast-dump.cc
@@ -21,6 +21,39 @@
 namespace Rust {
 namespace AST {
 
+Indent::Indent () : tabs (0) {}
+
+std::ofstream &
+Indent::operator<< (std::ofstream &stream)
+{
+  for (size_t i = 0; i < tabs; i++)
+    stream << '\t';
+
+  return stream;
+}
+
+void
+Indent::increment ()
+{
+  tabs++;
+}
+
+void
+Indent::decrement ()
+{
+  rust_assert (tabs != 0);
+  tabs--;
+}
+
+Dump::Dump (std::ofstream &stream) : stream (stream), indentation (Indent ()) {}
+
+void
+Dump::go (AST::Crate &crate)
+{
+  for (auto &item : crate.items)
+    item->accept_vis (*this);
+}
+
 void
 Dump::visit (Token &tok)
 {}
diff --git a/gcc/rust/ast/rust-ast-dump.h b/gcc/rust/ast/rust-ast-dump.h
index 166b511..99568c6 100644
--- a/gcc/rust/ast/rust-ast-dump.h
+++ b/gcc/rust/ast/rust-ast-dump.h
@@ -25,8 +25,35 @@
 namespace Rust {
 namespace AST {
 
+// TODO: We might want to reuse this class somewhere else
+class Indent
+{
+public:
+  Indent ();
+
+  std::ofstream &operator<< (std::ofstream &stream);
+
+  void increment ();
+  void decrement ();
+
+private:
+  size_t tabs;
+};
+
 class Dump : public ASTVisitor
 {
+public:
+  Dump (std::ofstream &stream);
+
+  /**
+   * Run the visitor on an entire crate and its items
+   */
+  void go (AST::Crate &crate);
+
+private:
+  std::ofstream &stream;
+  Indent indentation;
+
   // rust-ast.h
   void visit (Token &tok);
   void visit (DelimTokenTree &delim_tok_tree);