ruby_decorators 0.0.2 → 0.0.3

data/README.md

@@ -26,6 +26,12 @@ Or install it yourself as:
 ## Usage
 
 ```ruby
+class Hi < RubyDecorator
+  def call(this, *args, &blk)
+    this.call(*args, &blk).sub('hello', 'hi')
+  end
+end
+
 class Batman < RubyDecorator
   def call(this, *args, &blk)
     this.call(*args, &blk).sub('world', 'batman')
@@ -58,6 +64,7 @@ class World
     @greeting
   end
 
+  +Hi
   +Catwoman
   def hello_catwoman
     @greeting
@@ -73,7 +80,7 @@ world = World.new
 
 world.hello_world          # => "hello world"
 world.hello_batman         # => "hello batman"
-world.hello_catwoman       # => "hello catwoman"
+world.hello_catwoman       # => "hi catwoman"
 world.hello_super_catwoman # => "hello super catwoman"
 ```
 

data/lib/ruby_decorator.rb

@@ -1,9 +1,9 @@
 class RubyDecorator
   def self.+@
-    RubyDecorators::Stack.decorators << self
+    RubyDecorators::Stack.all << self
   end
 
   def +@
-    RubyDecorators::Stack.decorators << self
+    RubyDecorators::Stack.all << self
   end
 end

data/lib/ruby_decorators.rb

@@ -1,43 +1,41 @@
 require "ruby_decorators/version"
 require "ruby_decorator"
-require "ruby_decorators/stack"
 
 module RubyDecorators
-  def method_added(method_name)
-    @__decorated_methods ||= []
+  class Stack
+    def self.all
+      @all ||= []
+    end
+  end
 
-    return if RubyDecorators::Stack.decorators.empty?  ||
-              method_name.to_s =~ /__undecorated_/     ||
-              @__decorated_methods.include?(method_name)
+  def method_added(method_name)
+    super
 
-    current_decorator = RubyDecorators::Stack.decorators.pop
-    method_visibility = detect_method_visibility(method_name)
+    @methods    ||= {}
+    @decorators ||= {}
 
-    class_eval <<-RUBY_EVAL, __FILE__, __LINE__ + 1
-      alias_method :__undecorated_#{method_name}, :#{method_name}
+    return if RubyDecorators::Stack.all.empty?
 
-      @__decorators ||= {}
-      @__decorators["#{method_name}"] = current_decorator
+    @methods[method_name]    = instance_method(method_name)
+    @decorators[method_name] = RubyDecorators::Stack.all.pop(42)
 
-      #{method_visibility}
+    class_eval <<-RUBY_EVAL, __FILE__, __LINE__ + 1
+      #{method_visibility_for(method_name)}
       def #{method_name}(*args, &blk)
-        decorator = #{current_decorator}.new
-        decorator ||= self.class.instance_variable_get(:@__decorators)["#{method_name}"]
-
-        if args.any?
-          decorator.call(method(:__undecorated_#{method_name}), *args, &blk)
-        else
-          decorator.call(method(:__undecorated_#{method_name}), &blk)
-        end
+        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
-
-    @__decorated_methods << method_name
   end
 
   private
 
-  def detect_method_visibility(method_name)
+  def method_visibility_for(method_name)
     if private_method_defined?(method_name)
       :private
     elsif protected_method_defined?(method_name)

data/lib/ruby_decorators/version.rb

@@ -1,3 +1,3 @@
 module RubyDecorators
-  VERSION = "0.0.2"
+  VERSION = "0.0.3"
 end

data/spec/ruby_decorators_spec.rb

@@ -7,6 +7,12 @@ describe RubyDecorators do
     end
   end
 
+  class Hi < RubyDecorator
+    def call(this, *args, &blk)
+      this.call(*args, &blk).sub('hello', 'hi')
+    end
+  end
+
   class Batman < RubyDecorator
     def call(this, *args, &blk)
       this.call(*args, &blk).sub('world', 'batman')
@@ -23,6 +29,15 @@ describe RubyDecorators do
     end
   end
 
+  class DummyClass2
+    extend RubyDecorators
+
+    +Hi
+    def hi
+      'hello'
+    end
+  end
+
   class DummyClass
     extend RubyDecorators
 
@@ -68,6 +83,18 @@ describe RubyDecorators do
       @greeting
     end
 
+    +Hi
+    +Batman
+    def hi_batman
+      @greeting
+    end
+
+    +Hi
+    +Catwoman.new('super', 'catwoman')
+    def hi_super_catwoman(arg1)
+      "#{@greeting}#{arg1}"
+    end
+
     protected
 
     +Batman
@@ -86,6 +113,7 @@ describe RubyDecorators do
   subject { DummyClass.new }
 
   it "#hello_world" do
+    DummyClass2.new.hi.must_equal 'hi'
     subject.hello_world.must_equal 'hello world'
   end
 
@@ -126,4 +154,14 @@ describe RubyDecorators do
       subject.hello_super_catwoman.must_equal 'hello super catwoman'
     end
   end
+
+  describe "multiple decorators" do
+    it "decorates a simple method" do
+      subject.hi_batman.must_equal 'hi batman'
+    end
+
+    it "decorates a method with args" do
+      subject.hi_super_catwoman('!').must_equal 'hi super catwoman!'
+    end
+  end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: ruby_decorators
 version: !ruby/object:Gem::Version
-  version: 0.0.2
+  version: 0.0.3
   prerelease: 
 platform: ruby
 authors:
@@ -72,7 +72,6 @@ files:
 - Rakefile
 - lib/ruby_decorator.rb
 - lib/ruby_decorators.rb
-- lib/ruby_decorators/stack.rb
 - lib/ruby_decorators/version.rb
 - ruby_decorators.gemspec
 - spec/ruby_decorators_spec.rb
@@ -91,7 +90,7 @@ required_ruby_version: !ruby/object:Gem::Requirement
       version: '0'
       segments:
       - 0
-      hash: -142632259474802857
+      hash: -1045434834331850826
 required_rubygems_version: !ruby/object:Gem::Requirement
   none: false
   requirements:
@@ -100,7 +99,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
       version: '0'
       segments:
       - 0
-      hash: -142632259474802857
+      hash: -1045434834331850826
 requirements: []
 rubyforge_project: 
 rubygems_version: 1.8.24

data/lib/ruby_decorators/stack.rb

@@ -1,7 +0,0 @@
-module RubyDecorators
-  class Stack
-    def self.decorators
-      @decorators ||= []
-    end
-  end
-end