class FactoryBot::Linter

# File lib/factory_bot/linter.rb, line 3
def initialize(factories, strategy: :create, traits: false, verbose: false)
  @factories_to_lint = factories
  @factory_strategy = strategy
  @traits = traits
  @verbose = verbose
  @invalid_factories = calculate_invalid_factories
end

# File lib/factory_bot/linter.rb, line 11
def lint!
  if invalid_factories.any?
    raise InvalidFactoryError, error_message
  end
end

# File lib/factory_bot/linter.rb, line 21
def calculate_invalid_factories
  factories_to_lint.each_with_object(Hash.new([])) do |factory, result|
    errors = lint(factory)
    result[factory] |= errors unless errors.empty?
  end
end

# File lib/factory_bot/linter.rb, line 90
    def error_message
      lines = invalid_factories.map { |_factory, exceptions|
        exceptions.map(&error_message_type)
      }.flatten

      <<~ERROR_MESSAGE.strip
        The following factories are invalid:

        #{lines.join("\n")}
      ERROR_MESSAGE
    end

# File lib/factory_bot/linter.rb, line 102
def error_message_type
  if @verbose
    :verbose_message
  else
    :message
  end
end

# File lib/factory_bot/linter.rb, line 110
def in_transaction
  if defined?(ActiveRecord::Base)
    ActiveRecord::Base.transaction(joinable: false) do
      yield
      raise ActiveRecord::Rollback
    end
  else
    yield
  end
end

# File lib/factory_bot/linter.rb, line 62
def lint(factory)
  if @traits
    lint_factory(factory) + lint_traits(factory)
  else
    lint_factory(factory)
  end
end

# File lib/factory_bot/linter.rb, line 70
def lint_factory(factory)
  result = []
  begin
    in_transaction { FactoryBot.public_send(factory_strategy, factory.name) }
  rescue => e
    result |= [FactoryError.new(e, factory)]
  end
  result
end

# File lib/factory_bot/linter.rb, line 80
def lint_traits(factory)
  result = []
  factory.definition.defined_traits.map(&:name).each do |trait_name|
    in_transaction { FactoryBot.public_send(factory_strategy, factory.name, trait_name) }
  rescue => e
    result |= [FactoryTraitError.new(e, factory, trait_name)]
  end
  result
end