From 470fdacee41ae1f1e6185288eb772fafe73b587e Mon Sep 17 00:00:00 2001
From: Kanezoh <asfaltmanta114@gmail.com>
Date: Wed, 5 May 2021 15:53:39 +0900
Subject: [PATCH] save response cookies

---
 main.cr                     |  2 +-
 src/mechanize/http/agent.cr | 21 +++++++++++++++++++--
 2 files changed, 20 insertions(+), 3 deletions(-)

diff --git a/main.cr b/main.cr
index 619732d..4e53c72 100644
--- a/main.cr
+++ b/main.cr
@@ -7,4 +7,4 @@ page = agent.get("http://example.com/", params: params)
 #puts page.code
 #puts page.body
 #puts page.css("h1").first.inner_text
-puts page.title
+#puts page.title
diff --git a/src/mechanize/http/agent.cr b/src/mechanize/http/agent.cr
index c3b6754..92e57e6 100644
--- a/src/mechanize/http/agent.cr
+++ b/src/mechanize/http/agent.cr
@@ -13,11 +13,14 @@ module MechanizeCr
 
       def fetch(uri, method = :get, headers = HTTP::Headers.new, params = Hash(String,String).new)
         uri = URI.parse(uri)
-        add_request_headers(headers)
+        set_request_headers(headers)
         params = hash_to_params(params)
         response = http_request uri, method, params
         body = response.not_nil!.body
         page = response_parse(response, body, uri)
+        # save cookies
+        add_response_cookies(response, uri, page)
+        page
       end
 
       def http_request(uri, method, params)
@@ -32,7 +35,7 @@ module MechanizeCr
         end
       end
 
-      private def add_request_headers(headers)
+      private def set_request_headers(headers)
         headers.each do |k,v|
           request_headers[k] = v
         end
@@ -51,6 +54,20 @@ module MechanizeCr
       private def response_parse(response, body, uri)
         @context.not_nil!.parse uri, response, body
       end
+
+      private def add_response_cookies(response, uri, page)
+        #if page.body =~ /Set-Cookie/
+        #  page.css("//head/meta[@http-equiv=\"Set-Cookie\"]").each do |meta|
+        #    save_cookies(uri, meta["content"])
+        #  end
+        #end
+        header_cookies = response.try &.cookies
+        if header_cookies.try &.empty?
+          request_headers
+        else
+          header_cookies.not_nil!.add_request_headers(request_headers)
+        end
+      end
     end
   end
 end