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