1 #pragma once 2 3 #include "common.hpp" 4 #include "sessions.hpp" 5 6 #include <boost/asio/io_context.hpp> 7 #include <boost/asio/ip/address.hpp> 8 #include <boost/beast/http/message.hpp> 9 #include <boost/beast/http/string_body.hpp> 10 #include <boost/beast/websocket.hpp> 11 #include <boost/url/url.hpp> 12 13 #include <string> 14 #include <string_view> 15 #include <system_error> 16 17 namespace crow 18 { 19 20 struct Request 21 { 22 boost::beast::http::request<boost::beast::http::string_body> req; 23 24 private: 25 boost::urls::url urlBase{}; 26 27 public: 28 bool isSecure{false}; 29 30 boost::asio::io_context* ioService{}; 31 boost::asio::ip::address ipAddress{}; 32 33 std::shared_ptr<persistent_data::UserSession> session; 34 35 std::string userRole{}; 36 Request(boost::beast::http::request<boost::beast::http::string_body> reqIn, 37 std::error_code& ec) : 38 req(std::move(reqIn)) 39 { 40 if (!setUrlInfo()) 41 { 42 ec = std::make_error_code(std::errc::invalid_argument); 43 } 44 } 45 46 Request(std::string_view bodyIn, std::error_code& /*ec*/) : req({}, bodyIn) 47 {} 48 49 Request() = default; 50 51 Request(const Request& other) = default; 52 Request(Request&& other) = default; 53 54 Request& operator=(const Request&) = default; 55 Request& operator=(Request&&) = default; 56 ~Request() = default; 57 58 void addHeader(std::string_view key, std::string_view value) 59 { 60 req.insert(key, value); 61 } 62 63 void addHeader(boost::beast::http::field key, std::string_view value) 64 { 65 req.insert(key, value); 66 } 67 68 boost::beast::http::verb method() const 69 { 70 return req.method(); 71 } 72 73 std::string_view getHeaderValue(std::string_view key) const 74 { 75 return req[key]; 76 } 77 78 std::string_view getHeaderValue(boost::beast::http::field key) const 79 { 80 return req[key]; 81 } 82 83 std::string_view methodString() const 84 { 85 return req.method_string(); 86 } 87 88 std::string_view target() const 89 { 90 return req.target(); 91 } 92 93 boost::urls::url_view url() const 94 { 95 return {urlBase}; 96 } 97 98 const boost::beast::http::fields& fields() const 99 { 100 return req.base(); 101 } 102 103 const std::string& body() const 104 { 105 return req.body(); 106 } 107 108 bool target(std::string_view target) 109 { 110 req.target(target); 111 return setUrlInfo(); 112 } 113 114 unsigned version() const 115 { 116 return req.version(); 117 } 118 119 bool isUpgrade() const 120 { 121 return boost::beast::websocket::is_upgrade(req); 122 } 123 124 bool keepAlive() const 125 { 126 return req.keep_alive(); 127 } 128 129 private: 130 bool setUrlInfo() 131 { 132 auto result = boost::urls::parse_relative_ref(target()); 133 134 if (!result) 135 { 136 return false; 137 } 138 urlBase = *result; 139 return true; 140 } 141 }; 142 143 } // namespace crow 144