ihatov08 · March 1, 2017 00:08 · Oct 17, 2013 · Oct 17, 2013 · Oct 15, 2013 · Oct 15, 2013
diff --git a/csv_stream.rb b/csv_stream.rb
@@ -24,7 +24,7 @@ def index
 
     def render_csv      
       set_csv_headers
-      self_response_body = csv_lines
+      self.response_body = csv_lines
     end
 
     def set_csv_headers

diff --git a/active_admin.rb b/active_admin.rb
@@ -1,4 +1,4 @@
 # if you want to monkey patch every controller, put this in initializers/active_admin.rb
-class ActiveAdmin::ResourceController
+ActiveAdmin::ResourceController.class_eval do
   include ActiveAdmin::CSVStream
-end
+end
diff --git a/csv_stream.rb b/csv_stream.rb
@@ -1,11 +1,13 @@
 # put in an autoloaded path, like lib/active_admin
-
 require 'csv'
+
 module ActiveAdmin
 
   module CSVStream
 
-    include ActionController::Live
+    # using ActionController::Live breaks warden
+    # see https://github.com/plataformatec/devise/issues/2332
+    # include ActionController::Live
 
     def apply_pagination(chain)
       chain = super unless formats.include?(:csv)
@@ -22,7 +24,7 @@ def index
 
     def render_csv      
       set_csv_headers
-      csv_lines
+      self_response_body = csv_lines
     end
 
     def set_csv_headers
@@ -33,17 +35,21 @@ def csv_lines
       default = ActiveAdmin.application.csv_options
       options = default.merge active_admin_config.csv_builder.options
       columns = active_admin_config.csv_builder.columns
-
-      CSV.generate(options) do |csv|
-        response.stream.write columns.map(&:name).to_csv
-        collection.each do |resource|
-          row = columns.map do |column|
-            call_method_or_proc_on resource, column.data
-          end.to_csv
-          response.stream.write row
+
+      Enumerator.new do |y|
+        CSV.generate(options) do |csv|
+          y << columns.map(&:name).to_csv
+          #response.stream.write columns.map(&:name).to_csv
+          collection.each do |resource|
+            row = columns.map do |column|
+              call_method_or_proc_on resource, column.data
+            end.to_csv
+            y << row
+            #response.stream.write row
+          end
         end
       end
-      response.stream.close
+      #response.stream.close
     end
 
   end

diff --git a/active_admin.rb b/active_admin.rb
@@ -1,4 +1,4 @@
 # if you want to monkey patch every controller, put this in initializers/active_admin.rb
-module ActiveAdmin::ResourceController
+class ActiveAdmin::ResourceController
   include ActiveAdmin::CSVStream
 end
diff --git a/active_admin.rb b/active_admin.rb
@@ -0,0 +1,4 @@
+# if you want to monkey patch every controller, put this in initializers/active_admin.rb
+module ActiveAdmin::ResourceController
+  include ActiveAdmin::CSVStream
+end
diff --git a/csv_stream.rb b/csv_stream.rb
@@ -1,4 +1,6 @@
-# put in an autoload path, like lib/active_admin
+# put in an autoloaded path, like lib/active_admin
+
+require 'csv'
 module ActiveAdmin
 
   module CSVStream

diff --git a/users.rb b/users.rb
@@ -1,4 +1,4 @@
-#admin/users.rb
+#admin/users.rb to use in a single controller
 ActiveAdmin.register User do
   controller do
     include ActiveAdmin::CSVStream

diff --git a/csv_stream.rb b/csv_stream.rb
@@ -1,3 +1,4 @@
+# put in an autoload path, like lib/active_admin
 module ActiveAdmin
 
   module CSVStream

diff --git a/users.rb b/users.rb
@@ -0,0 +1,6 @@
+#admin/users.rb
+ActiveAdmin.register User do
+  controller do
+    include ActiveAdmin::CSVStream
+  end
+end
diff --git a/csv_stream.rb b/csv_stream.rb
@@ -0,0 +1,48 @@
+module ActiveAdmin
+
+  module CSVStream
+
+    include ActionController::Live
+
+    def apply_pagination(chain)
+      chain = super unless formats.include?(:csv)
+      chain
+    end
+
+    def index
+      index! do |format|
+        format.csv { render_csv }
+      end
+    end
+
+    private
+
+    def render_csv      
+      set_csv_headers
+      csv_lines
+    end
+
+    def set_csv_headers
+      headers["Content-Type"] = "text/csv"
+    end
+
+    def csv_lines
+      default = ActiveAdmin.application.csv_options
+      options = default.merge active_admin_config.csv_builder.options
+      columns = active_admin_config.csv_builder.columns
+
+      CSV.generate(options) do |csv|
+        response.stream.write columns.map(&:name).to_csv
+        collection.each do |resource|
+          row = columns.map do |column|
+            call_method_or_proc_on resource, column.data
+          end.to_csv
+          response.stream.write row
+        end
+      end
+      response.stream.close
+    end
+
+  end
+
+end