couchrest_model/lib/couchrest/model/casted_array.rb

84 lines
1.9 KiB
Ruby
Raw Normal View History

#
# Wrapper around Array so that the casted_by attribute is set in all
# elements of the array.
#
module CouchRest::Model
class CastedArray < Array
include CouchRest::Model::CastedBy
include CouchRest::Model::Dirty
attr_accessor :casted_by_property
2010-06-16 21:04:53 +02:00
def initialize(array, property, parent = nil)
self.casted_by_property = property
self.casted_by = parent unless parent.nil?
2010-06-16 21:04:53 +02:00
super(array)
end
# Adding new entries
def << obj
2010-06-16 21:04:53 +02:00
super(instantiate_and_cast(obj))
end
def push(obj)
2010-06-16 21:04:53 +02:00
super(instantiate_and_cast(obj))
end
def unshift(obj)
super(instantiate_and_cast(obj))
end
def []= index, obj
value = instantiate_and_cast(obj, false)
couchrest_parent_will_change! if use_dirty? && value != self[index]
super(index, value)
end
def pop
couchrest_parent_will_change! if use_dirty? && self.length > 0
super
end
def shift
couchrest_parent_will_change! if use_dirty? && self.length > 0
super
end
def clear
couchrest_parent_will_change! if use_dirty? && self.length > 0
super
end
def delete(obj)
couchrest_parent_will_change! if use_dirty? && self.length > 0
super(obj)
end
def delete_at(index)
couchrest_parent_will_change! if use_dirty? && self.length > 0
super(index)
end
def build(*args)
obj = casted_by_property.build(*args)
self.push(obj)
obj
end
2010-06-16 21:04:53 +02:00
protected
def instantiate_and_cast(obj, change = true)
property = casted_by_property
couchrest_parent_will_change! if change && use_dirty?
if casted_by && property && obj.class != property.type_class
property.cast_value(casted_by, obj)
2010-06-16 21:04:53 +02:00
else
obj.casted_by = casted_by if obj.respond_to?(:casted_by)
obj.casted_by_property = casted_by_property if obj.respond_to?(:casted_by_property)
2010-06-16 21:04:53 +02:00
obj
end
end
end
end