Class: Rubee::CookieSamesiteMiddleware

Inherits:
Object
  • Object
show all
Defined in:
lib/rubee/internal_middlewares/cookie_samesite_middleware.rb

Instance Method Summary collapse

Constructor Details

#initialize(app) ⇒ CookieSamesiteMiddleware

Returns a new instance of CookieSamesiteMiddleware.



3
4
5
 
# File 'lib/rubee/internal_middlewares/cookie_samesite_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
 
# File 'lib/rubee/internal_middlewares/cookie_samesite_middleware.rb', line 7

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

  cookie_header = headers["Set-Cookie"] || headers["set-cookie"]
  header_key = headers["Set-Cookie"] ? "Set-Cookie" : "set-cookie"

  if cookie_header
    headers[header_key] = patch_cookies(cookie_header)
  end

  [status, headers, body]
end