validates_timeliness/lib/validates_timeliness/orm/mongoid.rb

55 lines
1.5 KiB
Ruby

module ValidatesTimeliness
module ORM
module Mongoid
extend ActiveSupport::Concern
# You need define the fields before you define the validations.
# It is best to use the plugin parser to avoid errors on a bad
# field value in Mongoid. Parser will return nil rather than error.
module ClassMethods
# Mongoid has no bulk attribute method definition hook. It defines
# them with each field definition. So we likewise define them after
# each validation is defined.
#
def timeliness_validation_for(attr_names, type)
super
attr_names.each { |attr_name| define_timeliness_write_method(attr_name) }
end
def timeliness_type_cast_code(attr_name, var_name)
type = timeliness_attribute_type(attr_name)
"#{var_name} = Timeliness::Parser.parse(value, :#{type})"
end
def timeliness_attribute_type(attr_name)
{
Date => :date,
Time => :datetime,
DateTime => :datetime
}[fields[attr_name.to_s].type] || :datetime
end
end
def reload
_clear_timeliness_cache
super
end
end
end
end
module Mongoid::Document
include ValidatesTimeliness::AttributeMethods
include ValidatesTimeliness::ORM::Mongoid
# Pre-2.3 reload
if instance_methods.include?('reload')
def reload_with_timeliness
_clear_timeliness_cache
reload_without_timeliness
end
alias_method_chain :reload, :timeliness
end
end