Contents

# -*- coding: utf-8 -*-
require 'comma/extractor'
require 'active_support/core_ext/class/attribute'
require 'active_support/core_ext/date_time/conversions'
require 'active_support/core_ext/object/blank'
require 'active_support/core_ext/string/inflections'

module Comma
  class HeaderExtractor < Extractor

    class_attribute :value_humanizer

    DEFAULT_VALUE_HUMANIZER = lambda do |value, model_class|
      value.is_a?(String) ? value : value.to_s.humanize
    end
    self.value_humanizer = DEFAULT_VALUE_HUMANIZER

    def method_missing(sym, *args, &block)
      model_class = @instance.class
      @results << self.value_humanizer.call(sym, model_class) if args.blank?
      args.each do |arg|
        case arg
        when Hash
          arg.each do |k, v|
            @results << self.value_humanizer.call(v, get_association_class(model_class, sym))
          end
        when Symbol
          @results << self.value_humanizer.call(arg, get_association_class(model_class, sym))
        when String
          @results << self.value_humanizer.call(arg, model_class)
        else
          raise "Unknown header symbol #{arg.inspect}"
        end
      end
    end

    def __static_column__(header = '', &block)
      @results << header
    end

    private

    def get_association_class(model_class, association)
      model_class.respond_to?(:reflect_on_association) ?
        model_class.reflect_on_association(association).klass : nil
    end
  end
end

Version data entries

1 entries across 1 versions & 1 rubygems

Version	Path
comma-3.2.0	lib/comma/header_extractor.rb