diff options
author | Joshua Peek <josh@joshpeek.com> | 2009-01-13 16:09:51 -0600 |
---|---|---|
committer | Joshua Peek <josh@joshpeek.com> | 2009-01-13 16:09:51 -0600 |
commit | 1adc1496f9152c893e1f08abcb1e5e7272829899 (patch) | |
tree | 85affcfeea0fd298b474e90e7e9870fcd2e60f77 /actionpack/lib/action_controller/rewindable_input.rb | |
parent | 5a43908c7414996354ca427354d98d789e0210e7 (diff) | |
download | rails-1adc1496f9152c893e1f08abcb1e5e7272829899.tar.gz rails-1adc1496f9152c893e1f08abcb1e5e7272829899.tar.bz2 rails-1adc1496f9152c893e1f08abcb1e5e7272829899.zip |
Add RewindableInput wrapper to fix issues with middleware that impolitely eat up non-rewindable input
Diffstat (limited to 'actionpack/lib/action_controller/rewindable_input.rb')
-rw-r--r-- | actionpack/lib/action_controller/rewindable_input.rb | 40 |
1 files changed, 40 insertions, 0 deletions
diff --git a/actionpack/lib/action_controller/rewindable_input.rb b/actionpack/lib/action_controller/rewindable_input.rb new file mode 100644 index 0000000000..296d8aed22 --- /dev/null +++ b/actionpack/lib/action_controller/rewindable_input.rb @@ -0,0 +1,40 @@ +module ActionController + class RewindableInput + class RewindableIO < ActiveSupport::BasicObject + def initialize(io) + @io = io + end + + def read(*args) + read_original_io + @io.read(*args) + end + + def rewind + read_original_io + @io.rewind + end + + def method_missing(method, *args, &block) + @io.send(method, *args, &block) + end + + private + def read_original_io + unless @str + @str = @io.read + @io = StringIO.new(@str) + end + end + end + + def initialize(app) + @app = app + end + + def call(env) + env['rack.input'] = RewindableIO.new(env['rack.input']) + @app.call(env) + end + end +end |