diff options
author | Joshua Peek <josh@joshpeek.com> | 2009-01-03 23:02:29 -0600 |
---|---|---|
committer | Joshua Peek <josh@joshpeek.com> | 2009-01-03 23:02:29 -0600 |
commit | ed2e776bdec3f0764433a6dc4f592f9bebfea859 (patch) | |
tree | 97837b9eafad49ac2444dfbd081a77ce91e2df48 /actionpack/lib/action_controller | |
parent | f7ee082bb3cde977a199b81207e5d12e1d7101b3 (diff) | |
download | rails-ed2e776bdec3f0764433a6dc4f592f9bebfea859.tar.gz rails-ed2e776bdec3f0764433a6dc4f592f9bebfea859.tar.bz2 rails-ed2e776bdec3f0764433a6dc4f592f9bebfea859.zip |
Move metal above method piggybacking middleware and add some test coverage
Diffstat (limited to 'actionpack/lib/action_controller')
-rw-r--r-- | actionpack/lib/action_controller/middleware_stack.rb | 21 |
1 files changed, 21 insertions, 0 deletions
diff --git a/actionpack/lib/action_controller/middleware_stack.rb b/actionpack/lib/action_controller/middleware_stack.rb index 74f28565c0..2bccba2ba1 100644 --- a/actionpack/lib/action_controller/middleware_stack.rb +++ b/actionpack/lib/action_controller/middleware_stack.rb @@ -1,6 +1,14 @@ module ActionController class MiddlewareStack < Array class Middleware + def self.new(klass, *args, &block) + if klass.is_a?(self) + klass + else + super + end + end + attr_reader :args, :block def initialize(klass, *args, &block) @@ -65,6 +73,19 @@ module ActionController block.call(self) if block_given? end + def insert(index, *objs) + index = self.index(index) unless index.is_a?(Integer) + objs = objs.map { |obj| Middleware.new(obj) } + super(index, *objs) + end + + alias_method :insert_before, :insert + + def insert_after(index, *objs) + index = self.index(index) unless index.is_a?(Integer) + insert(index + 1, *objs) + end + def use(*args, &block) middleware = Middleware.new(*args, &block) push(middleware) |