141 lines
4.8 KiB
Ruby
141 lines
4.8 KiB
Ruby
require File.join(File.dirname(__FILE__), '..', 'more', 'property')
|
|
|
|
class Time
|
|
# returns a local time value much faster than Time.parse
|
|
def self.mktime_with_offset(string)
|
|
string =~ /(\d{4})\/(\d{2})\/(\d{2}) (\d{2}):(\d{2}):(\d{2}) ([\+\-])(\d{2})/
|
|
# $1 = year
|
|
# $2 = month
|
|
# $3 = day
|
|
# $4 = hours
|
|
# $5 = minutes
|
|
# $6 = seconds
|
|
# $7 = time zone direction
|
|
# $8 = tz difference
|
|
# utc time with wrong TZ info:
|
|
time = mktime($1, RFC2822_MONTH_NAME[$2.to_i - 1], $3, $4, $5, $6, $7)
|
|
tz_difference = ("#{$7 == '-' ? '+' : '-'}#{$8}".to_i * 3600)
|
|
time + tz_difference + zone_offset(time.zone)
|
|
end
|
|
end
|
|
|
|
module CouchRest
|
|
module Mixins
|
|
module Properties
|
|
|
|
class IncludeError < StandardError; end
|
|
|
|
def self.included(base)
|
|
base.class_eval <<-EOS, __FILE__, __LINE__
|
|
extlib_inheritable_accessor(:properties) unless self.respond_to?(:properties)
|
|
self.properties ||= []
|
|
EOS
|
|
base.extend(ClassMethods)
|
|
raise CouchRest::Mixins::Properties::IncludeError, "You can only mixin Properties in a class responding to [] and []=, if you tried to mixin CastedModel, make sure your class inherits from Hash or responds to the proper methods" unless (base.new.respond_to?(:[]) && base.new.respond_to?(:[]=))
|
|
end
|
|
|
|
def apply_defaults
|
|
return if self.respond_to?(:new_document?) && (new_document? == false)
|
|
return unless self.class.respond_to?(:properties)
|
|
return if self.class.properties.empty?
|
|
# TODO: cache the default object
|
|
self.class.properties.each do |property|
|
|
key = property.name.to_s
|
|
# let's make sure we have a default
|
|
unless property.default.nil?
|
|
if property.default.class == Proc
|
|
self[key] = property.default.call
|
|
else
|
|
self[key] = Marshal.load(Marshal.dump(property.default))
|
|
end
|
|
end
|
|
end
|
|
end
|
|
|
|
def cast_keys
|
|
return unless self.class.properties
|
|
self.class.properties.each do |property|
|
|
key = self.has_key?(property.name) ? property.name : property.name.to_sym
|
|
# Don't cast the property unless it has a value
|
|
next if (value = self[key]).nil?
|
|
write_property(property, value)
|
|
end
|
|
end
|
|
|
|
protected
|
|
|
|
def write_attribute(name, value)
|
|
unless (property = property(name)).nil?
|
|
write_property(property, value)
|
|
else
|
|
self[name] = value
|
|
end
|
|
end
|
|
|
|
def write_property(property, value)
|
|
value = property.typecast(value)
|
|
value.casted_by = self if value.respond_to?(:casted_by)
|
|
self[property.name] = value
|
|
end
|
|
|
|
def property(name)
|
|
properties.find {|p| p.name == name.to_s}
|
|
end
|
|
|
|
module ClassMethods
|
|
|
|
def property(name, options={})
|
|
existing_property = self.properties.find{|p| p.name == name.to_s}
|
|
if existing_property.nil? || (existing_property.default != options[:default])
|
|
define_property(name, options)
|
|
end
|
|
end
|
|
|
|
protected
|
|
|
|
# This is not a thread safe operation, if you have to set new properties at runtime
|
|
# make sure to use a mutex.
|
|
def define_property(name, options={})
|
|
# check if this property is going to casted
|
|
options[:casted] = options[:cast_as] ? options[:cast_as] : false
|
|
property = CouchRest::Property.new(name, (options.delete(:cast_as) || options.delete(:type)), options)
|
|
create_property_getter(property)
|
|
create_property_setter(property) unless property.read_only == true
|
|
properties << property
|
|
end
|
|
|
|
# defines the getter for the property (and optional aliases)
|
|
def create_property_getter(property)
|
|
# meth = property.name
|
|
class_eval <<-EOS, __FILE__, __LINE__
|
|
def #{property.name}
|
|
self['#{property.name}']
|
|
end
|
|
EOS
|
|
|
|
if property.alias
|
|
class_eval <<-EOS, __FILE__, __LINE__
|
|
alias #{property.alias.to_sym} #{property.name.to_sym}
|
|
EOS
|
|
end
|
|
end
|
|
|
|
# defines the setter for the property (and optional aliases)
|
|
def create_property_setter(property)
|
|
meth = property.name
|
|
class_eval <<-EOS
|
|
def #{meth}=(value)
|
|
write_attribute('#{meth}', value)
|
|
end
|
|
EOS
|
|
|
|
if property.alias
|
|
class_eval <<-EOS
|
|
alias #{property.alias.to_sym}= #{meth.to_sym}=
|
|
EOS
|
|
end
|
|
end
|
|
end # module ClassMethods
|
|
end
|
|
end
|
|
end |