-
Notifications
You must be signed in to change notification settings - Fork 1.2k
Expand file tree
/
Copy pathServerOffline.cpp
More file actions
103 lines (84 loc) · 3.02 KB
/
Copy pathServerOffline.cpp
File metadata and controls
103 lines (84 loc) · 3.02 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
/*
* ServerOffline.cpp
*
* Copyright (C) 2021 by RStudio, PBC
*
* Unless you have received this program directly from RStudio pursuant
* to the terms of a commercial license agreement with RStudio, then
* this program is licensed to you under the terms of version 3 of the
* GNU Affero General Public License. This program is distributed WITHOUT
* ANY EXPRESS OR IMPLIED WARRANTY, INCLUDING THOSE OF NON-INFRINGEMENT,
* MERCHANTABILITY OR FITNESS FOR A PARTICULAR PURPOSE. Please refer to the
* AGPL (http://www.gnu.org/licenses/agpl-3.0.txt) for more details.
*
*/
#include "ServerOffline.hpp"
#include <shared_core/Error.hpp>
#include <core/gwt/GwtFileHandler.hpp>
#include <core/http/Request.hpp>
#include <core/http/Response.hpp>
#include <core/json/JsonRpc.hpp>
#include <core/Log.hpp>
#include <core/text/TemplateFilter.hpp>
#include <server/ServerOptions.hpp>
#include <server/ServerUriHandlers.hpp>
using namespace rstudio::core;
namespace rstudio {
namespace server {
namespace offline {
namespace {
void handleOfflineRequest(const http::Request& request,
http::Response* pResponse)
{
// send error code for json responses
if (request.acceptsContentType(json::kJsonContentType))
{
json::setJsonRpcError(Error(json::errc::ServerOffline, ERROR_LOCATION), pResponse);
}
// send error page for html responses
else if (request.acceptsContentType("text/html"))
{
std::ostringstream os;
std::map<std::string, std::string> vars;
vars["request_uri"] = string_utils::jsLiteralEscape(request.uri());
vars["base_uri"] = string_utils::jsLiteralEscape(request.baseUri(core::http::BaseUriUse::External));
FilePath offlineTemplate = FilePath(options().wwwLocalPath()).completeChildPath("offline.htm");
core::Error err = core::text::renderTemplate(offlineTemplate, vars, os);
if (err)
{
// if we cannot display the page log the error
// note: this should never happen in a proper deployment
LOG_ERROR(err);
}
else
{
std::string body = os.str();
pResponse->setContentType("text/html");
pResponse->setBodyUnencoded(body);
}
// set 503 status even if there was an error showing the page
pResponse->setNoCacheHeaders();
pResponse->setStatusCode(core::http::status::ServiceUnavailable);
}
// other content types just get a plain 503 with no content
else
{
pResponse->setStatusCode(http::status::ServiceUnavailable);
}
}
}
Error httpServerAddHandlers()
{
// alias options
Options& options = server::options();
uri_handlers::addBlocking("/images",
gwt::fileHandlerFunction(options.wwwLocalPath(),
"/"));
// default handler sends back offline page or json error as appropriate
uri_handlers::setBlockingDefault(handleOfflineRequest);
// success
return Success();
}
} // namespace offline
} // namespace server
} // namespace rstudio