toptal/crystalball

# frozen_string_literal: true

require 'crystalball/predictor/strategy'

module Crystalball
  class Predictor
    # Used with `predictor.use Crystalball::Predictor::ModifiedSpecs.new`. Will find files that match spec regexp and
    # return all new or modified files. You can specify spec regexp using first parameter to `#initialize`.
    class ModifiedSpecs
      include Strategy

      # @param [Regexp] spec_pattern - regexp to filter specs files
      def initialize(spec_pattern = %r{spec/.*_spec\.rb\z})
        @spec_pattern = spec_pattern
      end

      # This strategy does not depend on a previously generated example group map.
      # It uses the spec pattern to determine which specs should run.
      # @param [Crystalball::SourceDiff] diff - the diff from which to predict
      #   which specs should run
      # @return [Array<String>] the spec paths associated with the changes
      def call(diff, _)
        super do
          diff.reject(&:deleted?).map(&:new_relative_path).grep(spec_pattern)
        end
      end

      private

      attr_reader :spec_pattern
    end
  end
end