Software /
code /
prosody
Comparison
plugins/mod_http_errors.lua @ 8366:272ff3ab25f3
Merge 0.10->trunk
author | Kim Alvefur <zash@zash.se> |
---|---|
date | Sun, 29 Oct 2017 02:05:19 +0200 |
parent | 8364:f91ab40a3105 |
child | 9760:88640b3ea6b8 |
comparison
equal
deleted
inserted
replaced
8352:6ff50541d2a6 | 8366:272ff3ab25f3 |
---|---|
1 module:set_global(); | 1 module:set_global(); |
2 | 2 |
3 local server = require "net.http.server"; | 3 local server = require "net.http.server"; |
4 local codes = require "net.http.codes"; | 4 local codes = require "net.http.codes"; |
5 local xml_escape = require "util.stanza".xml_escape; | |
6 local render = require "util.interpolation".new("%b{}", xml_escape); | |
5 | 7 |
6 local show_private = module:get_option_boolean("http_errors_detailed", false); | 8 local show_private = module:get_option_boolean("http_errors_detailed", false); |
7 local always_serve = module:get_option_boolean("http_errors_always_show", true); | 9 local always_serve = module:get_option_boolean("http_errors_always_show", true); |
8 local default_message = { module:get_option_string("http_errors_default_message", "That's all I know.") }; | 10 local default_message = { module:get_option_string("http_errors_default_message", "That's all I know.") }; |
9 local default_messages = { | 11 local default_messages = { |
19 | 21 |
20 local html = [[ | 22 local html = [[ |
21 <!DOCTYPE html> | 23 <!DOCTYPE html> |
22 <html> | 24 <html> |
23 <head> | 25 <head> |
24 <meta charset="utf-8"> | 26 <meta charset="utf-8"> |
25 <style> | 27 <title>{title}</title> |
26 body{ | 28 <style> |
27 margin-top:14%; | 29 body{ |
28 text-align:center; | 30 margin-top:14%; |
29 background-color:#F8F8F8; | 31 text-align:center; |
30 font-family:sans-serif; | 32 background-color:#F8F8F8; |
31 } | 33 font-family:sans-serif; |
32 h1{ | 34 } |
33 font-size:xx-large; | 35 h1{ |
34 } | 36 font-size:xx-large; |
35 p{ | 37 } |
36 font-size:x-large; | 38 p{ |
37 } | 39 font-size:x-large; |
38 p+p { font-size: large; font-family: courier } | 40 } |
39 </style> | 41 p+p { |
42 font-size:large; | |
43 font-family:courier; | |
44 } | |
45 </style> | |
40 </head> | 46 </head> |
41 <body> | 47 <body> |
42 <h1>$title</h1> | 48 <h1>{title}</h1> |
43 <p>$message</p> | 49 <p>{message}</p> |
44 <p>$extra</p> | 50 <p>{extra?}</p> |
45 </body> | 51 </body> |
46 </html> | 52 </html> |
47 ]]; | 53 ]]; |
48 html = html:gsub("%s%s+", ""); | |
49 | |
50 local entities = { | |
51 ["<"] = "<", [">"] = ">", ["&"] = "&", | |
52 ["'"] = "'", ["\""] = """, ["\n"] = "<br/>", | |
53 }; | |
54 | |
55 local function tohtml(plain) | |
56 return (plain:gsub("[<>&'\"\n]", entities)); | |
57 | |
58 end | |
59 | 54 |
60 local function get_page(code, extra) | 55 local function get_page(code, extra) |
61 local message = messages[code]; | 56 local message = messages[code]; |
62 if always_serve or message then | 57 if always_serve or message then |
63 message = message or default_message; | 58 message = message or default_message; |
64 return (html:gsub("$(%a+)", { | 59 return render(html, { |
65 title = rawget(codes, code) or ("Code "..tostring(code)); | 60 title = rawget(codes, code) or ("Code "..tostring(code)); |
66 message = message[1]:gsub("%%", function () | 61 message = message[1]:gsub("%%", function () |
67 return message[math.random(2, math.max(#message,2))]; | 62 return message[math.random(2, math.max(#message,2))]; |
68 end); | 63 end); |
69 extra = tohtml(extra or ""); | 64 extra = extra; |
70 })); | 65 }); |
71 end | 66 end |
72 end | 67 end |
73 | 68 |
74 module:hook_object_event(server, "http-error", function (event) | 69 module:hook_object_event(server, "http-error", function (event) |
70 if event.response then | |
71 event.response.headers.content_type = "text/html; charset=utf-8"; | |
72 end | |
75 return get_page(event.code, (show_private and event.private_message) or event.message); | 73 return get_page(event.code, (show_private and event.private_message) or event.message); |
76 end); | 74 end); |