Set default config output_style

mdespuits · Apr 7, 2012 · 739c6b3 · 739c6b3
1 parent e6a4bcc
commit 739c6b3
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/lib/transplant.rb b/lib/transplant.rb
@@ -3,3 +3,8 @@
 %w[version configuration stats planter].each do |klass|
   require "transplant/#{klass}"
 end
+
+Transplant::Configuration.setup do |config|
+  config.root_path = Rails.root if defined?(Rails)
+  config.output_style = :both
+end
diff --git a/lib/transplant/stats.rb b/lib/transplant/stats.rb
@@ -24,7 +24,9 @@ def output_to_file
     end
 
     def add_to_results(output)
-      @result_set << output
+      style = ::Transplant::Configuration.output_style
+      puts output if [:both, :output].include?(style)
+      @result_set << output if [:both, :file].include?(style)
     end
 
     def output(header, input = {}, depth = 0, sub_output = false)