summaryrefslogtreecommitdiff
path: root/app/models/concerns/id_in_ordered.rb
blob: b89409e6841b9c2f3e53d68a3f2d93db08d3e937 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
# frozen_string_literal: true

module IdInOrdered
  extend ActiveSupport::Concern

  included do
    scope :id_in_ordered, -> (ids) do
      raise ArgumentError, "ids must be an array of integers" unless ids.is_a?(Enumerable) && ids.all? { |id| id.is_a?(Integer) }

      # No need to sort if no more than 1 and the sorting code doesn't work
      # with an empty array
      return id_in(ids) unless ids.count > 1

      id_attribute = arel_table[:id]
      id_in(ids)
        .order(
          Arel.sql("array_position(ARRAY[#{ids.join(',')}], #{id_attribute.relation.name}.#{id_attribute.name})"))
    end
  end
end