Class: Ungarbled::Middleware

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

Constant Summary collapse

DISPOSITION_KEY = 
'Content-Disposition'
DISPOSITION_REGEX = 
/^(?:attachment|inline); filename="(.+)"$/

Instance Method Summary collapse

Constructor Details

#initialize(app, options = {}) ⇒ Middleware

Returns a new instance of Middleware.



6
7
8
9
 
# File 'lib/ungarbled/middleware.rb', line 6

def initialize(app, options = {})
  @app = app
  @options = options
end

Instance Method Details

#call(env) ⇒ Object 



11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
 
# File 'lib/ungarbled/middleware.rb', line 11

def call(env)
  req = Rack::Request.new(env)
  browser = Browser.new(
    ua: req.user_agent,
    accept_language: req.env['HTTP_ACCEPT_LANGUAGE']
  )
  encoder = Ungarbled::Encoder.new(browser, @options)
  status, headers, res = @app.call(env)
  disposition = headers[DISPOSITION_KEY]

  if disposition
    headers[DISPOSITION_KEY] =
      encode_disposition(encoder, disposition)
  end

  [status, headers, res]
end

#encode_disposition(encoder, disposition) ⇒ Object 



29
30
31
32
33
 
# File 'lib/ungarbled/middleware.rb', line 29

def encode_disposition(encoder, disposition)
  replacing = DISPOSITION_REGEX.match(disposition)[1]
  replaced = encoder.encode(replacing)
  disposition.gsub(replacing, replaced)
end