-
Notifications
You must be signed in to change notification settings - Fork 346
/
annotatable.rb
68 lines (61 loc) · 1.64 KB
/
annotatable.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
module Annotatable
def self.included(base)
base.extend ClassMethods
end
module ClassMethods
def self.extended(base)
base.instance_eval do
alias :inherited_without_annotatable :inherited
alias :inherited :inherited_with_annotatable
end
end
def annotate(*attrs)
options = {}
options = attrs.pop if attrs.last.kind_of?(Hash)
options.symbolize_keys!
inherit = options[:inherit]
if inherit
@inherited_annotations ||= []
@inherited_annotations += attrs
end
attrs.each do |attr|
class_eval %{
def self.#{attr}(string = nil)
@#{attr} = string || @#{attr}
end
def #{attr}(string = nil)
self.class.#{attr}(string)
end
def self.#{attr}=(string = nil)
#{attr}(string)
end
def #{attr}=(string = nil)
self.class.#{attr}=(string)
end
}
end
attrs
end
def inherited_with_annotatable(subclass)
inherited_without_annotatable(subclass)
(["inherited_annotations"] + (@inherited_annotations || [])).each do |t|
ivar = "@#{t}"
subclass.instance_variable_set(ivar, instance_variable_get(ivar))
end
end
end
end
# We don't necessarily have ActiveSupport loaded yet!
class Hash
# Return a new hash with all keys converted to symbols.
def symbolize_keys
inject({}) do |options, (key, value)|
options[key.to_sym || key] = value
options
end
end
# Destructively convert all keys to symbols.
def symbolize_keys!
self.replace(self.symbolize_keys)
end
end