lib/attractor/reporters/console_reporter.rb in attractor-2.3.0 vs lib/attractor/reporters/console_reporter.rb in attractor-2.4.0

@@ -1,28 +1,67 @@
 # frozen_string_literal: true
 
 module Attractor
   # console reporter
   class ConsoleReporter < BaseReporter
-    def report
-      super
-      puts "Calculated churn and complexity"
-      puts
-      puts "file_path#{" " * 53}complexity   churn"
-      puts "-" * 80
+    class TableFormatter
+      def call(calculators)
+        puts "Calculated churn and complexity"
+        puts
+        puts "file_path#{" " * 53}complexity   churn"
+        puts "-" * 80
 
-      @calculators.each do |calc|
-        # e.g. ['js', JsCalculator']
-        puts calc.last.type
+        calculators.each do |calc|
+          # e.g. ['js', JsCalculator']
+          puts calc.last.type
 
-        values = calc.last.calculate
-        suggester = Suggester.new(values)
+          values = calc.last.calculate
+          suggester = Suggester.new(values)
 
-        puts values&.map(&:to_s)
-        puts
-        puts "Suggestions for refactorings:"
-        suggester.suggest&.each { |sug| puts sug.file_path }
-        puts
+          puts values&.map(&:to_s)
+          puts
+          puts "Suggestions for refactorings:"
+          suggester.suggest&.each { |sug| puts sug.file_path }
+          puts
+        end
       end
+    end
+
+    class CSVFormatter
+      def call(calculators)
+        require "csv"
+
+        result = CSV.generate do |csv|
+          csv << %w[file_path score complexity churn type refactor]
+
+          calculators.each do |calc|
+            type = calc.last.type
+            values = calc.last.calculate
+            suggester = Suggester.new(values)
+            to_be_refactored = suggester.suggest.map(&:file_path)
+
+            values.each do |value|
+              csv << [value.file_path, value.score, value.complexity, value.churn, type, to_be_refactored.include?(value.file_path)]
+            end
+          end
+        end
+
+        puts result
+      end
+    end
+
+    def initialize(format:, **other)
+      super(**other)
+      @formatter = case format.to_sym
+                   when :csv
+                     CSVFormatter.new
+                   else
+                     TableFormatter.new
+      end
+    end
+
+    def report
+      super
+      @formatter.call(@calculators)
     end
   end
 end