4659
|
1 |
local array = require"util.array"; |
|
2 |
local st = require"util.stanza"; |
|
3 |
|
|
4 |
local tos_version = assert(module:get_option("tos_version"), "tos_version must be set") |
|
5 |
|
|
6 |
local status_storage; |
|
7 |
if prosody.process_type == "prosody" or prosody.shutdown then |
|
8 |
status_storage = module:open_store("tos_status", "keyval") |
|
9 |
end |
|
10 |
|
|
11 |
local documents = array{}; |
|
12 |
|
|
13 |
local function validate_doc(doc) |
|
14 |
if not doc.title or not doc.sources or #doc.sources < 1 then |
|
15 |
return false, "document needs to have a title and at least one source" |
|
16 |
end |
|
17 |
for _, source in ipairs(doc.sources) do |
|
18 |
if not source.url or not source.type then |
|
19 |
return false, "document " .. doc.title .. " has a source without url or type" |
|
20 |
end |
|
21 |
end |
|
22 |
return true, doc |
|
23 |
end |
|
24 |
|
|
25 |
for _, doc in ipairs(assert(module:get_option("tos_documents"), "tos_documents option is required")) do |
|
26 |
local ok, doc_or_err = validate_doc(doc) |
|
27 |
if not ok then |
|
28 |
error("invalid TOS document: "..doc_or_err) |
|
29 |
end |
|
30 |
documents:push(doc); |
|
31 |
end |
|
32 |
|
|
33 |
local function send_tos_push(session) |
|
34 |
local to = session.username .. "@" .. session.host .. "/" .. session.resource; |
|
35 |
local push = st.message({ |
|
36 |
type = "headline", |
|
37 |
to = to, |
|
38 |
}, "In order to continue to use the service, you have to accept the current version of the Terms of Service."); |
|
39 |
local tos = push:tag("tos-push", { xmlns = "urn:xmpp:tos:0" }):tag("tos", { version = tos_version }); |
|
40 |
for _, doc in ipairs(documents) do |
|
41 |
local doc_tag = tos:tag("document"):text_tag("title", doc.title); |
|
42 |
for _, source in ipairs(doc.sources) do |
|
43 |
doc_tag:tag("source", { url = source.url, type = source.type }):up(); |
|
44 |
end |
|
45 |
doc_tag:up(); |
|
46 |
end |
|
47 |
tos:up():up(); |
|
48 |
module:send(push); |
|
49 |
end |
|
50 |
|
|
51 |
local function check_tos(event) |
|
52 |
local user = event.origin.username |
|
53 |
assert(user) |
|
54 |
local tos_status = status_storage:get(user) |
|
55 |
if tos_status and tos_status.version and tos_status.version == tos_version then |
|
56 |
module:log("debug", "user %s has signed the current tos", user); |
|
57 |
return |
|
58 |
end |
|
59 |
module:log("debug", "user %s has not signed the current tos, sending tos push", user); |
|
60 |
send_tos_push(event.origin) |
|
61 |
end |
|
62 |
|
|
63 |
local function handle_accept_tos_iq(event) |
|
64 |
local user = event.origin.username; |
|
65 |
assert(user); |
|
66 |
local accept = event.stanza.tags[1]; |
|
67 |
local version = accept.attr["version"]; |
|
68 |
module:log("debug", "user %s has accepted ToS version %s", user, version); |
|
69 |
if version ~= tos_version then |
|
70 |
local reply = st.error_reply(event.stanza, "modify", "not-allowed", "Only the most recent version of the ToS can be accepted"); |
|
71 |
module:log("debug", "%s is not the most recent version (%s), rejecting", version, tos_version); |
|
72 |
event.origin.send(reply); |
|
73 |
return true; |
|
74 |
end |
|
75 |
|
|
76 |
status_storage:set(user, { version = tos_version }); |
|
77 |
local reply = st.reply(event.stanza); |
|
78 |
event.origin.send(reply); |
|
79 |
return true; |
|
80 |
end |
|
81 |
|
|
82 |
module:hook("presence/initial", check_tos, -100); |
|
83 |
module:hook("iq-set/bare/urn:xmpp:tos:0:accept", handle_accept_tos_iq); |