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 boost::beast::http::verb method() const 59 { 60 return req.method(); 61 } 62 63 std::string_view getHeaderValue(std::string_view key) const 64 { 65 return req[key]; 66 } 67 68 std::string_view getHeaderValue(boost::beast::http::field key) const 69 { 70 return req[key]; 71 } 72 73 std::string_view methodString() const 74 { 75 return req.method_string(); 76 } 77 78 std::string_view target() const 79 { 80 return req.target(); 81 } 82 83 boost::urls::url_view url() const 84 { 85 return {urlBase}; 86 } 87 88 const boost::beast::http::fields& fields() const 89 { 90 return req.base(); 91 } 92 93 const std::string& body() const 94 { 95 return req.body(); 96 } 97 98 bool target(std::string_view target) 99 { 100 req.target(target); 101 return setUrlInfo(); 102 } 103 104 unsigned version() const 105 { 106 return req.version(); 107 } 108 109 bool isUpgrade() const 110 { 111 return boost::beast::websocket::is_upgrade(req); 112 } 113 114 bool keepAlive() const 115 { 116 return req.keep_alive(); 117 } 118 119 private: 120 bool setUrlInfo() 121 { 122 auto result = boost::urls::parse_relative_ref(target()); 123 124 if (!result) 125 { 126 return false; 127 } 128 urlBase = *result; 129 return true; 130 } 131 }; 132 133 } // namespace crow 134