Contents

class ArrayToAssociationWrapper < Array

  def defer_association_methods_to owner, association_name
    @association_owner = owner
    @association_name = association_name
  end

  # trick collection_name.include?(obj)
  # If you use a collection of SingleTableInheritance and didn't :select 'type' the
  # include? method will not find any subclassed object.
  def include_with_deferred_save?(obj)
    if @association_owner.present?
      if self.detect { |itm| itm == obj || (itm[:id] == obj[:id] && obj.is_a?(itm.class)) }
        return true
      else
        return false
      end
    else
      include_without_deferred_save?(obj)
    end
  end

  alias_method_chain :include?, 'deferred_save'

  def find_with_deferred_save *args
    if @association_owner.present?
      collection_without_deferred_save.send(:find, *args)
    else
      find_without_deferred_save
    end
  end

  alias_method_chain :find, :deferred_save

  def first_with_deferred_save *args
    if @association_owner.present?
      collection_without_deferred_save.send(:first, *args)
    else
      first_without_deferred_save
    end
  end

  alias_method_chain :first, :deferred_save

  def last_with_deferred_save *args
    if @association_owner.present?
      collection_without_deferred_save.send(:last, *args)
    else
      last_without_deferred_save
    end
  end

  alias_method_chain :last, :deferred_save

  define_method :method_missing do |method, *args|
    #puts "#{self.class}.method_missing(#{method}) (#{collection_without_deferred_save.inspect})"
    if @association_owner.present?
      collection_without_deferred_save.send(method, *args) unless method == :set_inverse_instance
    else
      super
    end
  end

  def collection_without_deferred_save
    @association_owner.send("#{@association_name}_without_deferred_save")
  end
end

Version data entries

8 entries across 8 versions & 1 rubygems

Version	Path
deferred_associations-0.5.7	lib/array_to_association_wrapper.rb
deferred_associations-0.5.6	lib/array_to_association_wrapper.rb
deferred_associations-0.5.5	lib/array_to_association_wrapper.rb
deferred_associations-0.5.4	lib/array_to_association_wrapper.rb
deferred_associations-0.5.3	lib/array_to_association_wrapper.rb
deferred_associations-0.5.2	lib/array_to_association_wrapper.rb
deferred_associations-0.5.1	lib/array_to_association_wrapper.rb
deferred_associations-0.5.0	lib/array_to_association_wrapper.rb