lib/ransack/helpers/form_helper.rb in ransack-0.5.4 vs lib/ransack/helpers/form_helper.rb in ransack-0.5.5

@@ -1,8 +1,9 @@
 module Ransack
   module Helpers
     module FormHelper
+
       def search_form_for(record, options = {}, &proc)
         if record.is_a?(Ransack::Search)
           search = record
           options[:url] ||= polymorphic_path(search.klass)
         elsif record.is_a?(Array) && (search = record.detect {|o| o.is_a?(Ransack::Search)})
@@ -19,9 +20,55 @@
         options[:as] ||= 'q'
         options[:html].reverse_merge!(html_options)
         options[:builder] ||= FormBuilder
 
         form_for(record, options, &proc)
+      end
+
+      def sort_link(search, attribute, *args)
+        raise TypeError, "First argument must be a Ransack::Search!" unless Search === search
+
+        search_params = params[:q] || {}.with_indifferent_access
+
+        attr_name = attribute.to_s
+
+        name = (args.size > 0 && !args.first.is_a?(Hash)) ? args.shift.to_s : Translate.attribute(attr_name, :context => search.context)
+
+        if existing_sort = search.sorts.detect {|s| s.name == attr_name}
+          prev_attr, prev_dir = existing_sort.name, existing_sort.dir
+        end
+
+        options = args.first.is_a?(Hash) ? args.shift.dup : {}
+        default_order = options.delete :default_order
+        current_dir = prev_attr == attr_name ? prev_dir : nil
+
+        if current_dir
+          new_dir = current_dir == 'desc' ? 'asc' : 'desc'
+        else
+          new_dir = default_order || 'asc'
+        end
+
+        html_options = args.first.is_a?(Hash) ? args.shift.dup : {}
+        css = ['sort_link', current_dir].compact.join(' ')
+        html_options[:class] = [css, html_options[:class]].compact.join(' ')
+        options.merge!(
+          :q => search_params.merge(:s => "#{attr_name} #{new_dir}")
+        )
+        link_to [ERB::Util.h(name), order_indicator_for(current_dir)].compact.join(' ').html_safe,
+                url_for(options),
+                html_options
+      end
+
+      private
+
+      def order_indicator_for(order)
+        if order == 'asc'
+          '▲'
+        elsif order == 'desc'
+          '▼'
+        else
+          nil
+        end
       end
 
     end
   end
 end
\ No newline at end of file