Contents

require 'csv'

module AbAdmin
  module Utils
    class CsvDocument
      include AbAdmin::Utils::EvalHelpers

      def initialize(source, options = {})
        @source = source
        @options = options
        @klass = @options.delete(:klass) || extract_klass
      end

      def default_columns
        @default_columns ||= @klass.column_names
      end

      def column_data
        @columns_names ||= @options[:column_data] || default_columns
      end

      def columns_names
        (@options[:column_names] || default_columns).map { |column| column.is_a?(Symbol) ? @klass.human_attribute_name(column) : column }
      end

      def filename
        @filename ||= [@options[:filename] || "#{@klass.model_name.plural}-#{Time.now.strftime('%Y-%m-%d')}", '.csv'].join
      end

      def render
        ::CSV.generate(:col_sep => @options[:column_separator] || ',') do |csv|
          csv << columns_names

          each_record do |item|
            csv << column_data.map { |column| AbAdmin.pretty_data call_method_or_proc_on(item, column, :exec => false) }
          end
        end
      end

      def each_record
        if @source.respond_to?(:find_each)
          @source.find_each do |item|
            yield item
          end
        else
          Array(@source).each do |item|
            yield item
          end
        end
      end

      protected

      def extract_klass
        if @source.respond_to?(:klass)
          @source.klass
        elsif @source.is_a?(Array)
          @source.first.try(:class)
        else
          @source.class
        end
      end

    end
  end
end

Version data entries

7 entries across 7 versions & 1 rubygems

Version	Path
ab_admin-0.2.3	lib/ab_admin/utils/csv_document.rb
ab_admin-0.2.2	lib/ab_admin/utils/csv_document.rb
ab_admin-0.2.1	lib/ab_admin/utils/csv_document.rb
ab_admin-0.2.0	lib/ab_admin/utils/csv_document.rb
ab_admin-0.1.2	lib/ab_admin/utils/csv_document.rb
ab_admin-0.1.1	lib/ab_admin/utils/csv_document.rb
ab_admin-0.1.0	lib/ab_admin/utils/csv_document.rb