lib/typus/controller/format.rb in typus-3.1.8 vs lib/typus/controller/format.rb in typus-3.1.9

@@ -4,41 +4,42 @@
   module Controller
     module Format
 
       protected
 
+      # This is crazy, but I want to have support for Kaminari, WillPaginate
+      # and whatever other pagination system which comes.
       def get_paginated_data
         items_per_page = @resource.typus_options_for(:per_page)
-        @items = @resource.page(params[:page]).per(items_per_page)
+
+        @items = if defined?(Kaminari)
+          @resource.page(params[:page]).per(items_per_page)
+        elsif defined?(WillPaginate)
+          @resource.paginate(:page => params[:page], :per_page => items_per_page)
+        else
+          @resource.limit(items_per_page) # Pagination is disabled, so just in case limit to 50 records.
+        end
       end
 
       alias_method :generate_html, :get_paginated_data
 
-      #--
-      # TODO: Find in batches only works properly if it's used on models, not
-      #       controllers, so in this action does nothing. We should find a way
-      #       to be able to process large amounts of data.
-      #++
       def generate_csv
         fields = @resource.typus_fields_for(:csv)
-        options = { :conditions => @conditions, :batch_size => 1000 }
 
         data = ::CSV.generate do |csv|
           csv << fields.keys.map { |k| @resource.human_attribute_name(k) }
-          @resource.find_in_batches(options) do |records|
-            records.each do |record|
-              csv << fields.map do |key, value|
-                       case value
-                       when :transversal
-                         a, b = key.split(".")
-                         record.send(a).send(b)
-                       when :belongs_to
-                         record.send(key).try(:to_label)
-                       else
-                         record.send(key)
-                       end
+          @resource.all.each do |record|
+            csv << fields.map do |key, value|
+                     case value
+                     when :transversal
+                       a, b = key.split(".")
+                       record.send(a).send(b)
+                     when :belongs_to
+                       record.send(key).try(:to_label)
+                     else
+                       record.send(key)
                      end
-            end
+                   end
           end
         end
 
         send_data data, :filename => "export-#{@resource.to_resource}-#{Time.zone.now.to_s(:number)}.csv"
       end