summaryrefslogtreecommitdiff
path: root/lib/core_ext/enumerable.rb
blob: fd44b8c368989c68ef80a2a50f92c70841c7ba7b (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
# frozen_string_literal: true

module Enumerable
  # Returns duplicate elements from +self+.
  #
  # If a block is given, it will use the return value of the
  # given block for comparison.
  #
  # @return [Array<Array>]
  # @example
  #   [1, 2, 3].non_uniq
  #   #=> []
  #
  #   [1, 1, 2, 3].non_uniq
  #   #=> [[1, 1]]
  #
  #   [1, 1, 1, 2, 3].non_uniq
  #   #=> [[1, 1, 1]]
  #
  #   [1, 2, 3, 4].non_uniq { |x| x * 0 }
  #   #=> [[1, 2, 3, 4]]
  #
  def non_uniq(&block)
    block ||= proc { |item| item }
    group_by(&block).select { |_, occurrences| occurrences.count > 1 }.values
  end
end