-
Notifications
You must be signed in to change notification settings - Fork 10
/
ruby_decorators.rb
47 lines (38 loc) · 1.16 KB
/
ruby_decorators.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
require "ruby_decorators/version"
require "ruby_decorator"
module RubyDecorators
class Stack
def self.all
@all ||= []
end
end
def method_added(method_name)
super
@methods ||= {}
@decorators ||= {}
return if RubyDecorators::Stack.all.empty?
@methods[method_name] = instance_method(method_name)
@decorators[method_name] = RubyDecorators::Stack.all.pop(42)
class_eval <<-RUBY_EVAL, __FILE__, __LINE__ + 1
#{method_visibility_for(method_name)}
def #{method_name}(*args, &blk)
decorators = self.class.instance_variable_get(:@decorators)[:#{method_name}]
method = self.class.instance_variable_get(:@methods)[:#{method_name}]
decorators.inject(method.bind(self)) do |method, decorator|
decorator = decorator.new if decorator.respond_to?(:new)
lambda { |*a, &b| decorator.call(method, *a, &b) }
end.call(*args, &blk)
end
RUBY_EVAL
end
private
def method_visibility_for(method_name)
if private_method_defined?(method_name)
:private
elsif protected_method_defined?(method_name)
:protected
else
:public
end
end
end