Skip to content

HTTPS clone URL

Subversion checkout URL

You can clone with
or
.
Download ZIP
Fetching contributors…

Cannot retrieve contributors at this time

132 lines (116 sloc) 3.391 kB
require 'twitter/error/identity_map_key_error'
module Twitter
class Base
# Define methods that retrieve the value from an initialized instance variable Hash, using the attribute as a key
#
# @param attrs [Array, Set, Symbol]
def self.attr_reader(*attrs)
mod = Module.new do
attrs.each do |attribute|
define_method attribute do
@attrs[attribute.to_sym]
end
define_method "#{attribute}?" do
!!@attrs[attribute.to_sym]
end
end
end
const_set(:Attributes, mod)
include mod
end
# return [Twitter::IdentityMap]
def self.identity_map
return unless Twitter.identity_map
@identity_map = Twitter.identity_map.new unless defined?(@identity_map) && @identity_map.class == Twitter.identity_map
@identity_map
end
# Retrieves an object from the identity map.
#
# @param attrs [Hash]
# @return [Twitter::Base]
def self.fetch(attrs)
return unless identity_map
if object = identity_map.fetch(Marshal.dump(attrs))
return object
end
return yield if block_given?
raise Twitter::Error::IdentityMapKeyError, "key not found"
end
# Stores an object in the identity map.
#
# @param object [Object]
# @return [Twitter::Base]
def self.store(object)
return object unless identity_map
identity_map.store(Marshal.dump(object.attrs), object)
end
# Returns a new object based on the response hash
#
# @param response [Hash]
# @return [Twitter::Base]
def self.from_response(response={})
fetch_or_new(response[:body])
end
# Retrieves an object from the identity map, or stores it in the
# identity map if it doesn't already exist.
#
# @param attrs [Hash]
# @return [Twitter::Base]
def self.fetch_or_new(attrs={})
return unless attrs
return new(attrs) unless identity_map
fetch(attrs) do
object = new(attrs)
store(object)
end
end
# Initializes a new object
#
# @param attrs [Hash]
# @return [Twitter::Base]
def initialize(attrs={})
@attrs = attrs
end
# Fetches an attribute of an object using hash notation
#
# @param method [String, Symbol] Message to send to the object
def [](method)
send(method.to_sym)
rescue NoMethodError
nil
end
# Retrieve the attributes of an object
#
# @return [Hash]
def attrs
@attrs.inject({}) do |attrs, (key, value)|
if value.respond_to?(:attrs)
attrs.merge!(key => value.attrs)
else
attrs.merge!(key => respond_to?(key) && method(key).arity.zero? ? send(key) : value)
end
end
end
alias to_hash attrs
# Update the attributes of an object
#
# @param attrs [Hash]
# @return [Twitter::Base]
def update(attrs)
@attrs.update(attrs)
self
end
protected
# @param attr [Symbol]
# @param other [Twitter::Base]
# @return [Boolean]
def attr_equal(attr, other)
self.class == other.class && !other.send(attr).nil? && send(attr) == other.send(attr)
end
# @param other [Twitter::Base]
# @return [Boolean]
def attrs_equal(other)
self.class == other.class && !other.attrs.empty? && attrs == other.attrs
end
end
end
Jump to Line
Something went wrong with that request. Please try again.