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 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91
|
require 'hashie/dash'
module Hashie
# A Trash is a 'translated' Dash where the keys can be remapped from a source
# hash.
#
# Trashes are useful when you need to read data from another application,
# such as a Java api, where the keys are named differently from how we would
# in Ruby.
class Trash < Dash
# Defines a property on the Trash. Options are as follows:
#
# * <tt>:default</tt> - Specify a default value for this property, to be
# returned before a value is set on the property in a new Dash.
# * <tt>:from</tt> - Specify the original key name that will be write only.
# * <tt>:with</tt> - Specify a lambda to be used to convert value.
# * <tt>:transform_with</tt> - Specify a lambda to be used to convert value
# without using the :from option. It transform the property itself.
def self.property(property_name, options = {})
super
if options[:from]
if property_name.to_sym == options[:from].to_sym
raise ArgumentError, "Property name (#{property_name}) and :from option must not be the same"
end
translations << options[:from].to_sym
if options[:with].respond_to? :call
class_eval do
define_method "#{options[:from]}=" do |val|
self[property_name.to_sym] = options[:with].call(val)
end
end
else
class_eval <<-RUBY
def #{options[:from]}=(val)
self[:#{property_name}] = val
end
RUBY
end
elsif options[:transform_with].respond_to? :call
transforms[property_name.to_sym] = options[:transform_with]
end
end
# Set a value on the Dash in a Hash-like way. Only works
# on pre-existing properties.
def []=(property, value)
if self.class.translations.include? property.to_sym
send("#{property}=", value)
elsif self.class.transforms.key? property.to_sym
super property, self.class.transforms[property.to_sym].call(value)
elsif property_exists? property
super
end
end
private
def self.translations
@translations ||= []
end
def self.transforms
@transforms ||= {}
end
# Raises an NoMethodError if the property doesn't exist
#
def property_exists?(property)
unless self.class.property?(property.to_sym)
raise NoMethodError, "The property '#{property}' is not defined for this Trash."
end
true
end
private
# Deletes any keys that have a translation
def initialize_attributes(attributes)
return unless attributes
attributes_copy = attributes.dup.delete_if do |k,v|
if self.class.translations.include?(k.to_sym)
self[k] = v
true
end
end
super attributes_copy
end
end
end
|