Class: DevToolbar::Middleware

Inherits:
Object
  • Object
show all
Defined in:
lib/dev_toolbar/middleware.rb

Instance Method Summary collapse

Constructor Details

#initialize(app) ⇒ Middleware

Returns a new instance of Middleware.



3
4
5
 
# File 'lib/dev_toolbar/middleware.rb', line 3

def initialize(app)
  @app = app
end

Instance Method Details

#call(env) ⇒ Object 



7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
 
# File 'lib/dev_toolbar/middleware.rb', line 7

def call(env)
  status, headers, response = @app.call(env)

  if Rails.env.development? && headers["Content-Type"]&.include?("text/html")
    response_body = response.body
    toolbar_html = <<-HTML
      <meta name="dev_toolbar_config" content='#{toolbar_links_content}'>
    HTML

    response_body.sub!('</head>', "#{toolbar_html}</head>")
    headers["Content-Length"] = response_body.bytesize.to_s

    response = [response_body]
  end

  [status, headers, response]
end