Contents

require 'active_record'
require 'csv'

module RailsCsvRenderer
  module Renderable
    # Converts an collection of active records to CSV formatted string
    # Options is configuration set for generated CSV file

    def to_custom_csv(options = {})
      csv_options = default_csv_options.merge(options)

      if is_active_record?
        if !(model.respond_to?(:csv_header) || model.respond_to?(:csv_row)) || model.class_variable_defined?(:@@dynamic_generated_csv_methods)
          define_csv_methods(options)
        end
      end

      CSV.generate(csv_options) do |csv|
        if is_active_record?
          csv << model.csv_header
          self.each do |obj|
            csv << obj.csv_row
          end
        else
          csv << self if respond_to?(:to_csv)
        end
      end
    end

    private

    def define_csv_methods(options)
      columns = model.column_names

      model.class_variable_set(:@@dynamic_generated_csv_methods, true)
      model.class_eval(%Q{
        class << self
          def csv_header
            ["#{ columns.map { |column_name| model.human_attribute_name(column_name) }.join('", "') }"]
          end
        end
      })
      model.class_eval(%Q{
        def csv_row
         [#{ columns.join(', ') }]
       end
      })
    end

    def is_active_record?
      is_a?(ActiveRecord::Relation) || (present? && first.is_a?(ActiveRecord::Base))
    end

    def model
      @model ||= is_a?(ActiveRecord::Relation) ? klass : first.class
    end

    def default_csv_options
      { encoding: 'utf-8' }
    end
  end
end

Version data entries

1 entries across 1 versions & 1 rubygems

Version	Path
rails_csv_renderer-0.1.0	lib/rails_csv_renderer/renderable.rb