#include #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; std::string css_url = get_origin(req, config) + "/style.css"; res.set_header("Content-Security-Policy", "default-src 'none'; style-src "s + css_url + "; img-src https://s.pximg.net " + config.image_proxy_url.get_origin()); Element html("html", { Element("head", { Element("meta", {{"charset", "utf-8"}}, {}), Element("title", {std::move(title)}), Element("link", {{"rel", "stylesheet"}, {"href", std::move(css_url)}}, {}), Element("meta", {{"name", "viewport"}, {"content", "width=device-width,initial-scale=1"}}, {}) }), std::move(element) }); res.set_content(""s + html.serialize(), "text/html"); } void serve_error(const httplib::Request& req, httplib::Response& res, const Config& config, std::string title, std::optional subtitle, std::optional info) { Element error_div("div", {{"class", "error"}}, { Element("h2", {title}) }); if (subtitle) { error_div.nodes.push_back(Element("p", { std::move(*subtitle) })); } 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)); } 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)); } std::string get_origin(const httplib::Request& req, const Config& config) { 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; } std::string proxy_image_url(const Config& config, blankie::murl::Url url) { if (url.hostname == "s.pximg.net") { return url.to_string(); } blankie::murl::Url new_url = config.image_proxy_url; if (!url.path.empty() && url.path[0] != '/') { new_url.path += '/'; } new_url.path += blankie::murl::normalize_path(std::move(url.path)); if (!new_url.query.empty() && !url.query.empty()) { new_url.query += '&'; new_url.query += std::move(url.query); } new_url.fragment = std::move(url.fragment); return new_url.to_string(); }