factory_trace 0.1.2 → 0.2.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 26b7253d7d994cb1f591977dc3bcadc38d6f6cfe80526ef4377d85d411fbd380
-  data.tar.gz: bcde5b14ecfcfbe16b9cada40618ae0c79e38be424b96b9b2967996a6ab1f1c2
+  metadata.gz: f8910590a0b5aab2a00f31e0aa41cc244161e8c387b47341b0f6c2a96d7e40c8
+  data.tar.gz: e6902745419605702ccc35def7d110b7c18b23b1407af3560b2d2e3f75a43e96
 SHA512:
-  metadata.gz: 83ef3b6436248a5ed5c440966337b89e7e1dcbd72b025bbcfb6448cbee188aac24f57d3122c0f41e81f2d9af138aed04ed24d4d5ac5ec0a87ad0ba068dce085a
-  data.tar.gz: fa02e34299d009aea40d99368e67e575152b30479e79a903464f45279ee33afdaa59d5bc51fd42fa1edee3ca7ecfea171705278778c0b1aa41522c81ce203a7f
+  metadata.gz: 15b006db74f1f06d6b8f5a16dff18158a407ed292e2a90b1859c7c6bd6fc503d1016e14516e2fa2e1081a4d6dd873fff0096e4fd2d98ab39c794e63c986a7bb5
+  data.tar.gz: 69a82579dc197d590f1578e04fcbd818e48c2bf5750d01f86ffb62e77d1fcb6341792dcae76ae0c50bfb8acb48d562fdac1f0caa9b1e6d6a292fc5853d9aae1b

data/exe/factory_trace

@@ -0,0 +1,14 @@
+#!/usr/bin/env ruby
+
+$LOAD_PATH.unshift("#{__dir__}/../lib")
+require 'factory_trace'
+
+fail "You should pass at least one file with traced information.\nYou can generate it using only_trace mode." if ARGV.empty?
+
+config = FactoryTrace.configuration
+data = FactoryTrace::TraceReader.read_from_files(*ARGV)
+data = FactoryTrace::FindUnused.call(data)
+code = data.size > 2 ? 1 : 0
+
+FactoryTrace::ReportWriter.new(config.out, config: config).write(data)
+exit(code)

data/lib/factory_trace/configuration.rb

@@ -1,11 +1,12 @@
 module FactoryTrace
   class Configuration
-    attr_accessor :path, :enabled, :color
+    attr_accessor :path, :enabled, :color, :mode
 
     def initialize
       @enabled = ENV.key?('FB_TRACE') || ENV.key?('FB_TRACE_FILE')
       @path = ENV['FB_TRACE_FILE']
       @color = path.nil?
+      @mode = extract_mode(ENV['FB_TRACE']) || :full
     end
 
     def out
@@ -13,5 +14,16 @@ module FactoryTrace
 
       File.open(path, 'w')
     end
+
+    def mode?(*args)
+      args.include?(mode)
+    end
+
+    private
+
+    def extract_mode(value)
+      matcher = value && value.match(/full|trace_only/)
+      matcher && matcher[0].to_sym
+    end
   end
 end

data/lib/factory_trace/find_unused.rb

@@ -1,13 +1,8 @@
 module FactoryTrace
   class FindUnused
-
-    # @param [Hash<Symbol, Set<Symbol>>]
-    def initialize(data)
-      @initial_data = data
-    end
-
+    # @param [Hash<Symbol, Set<Symbol>>] initial_data
     # @return [Array<Hash>]
-    def check!
+    def self.call(initial_data)
       # This is required to exclude parent traits from +defined_traits+
       FactoryBot.reload
 
@@ -39,9 +34,8 @@ module FactoryTrace
     # Return new structure where each trait is moved to its own factory
     #
     # @param [Hash<Symbol, Set<Symbol>>]
-    #
     # @return [Hash<String, Set<String>>]
-    def prepare(data)
+    def self.prepare(data)
       # +_traits+ is for global traits
       output = {'_traits' => Set.new, '_total_used' => count_total(data)}
 
@@ -75,10 +69,8 @@ module FactoryTrace
     end
 
     # @param [Hash<Symbol, Set<Symbol>>]
-    def count_total(data)
+    def self.count_total(data)
       data.reduce(0) { |result, (_factory, traits)| result + 1 + traits.size }
     end
-
-    attr_reader :initial_data
   end
 end

data/lib/factory_trace/readers/trace_reader.rb

@@ -0,0 +1,36 @@
+require 'set'
+
+module FactoryTrace
+  class TraceReader
+    def self.read_from_files(*file_names, config: Configuration.new)
+      file_names.reduce({}) do |hash, file_name|
+        reader = new(File.open(file_name, 'r'), config: config)
+        hash.merge(reader.read) { |_key, v1, v2| v1 | v2 }
+      end
+    end
+
+    def initialize(io, config: Configuration.new)
+      @io = io
+      @config = config
+    end
+
+    def read
+      @data ||= {}
+
+      io.each_line do |line|
+        factory, *traits = line.strip.split(',').map(&:to_sym)
+
+        if factory
+          @data[factory] ||= Set.new
+          @data[factory] |= traits
+        end
+      end
+
+      @data
+    end
+
+    private
+
+    attr_reader :io, :config
+  end
+end

data/lib/factory_trace/version.rb

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

data/lib/factory_trace/{printer.rb → writers/report_writer.rb}

@@ -1,18 +1,13 @@
 module FactoryTrace
-  class Printer
+  class ReportWriter < Writer
     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)
+    def write(results)
       total_color = results.size == 2 ? :green : :red
 
       results.each do |result|
@@ -42,7 +37,5 @@ module FactoryTrace
 
       "#{COLORS[color]}#{msg}\e[0m"
     end
-
-    attr_reader :io, :config
   end
 end

data/lib/factory_trace/writers/trace_writer.rb

@@ -0,0 +1,11 @@
+module FactoryTrace
+  class TraceWriter < Writer
+    # @param [Hash<Symbol, Set<Hash>>] results
+    def write(results)
+      results.each do |key, set|
+        line = [key, set.to_a.join(',')].join(',')
+        io.puts(line)
+      end
+    end
+  end
+end

data/lib/factory_trace/writers/writer.rb

@@ -0,0 +1,23 @@
+module FactoryTrace
+  class Writer
+    def self.factory(io, config: Configuration.new)
+      writer =
+        if config.mode?(:full)
+          ReportWriter
+        elsif config.mode?(:trace_only)
+          TraceWriter
+        end
+
+      writer.new(io, config: config)
+    end
+
+    def initialize(io, config: Configuration.new)
+      @io = io
+      @config = config
+    end
+
+    private
+
+    attr_reader :io, :config
+  end
+end

data/lib/factory_trace.rb

@@ -5,7 +5,10 @@ require 'factory_trace/configuration'
 require 'factory_trace/version'
 require 'factory_trace/tracker'
 require 'factory_trace/find_unused'
-require 'factory_trace/printer'
+require 'factory_trace/readers/trace_reader'
+require 'factory_trace/writers/writer'
+require 'factory_trace/writers/report_writer'
+require 'factory_trace/writers/trace_writer'
 # Integrations
 require 'integrations/rspec' if defined?(RSpec::Core)
 
@@ -20,27 +23,33 @@ module FactoryTrace
     def stop
       return unless configuration.enabled
 
-      result = FindUnused.new(tracker.storage).check!
-
-      printer.print(result)
+      writer.write(results)
     end
 
     def configure
       yield(configuration)
     end
 
+    def configuration
+      @configuration ||= Configuration.new
+    end
+
     private
 
-    def tracker
-      @tracker ||= Tracker.new
+    def results
+      if configuration.mode?(:full)
+        FindUnused.call(tracker.storage)
+      elsif configuration.mode?(:trace_only)
+        tracker.storage
+      end
     end
 
-    def printer
-      @printer ||= Printer.new(configuration.out, config: configuration)
+    def tracker
+      @tracker ||= Tracker.new
     end
 
-    def configuration
-      @configuration ||= Configuration.new
+    def writer
+      @writer ||= Writer.factory(configuration.out, config: configuration)
     end
   end
 end

metadata

@@ -1,12 +1,12 @@
 --- !ruby/object:Gem::Specification
 name: factory_trace
 version: !ruby/object:Gem::Version
-  version: 0.1.2
+  version: 0.2.1
 platform: ruby
 authors:
 - djezzzl
 autorequire: 
-bindir: bin
+bindir: exe
 cert_chain: []
 date: 2019-04-30 00:00:00.000000000 Z
 dependencies:
@@ -69,16 +69,21 @@ dependencies:
 description: 
 email:
 - lawliet.djez@gmail.com
-executables: []
+executables:
+- factory_trace
 extensions: []
 extra_rdoc_files: []
 files:
+- exe/factory_trace
 - lib/factory_trace.rb
 - lib/factory_trace/configuration.rb
 - lib/factory_trace/find_unused.rb
-- lib/factory_trace/printer.rb
+- lib/factory_trace/readers/trace_reader.rb
 - lib/factory_trace/tracker.rb
 - lib/factory_trace/version.rb
+- lib/factory_trace/writers/report_writer.rb
+- lib/factory_trace/writers/trace_writer.rb
+- lib/factory_trace/writers/writer.rb
 - lib/integrations/rspec.rb
 homepage: https://github.com/djezzzl/factory_trace
 licenses: