module Enumerable

def as_json(options = nil) # :nodoc:
  to_a.as_json(options)
end

def compact_blank
  reject(&:blank?)
end

def exclude?(object)
  !include?(object)
end

def excluding(*elements)
  elements.flatten!(1)
  reject { |element| elements.include?(element) }
end

def in_order_of(key, series)
  group_by(&key).values_at(*series).flatten(1).compact
end

def including(*elements)
  to_a.including(*elements)
end

def index_by
  if block_given?
    result = {}
    each { |elem| result[yield(elem)] = elem }
    result
  else
    to_enum(:index_by) { size if respond_to?(:size) }
  end
end

def index_with(default = (no_default = true))
  if block_given?
    result = {}
    each { |elem| result[elem] = yield(elem) }
    result
  elsif no_default
    to_enum(:index_with) { size if respond_to?(:size) }
  else
    result = {}
    each { |elem| result[elem] = default }
    result
  end
end

def many?
  cnt = 0
  if block_given?
    any? do |*args|
      cnt += 1 if yield(*args)
      cnt > 1
    end
  else
    any? { (cnt += 1) > 1 }
  end
end

def maximum(key)
  map(&key).max
end

def minimum(key)
  map(&key).min
end

def pick(*keys)
  return if none?
  if keys.many?
    keys.map { |key| first[key] }
  else
    first[keys.first]
  end
end

def pluck(*keys)
  if keys.many?
    map { |element| keys.map { |key| element[key] } }
  else
    key = keys.first
    map { |element| element[key] }
  end
end

def sole
  case count
  when 1   then return first # rubocop:disable Style/RedundantReturn
  when 0   then raise ActiveSupport::EnumerableCoreExt::SoleItemExpectedError, "no item found"
  when 2.. then raise ActiveSupport::EnumerableCoreExt::SoleItemExpectedError, "multiple items found"
  end
end

def sum(identity = nil, &block)
  if identity
    _original_sum_with_required_identity(identity, &block)
  elsif block_given?
    map(&block).sum
  # we check `first(1) == []` to check if we have an
  # empty Enumerable; checking `empty?` would return
  # true for `[nil]`, which we want to deprecate to
  # keep consistent with Ruby
  elsif first.is_a?(Numeric) || first(1) == [] || first.respond_to?(:coerce)
    identity ||= 0
    _original_sum_with_required_identity(identity, &block)
  else
    ActiveSupport::Deprecation.warn(<<-MSG.squish)
      Rails 7.0 has deprecated Enumerable.sum in favor of Ruby's native implementation available since 2.4.
      Sum of non-numeric elements requires an initial argument.
    MSG
    inject(:+) || 0
  end
end