lib/typus/controller/format.rb in typus-3.1.0.rc18 vs lib/typus/controller/format.rb in typus-3.1.0.rc19

@@ -4,61 +4,74 @@
   module Controller
     module Format
 
       protected
 
-      def generate_html
+      def get_paginated_data
         items_per_page = @resource.typus_options_for(:per_page)
         @items = @resource.page(params[:page]).per(items_per_page)
       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)
+        if can_export?(:csv)
+          fields = @resource.typus_fields_for(:csv)
 
-        filename = Rails.root.join("tmp", "export-#{@resource.to_resource}-#{Time.zone.now.to_s(:number)}.csv")
+          filename = Rails.root.join("tmp", "export-#{@resource.to_resource}-#{Time.zone.now.to_s(:number)}.csv")
 
-        options = { :conditions => @conditions, :batch_size => 1000 }
+          options = { :conditions => @conditions, :batch_size => 1000 }
 
-        ::CSV.open(filename, 'w') do |csv|
-          csv << fields.keys
-          @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)
+          ::CSV.open(filename, 'w') do |csv|
+            csv << fields.keys
+            @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
                        end
-                     end
+              end
             end
           end
-        end
 
-        send_file filename
+          send_file filename
+        else
+          not_allowed
+        end
       end
 
       def generate_json
         export(:json)
       end
 
       def generate_xml
-        export(:xml)
+        can_export?(:xml) ? export(:xml) : not_allowed
       end
 
       def export(format)
         fields = @resource.typus_fields_for(format).map { |i| i.first }
         methods = fields - @resource.column_names
         except = @resource.column_names - fields
-        render format => @resource.send("to_#{format}", :methods => methods, :except => except)
+
+        get_paginated_data
+
+        render format => @items.send("to_#{format}", :methods => methods, :except => except)
+      end
+
+      def can_export?(format)
+        @resource.typus_options_for(:export).extract_settings.include?(format.to_s)
       end
 
     end
   end
 end