2023-04-04 17:01:24 +00:00
|
|
|
#include <regex>
|
|
|
|
|
|
2023-04-03 09:32:26 +00:00
|
|
|
#include "config.h"
|
|
|
|
|
#include "servehelper.h"
|
|
|
|
|
|
|
|
|
|
void serve(const httplib::Request& req, httplib::Response& res, const Config& config, std::string title, Element element) {
|
|
|
|
|
using namespace std::string_literals;
|
|
|
|
|
|
2023-04-06 12:24:09 +00:00
|
|
|
std::string css_url = get_origin(req, config) + "/style.css";
|
2023-04-04 17:01:24 +00:00
|
|
|
res.set_header("Content-Security-Policy", "default-src 'none'; style-src "s + css_url
|
2023-04-09 08:17:06 +00:00
|
|
|
+ "; img-src https://s.pximg.net " + config.image_proxy_url.get_origin());
|
2023-04-03 09:32:26 +00:00
|
|
|
|
|
|
|
|
Element html("html", {
|
|
|
|
|
Element("head", {
|
2023-04-03 14:40:34 +00:00
|
|
|
Element("meta", {{"charset", "utf-8"}}, {}),
|
2023-04-03 09:32:26 +00:00
|
|
|
Element("title", {std::move(title)}),
|
2023-04-06 15:24:45 +00:00
|
|
|
Element("link", {{"rel", "stylesheet"}, {"href", std::move(css_url)}}, {}),
|
|
|
|
|
Element("meta", {{"name", "viewport"}, {"content", "width=device-width,initial-scale=1"}}, {})
|
2023-04-03 09:32:26 +00:00
|
|
|
}),
|
|
|
|
|
std::move(element)
|
|
|
|
|
});
|
|
|
|
|
res.set_content("<!DOCTYPE html>"s + html.serialize(), "text/html");
|
|
|
|
|
}
|
|
|
|
|
|
2023-04-03 15:27:07 +00:00
|
|
|
void serve_error(const httplib::Request& req, httplib::Response& res, const Config& config,
|
|
|
|
|
std::string title, std::optional<std::string> subtitle, std::optional<std::string> info) {
|
|
|
|
|
|
|
|
|
|
Element error_div("div", {{"class", "error"}}, {
|
|
|
|
|
Element("h2", {title})
|
|
|
|
|
});
|
|
|
|
|
if (subtitle) {
|
2023-04-04 08:33:48 +00:00
|
|
|
error_div.nodes.push_back(Element("p", {
|
|
|
|
|
std::move(*subtitle)
|
|
|
|
|
}));
|
2023-04-03 15:27:07 +00:00
|
|
|
}
|
|
|
|
|
if (info) {
|
|
|
|
|
error_div.nodes.push_back(Element("pre", {
|
|
|
|
|
Element("code", {std::move(*info)})
|
|
|
|
|
}));
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
Element body("body", {std::move(error_div)});
|
|
|
|
|
serve(req, res, config, std::move(title), std::move(body));
|
|
|
|
|
}
|
|
|
|
|
|
2023-04-04 08:33:48 +00:00
|
|
|
void serve_redirect(const httplib::Request& req, httplib::Response& res, const Config& config, std::string url) {
|
|
|
|
|
using namespace std::string_literals;
|
|
|
|
|
|
|
|
|
|
Element body("body", {
|
|
|
|
|
"Redirecting to ",
|
|
|
|
|
Element("a", {{"href", url}}, {url}),
|
|
|
|
|
"…"
|
|
|
|
|
});
|
|
|
|
|
res.set_redirect(url);
|
|
|
|
|
serve(req, res, config, "Redirecting to "s + std::move(url) + "…", std::move(body));
|
|
|
|
|
}
|
|
|
|
|
|
2023-04-03 14:40:34 +00:00
|
|
|
std::string get_origin(const httplib::Request& req, const Config& config) {
|
2023-04-03 09:32:26 +00:00
|
|
|
if (req.has_header("X-Canonical-Origin")) {
|
|
|
|
|
return req.get_header_value("X-Canonical-Origin");
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
std::string origin = "http://";
|
|
|
|
|
if (req.has_header("Host")) {
|
|
|
|
|
origin += req.get_header_value("Host");
|
|
|
|
|
} else {
|
|
|
|
|
origin += config.bind_host;
|
|
|
|
|
if (config.bind_port != 80) {
|
|
|
|
|
origin += ':' + std::to_string(config.bind_port);
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
return origin;
|
|
|
|
|
}
|
2023-04-04 17:01:24 +00:00
|
|
|
|
2023-04-09 17:34:13 +00:00
|
|
|
std::string proxy_url(blankie::murl::Url base, blankie::murl::Url url) {
|
2023-04-06 12:24:09 +00:00
|
|
|
if (!url.path.empty() && url.path[0] != '/') {
|
2023-04-09 17:34:13 +00:00
|
|
|
base.path += '/';
|
|
|
|
|
}
|
|
|
|
|
base.path += blankie::murl::normalize_path(std::move(url.path));
|
|
|
|
|
if (!base.query.empty() && !url.query.empty()) {
|
|
|
|
|
base.query += '&';
|
2023-04-05 16:36:20 +00:00
|
|
|
}
|
2023-04-09 17:34:13 +00:00
|
|
|
base.query += std::move(url.query);
|
|
|
|
|
base.fragment = std::move(url.fragment);
|
|
|
|
|
return base.to_string();
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
std::string proxy_image_url(const Config& config, blankie::murl::Url url) {
|
|
|
|
|
if (url.is_host_equal("s.pximg.net")) {
|
|
|
|
|
return url.to_string();
|
2023-04-04 17:01:24 +00:00
|
|
|
}
|
2023-04-09 17:34:13 +00:00
|
|
|
return proxy_url(config.image_proxy_url, std::move(url));
|
2023-04-04 17:01:24 +00:00
|
|
|
}
|
