factory_trace 0.1.1 → 0.1.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 7f8ffd7275b885d4354634c3a47c9697d96e40774e065526631c229263a52a80
-  data.tar.gz: 392e63586adb5d09b04f44ff80f3bd71606508e58ed21dc72525e10454f0b705
+  metadata.gz: 26b7253d7d994cb1f591977dc3bcadc38d6f6cfe80526ef4377d85d411fbd380
+  data.tar.gz: bcde5b14ecfcfbe16b9cada40618ae0c79e38be424b96b9b2967996a6ab1f1c2
 SHA512:
-  metadata.gz: 3017d866a2b4d76f812e9ebf14a2c476a2d716fea6c5696b03ce83947e008431f340186caff533be69fe73d7da95b7f2f2b7c08d9cf257af4921ebbdf5d33d2f
-  data.tar.gz: 601ae2b6b8d370a3be34abe13404b50a3516789519548de30b15a60c66c712c3b4e2ad6bedd128dd75b53b7e08954d783142e85eca1e89e1f7cc4b70110f7e54
+  metadata.gz: 83ef3b6436248a5ed5c440966337b89e7e1dcbd72b025bbcfb6448cbee188aac24f57d3122c0f41e81f2d9af138aed04ed24d4d5ac5ec0a87ad0ba068dce085a
+  data.tar.gz: fa02e34299d009aea40d99368e67e575152b30479e79a903464f45279ee33afdaa59d5bc51fd42fa1edee3ca7ecfea171705278778c0b1aa41522c81ce203a7f

data/lib/factory_trace/configuration.rb

@@ -1,10 +1,11 @@
 module FactoryTrace
   class Configuration
-    attr_accessor :path, :enabled
+    attr_accessor :path, :enabled, :color
 
     def initialize
       @enabled = ENV.key?('FB_TRACE') || ENV.key?('FB_TRACE_FILE')
       @path = ENV['FB_TRACE_FILE']
+      @color = path.nil?
     end
 
     def out

data/lib/factory_trace/find_unused.rb

@@ -8,6 +8,9 @@ module FactoryTrace
 
     # @return [Array<Hash>]
     def check!
+      # This is required to exclude parent traits from +defined_traits+
+      FactoryBot.reload
+
       data = prepare(initial_data)
 
       output = []

data/lib/factory_trace/printer.rb

@@ -1,32 +1,48 @@
 module FactoryTrace
   class Printer
-    def initialize(io)
+    COLORS = {
+      blue: "\e[34m",
+      green: "\e[32m",
+      red: "\e[31m"
+    }.freeze
+
+    def initialize(io, config: Configuration.new)
       @io = io
+      @config = config
     end
 
     # @param [Array<Hash>] results
     def print(results)
+      total_color = results.size == 2 ? :green : :red
+
       results.each do |result|
-        io.puts(convert(result))
+        io.puts(convert(result, total_color: total_color))
       end
     end
 
     private
 
     # @param [Hash<Symbol, Object>] result
-    def convert(result)
+    # @param [Symbol] total_color
+    def convert(result, total_color:)
       case
       when result[:value]
-        "total number of unique #{result[:code]} factories & traits: #{result[:value]}"
+        colorize(total_color, "total number of unique #{result[:code]} factories & traits: #{result[:value]}")
       when result[:factory] && result[:trait]
-        "#{result[:code]} trait '#{result[:trait].name}' of factory '#{result[:factory].name}'"
+        "#{result[:code]} trait #{colorize(:blue, result[:trait].name)} of factory #{colorize(:blue, result[:factory].name)}"
       when result[:factory]
-        "#{result[:code]} factory '#{result[:factory].name}'"
+        "#{result[:code]} factory #{colorize(:blue, result[:factory].name)}"
       else
-        "#{result[:code]} global trait '#{result[:trait].name}'"
+        "#{result[:code]} global trait #{colorize(:blue, result[:trait].name)}"
       end
     end
 
-    attr_reader :io
+    def colorize(color, msg)
+      return msg unless config.color
+
+      "#{COLORS[color]}#{msg}\e[0m"
+    end
+
+    attr_reader :io, :config
   end
 end

data/lib/factory_trace/version.rb

@@ -1,3 +1,3 @@
 module FactoryTrace
-  VERSION = '0.1.1'
+  VERSION = '0.1.2'
 end

data/lib/factory_trace.rb

@@ -36,7 +36,7 @@ module FactoryTrace
     end
 
     def printer
-      @printer ||= Printer.new(configuration.out)
+      @printer ||= Printer.new(configuration.out, config: configuration)
     end
 
     def configuration

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: factory_trace
 version: !ruby/object:Gem::Version
-  version: 0.1.1
+  version: 0.1.2
 platform: ruby
 authors:
 - djezzzl
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2019-04-28 00:00:00.000000000 Z
+date: 2019-04-30 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: factory_bot