You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
129 lines
3.2 KiB
C++
129 lines
3.2 KiB
C++
#ifndef WEBCC_HTTP_MESSAGE_H_
|
|
#define WEBCC_HTTP_MESSAGE_H_
|
|
|
|
#include <cassert>
|
|
#include <string>
|
|
#include <utility> // for move()
|
|
#include <vector>
|
|
|
|
#include "boost/asio/buffer.hpp" // for const_buffer
|
|
|
|
#include "webcc/globals.h"
|
|
|
|
namespace webcc {
|
|
|
|
// -----------------------------------------------------------------------------
|
|
|
|
typedef std::pair<std::string, std::string> HttpHeader;
|
|
|
|
class HttpMessage;
|
|
|
|
std::ostream& operator<<(std::ostream& os, const HttpMessage& message);
|
|
|
|
// -----------------------------------------------------------------------------
|
|
|
|
class HttpHeaderDict {
|
|
public:
|
|
std::size_t size() const {
|
|
return headers_.size();
|
|
}
|
|
|
|
void Add(const std::string& key, const std::string& value);
|
|
|
|
void Add(std::string&& key, std::string&& value);
|
|
|
|
bool Has(const std::string& key) const;
|
|
|
|
const HttpHeader& Get(std::size_t i) const {
|
|
assert(i < size());
|
|
return headers_[i];
|
|
}
|
|
|
|
const std::vector<HttpHeader>& data() const {
|
|
return headers_;
|
|
}
|
|
|
|
private:
|
|
std::vector<HttpHeader> headers_;
|
|
};
|
|
|
|
// -----------------------------------------------------------------------------
|
|
|
|
// Base class for HTTP request and response messages.
|
|
class HttpMessage {
|
|
public:
|
|
HttpMessage() : content_length_(kInvalidLength) {
|
|
}
|
|
|
|
virtual ~HttpMessage() = default;
|
|
|
|
const std::string& start_line() const {
|
|
return start_line_;
|
|
}
|
|
|
|
void set_start_line(const std::string& start_line) {
|
|
start_line_ = start_line;
|
|
}
|
|
|
|
std::size_t content_length() const {
|
|
return content_length_;
|
|
}
|
|
|
|
const std::string& content() const {
|
|
return content_;
|
|
}
|
|
|
|
void SetHeader(const std::string& key, const std::string& value) {
|
|
headers_.Add(key, value);
|
|
}
|
|
|
|
void SetHeader(std::string&& key, std::string&& value) {
|
|
headers_.Add(std::move(key), std::move(value));
|
|
}
|
|
|
|
// E.g., "text/html", "application/json; charset=utf-8", etc.
|
|
void SetContentType(const std::string& media_type,
|
|
const std::string& charset);
|
|
|
|
// TODO: Remove parameter |set_length|.
|
|
void SetContent(std::string&& content, bool set_length);
|
|
|
|
void SetContentInAppJsonUtf8(std::string&& content, bool set_length);
|
|
|
|
// Make the message (e.g., update start line).
|
|
// Must be called before ToBuffers()!
|
|
virtual bool Prepare() = 0;
|
|
|
|
// Convert the message into a vector of buffers. The buffers do not own the
|
|
// underlying memory blocks, therefore the message object must remain valid
|
|
// and not be changed until the write operation has completed.
|
|
std::vector<boost::asio::const_buffer> ToBuffers() const;
|
|
|
|
// Dump to output stream.
|
|
void Dump(std::ostream& os, std::size_t indent = 0,
|
|
const std::string& prefix = "") const;
|
|
|
|
// Dump to string, only used by logger.
|
|
std::string Dump(std::size_t indent = 0,
|
|
const std::string& prefix = "") const;
|
|
|
|
protected:
|
|
void SetContentLength(std::size_t content_length) {
|
|
content_length_ = content_length;
|
|
SetHeader(http::headers::kContentLength, std::to_string(content_length));
|
|
}
|
|
|
|
// Start line with trailing CRLF.
|
|
std::string start_line_;
|
|
|
|
std::size_t content_length_;
|
|
|
|
HttpHeaderDict headers_;
|
|
|
|
std::string content_;
|
|
};
|
|
|
|
} // namespace webcc
|
|
|
|
#endif // WEBCC_HTTP_MESSAGE_H_
|