112 lines
3.6 KiB
C++
112 lines
3.6 KiB
C++
// Copyright 2015 The Android Open Source Project
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
#include <libwebserv/response_impl.h>
|
|
|
|
#include <algorithm>
|
|
|
|
#include <base/json/json_writer.h>
|
|
#include <base/logging.h>
|
|
#include <base/values.h>
|
|
#include <brillo/http/http_request.h>
|
|
#include <brillo/mime_utils.h>
|
|
#include <brillo/streams/memory_stream.h>
|
|
#include <brillo/strings/string_utils.h>
|
|
#include <libwebserv/dbus_protocol_handler.h>
|
|
|
|
namespace libwebserv {
|
|
|
|
ResponseImpl::ResponseImpl(DBusProtocolHandler* handler,
|
|
const std::string& request_id)
|
|
: handler_{handler}, request_id_{request_id} {
|
|
}
|
|
|
|
ResponseImpl::~ResponseImpl() {
|
|
if (!reply_sent_) {
|
|
ReplyWithError(brillo::http::status_code::InternalServerError,
|
|
"Internal server error");
|
|
}
|
|
}
|
|
|
|
void ResponseImpl::AddHeader(const std::string& header_name,
|
|
const std::string& value) {
|
|
headers_.emplace(header_name, value);
|
|
}
|
|
|
|
void ResponseImpl::AddHeaders(
|
|
const std::vector<std::pair<std::string, std::string>>& headers) {
|
|
headers_.insert(headers.begin(), headers.end());
|
|
}
|
|
|
|
void ResponseImpl::Reply(int status_code,
|
|
brillo::StreamPtr data_stream,
|
|
const std::string& mime_type) {
|
|
CHECK(data_stream);
|
|
status_code_ = status_code;
|
|
data_stream_ = std::move(data_stream);
|
|
AddHeader(brillo::http::response_header::kContentType, mime_type);
|
|
SendResponse();
|
|
}
|
|
|
|
void ResponseImpl::ReplyWithText(int status_code,
|
|
const std::string& text,
|
|
const std::string& mime_type) {
|
|
Reply(status_code, brillo::MemoryStream::OpenCopyOf(text, nullptr),
|
|
mime_type);
|
|
}
|
|
|
|
void ResponseImpl::ReplyWithJson(int status_code, const base::Value* json) {
|
|
std::string text;
|
|
base::JSONWriter::WriteWithOptions(
|
|
*json, base::JSONWriter::OPTIONS_PRETTY_PRINT, &text);
|
|
std::string mime_type = brillo::mime::AppendParameter(
|
|
brillo::mime::application::kJson,
|
|
brillo::mime::parameters::kCharset,
|
|
"utf-8");
|
|
ReplyWithText(status_code, text, mime_type);
|
|
}
|
|
|
|
void ResponseImpl::ReplyWithJson(
|
|
int status_code, const std::map<std::string, std::string>& json) {
|
|
base::DictionaryValue json_value;
|
|
for (const auto& pair : json) {
|
|
json_value.SetString(pair.first, pair.second);
|
|
}
|
|
ReplyWithJson(status_code, &json_value);
|
|
}
|
|
|
|
void ResponseImpl::Redirect(int status_code, const std::string& redirect_url) {
|
|
AddHeader(brillo::http::response_header::kLocation, redirect_url);
|
|
ReplyWithError(status_code, "");
|
|
}
|
|
|
|
void ResponseImpl::ReplyWithError(int status_code,
|
|
const std::string& error_text) {
|
|
status_code_ = status_code;
|
|
data_stream_ = brillo::MemoryStream::OpenCopyOf(error_text, nullptr);
|
|
SendResponse();
|
|
}
|
|
|
|
void ResponseImpl::ReplyWithErrorNotFound() {
|
|
ReplyWithError(brillo::http::status_code::NotFound, "Not Found");
|
|
}
|
|
|
|
void ResponseImpl::SendResponse() {
|
|
CHECK(!reply_sent_) << "Response already sent";
|
|
reply_sent_ = true;
|
|
handler_->CompleteRequest(request_id_, status_code_, headers_,
|
|
std::move(data_stream_));
|
|
}
|
|
|
|
} // namespace libwebserv
|