# frozen_string_literal: true

module RuboCop
  module RSpec
    # RSpec public API methods that are commonly used in cops
    module Language
      # Set of method selectors
      class SelectorSet
        def initialize(selectors)
          @selectors = selectors
        end

        def ==(other)
          selectors.eql?(other.selectors)
        end

        def +(other)
          self.class.new(selectors + other.selectors)
        end

        def include?(selector)
          selectors.include?(selector)
        end

        def to_node_pattern
          selectors.map(&:inspect).join(' ')
        end

        protected

        attr_reader :selectors
      end

      module ExampleGroups
        GROUPS  = SelectorSet.new(%i(describe context feature example_group))
        SKIPPED = SelectorSet.new(%i(xdescribe xcontext xfeature))
        FOCUSED = SelectorSet.new(%i(fdescribe fcontext ffeature))

        ALL = GROUPS + SKIPPED + FOCUSED
      end

      module SharedGroups
        ALL = SelectorSet.new(
          %i(shared_examples shared_context shared_examples_for)
        )
      end

      module Examples
        EXAMPLES = SelectorSet.new(%i(it specify example scenario its))
        FOCUSED  = SelectorSet.new(%i(fit fspecify fexample fscenario focus))
        SKIPPED  = SelectorSet.new(%i(xit xspecify xexample xscenario skip))
        PENDING  = SelectorSet.new(%i(pending))

        ALL = EXAMPLES + FOCUSED + SKIPPED + PENDING
      end

      module Hooks
        ALL = SelectorSet.new(%i(after around before))
      end

      module Helpers
        ALL = SelectorSet.new(%i(let let!))
      end

      ALL =
        ExampleGroups::ALL +
        SharedGroups::ALL  +
        Examples::ALL      +
        Hooks::ALL         +
        Helpers::ALL
    end
  end
end