Module: HashParam::ClassMethods

Defined in:
lib/hash_param.rb

Instance Method Summary collapse

Instance Method Details

#hash_param(*method_names, transform_key: nil, from: nil) ⇒ Object 



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
 
# File 'lib/hash_param.rb', line 9

def hash_param(*method_names, transform_key: nil, from: nil)
  method_names.each do |method_name|
    visibility = instance_method_visibility(method_name)
    hidden_name = "#{method_name}_before_hash_param"

    alias_method hidden_name, method_name
    private hidden_name

    if from.nil?
      # We're passed the argument, e.g., caller expects f(data)
      define_method(method_name) do |arg|
        hash_param_dispatch(hidden_name, arg, transform_key: transform_key)
      end
    elsif from.is_a?(Symbol) && from.to_s[0] == '@'
      # Pull the data from an instance variable
      define_method(method_name) do
        hash_param_dispatch(hidden_name, instance_variable_get(from),
                            transform_key: transform_key)
      end
    elsif from.is_a?(Symbol)
      # Pull the Hash from a method, e.g., Rails' `params`
      define_method(method_name) do
        hash_param_dispatch(hidden_name, send(from),
                            transform_key: transform_key)
      end
    end

    # Set the new wrapping method's visibility to what the original was.
    send(visibility, method_name)
  end
end