forked from sferik/twitter-ruby
/
base.rb
123 lines (108 loc) · 3.01 KB
/
base.rb
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
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
require 'twitter/core_ext/kernel'
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
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
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