Add public #to_source interface

txus · Feb 23, 2012 · 7cb2bd6 · 7cb2bd6
1 parent f47c9d9
commit 7cb2bd6
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 4 deletions.
diff --git a/lib/to_source.rb b/lib/to_source.rb
@@ -3,4 +3,14 @@
 require "to_source/visitor"
 
 module ToSource
+  # Public: Converts the node back to its original source code.
+  #
+  # Returns the String output.
+  def to_source
+    visitor = Visitor.new
+    lazy_visit(visitor)
+    visitor.output
+  end
 end
+
+Rubinius::AST::Node.send :include, ToSource
diff --git a/test/to_source/visitor_test.rb b/test/to_source/visitor_test.rb
@@ -4,10 +4,7 @@
 module ToSource
   class VisitorTest < MiniTest::Unit::TestCase
     def visit(code)
-      visitor = Visitor.new
-      ast     = code.to_ast
-      ast.lazy_visit(visitor)
-      visitor.output
+      code.to_ast.to_source
     end
 
     def assert_source(code)