/
cookie_jar.rb
37 lines (33 loc) · 1.02 KB
/
cookie_jar.rb
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
require "faraday"
require "http/cookie_jar"
module Faraday
class CookieJar < Faraday::Middleware
def initialize(app, options = {})
super(app)
@jar = options[:jar] || HTTP::CookieJar.new
end
def call(env)
cookies = @jar.cookies(env[:url])
unless cookies.empty?
cookie_value = HTTP::Cookie.cookie_value(cookies)
if env[:request_headers]["Cookie"]
unless env[:request_headers]["Cookie"] == cookie_value
env[:request_headers]["Cookie"] = cookie_value + ';' + env[:request_headers]["Cookie"]
end
else
env[:request_headers]["Cookie"] = cookie_value
end
end
@app.call(env).on_complete do |res|
if res[:response_headers]
if set_cookie = res[:response_headers]["Set-Cookie"]
@jar.parse(set_cookie, env[:url])
end
end
end
end
end
end
if Faraday::Middleware.respond_to? :register_middleware
Faraday::Middleware.register_middleware :cookie_jar => Faraday::CookieJar
end