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
|
# File 'lib/croatia/utils/enum.rb', line 9
def enum(name, values, **options)
name = name.to_s.to_sym unless name.is_a?(Symbol)
values = values.zip(values) if values.is_a?(Array)
unless values.respond_to?(:to_h)
raise ArgumentError, "Enum values must be defined as a Hash or as an Array"
end
values = values.to_h.freeze
values_method_name = "#{name}_values".to_sym
define_singleton_method(values_method_name) do
values
end
define_method("#{name}=") do |value|
values = self.class.public_send(values_method_name)
enum_value = values[value]
enum_value = value if enum_value.nil? && values.has_value?(value)
if enum_value.nil? && !options[:allow_nil]
raise ArgumentError, "Invalid value for enum #{name}: #{value.inspect}"
end
instance_variable_set("@#{name}", enum_value)
end
define_method(name) do
instance_variable_get("@#{name}")
end
prefix = options[:prefix] ? "#{options[:prefix]}_" : ""
suffix = options[:suffix] ? "_#{options[:suffix]}" : ""
values.each do |key, value|
value_method_name = "#{prefix}#{key}#{suffix}"
define_method("#{value_method_name}?") do
instance_variable_get("@#{name}") == value
end
define_method("#{value_method_name}!") do
public_send("#{name}=", value)
end
end
end
|