lib/todo_next.rb in todo_next-0.0.1 vs lib/todo_next.rb in todo_next-0.0.2

@@ -4,20 +4,17 @@
   # Your code goes here...
 end
 require File.dirname(__FILE__) + '/todo_next/parser'
 
 main = TOPLEVEL_BINDING.eval('self')
-def main.todo_next(text, puts_code=false)
+def main.todo_next(text, puts_code=true)
   code = TodoNext(text)
   if puts_code
     puts '---' ; puts code  ; puts '---'
   end
   eval code
 end
 
 def TodoNext(source)
-  TodoNext::Parser.
-      parse(source).
-      visit(TodoNext::RspecGeneratorVisitor.new).
-      flatten.
-      join("\n")
+  tree = TodoNext::Parser.parse(source)
+  tree.to_rspec
 end
\ No newline at end of file