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
|
module:set_global();
local server = require "net.http.server";
local codes = require "net.http.codes";
local xml_escape = require "util.stanza".xml_escape;
local render = require "util.interpolation".new("%b{}", xml_escape);
local show_private = module:get_option_boolean("http_errors_detailed", false);
local always_serve = module:get_option_boolean("http_errors_always_show", true);
local default_message = { module:get_option_string("http_errors_default_message", "That's all I know.") };
local default_messages = {
[400] = { "What kind of request do you call that??" };
[403] = { "You're not allowed to do that." };
[404] = { "Whatever you were looking for is not here. %";
"Where did you put it?", "It's behind you.", "Keep looking." };
[500] = { "% Check your error log for more info.";
"Gremlins.", "It broke.", "Don't look at me." };
};
local messages = setmetatable(module:get_option("http_errors_messages", {}), { __index = default_messages });
local html = [[
<!DOCTYPE html>
<html>
<head>
<meta charset="utf-8">
<title>{title}</title>
<style>
body{margin-top:14%;text-align:center;background-color:#f8f8f8;font-family:sans-serif}
h1{font-size:xx-large}
p{font-size:x-large}
p.warning{font-size:large;background-color:yellow}
p.extra{font-size:large;font-family:courier}
@media(prefers-color-scheme:dark){
body{background-color:#161616;color:#eee}
p.warning{background-color:inherit;color:yellow}
}
</style>
</head>
<body>
<h1>{title}</h1>
<p>{message}</p>
{warning&<p class="warning">⚠ {warning?} ⚠</p>}
{extra&<p class="extra">{extra?}</p>}
</body>
</html>
]];
local function get_page(code, extra)
local message = messages[code];
if always_serve or message then
message = message or default_message;
return render(html, {
title = rawget(codes, code) or ("Code "..tostring(code));
message = message[1]:gsub("%%", function ()
return message[math.random(2, math.max(#message,2))];
end);
extra = extra;
});
end
end
module:hook_object_event(server, "http-error", function (event)
if event.response then
event.response.headers.content_type = "text/html; charset=utf-8";
end
return get_page(event.code, (show_private and event.private_message) or event.message or (event.error and event.error.text));
end);
module:hook("http-message", function (event)
if event.response then
event.response.headers.content_type = "text/html; charset=utf-8";
end
return render(html, event);
end);
module:hook_object_event(server, "http-error", function (event)
local request, response = event.request, event.response;
if request and response and request.path == "/" and response.status_code == 404 then
response.headers.content_type = "text/html; charset=utf-8";
return render(html, {
title = "Prosody is running!";
message = "Welcome to the XMPP world!";
});
end
end, 1);
|