Mercurial > prosody-modules
annotate mod_muc_markers/mod_muc_markers.lua @ 4029:b9bd56790286
mod_measure_message_e2ee/README: Fix typo
author | Kim Alvefur <zash@zash.se> |
---|---|
date | Wed, 20 May 2020 17:24:15 +0200 |
parents | e3964f876b5d |
children | 787fc3030087 |
rev | line source |
---|---|
3972
45c5603a6c07
mod_muc_markers: New module for server-side receipt tracking in MUCs
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
1 -- Track messages received by users of the MUC |
45c5603a6c07
mod_muc_markers: New module for server-side receipt tracking in MUCs
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
2 |
45c5603a6c07
mod_muc_markers: New module for server-side receipt tracking in MUCs
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
3 -- We rewrite the 'id' attribute of outgoing stanzas to match the stanza (archive) id |
45c5603a6c07
mod_muc_markers: New module for server-side receipt tracking in MUCs
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
4 -- This module is therefore incompatible with the muc#stable_id feature |
45c5603a6c07
mod_muc_markers: New module for server-side receipt tracking in MUCs
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
5 -- We rewrite the id because XEP-0333 doesn't tell clients explicitly which id to use |
45c5603a6c07
mod_muc_markers: New module for server-side receipt tracking in MUCs
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
6 -- in marker reports. However it implies the 'id' attribute through examples, and this |
45c5603a6c07
mod_muc_markers: New module for server-side receipt tracking in MUCs
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
7 -- is what some clients implement. |
45c5603a6c07
mod_muc_markers: New module for server-side receipt tracking in MUCs
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
8 -- Notably Conversations will ack the origin-id instead. We need to update the XEP to |
45c5603a6c07
mod_muc_markers: New module for server-side receipt tracking in MUCs
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
9 -- clarify the correct behaviour. |
45c5603a6c07
mod_muc_markers: New module for server-side receipt tracking in MUCs
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
10 |
4026
e3964f876b5d
mod_muc_markers: Broadcast current markers on join
Kim Alvefur <zash@zash.se>
parents:
4025
diff
changeset
|
11 local st = require "util.stanza"; |
e3964f876b5d
mod_muc_markers: Broadcast current markers on join
Kim Alvefur <zash@zash.se>
parents:
4025
diff
changeset
|
12 |
3972
45c5603a6c07
mod_muc_markers: New module for server-side receipt tracking in MUCs
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
13 local xmlns_markers = "urn:xmpp:chat-markers:0"; |
45c5603a6c07
mod_muc_markers: New module for server-side receipt tracking in MUCs
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
14 |
4024
95882b487ed2
mod_muc_markers: Allow configuration of which marker to track, default to displayed
Matthew Wild <mwild1@gmail.com>
parents:
3972
diff
changeset
|
15 local marker_element_name = module:get_option_string("muc_marker_type", "displayed"); |
95882b487ed2
mod_muc_markers: Allow configuration of which marker to track, default to displayed
Matthew Wild <mwild1@gmail.com>
parents:
3972
diff
changeset
|
16 |
3972
45c5603a6c07
mod_muc_markers: New module for server-side receipt tracking in MUCs
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
17 local muc_marker_map_store = module:open_store("muc_markers", "map"); |
45c5603a6c07
mod_muc_markers: New module for server-side receipt tracking in MUCs
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
18 |
45c5603a6c07
mod_muc_markers: New module for server-side receipt tracking in MUCs
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
19 local function get_stanza_id(stanza, by_jid) |
45c5603a6c07
mod_muc_markers: New module for server-side receipt tracking in MUCs
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
20 for tag in stanza:childtags("stanza-id", "urn:xmpp:sid:0") do |
45c5603a6c07
mod_muc_markers: New module for server-side receipt tracking in MUCs
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
21 if tag.attr.by == by_jid then |
45c5603a6c07
mod_muc_markers: New module for server-side receipt tracking in MUCs
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
22 return tag.attr.id; |
45c5603a6c07
mod_muc_markers: New module for server-side receipt tracking in MUCs
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
23 end |
45c5603a6c07
mod_muc_markers: New module for server-side receipt tracking in MUCs
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
24 end |
45c5603a6c07
mod_muc_markers: New module for server-side receipt tracking in MUCs
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
25 return nil; |
45c5603a6c07
mod_muc_markers: New module for server-side receipt tracking in MUCs
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
26 end |
45c5603a6c07
mod_muc_markers: New module for server-side receipt tracking in MUCs
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
27 |
45c5603a6c07
mod_muc_markers: New module for server-side receipt tracking in MUCs
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
28 module:hook("muc-broadcast-message", function (event) |
45c5603a6c07
mod_muc_markers: New module for server-side receipt tracking in MUCs
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
29 local stanza = event.stanza; |
45c5603a6c07
mod_muc_markers: New module for server-side receipt tracking in MUCs
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
30 |
45c5603a6c07
mod_muc_markers: New module for server-side receipt tracking in MUCs
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
31 local archive_id = get_stanza_id(stanza, event.room.jid); |
45c5603a6c07
mod_muc_markers: New module for server-side receipt tracking in MUCs
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
32 -- We are not interested in stanzas that didn't get archived |
45c5603a6c07
mod_muc_markers: New module for server-side receipt tracking in MUCs
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
33 if not archive_id then return; end |
45c5603a6c07
mod_muc_markers: New module for server-side receipt tracking in MUCs
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
34 |
45c5603a6c07
mod_muc_markers: New module for server-side receipt tracking in MUCs
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
35 -- Add stanza id as id attribute |
45c5603a6c07
mod_muc_markers: New module for server-side receipt tracking in MUCs
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
36 stanza.attr.id = archive_id; |
45c5603a6c07
mod_muc_markers: New module for server-side receipt tracking in MUCs
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
37 -- Add markable element to request markers from clients |
45c5603a6c07
mod_muc_markers: New module for server-side receipt tracking in MUCs
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
38 stanza:tag("markable", { xmlns = xmlns_markers }):up(); |
45c5603a6c07
mod_muc_markers: New module for server-side receipt tracking in MUCs
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
39 end, -1); |
45c5603a6c07
mod_muc_markers: New module for server-side receipt tracking in MUCs
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
40 |
45c5603a6c07
mod_muc_markers: New module for server-side receipt tracking in MUCs
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
41 module:hook("muc-occupant-groupchat", function (event) |
4024
95882b487ed2
mod_muc_markers: Allow configuration of which marker to track, default to displayed
Matthew Wild <mwild1@gmail.com>
parents:
3972
diff
changeset
|
42 local marker = event.stanza:get_child(marker_element_name, xmlns_markers); |
3972
45c5603a6c07
mod_muc_markers: New module for server-side receipt tracking in MUCs
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
43 if not marker then return; end |
45c5603a6c07
mod_muc_markers: New module for server-side receipt tracking in MUCs
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
44 |
45c5603a6c07
mod_muc_markers: New module for server-side receipt tracking in MUCs
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
45 -- Store the id that the user has received to |
45c5603a6c07
mod_muc_markers: New module for server-side receipt tracking in MUCs
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
46 module:log("warn", "New marker for %s: %s", event.occupant.bare_jid, marker.attr.id); |
45c5603a6c07
mod_muc_markers: New module for server-side receipt tracking in MUCs
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
47 muc_marker_map_store:set(event.occupant.bare_jid, event.room.jid, marker.attr.id); |
45c5603a6c07
mod_muc_markers: New module for server-side receipt tracking in MUCs
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
48 |
4025
57b4cdeba318
mod_muc_markers: Prevent storage instead of broadcast
Kim Alvefur <zash@zash.se>
parents:
4024
diff
changeset
|
49 end); |
57b4cdeba318
mod_muc_markers: Prevent storage instead of broadcast
Kim Alvefur <zash@zash.se>
parents:
4024
diff
changeset
|
50 |
57b4cdeba318
mod_muc_markers: Prevent storage instead of broadcast
Kim Alvefur <zash@zash.se>
parents:
4024
diff
changeset
|
51 module:hook("muc-message-is-historic", function (event) |
57b4cdeba318
mod_muc_markers: Prevent storage instead of broadcast
Kim Alvefur <zash@zash.se>
parents:
4024
diff
changeset
|
52 local marker = event.stanza:get_child("received", xmlns_markers); |
57b4cdeba318
mod_muc_markers: Prevent storage instead of broadcast
Kim Alvefur <zash@zash.se>
parents:
4024
diff
changeset
|
53 |
57b4cdeba318
mod_muc_markers: Prevent storage instead of broadcast
Kim Alvefur <zash@zash.se>
parents:
4024
diff
changeset
|
54 -- Prevent stanza from reaching the archive (it's just noise) |
57b4cdeba318
mod_muc_markers: Prevent storage instead of broadcast
Kim Alvefur <zash@zash.se>
parents:
4024
diff
changeset
|
55 if marker then |
57b4cdeba318
mod_muc_markers: Prevent storage instead of broadcast
Kim Alvefur <zash@zash.se>
parents:
4024
diff
changeset
|
56 return false |
57b4cdeba318
mod_muc_markers: Prevent storage instead of broadcast
Kim Alvefur <zash@zash.se>
parents:
4024
diff
changeset
|
57 end |
3972
45c5603a6c07
mod_muc_markers: New module for server-side receipt tracking in MUCs
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
58 end); |
45c5603a6c07
mod_muc_markers: New module for server-side receipt tracking in MUCs
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
59 |
4026
e3964f876b5d
mod_muc_markers: Broadcast current markers on join
Kim Alvefur <zash@zash.se>
parents:
4025
diff
changeset
|
60 local function find_nickname(room, user_jid) |
e3964f876b5d
mod_muc_markers: Broadcast current markers on join
Kim Alvefur <zash@zash.se>
parents:
4025
diff
changeset
|
61 -- Find their current nickname |
e3964f876b5d
mod_muc_markers: Broadcast current markers on join
Kim Alvefur <zash@zash.se>
parents:
4025
diff
changeset
|
62 for nick, occupant in pairs(room._occupants) do |
e3964f876b5d
mod_muc_markers: Broadcast current markers on join
Kim Alvefur <zash@zash.se>
parents:
4025
diff
changeset
|
63 if occupant.bare_jid == user_jid then |
e3964f876b5d
mod_muc_markers: Broadcast current markers on join
Kim Alvefur <zash@zash.se>
parents:
4025
diff
changeset
|
64 return nick; |
e3964f876b5d
mod_muc_markers: Broadcast current markers on join
Kim Alvefur <zash@zash.se>
parents:
4025
diff
changeset
|
65 end |
e3964f876b5d
mod_muc_markers: Broadcast current markers on join
Kim Alvefur <zash@zash.se>
parents:
4025
diff
changeset
|
66 end |
e3964f876b5d
mod_muc_markers: Broadcast current markers on join
Kim Alvefur <zash@zash.se>
parents:
4025
diff
changeset
|
67 -- Or if they're not here |
e3964f876b5d
mod_muc_markers: Broadcast current markers on join
Kim Alvefur <zash@zash.se>
parents:
4025
diff
changeset
|
68 local nickname = room:get_affiliation_data(user_jid, "reserved_nickname"); |
e3964f876b5d
mod_muc_markers: Broadcast current markers on join
Kim Alvefur <zash@zash.se>
parents:
4025
diff
changeset
|
69 if nickname then return room.jid.."/"..nickname; end |
e3964f876b5d
mod_muc_markers: Broadcast current markers on join
Kim Alvefur <zash@zash.se>
parents:
4025
diff
changeset
|
70 end |
e3964f876b5d
mod_muc_markers: Broadcast current markers on join
Kim Alvefur <zash@zash.se>
parents:
4025
diff
changeset
|
71 |
e3964f876b5d
mod_muc_markers: Broadcast current markers on join
Kim Alvefur <zash@zash.se>
parents:
4025
diff
changeset
|
72 -- Synthesize markers |
e3964f876b5d
mod_muc_markers: Broadcast current markers on join
Kim Alvefur <zash@zash.se>
parents:
4025
diff
changeset
|
73 if muc_marker_map_store.get_all then |
e3964f876b5d
mod_muc_markers: Broadcast current markers on join
Kim Alvefur <zash@zash.se>
parents:
4025
diff
changeset
|
74 module:hook("muc-occupant-session-new", function (event) |
e3964f876b5d
mod_muc_markers: Broadcast current markers on join
Kim Alvefur <zash@zash.se>
parents:
4025
diff
changeset
|
75 local room, to = event.room, event.stanza.attr.from; |
e3964f876b5d
mod_muc_markers: Broadcast current markers on join
Kim Alvefur <zash@zash.se>
parents:
4025
diff
changeset
|
76 local markers = muc_marker_map_store:get_all(room.jid); |
e3964f876b5d
mod_muc_markers: Broadcast current markers on join
Kim Alvefur <zash@zash.se>
parents:
4025
diff
changeset
|
77 if not markers then return end |
e3964f876b5d
mod_muc_markers: Broadcast current markers on join
Kim Alvefur <zash@zash.se>
parents:
4025
diff
changeset
|
78 for user_jid, id in pairs(markers) do |
e3964f876b5d
mod_muc_markers: Broadcast current markers on join
Kim Alvefur <zash@zash.se>
parents:
4025
diff
changeset
|
79 local room_nick = find_nickname(room, user_jid); |
e3964f876b5d
mod_muc_markers: Broadcast current markers on join
Kim Alvefur <zash@zash.se>
parents:
4025
diff
changeset
|
80 if room_nick then |
e3964f876b5d
mod_muc_markers: Broadcast current markers on join
Kim Alvefur <zash@zash.se>
parents:
4025
diff
changeset
|
81 local recv_marker = st.message({ type = "groupchat", from = room_nick, to = to }) |
e3964f876b5d
mod_muc_markers: Broadcast current markers on join
Kim Alvefur <zash@zash.se>
parents:
4025
diff
changeset
|
82 :tag("received", { xmlns = xmlns_markers, id = id }); |
e3964f876b5d
mod_muc_markers: Broadcast current markers on join
Kim Alvefur <zash@zash.se>
parents:
4025
diff
changeset
|
83 room:route_stanza(recv_marker); |
e3964f876b5d
mod_muc_markers: Broadcast current markers on join
Kim Alvefur <zash@zash.se>
parents:
4025
diff
changeset
|
84 end |
e3964f876b5d
mod_muc_markers: Broadcast current markers on join
Kim Alvefur <zash@zash.se>
parents:
4025
diff
changeset
|
85 end |
e3964f876b5d
mod_muc_markers: Broadcast current markers on join
Kim Alvefur <zash@zash.se>
parents:
4025
diff
changeset
|
86 end); |
e3964f876b5d
mod_muc_markers: Broadcast current markers on join
Kim Alvefur <zash@zash.se>
parents:
4025
diff
changeset
|
87 end |
e3964f876b5d
mod_muc_markers: Broadcast current markers on join
Kim Alvefur <zash@zash.se>
parents:
4025
diff
changeset
|
88 |
3972
45c5603a6c07
mod_muc_markers: New module for server-side receipt tracking in MUCs
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
89 -- Public API |
45c5603a6c07
mod_muc_markers: New module for server-side receipt tracking in MUCs
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
90 |
45c5603a6c07
mod_muc_markers: New module for server-side receipt tracking in MUCs
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
91 function get_user_read_marker(user_jid, room_jid) |
45c5603a6c07
mod_muc_markers: New module for server-side receipt tracking in MUCs
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
92 return muc_marker_map_store:get(user_jid, room_jid); |
45c5603a6c07
mod_muc_markers: New module for server-side receipt tracking in MUCs
Matthew Wild <mwild1@gmail.com>
parents:
diff
changeset
|
93 end |