Module: ActiveRecordEx::AssumeDestroy::ClassMethods

Defined in:
lib/active_record_ex/assume_destroy.rb

Instance Method Summary collapse

Instance Method Details

#accepts_nested_attributes_for(assoc_name, options = {}) ⇒ Object



13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
# File 'lib/active_record_ex/assume_destroy.rb', line 13

def accepts_nested_attributes_for(assoc_name, options={})
  assume_destroy = options[:assume_destroy]
  options.delete(:assume_destroy)
  options[:allow_destroy] = assume_destroy

  super assoc_name, options

  return unless assume_destroy
  attrs_name = ("#{assoc_name.to_s}_attributes").to_sym
  setter_name = ("#{attrs_name.to_s}=").to_sym
  unassuming_setter_name = ("#{attrs_name.to_s}_without_assume=").to_sym
  assuming_setter_name = ("#{attrs_name.to_s}_with_assume=").to_sym

  define_method(assuming_setter_name) do |attrs|
    ids = attrs.map { |a| a['id'] || a[:id] }.compact
    assocs = self.send(assoc_name)

    dead_assocs = []
    # the ternary's 'cause Arel doesn't do the right thing with an empty array
    dead_assocs = assocs.where('id NOT IN (?)', ids.any? ? ids : '') unless self.new_record?
    dead_attrs = dead_assocs.map {|assoc| {id: assoc.id, _destroy: true }}

    attrs = attrs.concat(dead_attrs)
    self.send(unassuming_setter_name, attrs)
  end
  alias_method_chain setter_name, :assume
end