summaryrefslogtreecommitdiff
path: root/lib/milestone_array.rb
blob: 4ed8485b36a6e9a73bc4cbac8bda9eaf0a85036e (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
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