From 6b4df41b309074bb93b7f34abd697e6719bd1347 Mon Sep 17 00:00:00 2001
From: yhirose <yuji.hirose.bug@gmail.com>
Date: Tue, 14 Jan 2020 17:02:25 -0500
Subject: [PATCH] Fix #330

---
 httplib.h    | 23 +++++++++++++++++++++++
 test/test.cc | 29 +++++++++++++++++++++++++++++
 2 files changed, 52 insertions(+)

diff --git a/httplib.h b/httplib.h
index 7577118..6a1c228 100644
--- a/httplib.h
+++ b/httplib.h
@@ -631,6 +631,11 @@ public:
                                 ContentProvider content_provider,
                                 const char *content_type);
 
+  std::shared_ptr<Response> Put(const char *path, const Params &params);
+
+  std::shared_ptr<Response> Put(const char *path, const Headers &headers,
+                                const Params &params);
+
   std::shared_ptr<Response> Patch(const char *path, const std::string &body,
                                   const char *content_type);
 
@@ -4087,6 +4092,24 @@ Client::Put(const char *path, const Headers &headers, size_t content_length,
                                     content_type);
 }
 
+inline std::shared_ptr<Response> Client::Put(const char *path,
+                                             const Params &params) {
+  return Put(path, Headers(), params);
+}
+
+inline std::shared_ptr<Response>
+Client::Put(const char *path, const Headers &headers, const Params &params) {
+  std::string query;
+  for (auto it = params.begin(); it != params.end(); ++it) {
+    if (it != params.begin()) { query += "&"; }
+    query += it->first;
+    query += "=";
+    query += detail::encode_url(it->second);
+  }
+
+  return Put(path, headers, query, "application/x-www-form-urlencoded");
+}
+
 inline std::shared_ptr<Response> Client::Patch(const char *path,
                                                const std::string &body,
                                                const char *content_type) {
diff --git a/test/test.cc b/test/test.cc
index bca6f6a..471b6a5 100644
--- a/test/test.cc
+++ b/test/test.cc
@@ -694,6 +694,15 @@ protected:
                   res.status = 400;
                 }
               })
+        .Put("/person",
+             [&](const Request &req, Response &res) {
+               if (req.has_param("name") && req.has_param("note")) {
+                 persons_[req.get_param_value("name")] =
+                     req.get_param_value("note");
+               } else {
+                 res.status = 400;
+               }
+             })
         .Get("/person/(.*)",
              [&](const Request &req, Response &res) {
                string name = req.matches[1];
@@ -1089,6 +1098,26 @@ TEST_F(ServerTest, PostMethod2) {
   ASSERT_EQ("coder", res->body);
 }
 
+TEST_F(ServerTest, PutMethod3) {
+  auto res = cli_.Get("/person/john3");
+  ASSERT_TRUE(res != nullptr);
+  ASSERT_EQ(404, res->status);
+
+  Params params;
+  params.emplace("name", "john3");
+  params.emplace("note", "coder");
+
+  res = cli_.Put("/person", params);
+  ASSERT_TRUE(res != nullptr);
+  ASSERT_EQ(200, res->status);
+
+  res = cli_.Get("/person/john3");
+  ASSERT_TRUE(res != nullptr);
+  ASSERT_EQ(200, res->status);
+  ASSERT_EQ("text/plain", res->get_header_value("Content-Type"));
+  ASSERT_EQ("coder", res->body);
+}
+
 TEST_F(ServerTest, PostWwwFormUrlEncodedJson) {
   Params params;
   params.emplace("json", JSON_DATA);