init project

app/libs/custom_sortable.rb

@@ -0,0 +1,45 @@
+module CustomSortable
+  extend ActiveSupport::Concern
+
+  included do |base|
+    base.instance_variable_set("@_sort_options", {})
+    base.instance_variable_set("@_sort_columns", [])
+    base.instance_variable_set("@_sort_directions", %w(asc desc))
+  end
+
+  def custom_sort(relations, sort_by, sort_direction)
+    sort_by = self.class.sort_options[:default_by] if sort_by.blank?
+    sort_direction = self.class.sort_options[:default_direction] if sort_direction.blank?
+
+    return relations unless self.class.check_sort_parameter_validate(sort_by.to_s, sort_direction.to_s)
+
+    order_method = self.class.sort_options[:reorder] ? :reorder : :order
+
+    default_table = self.class.sort_options[:default_table]
+    if default_table.present?
+      relations.send(order_method, "#{default_table}.#{sort_by} #{sort_direction}")
+    else
+      relations.send(order_method, "#{sort_by} #{sort_direction}")
+    end
+  end
+
+  module ClassMethods
+    def sort_columns(*columns)
+      opts = columns.extract_options!
+      @_sort_options[:default_by]        = opts[:default_by].to_s
+      @_sort_options[:default_direction] = opts[:default_direction].to_s
+      @_sort_options[:reorder]           = opts[:reorder]
+      @_sort_options[:default_table]     = opts[:default_table]
+
+      @_sort_columns = columns.map(&:to_s)
+    end
+
+    def check_sort_parameter_validate(sort_by, sort_direction)
+      (sort_by.blank? || @_sort_columns.include?(sort_by)) && @_sort_directions.include?(sort_direction)
+    end
+
+    def sort_options
+      @_sort_options
+    end
+  end
+end