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.
127 lines
3.5 KiB
C++
127 lines
3.5 KiB
C++
#include "webcc/request_builder.h"
|
|
|
|
#include "webcc/base64.h"
|
|
#include "webcc/logger.h"
|
|
#include "webcc/string.h"
|
|
#include "webcc/utility.h"
|
|
|
|
#ifdef WEBCC_ENABLE_GZIP
|
|
#include "webcc/gzip.h"
|
|
#endif
|
|
|
|
namespace bfs = boost::filesystem;
|
|
|
|
namespace webcc {
|
|
|
|
RequestPtr RequestBuilder::operator()() {
|
|
assert(headers_.size() % 2 == 0);
|
|
|
|
auto request = std::make_shared<Request>(method_);
|
|
|
|
request->set_url(std::move(url_));
|
|
|
|
for (std::size_t i = 1; i < headers_.size(); i += 2) {
|
|
request->SetHeader(std::move(headers_[i - 1]), std::move(headers_[i]));
|
|
}
|
|
|
|
// If no Keep-Alive, explicitly set `Connection` to "Close".
|
|
if (!keep_alive_) {
|
|
request->SetHeader(headers::kConnection, "Close");
|
|
}
|
|
|
|
if (body_) {
|
|
request->SetContentType(media_type_, charset_);
|
|
|
|
#ifdef WEBCC_ENABLE_GZIP
|
|
if (gzip_ && body_->Compress()) {
|
|
request->SetHeader(headers::kContentEncoding, "gzip");
|
|
}
|
|
#endif
|
|
} else if (!form_parts_.empty()) {
|
|
// Another choice to generate the boundary is like what Apache does.
|
|
// See: https://stackoverflow.com/a/5686863
|
|
auto boundary = random_string(30);
|
|
|
|
request->SetContentType("multipart/form-data; boundary=" + boundary);
|
|
|
|
body_ = std::make_shared<FormBody>(form_parts_, boundary);
|
|
}
|
|
|
|
if (body_) {
|
|
request->SetBody(body_, true);
|
|
}
|
|
|
|
return request;
|
|
}
|
|
|
|
#ifdef WEBCC_ENABLE_GZIP
|
|
|
|
// Accept Gzip compressed response data or not.
|
|
RequestBuilder& RequestBuilder::AcceptGzip(bool gzip) {
|
|
if (gzip) {
|
|
return Header(headers::kAcceptEncoding, "gzip, deflate");
|
|
} else {
|
|
return Header(headers::kAcceptEncoding, "identity");
|
|
}
|
|
}
|
|
|
|
#endif // WEBCC_ENABLE_GZIP
|
|
|
|
RequestBuilder& RequestBuilder::File(const bfs::path& path,
|
|
bool infer_media_type,
|
|
std::size_t chunk_size) {
|
|
body_.reset(new FileBody{ path, chunk_size });
|
|
|
|
if (infer_media_type) {
|
|
media_type_ = media_types::FromExtension(path.extension().string());
|
|
}
|
|
|
|
return *this;
|
|
}
|
|
|
|
RequestBuilder& RequestBuilder::FormFile(const std::string& name,
|
|
const bfs::path& path,
|
|
const std::string& media_type) {
|
|
assert(!name.empty());
|
|
return Form(FormPart::NewFile(name, path, media_type));
|
|
}
|
|
|
|
RequestBuilder& RequestBuilder::FormData(const std::string& name,
|
|
std::string&& data,
|
|
const std::string& media_type) {
|
|
assert(!name.empty());
|
|
return Form(FormPart::New(name, std::move(data), media_type));
|
|
}
|
|
|
|
RequestBuilder& RequestBuilder::Header(const std::string& key,
|
|
const std::string& value) {
|
|
headers_.push_back(key);
|
|
headers_.push_back(value);
|
|
return *this;
|
|
}
|
|
|
|
RequestBuilder& RequestBuilder::Auth(const std::string& type,
|
|
const std::string& credentials) {
|
|
headers_.push_back(headers::kAuthorization);
|
|
headers_.push_back(type + " " + credentials);
|
|
return *this;
|
|
}
|
|
|
|
RequestBuilder& RequestBuilder::AuthBasic(const std::string& login,
|
|
const std::string& password) {
|
|
auto credentials = Base64Encode(login + ":" + password);
|
|
return Auth("Basic", credentials);
|
|
}
|
|
|
|
RequestBuilder& RequestBuilder::AuthToken(const std::string& token) {
|
|
return Auth("Token", token);
|
|
}
|
|
|
|
RequestBuilder& RequestBuilder::Date() {
|
|
headers_.push_back(headers::kDate);
|
|
headers_.push_back(utility::HttpDate());
|
|
return *this;
|
|
}
|
|
|
|
} // namespace webcc
|