Class: CommandDeck::Injector

Inherits:
Object
  • Object
show all
Defined in:
lib/command_deck/injector.rb

Overview

Tiny middleware that injects a tiny floating UI into HTML responses.

Instance Method Summary collapse

Constructor Details

#initialize(body, headers) ⇒ Injector

Returns a new instance of Injector.



8
9
10
11
12
13
 
# File 'lib/command_deck/injector.rb', line 8

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
 
# File 'lib/command_deck/injector.rb', line 15

def inject(content)
  @headers[Rack::CONTENT_LENGTH] = (@body.bytesize + content.bytesize).to_s if @headers[Rack::CONTENT_LENGTH]

  [
    if (position = @body.rindex("</body>"))
      [@body.insert(position, content)]
    else
      [@body << content]
    end,
    @headers
  ]
end