karchnu
/
mechanize.cr
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

select cookies depending on domain

master
Kanezoh 2021-05-31 14:46:09 +09:00
parent 813ba7248b
commit 88da2f28cc
1 changed files with 17 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
src/mechanize/http/agent.cr
View File

@ -4,17 +4,18 @@ require "http/client"
module MechanizeCr module MechanizeCr
module HTTP module HTTP
class Agent class Agent
property :request_headers, :context property :request_headers, :context, :cookies
property history : Array(MechanizeCr::Page) property history : Array(MechanizeCr::Page)
def initialize(@context : Mechanize | Nil = nil, history = Array(MechanizeCr::Page).new) def initialize(@context : Mechanize | Nil = nil, history = Array(MechanizeCr::Page).new)
@history = history @history = history
@request_headers = ::HTTP::Headers.new @request_headers = ::HTTP::Headers.new
@context = context @context = context
@cookies = Hash(String, ::HTTP::Cookies).new
end end
def fetch(uri, method = :get, headers = HTTP::Headers.new, params = Hash(String,String).new) def fetch(uri, method = :get, headers = HTTP::Headers.new, params = Hash(String,String).new)
uri = resolve(uri) uri = resolve(uri)
set_request_headers(headers) set_request_headers(uri, headers)
uri, params = resolve_parameters(uri, method, params) uri, params = resolve_parameters(uri, method, params)
response = http_request(uri, method, params) response = http_request(uri, method, params)
body = response.not_nil!.body body = response.not_nil!.body
@ -36,10 +37,16 @@ module MechanizeCr
end end
end end
private def set_request_headers(headers) private def set_request_headers(uri, headers)
headers.each do |k,v| headers.each do |k,v|
request_headers[k] = v request_headers[k] = v
end end
if cookies.fetch(uri.host.to_s, nil).nil?
return
else
valid_cookies = cookies[uri.host.to_s]
valid_cookies.not_nil!.add_request_headers(request_headers)
end
end end
private def resolve_parameters(uri, method, params) private def resolve_parameters(uri, method, params)
@ -66,10 +73,15 @@ module MechanizeCr
# end # end
#end #end
header_cookies = response.try &.cookies header_cookies = response.try &.cookies
if header_cookies.try &.empty? if (header_cookies.nil?)
request_headers request_headers
else else
header_cookies.not_nil!.add_request_headers(request_headers) if cookies.fetch(uri.host.to_s, ::HTTP::Cookies.new).empty?
cookies[uri.host.to_s] = ::HTTP::Cookies.new
end
header_cookies.each do |cookie|
cookies[uri.host.to_s] << cookie
end
end end
end end