diff --git a/spec/http_spec.cr b/spec/http_spec.cr index f1157e3..728fb7a 100644 --- a/spec/http_spec.cr +++ b/spec/http_spec.cr @@ -6,6 +6,12 @@ WebMock.stub(:post, "http://example.com/post") WebMock.stub(:get, "example.com/%E3%81%82%E3%81%82%E3%81%82") WebMock.stub(:get, "https://example.com/") WebMock.stub(:get, "https://example.com/post") +WebMock.stub(:put, "http://example.com/put") + .with(body: "hello") + .to_return(body: "success") +WebMock.stub(:delete, "http://example.com/delete") + .with(body: "hello") + .to_return(body: "success") describe "Mechanize HTTP test" do it "simple GET" do @@ -33,6 +39,28 @@ describe "Mechanize HTTP test" do page.uri.to_s.should eq uri end + it "simple POST" do + agent = Mechanize.new + query = {"email" => "foobar"} + page = agent.post("http://example.com/post", query: query) + page.body.should eq "success" + page.code.should eq 200 + end + + it "PUT" do + agent = Mechanize.new + page = agent.put("http://example.com/put", body: "hello") + page.body.should eq "success" + page.code.should eq 200 + end + + it "DELETE" do + agent = Mechanize.new + page = agent.delete("http://example.com/delete", body: "hello") + page.body.should eq "success" + page.code.should eq 200 + end + it "can escape non-ascii character" do agent = Mechanize.new page = agent.get("http://example.com/あああ") @@ -49,14 +77,6 @@ describe "Mechanize HTTP test" do agent.request_headers["Foo"].should eq "Bar" end - it "simple POST" do - agent = Mechanize.new - query = {"email" => "foobar"} - page = agent.post("http://example.com/post", query: query) - page.body.should eq "success" - page.code.should eq 200 - end - it "can set user agent" do agent = Mechanize.new mac_chrome_agent = "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_13_6) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/69.0.3497.100 Safari/537.36" diff --git a/src/mechanize.cr b/src/mechanize.cr index 18c54b1..76b7417 100644 --- a/src/mechanize.cr +++ b/src/mechanize.cr @@ -11,8 +11,6 @@ require "./mechanize/errors/*" # This class is main class of Mechanize.cr, # using this class' instance to start web interaction. # -# now only supports GET, POST. other HTTP methods will be implemented soon... -# # Examples: # # ``` @@ -91,6 +89,52 @@ class Mechanize post_form(uri, form, headers) end + # Send PUT request to specified uri with headers, and body. + # + # Examples (send put request whose body is "hello") + # + # ``` + # agent = Mechanize.new + # agent.put("http://example.com", + # body: "hello") + # ``` + def put(uri : String | URI, + body : String?, + headers = ::HTTP::Headers.new) : Mechanize::Page + method = :put + headers.merge!({ + "Content-Type" => "application/octet-stream", + "Content-Length" => body.size.to_s, + }) + + page = @agent.fetch(uri, method, headers: headers, body: body) + request_headers.delete("Content-Type") + request_headers.delete("Content-Length") + add_to_history(page) + # yield page if block_given? + page + end + + # Send DELETE request to specified uri with headers, and body. + # + # Examples (send delete request whose body is "hello") + # + # ``` + # agent = Mechanize.new + # agent.delete("http://example.com", + # body: "hello") + # ``` + def delete(uri : String | URI, + body : String?, + headers = ::HTTP::Headers.new) : Mechanize::Page + method = :delete + + page = @agent.fetch(uri, method, headers: headers, body: body) + add_to_history(page) + # yield page if block_given? + page + end + # get the value of request headers. # # ``` @@ -263,8 +307,8 @@ class Mechanize # fetch the page page = @agent.fetch(uri, :post, headers: headers, params: {"value" => request_data}, referer: cur_page) - headers.delete("Content-Type") - headers.delete("Content-Length") + request_headers.delete("Content-Type") + request_headers.delete("Content-Length") add_to_history(page) page end diff --git a/src/mechanize/http/agent.cr b/src/mechanize/http/agent.cr index c8905a3..33408d3 100644 --- a/src/mechanize/http/agent.cr +++ b/src/mechanize/http/agent.cr @@ -22,14 +22,18 @@ class Mechanize # send http request and return page. # This method is called from Mechanize#get, #post and other methods. # There's no need to call this method directly. - 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, + body : String? = nil, referer = (current_page unless history.empty?)) uri = resolve_url(uri, referer) set_request_headers(uri, headers) set_user_agent set_request_referer(referer) uri, params = resolve_parameters(uri, method, params) - response = http_request(uri, method, params) + response = http_request(uri, method, params, body) body = response.not_nil!.body page = response_parse(response, body, uri) response_log(response) @@ -58,7 +62,7 @@ class Mechanize end # send http request - private def http_request(uri, method, params) : ::HTTP::Client::Response? + private def http_request(uri, method, params, body) : ::HTTP::Client::Response? request_log(uri, method) case uri.scheme.not_nil!.downcase @@ -68,6 +72,10 @@ class Mechanize ::HTTP::Client.get(uri, headers: request_headers) when :post ::HTTP::Client.post(uri, headers: request_headers, form: params.not_nil!.fetch("value", "")) + when :put + ::HTTP::Client.put(uri, headers: request_headers, body: body) + when :delete + ::HTTP::Client.delete(uri, headers: request_headers, body: body) end end end