-
Notifications
You must be signed in to change notification settings - Fork 418
/
abstract_struct.rb
159 lines (142 loc) · 4.17 KB
/
abstract_struct.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
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
module Concurrent
module Synchronization
# @!visibility private
# @!macro internal_implementation_note
module AbstractStruct
# @!visibility private
def initialize(*values)
super()
ns_initialize(*values)
end
# @!macro struct_length
#
# Returns the number of struct members.
#
# @return [Fixnum] the number of struct members
def length
self.class::MEMBERS.length
end
alias_method :size, :length
# @!macro struct_members
#
# Returns the struct members as an array of symbols.
#
# @return [Array] the struct members as an array of symbols
def members
self.class::MEMBERS.dup
end
protected
# @!macro struct_values
#
# @!visibility private
def ns_values
@values.dup
end
# @!macro struct_values_at
#
# @!visibility private
def ns_values_at(indexes)
@values.values_at(*indexes)
end
# @!macro struct_to_h
#
# @!visibility private
def ns_to_h
length.times.reduce({}){|memo, i| memo[self.class::MEMBERS[i]] = @values[i]; memo}
end
# @!macro struct_get
#
# @!visibility private
def ns_get(member)
if member.is_a? Integer
if member >= @values.length
raise IndexError.new("offset #{member} too large for struct(size:#{@values.length})")
end
@values[member]
else
send(member)
end
rescue NoMethodError
raise NameError.new("no member '#{member}' in struct")
end
# @!macro struct_equality
#
# @!visibility private
def ns_equality(other)
self.class == other.class && self.values == other.values
end
# @!macro struct_each
#
# @!visibility private
def ns_each
values.each{|value| yield value }
end
# @!macro struct_each_pair
#
# @!visibility private
def ns_each_pair
@values.length.times do |index|
yield self.class::MEMBERS[index], @values[index]
end
end
# @!macro struct_select
#
# @!visibility private
def ns_select
values.select{|value| yield value }
end
# @!macro struct_inspect
#
# @!visibility private
def ns_inspect
struct = pr_underscore(self.class.ancestors[1])
clazz = ((self.class.to_s =~ /^#<Class:/) == 0) ? '' : " #{self.class}"
"#<#{struct}#{clazz} #{ns_to_h}>"
end
# @!macro struct_merge
#
# @!visibility private
def ns_merge(other, &block)
self.class.new(*self.to_h.merge(other, &block).values)
end
# @!visibility private
def pr_underscore(clazz)
word = clazz.to_s.dup # dup string to workaround JRuby 9.2.0.0 bug https://github.com/jruby/jruby/issues/5229
word.gsub!(/::/, '/')
word.gsub!(/([A-Z]+)([A-Z][a-z])/,'\1_\2')
word.gsub!(/([a-z\d])([A-Z])/,'\1_\2')
word.tr!("-", "_")
word.downcase!
word
end
# @!visibility private
def self.define_struct_class(parent, base, name, members, &block)
clazz = Class.new(base || Object) do
include parent
self.const_set(:MEMBERS, members.collect{|member| member.to_s.to_sym}.freeze)
def ns_initialize(*values)
raise ArgumentError.new('struct size differs') if values.length > length
@values = values.fill(nil, values.length..length-1)
end
end
unless name.nil?
begin
parent.send :remove_const, name if parent.const_defined?(name, false)
parent.const_set(name, clazz)
clazz
rescue NameError
raise NameError.new("identifier #{name} needs to be constant")
end
end
members.each_with_index do |member, index|
clazz.send :remove_method, member if clazz.instance_methods.include? member
clazz.send(:define_method, member) do
@values[index]
end
end
clazz.class_exec(&block) unless block.nil?
clazz
end
end
end
end