4655
|
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); |