1 2 3 4 5 6 7 8 9 10 11 12 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 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74
|
module AttrRequired
class AttrMissing < StandardError; end
def self.included(klass)
klass.send :extend, ClassMethods
end
module ClassMethods
def inherited(klass)
super
unless required_attributes.empty?
klass.attr_required *required_attributes
end
end
def attr_required(*keys)
if defined? undef_optional_attributes
undef_optional_attributes *keys
end
required_attributes.concat keys
attr_accessor *keys
end
def attr_required?(key)
required_attributes.include? key
end
def required_attributes
@required_attributes ||= []
end
def undef_required_attributes(*keys)
keys.each do |key|
if attr_required?(key)
undef_method key, :"#{key}="
required_attributes.delete key
end
end
end
end
def required_attributes
self.class.required_attributes
end
def attr_required?(key)
self.class.attr_required? key
end
def attr_missing?
!attr_missing.empty?
end
def attr_missing!
if attr_missing?
raise AttrMissing.new("'#{attr_missing.join('\', \'')}' required.")
end
end
def attr_missing
required_attributes.select do |key|
value = send(key)
if value.respond_to?(:empty?)
value.empty?
else
value.nil?
end
end
end
end
|