to_csv_rails 0.1

data/Rakefile

@@ -0,0 +1,8 @@
+require 'rake/testtask'
+
+Rake::TestTask.new do |t|
+  t.libs << 'test'
+end
+
+desc "Run tests"
+task :default => :test

data/lib/to_csv_rails/array.rb

@@ -0,0 +1,28 @@
+class Array
+  def to_csv(options = {})
+    return '' unless self.any?
+
+    columns = self.first.attributes.keys.map(&:to_sym)
+
+    if options[:only]
+      columns.delete_if{|column| !options[:only].include?(column)}
+    end
+
+    if options[:except]
+      columns.delete_if{|column| options[:except].include?(column)}
+    end
+
+    columns += Array(options[:methods])
+
+    output = CSV_HANDLER.generate(Hash.new(options[:csv_options])) do |csv|
+
+      csv << options[:headers] unless options[:headers].nil?
+
+      self.each do |row|
+        csv << columns.collect{|column| row.send(column) }
+      end
+    end
+
+    output
+  end
+end

data/lib/to_csv_rails/version.rb

@@ -0,0 +1,3 @@
+module ToCsvRails
+  VERSION = "0.1"
+end

data/lib/to_csv_rails.rb

@@ -0,0 +1,17 @@
+
+if RUBY_VERSION >= '1.9'
+  require 'csv'
+  CSV_HANDLER = CSV
+else
+  begin
+    gem 'fastercsv'
+    require 'fastercsv'
+
+    CSV_HANDLER = FasterCSV
+  rescue LoadError => e
+    raise "Error : FasterCSV not installed, please run: gem install fastercsv"
+  end
+end
+
+require "to_csv_rails/version"
+require "to_csv_rails/array.rb"

data/test/test_csv.rb

@@ -0,0 +1,54 @@
+require 'test_helper'
+
+class CsvTest < Test::Unit::TestCase
+  def teardown
+    User.destroy_all
+  end
+
+  def test_simple_export
+    user1 = User.create! :name => 'User 1', :email => 'user1@email.com'
+    user2 = User.create! :name => 'User 2', :email => 'user2@email.com'
+
+    data = User.all
+    
+    assert_equal data.to_csv_rails,
+      "#{user1.id},User 1,user1@email.com\n#{user2.id},User 2,user2@email.com\n"
+  end
+
+  def test_export_with_empty_data
+    data = User.all
+
+    assert_equal data.to_csv_rails,
+      ""
+  end
+
+  def test_export_with_only_option
+    User.create! :name => 'User 1', :email => 'user1@email.com'
+    User.create! :name => 'User 2', :email => 'user2@email.com'
+
+    data = User.all
+
+    assert_equal data.to_csv_rails(:only => [:name]),
+      "User 1\nUser 2\n"
+  end
+
+  def test_export_with_except_option
+    User.create! :name => 'User 1', :email => 'user1@email.com'
+    User.create! :name => 'User 2', :email => 'user2@email.com'
+
+    data = User.all
+
+    assert_equal data.to_csv_rails(:except => [:id, :name]),
+      "user1@email.com\nuser2@email.com\n"
+  end
+
+  def test_export_with_headers_option
+    user1 = User.create! :name => 'User 1', :email => 'user1@email.com'
+    user2 = User.create! :name => 'User 2', :email => 'user2@email.com'
+
+    data = User.all
+
+    assert_equal data.to_csv_rails(:headers => [:id, :name, :email]),
+      "id,name,email\n#{user1.id},User 1,user1@email.com\n#{user2.id},User 2,user2@email.com\n"
+  end
+end

data/test/test_helper.rb

@@ -0,0 +1,17 @@
+require 'rubygems'
+require 'active_record'
+require 'test/unit'
+require 'to_csv_rails'
+
+ActiveRecord::Base.establish_connection(:adapter => "sqlite3", :database => ":memory:")
+
+ActiveRecord::Schema.define(:version => 1) do
+  create_table :users do |t|
+    t.string :name
+    t.string :email
+  end
+end
+
+class User < ActiveRecord::Base
+  
+end

data/to_csv_rails.gemspec

@@ -0,0 +1,17 @@
+$:.push File.expand_path("../lib", __FILE__)
+require "to_csv_rails/version"
+
+Gem::Specification.new do |s|
+  s.name        = "to_csv_rails"
+  s.version     = ToCsvRails::VERSION
+  s.authors     = ["Fernando Morgenstern"]
+  s.email       = ["contato@fernandomarcelo.com"]
+  s.homepage    = "http://fernandomarcelo.com/"
+  s.summary     = %q{An to_csv implementation for Rails framework}
+  s.description = %q{An to_csv implementation focused on making it easier for Rails developers to export data to CSV format.}
+  s.files = Dir['**/*'].select{|f| File.file?(f)}
+
+  if RUBY_VERSION < "1.9.0"
+    s.add_dependency('fastercsv')
+  end
+end

metadata

@@ -0,0 +1,54 @@
+--- !ruby/object:Gem::Specification
+name: to_csv_rails
+version: !ruby/object:Gem::Version
+  version: '0.1'
+  prerelease: 
+platform: ruby
+authors:
+- Fernando Morgenstern
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2012-12-29 00:00:00.000000000 Z
+dependencies: []
+description: An to_csv implementation focused on making it easier for Rails developers
+  to export data to CSV format.
+email:
+- contato@fernandomarcelo.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- lib/to_csv_rails/array.rb
+- lib/to_csv_rails/version.rb
+- lib/to_csv_rails.rb
+- Rakefile
+- test/test_csv.rb
+- test/test_helper.rb
+- to_csv_rails-0.1.gem
+- to_csv_rails.gemspec
+homepage: http://fernandomarcelo.com/
+licenses: []
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 1.8.24
+signing_key: 
+specification_version: 3
+summary: An to_csv implementation for Rails framework
+test_files: []