Added Post Body Parsing

CMakeLists.txt

@@ -17,13 +17,14 @@ set(SOURCE_FILES
         Source/StringUtils.cpp
         Source/Cookie.cpp
         Source/Session.cpp
-        Source/Response.cpp Source/InbuildMiddleWare.cpp)
+        Source/Response.cpp
+        Source/InbuildMiddleWare.cpp)
 include_directories(${CMAKE_SOURCE_DIR}/)
 add_library(VWeb ${SOURCE_FILES})
 
-set(mode release)
+set(mode Release)
 if (CMAKE_BUILD_TYPE STREQUAL "Debug")
-    set(mode debug)
+    set(mode Debug)
 endif ()
 set(target_file ${CMAKE_SOURCE_DIR}/dist/libVWeb.${mode}.a)
 add_custom_command(TARGET VWeb POST_BUILD COMMAND ${CMAKE_COMMAND} -E copy $<TARGET_FILE:VWeb> ${target_file})

Source/Cookie.cpp

@@ -103,5 +103,6 @@ void Cookies::CookieToString(std::stringstream &stream, Cookie &cookie) {
     view << "; SameSite=" << sameSiteValue;
   }
   view << "\n";
+  stream << view.str();
 }
 } // namespace VWeb

Source/InbuildMiddleWare.cpp

@@ -68,7 +68,7 @@ PreMiddleWareReturn SessionManager::PreHandle(Request &request) {
   return {};
 }
 
-bool SessionManager::PostHandle(const Request &request, Response &response) {
+bool SessionManager::PostHandle(Request &request, Response &response) {
   if (response.SessionData->Id.empty() &&
       response.SessionData->ContainsData()) {
     response.SessionData->Update();
@@ -80,8 +80,9 @@ bool SessionManager::PostHandle(const Request &request, Response &response) {
     }
     m_Sessions[response.SessionData->Id] = response.SessionData;
     auto &sidCookie = response.CookieData->Get("sid");
+    sidCookie.Name = "sid";
     sidCookie.HttpOnly = true;
-    sidCookie.Secure = true;
+    sidCookie.SameSite = CookieSameSite::Strict;
     sidCookie.Value = response.SessionData->Id;
   }
   return true;
@@ -101,7 +102,7 @@ void SessionManager::GC() {
 #pragma endregion SESSION
 #pragma region COOKIES
 PreMiddleWareReturn CookieManager::PreHandle(Request &request) {
-  auto &cookieHeaders = request.Header("Cookie");
+  auto &cookieHeaders = request.Header("cookie");
   auto &cookies = request.CookieData;
   if (cookieHeaders.Size() > 0) {
     auto &values = cookieHeaders.Values();

Source/RequestHandler.cpp

@@ -27,6 +27,7 @@ bool ParseRequest(Ref<Request> &request) {
     if (index != std::string::npos) {
       auto key = line.substr(0, index);
       String::Trim(key);
+      String::ToLowerCase(key);
       request->Header(key).Add(String::TrimCopy(line.substr(index + 1)));
     } else if (line.find("HTTP")) {
       auto headers = String::Split(line, " ");
@@ -41,6 +42,49 @@ bool ParseRequest(Ref<Request> &request) {
   return true;
 }
 
+void ParseParameterString(Request &req, const std::string &toParse) {
+  auto data = String::Split(toParse, "&");
+  for (auto &item : data) {
+    auto d = String::Split(item, "=", 1);
+    req.Parameter(d[0]).Add(d.size() > 1 ? d[1] : "true");
+  }
+}
+
+std::string GetPostBody(const std::string& originalBody)
+{
+  auto body = String::Split(originalBody, "\r\n\r\n", 1);
+  if (body.size() > 1 && ! body[body.size() - 1].empty())
+    return String::TrimCopy(String::UrlDecode(body[body.size() - 1]));
+  return {};
+}
+
+void ParseParameters(Request &request, RequestHandler &requestHandler) {
+  auto &uri = request.URI;
+  size_t hasURLParameters = uri.find('?');
+  if (hasURLParameters != std::string::npos) {
+    ParseParameterString(request, uri.substr(hasURLParameters + 1));
+    request.URI = uri.substr (0, hasURLParameters);
+  }
+
+  if (request.Method == HttpMethod::HEAD || request.Method == HttpMethod::GET ||
+      request.Method == HttpMethod::OPTIONS)
+    return;
+
+  // POST :D
+  auto &contentType = request.Header("content-type");
+  if (contentType.Size() == 0)
+    return;
+  auto &key = contentType.GetFirst();
+  if (key == "application/x-www-form-urlencoded") {
+    ParseParameterString(request, GetPostBody(request.Body));
+  } else {
+    if (requestHandler.HasBodyParser(key)) {
+      auto &func = requestHandler.GetBodyParser(key);
+      func(request, GetPostBody(request.Body));
+    }
+  }
+}
+
 struct RequestJob : public WorkerJob {
   Ref<Request> MRequest{};
   Ref<Router> MRouter{};
@@ -53,6 +97,7 @@ struct RequestJob : public WorkerJob {
       SocketUtils::Close(MRequest->ID);
       return;
     }
+    ParseParameters(*MRequest, *MRequestHandler);
     MRequest->CookieData = CreateRef<Cookies>();
     MRequest->SessionData = CreateRef<Session>();
     Ref<Response> response;
@@ -72,7 +117,8 @@ struct RequestJob : public WorkerJob {
   }
 };
 RequestHandler::RequestHandler(const Ref<SocketManager> &manager)
-    : m_SocketManager(manager), m_MiddleWare(CreateRef<MiddleWareHandler>()) {
+    : m_SocketManager(manager),
+      m_MiddleWare(CreateRef<MiddleWareHandler>()) {
   m_MiddleWare = CreateRef<MiddleWareHandler>();
 }
 void RequestHandler::InitThreads(int count) {
@@ -89,12 +135,8 @@ void RequestHandler::AddRequest(Ref<Request> &request) {
     m_Pool.Dispatch(job);
   }
 }
-void RequestHandler::Stop() {
-  m_Pool.Stop();
-}
-void RequestHandler::SetRouter(Ref<Router> &router) {
-  m_Router = router;
-}
+void RequestHandler::Stop() { m_Pool.Stop(); }
+void RequestHandler::SetRouter(Ref<Router> &router) { m_Router = router; }
 void RequestHandler::AddSendResponse(SendData sendData) {
   auto id = sendData.SocketID;
   m_Server->m_OutRequest.Add(id, sendData);

Source/StringUtils.cpp

@@ -19,12 +19,14 @@ static void StringRightTrim(std::string &s) {
 }
 
 std::vector<std::string> String::Split(const std::string &s,
-                                       const std::string &delimiter) {
+                                       const std::string &delimiter,
+                                       int limit) {
   size_t pos_start = 0, pos_end, delim_len = delimiter.length();
   std::string token;
   std::vector<std::string> res;
 
-  while ((pos_end = s.find(delimiter, pos_start)) != std::string::npos) {
+  while ((pos_end = s.find(delimiter, pos_start)) != std::string::npos &&
+         (limit == -1 || res.size() < (size_t)limit)) {
     token = s.substr(pos_start, pos_end - pos_start);
     pos_start = pos_end + delim_len;
     res.push_back(token);

Source/StringUtils.h

@@ -7,7 +7,8 @@ namespace VWeb {
 class String {
 public:
   static std::vector<std::string> Split(const std::string& s,
-                                        const std::string& delimiter);
+                                        const std::string& delimiter,
+                                        int limit = -1);
   static std::string Join(const std::vector<std::string>& items, const std::string& delimiter);
   static int ToNumber(std::string& string, int def);
   static void ToLowerCase(std::string& value);

VWeb.h

@@ -3,6 +3,7 @@
 #include <cassert>
 #include <condition_variable>
 #include <fcntl.h>
+#include <functional>
 #include <map>
 #include <memory>
 #include <mutex>
@@ -276,6 +277,7 @@ public:
   void RemoveRoute(const std::string &path);
 
   Ref<MiddleWareHandler> &Middleware();
+
 protected:
   void Execute();
   void OutgoingExecute(epoll_event &event);
@@ -300,6 +302,7 @@ protected:
 using ms = std::chrono::duration<double, std::milli>;
 struct SessionData {
   virtual ~SessionData() = default;
+  template <class T> T *As() { return reinterpret_cast<T *>(this); }
 };
 struct Session {
   std::string Id;
@@ -309,7 +312,9 @@ struct Session {
   void Remove(const std::string &key);
   bool Has(const std::string &key);
   Ref<SessionData> &operator[](const std::string &key) { return m_Data[key]; }
-  void SetSessionData(const std::string& key, const Ref<SessionData>& data) { m_Data[key] = data; }
+  void SetSessionData(const std::string &key, const Ref<SessionData> &data) {
+    m_Data[key] = data;
+  }
   bool ContainsData() { return !m_Data.empty(); }
 
 protected:
@@ -391,15 +396,15 @@ enum class HttpMethod {
 };
 enum class CookieSameSite { Invalid = 0, Strict, Lax, None };
 struct Cookie {
-  std::string Name;
-  std::string Value;
-  std::string Expires;
+  std::string Name{};
+  std::string Value{};
+  std::string Expires{};
   int MaxAge{};
-  std::string Domain;
-  std::string Path;
+  std::string Domain{};
+  std::string Path{"/"};
   bool Secure{};
   bool HttpOnly{};
-  CookieSameSite SameSite{CookieSameSite::Invalid};
+  CookieSameSite SameSite{CookieSameSite::Strict};
   bool IsOld = false;
 };
 struct Cookies {
@@ -442,9 +447,13 @@ public:
   Ref<Cookies> CookieData;
   Cookie &GetCookie(const std::string &key) { return CookieData->Get(key); };
   ParameterValue &Parameter(const std::string &key) { return Parameters[key]; }
-  bool HasParameter(const std::string &key) const { return Parameters.contains(key); }
+  bool HasParameter(const std::string &key) const {
+    return Parameters.contains(key);
+  }
   bool HasHeader(const std::string &key) const { return Headers.contains(key); }
-  std::string &FirstOf(const std::string &key) { return Parameters[key].GetFirst(); }
+  std::string &FirstOf(const std::string &key) {
+    return Parameters[key].GetFirst();
+  }
   ParameterValue &Header(const std::string &key) { return Headers[key]; }
   std::unordered_map<std::string, ParameterValue> Parameters;
   std::unordered_map<std::string, ParameterValue> Headers;
@@ -483,6 +492,7 @@ protected:
   std::unordered_map<std::string, ParameterValue> m_Headers;
 };
 class MiddleWareHandler;
+typedef std::function<const void(Request &, const std::string& body)> ParseFunction;
 class RequestHandler {
 public:
   explicit RequestHandler(const Ref<SocketManager> &manager);
@@ -492,8 +502,15 @@ public:
   void SetRouter(Ref<Router> &router);
   void AddSendResponse(SendData);
   Ref<MiddleWareHandler> &Middleware() { return m_MiddleWare; }
+  bool HasBodyParser(const std::string &key) {
+    return m_ParseFunctions.contains(key);
+  }
+  ParseFunction &GetBodyParser(const std::string &key) {
+    return m_ParseFunctions.at(key);
+  }
 
 protected:
+  std::unordered_map<std::string, ParseFunction> m_ParseFunctions;
   Ref<Router> m_Router{nullptr};
   Ref<SocketManager> m_SocketManager{nullptr};
   Server *m_Server{nullptr};
@@ -552,8 +569,8 @@ typedef std::optional<Ref<Response>> PreMiddleWareReturn;
 struct MiddleWare {
   int Pos{0};
   virtual PreMiddleWareReturn PreHandle(Request &) { return {}; }
-  virtual bool PostHandle(const Request &, Response &){ return true; }
-  virtual void Shutdown(const Request &, const Response &){};
+  virtual bool PostHandle(Request &, Response &) { return true; }
+  virtual void Shutdown(Request &, const Response &){};
   bool operator<(const MiddleWare *rhs) const { return Pos < rhs->Pos; }
 };
 
@@ -564,8 +581,12 @@ public:
   void Shutdown(Ref<Request> &, Ref<Response> &);
 
 public:
-  template <class T> Ref<MiddleWare> GetRef() { return GetById(typeid(T).name()); }
-  template <class T> T& Get() { return static_cast<T&>(*GetById(typeid(T).name())); }
+  template <class T> Ref<MiddleWare> GetRef() {
+    return GetById(typeid(T).name());
+  }
+  template <class T> T &Get() {
+    return static_cast<T &>(*GetById(typeid(T).name()));
+  }
   template <class T> void Set(Ref<MiddleWare> &instance) {
     auto &type = typeid(T);
     if (type.before(typeid(MiddleWare)))
@@ -575,6 +596,7 @@ public:
     return static_cast<T &>(*CreateMiddleWare<T>());
   }
   template <class T> void Remove() { RemoveById(typeid(T).name()); }
+
 protected:
   template <class T> Ref<MiddleWare> CreateMiddleWare() {
     return SetById(typeid(T).name(), CreateRef<T>());
@@ -591,7 +613,10 @@ public:
   AuthWare() = default;
   ~AuthWare() = default;
   PreMiddleWareReturn PreHandle(Request &request) override;
-  void SetAuthMethod(AuthFunction function) { m_AuthFunction = std::move(function); }
+  void SetAuthMethod(AuthFunction function) {
+    m_AuthFunction = std::move(function);
+  }
+
 protected:
   AuthFunction m_AuthFunction{nullptr};
 };
@@ -601,9 +626,11 @@ public:
   SessionManager();
   ~SessionManager();
   PreMiddleWareReturn PreHandle(Request &request) override;
-  bool PostHandle(const Request& request, Response& response) override;
+  bool PostHandle(Request &request, Response &response) override;
+
 protected:
   void GC();
+
 protected:
   Ref<std::thread> m_GCThread;
   std::mutex m_Mutex;

example/main.cpp

@@ -1,5 +1,20 @@
 #include <VWeb.h>
 
+class MyCompleteController : public VWeb::Route {
+public:
+  bool Get(const VWeb::Request&, VWeb::Response& response) {
+    response << "MyCompleteController: GET";
+    return true;
+  }
+  bool Post(const VWeb::Request&, VWeb::Response& response) {
+    response << "MyCompleteController: POST";
+    return true;
+  }
+
+  bool IsAllowed(const VWeb::Request& request) {
+    return request.HasHeader("Auth");
+  }
+};
 bool Ping(const VWeb::Request&, VWeb::Response& response) {
   response << "Pong";
   return true;