summaryrefslogtreecommitdiff
path: root/lib/milestone_array.rb
blob: 107abe2913d1fc70ea7bca04a334a1d809541af1 (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
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
# frozen_string_literal: true

module MilestoneArray
  class << self
    def sort(array, sort_method)
      case sort_method
      when "due_date_asc"
        sort_asc_nulls_last(array, "due_date")
      when "due_date_desc"
        sort_desc_nulls_last(array, "due_date")
      when "start_date_asc"
        sort_asc_nulls_last(array, "start_date")
      when "start_date_desc"
        sort_desc_nulls_last(array, "start_date")
      when "name_asc"
        sort_asc(array, "title")
      when "name_desc"
        sort_asc(array, "title").reverse
      else
        array
      end
    end

    private

    def sort_asc_nulls_last(array, attribute)
      attribute = attribute.to_sym

      array.select(&attribute).sort_by(&attribute) + array.reject(&attribute)
    end

    def sort_desc_nulls_last(array, attribute)
      attribute = attribute.to_sym

      array.select(&attribute).sort_by(&attribute).reverse + array.reject(&attribute)
    end

    def sort_asc(array, attribute)
      array.sort_by(&attribute.to_sym)
    end
  end
end