CSV customization

app/controllers/spree/admin/base_controller_decorator.rb

@@ -0,0 +1,9 @@
+Spree::Admin::BaseController.class_eval do
+  def csv_filename args
+    args = {:name => 'report'}.merge(args)
+    fmt = '%Y-%m-%d'
+    parts = [args[:name]]
+    [:date_start, :date_end].each { |k| parts << args[k].strftime(fmt) if args.has_key?(k) }
+    parts.join('_') + '.csv'
+  end
+end

app/controllers/spree/admin/orders_controller_decorator.rb

@@ -4,6 +4,6 @@ def export_csv
     created_at_lt = Time.zone.parse(params[:created_at_lt]).end_of_day rescue ""
 
     @orders = Spree::Order.paid_and_ready_to_ship.completed_between(created_at_gt, created_at_lt).order("created_at")
-    send_data @orders.export_csv
+    send_data @orders.export_csv.encode(SpreeExport.output_encoding, :invalid => :replace, :replace => '?'), :filename => csv_filename(:name => 'orders', :date_start => created_at_gt, :date_end => created_at_lt)
   end
 end

app/controllers/spree/admin/users_controller_decorator.rb

@@ -1,6 +1,6 @@
 Spree::Admin::UsersController.class_eval do
   def export_csv
     @users = Spree::User.order("created_at")
-    send_data @users.export_csv
+    send_data @users.export_csv.encode(SpreeExport.output_encoding, :invalid => :replace, :replace => '?'), :filename => csv_filename(:name => 'users', :date_end => Time.now)
   end
 end

app/models/spree/order_decorator.rb

@@ -10,14 +10,22 @@ def self.paid_and_ready_to_ship
   end
 
   def self.export_csv(options = {})
+    options = SpreeExport.csv_options.merge options
+
     CSV.generate(options) do |csv|
-      address_column_names = ["firstname", "lastname", "address1", "address2", "city", "zipcode", "phone", "state_name", "alternative_phone", "company"]
-      order_column_names = ["id", "number", "total", "state", "shipment_state", "email", "user_id", "completed_at", "payment_state", "special_instructions"]
-      column_headers = order_column_names + address_column_names
-      csv << column_headers
-      all.each do |order|
-        values = order.attributes.values_at(*order_column_names) + order.ship_address.attributes.values_at(*address_column_names)
-        csv << values
+
+      if SpreeExport.order_csv_block.nil?
+
+        address_column_names = ["firstname", "lastname", "address1", "address2", "city", "zipcode", "phone", "state_name", "alternative_phone", "company"]
+        order_column_names = ["id", "number", "total", "state", "shipment_state", "email", "user_id", "completed_at", "payment_state", "special_instructions"]
+        column_headers = order_column_names + address_column_names
+        csv << column_headers
+        all.each do |order|
+          values = order.attributes.values_at(*order_column_names) + order.ship_address.attributes.values_at(*address_column_names)
+          csv << values
+        end
+      else
+        SpreeExport.order_csv_block.call csv, all
       end
     end
   end

app/models/spree/user_decorator.rb

@@ -2,13 +2,19 @@
 Spree::User.instance_eval do
 
   def self.export_csv(options = {})
+    options = SpreeExport.csv_options.merge options
+
     CSV.generate(options) do |csv|
-      user_column_names = ["id", "email", "login"]
-      column_headers = user_column_names
-      csv << column_headers
-      all.each do |user|
-        values = user.attributes.values_at(*user_column_names)
-        csv << values
+      if SpreeExport.user_csv_block.nil?
+        user_column_names = ["id", "email", "login"]
+        column_headers = user_column_names
+        csv << column_headers
+        all.each do |user|
+          values = user.attributes.values_at(*user_column_names)
+          csv << values
+        end
+      else
+        SpreeExport.user_csv_block.call csv, all
       end
     end
   end

lib/generators/spree_export/install/install_generator.rb

@@ -14,6 +14,19 @@ def add_stylesheets
         inject_into_file 'app/assets/stylesheets/admin/all.css', " *= require admin/spree_export\n", :before => /\*\//, :verbose => true
       end
 
+      def create_initializer
+        create_file 'config/initializers/spree_export.rb' do
+          string = <<-RUBY.gsub /^ {12}/, ''
+            # config/initializers/spree_export.rb
+
+            SpreeExport.config do |config|
+              config.csv_options = {}
+              config.output_encoding = Encoding::UTF_8
+            end
+          RUBY
+        end
+      end
+
       def add_migrations
         run 'bundle exec rake railties:install:migrations FROM=spree_export'
       end

lib/spree_export.rb

@@ -1,2 +1,19 @@
 require 'spree_core'
 require 'spree_export/engine'
+
+module SpreeExport
+
+  mattr_accessor :csv_options
+  self.csv_options = {}
+
+  mattr_accessor :output_encoding
+  self.output_encoding = Encoding::UTF_8
+
+  mattr_accessor :order_csv_block
+  mattr_accessor :user_csv_block
+
+  def self.config &block
+    yield(self)
+  end
+
+end

spree_export.gemspec

@@ -20,7 +20,7 @@ Gem::Specification.new do |s|
 
   s.add_development_dependency 'capybara', '~> 2.1'
   s.add_development_dependency 'coffee-rails'
-  s.add_development_dependency 'database_cleaner'
+  s.add_development_dependency 'database_cleaner', '< 1.1.0'
   s.add_development_dependency 'factory_girl', '~> 4.2'
   s.add_development_dependency 'ffaker'
   s.add_development_dependency 'rspec-rails',  '~> 2.13'