Module: MethodStruct

Defined in:
lib/method_struct.rb,
lib/method_struct/version.rb

Constant Summary collapse

VERSION =
"0.1.5"

Class Method Summary collapse

Class Method Details

.new(*fields, &block) ⇒ Object



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

def self.new(*fields, &block)
  if fields.last.is_a?(Hash)
    method_name = fields.last[:method_name]
    fields = fields.take(fields.size - 1)
  else
    method_name = :call
  end

  Class.new do
    singleton_class = (class << self; self; end)

    singleton_class.instance_eval do
      define_method(method_name) do |*field_values, &blk|
        new(*field_values).send(method_name, &blk)
      end

      define_method(:[]) do |*args|
        send(method_name, *args)
      end
    end

    define_method(:[]) do |&blk|
      send(method_name, &blk)
    end

    define_method(:initialize) do |*values|
      if fields.size > 1 && values.size == 1 && values.first.is_a?(Hash)
        fields.each do |field|
          instance_variable_set("@#{field}", values.first[field])
        end
      else
        fields.zip(values).each do |field, value|
          instance_variable_set("@#{field}", value)
        end
      end
    end

    define_method(:==) do |other|
      self.class == other.class && fields.all? do |field|
        send(field) == other.send(field)
      end
    end
    alias_method :eql?, :==

    define_method(:hash) do
      fields.map { |field| send(field).hash }.inject(&:^)
    end

    class_eval(&block) if block_given?

    protected
    fields.each do |field|
      attr_reader(field)
    end
  end
end