1 // Copyright 2015 The Android Open Source Project
2 //
3 // Licensed under the Apache License, Version 2.0 (the "License");
4 // you may not use this file except in compliance with the License.
5 // You may obtain a copy of the License at
6 //
7 // http://www.apache.org/licenses/LICENSE-2.0
8 //
9 // Unless required by applicable law or agreed to in writing, software
10 // distributed under the License is distributed on an "AS IS" BASIS,
11 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12 // See the License for the specific language governing permissions and
13 // limitations under the License.
14
15 #include <libwebserv/response_impl.h>
16
17 #include <algorithm>
18
19 #include <base/json/json_writer.h>
20 #include <base/logging.h>
21 #include <base/values.h>
22 #include <brillo/http/http_request.h>
23 #include <brillo/mime_utils.h>
24 #include <brillo/streams/memory_stream.h>
25 #include <brillo/strings/string_utils.h>
26 #include <libwebserv/dbus_protocol_handler.h>
27
28 namespace libwebserv {
29
ResponseImpl(DBusProtocolHandler * handler,const std::string & request_id)30 ResponseImpl::ResponseImpl(DBusProtocolHandler* handler,
31 const std::string& request_id)
32 : handler_{handler}, request_id_{request_id} {
33 }
34
~ResponseImpl()35 ResponseImpl::~ResponseImpl() {
36 if (!reply_sent_) {
37 ReplyWithError(brillo::http::status_code::InternalServerError,
38 "Internal server error");
39 }
40 }
41
AddHeader(const std::string & header_name,const std::string & value)42 void ResponseImpl::AddHeader(const std::string& header_name,
43 const std::string& value) {
44 headers_.emplace(header_name, value);
45 }
46
AddHeaders(const std::vector<std::pair<std::string,std::string>> & headers)47 void ResponseImpl::AddHeaders(
48 const std::vector<std::pair<std::string, std::string>>& headers) {
49 headers_.insert(headers.begin(), headers.end());
50 }
51
Reply(int status_code,brillo::StreamPtr data_stream,const std::string & mime_type)52 void ResponseImpl::Reply(int status_code,
53 brillo::StreamPtr data_stream,
54 const std::string& mime_type) {
55 CHECK(data_stream);
56 status_code_ = status_code;
57 data_stream_ = std::move(data_stream);
58 AddHeader(brillo::http::response_header::kContentType, mime_type);
59 SendResponse();
60 }
61
ReplyWithText(int status_code,const std::string & text,const std::string & mime_type)62 void ResponseImpl::ReplyWithText(int status_code,
63 const std::string& text,
64 const std::string& mime_type) {
65 Reply(status_code, brillo::MemoryStream::OpenCopyOf(text, nullptr),
66 mime_type);
67 }
68
ReplyWithJson(int status_code,const base::Value * json)69 void ResponseImpl::ReplyWithJson(int status_code, const base::Value* json) {
70 std::string text;
71 base::JSONWriter::WriteWithOptions(
72 *json, base::JSONWriter::OPTIONS_PRETTY_PRINT, &text);
73 std::string mime_type = brillo::mime::AppendParameter(
74 brillo::mime::application::kJson,
75 brillo::mime::parameters::kCharset,
76 "utf-8");
77 ReplyWithText(status_code, text, mime_type);
78 }
79
ReplyWithJson(int status_code,const std::map<std::string,std::string> & json)80 void ResponseImpl::ReplyWithJson(
81 int status_code, const std::map<std::string, std::string>& json) {
82 base::DictionaryValue json_value;
83 for (const auto& pair : json) {
84 json_value.SetString(pair.first, pair.second);
85 }
86 ReplyWithJson(status_code, &json_value);
87 }
88
Redirect(int status_code,const std::string & redirect_url)89 void ResponseImpl::Redirect(int status_code, const std::string& redirect_url) {
90 AddHeader(brillo::http::response_header::kLocation, redirect_url);
91 ReplyWithError(status_code, "");
92 }
93
ReplyWithError(int status_code,const std::string & error_text)94 void ResponseImpl::ReplyWithError(int status_code,
95 const std::string& error_text) {
96 status_code_ = status_code;
97 data_stream_ = brillo::MemoryStream::OpenCopyOf(error_text, nullptr);
98 SendResponse();
99 }
100
ReplyWithErrorNotFound()101 void ResponseImpl::ReplyWithErrorNotFound() {
102 ReplyWithError(brillo::http::status_code::NotFound, "Not Found");
103 }
104
SendResponse()105 void ResponseImpl::SendResponse() {
106 CHECK(!reply_sent_) << "Response already sent";
107 reply_sent_ = true;
108 handler_->CompleteRequest(request_id_, status_code_, headers_,
109 std::move(data_stream_));
110 }
111
112 } // namespace libwebserv
113