Class: Module

Inherits:
Object
  • Object
show all
Defined in:
lib/spirit_hands/mattr_accessor_with_default.rb

Instance Method Summary collapse

Instance Method Details

#mattr_accessor_with_default(property, default = nil, &block) ⇒ Object



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
# File 'lib/spirit_hands/mattr_accessor_with_default.rb', line 2

def mattr_accessor_with_default(property, default = nil, &block)
  if block_given?
    if default
      raise ArgumentError, 'Must specify only one of +default+ or &block'
    end
    default = block
  end

  getter = property.to_sym
  setter = "#{property}=".to_sym
  class_var = "@@#{property}".to_sym

  # self.property=
  define_singleton_method(setter) do |value|
    if value.nil?
      remove_class_variable(class_var)
    else
      class_variable_set(class_var, value)
    end
  end

  # self.property
  if default.respond_to?(:call) # default is callable
    define_singleton_method(getter) do
      if class_variable_defined?(class_var)
        class_variable_get(class_var)
      else
        default.()
      end
    end
  else # default is literal object
    default_value_name = "DEFAULT_#{property}".upcase.to_sym
    const_set(default_value_name, default)
    define_singleton_method(getter) do
      if class_variable_defined?(class_var)
        class_variable_get(class_var)
      else
        const_get(default_value_name)
      end
    end
  end
end