Class: WebConsole::Injector
- Inherits:
-
Object
- Object
- WebConsole::Injector
- Defined in:
- lib/web_console/injector.rb
Overview
Injects content into a Rack body.
Instance Method Summary collapse
-
#initialize(body, headers) ⇒ Injector
constructor
A new instance of Injector.
- #inject(content) ⇒ Object
Constructor Details
#initialize(body, headers) ⇒ Injector
Returns a new instance of Injector.
6 7 8 9 10 11 12 13 |
# File 'lib/web_console/injector.rb', line 6 def initialize(body, headers) @body = "".dup body.each { |part| @body << part } body.close if body.respond_to?(:close) @headers = headers end |
Instance Method Details
#inject(content) ⇒ Object
15 16 17 18 19 20 21 22 23 24 25 26 27 28 |
# File 'lib/web_console/injector.rb', line 15 def inject(content) # Remove any previously set Content-Length header because we modify # the body. Otherwise the response will be truncated. @headers.delete("Content-Length") [ if position = @body.rindex("</body>") [ @body.insert(position, content) ] else [ @body << content ] end, @headers ] end |