From aa2bb5701af9f2b83020be61d00c7d651e41c8e8 Mon Sep 17 00:00:00 2001
From: Kanezoh <asfaltmanta114@gmail.com>
Date: Sat, 1 May 2021 23:22:37 +0900
Subject: [PATCH] add request headers

---
 main.cr           |  3 ++-
 src/http/agent.cr | 16 +++++++++++++---
 src/mechanize.cr  |  8 ++++++++
 3 files changed, 23 insertions(+), 4 deletions(-)

diff --git a/main.cr b/main.cr
index aabcfb2..d47039f 100644
--- a/main.cr
+++ b/main.cr
@@ -1,5 +1,6 @@
 require "./src/mechanize.cr"
 
 agent = Mechanize.new
-page = agent.get "http://example.com/"
+agent.request_headers = HTTP::Headers{"Foo" => "Bar"}
+page = agent.get("https://example.com")
 puts page
diff --git a/src/http/agent.cr b/src/http/agent.cr
index 87df571..68c921d 100644
--- a/src/http/agent.cr
+++ b/src/http/agent.cr
@@ -4,25 +4,35 @@ require "http/client"
 module MechanizeCr
   module HTTP
     class Agent
+      property :request_headers
+
       def initialize()
+        @request_headers = ::HTTP::Headers.new
       end
 
       def fetch(uri, method = :get, headers = HTTP::Headers.new)
-        response = http_request uri, method, headers
+        add_request_headers(headers)
+        response = http_request uri, method
         puts response.not_nil!.body
       end
 
-      def http_request(uri, method, headers)
+      def http_request(uri, method)
         uri = URI.parse(uri)
         request = ::HTTP::Client.new(uri.host.not_nil!)
         case uri.scheme.not_nil!.downcase
         when "http", "https" then
           case method
           when :get
-            request.get(uri.path, headers: headers)
+            request.get(uri.path, headers: request_headers)
           end
         end
       end
+
+      private def add_request_headers(headers)
+        headers.each do |k,v|
+          request_headers[k] = v
+        end
+      end
     end
   end
 end
diff --git a/src/mechanize.cr b/src/mechanize.cr
index 5550706..fe9ec03 100644
--- a/src/mechanize.cr
+++ b/src/mechanize.cr
@@ -14,4 +14,12 @@ class Mechanize
     #yield page if block_given?
     page
   end
+
+  def request_headers
+    @agent.request_headers
+  end
+
+  def request_headers=(request_headers)
+    @agent.request_headers = request_headers
+  end
 end