Software /
code /
prosody
Comparison
plugins/mod_carbons.lua @ 6546:3426e903c48d
mod_carbons: Import XEP-0280 implementation from prosody-modules (sans compat with older versions of the protocol)
author | Kim Alvefur <zash@zash.se> |
---|---|
date | Fri, 16 Jan 2015 14:02:14 +0100 |
child | 6803:7ed87299dbf9 |
comparison
equal
deleted
inserted
replaced
6545:ec566d7cd518 | 6546:3426e903c48d |
---|---|
1 -- XEP-0280: Message Carbons implementation for Prosody | |
2 -- Copyright (C) 2011 Kim Alvefur | |
3 -- | |
4 -- This file is MIT/X11 licensed. | |
5 | |
6 local st = require "util.stanza"; | |
7 local jid_bare = require "util.jid".bare; | |
8 local xmlns_carbons = "urn:xmpp:carbons:2"; | |
9 local xmlns_forward = "urn:xmpp:forward:0"; | |
10 local full_sessions, bare_sessions = full_sessions, bare_sessions; | |
11 | |
12 local function toggle_carbons(event) | |
13 local origin, stanza = event.origin, event.stanza; | |
14 local state = stanza.tags[1].name; | |
15 module:log("debug", "%s %sd carbons", origin.full_jid, state); | |
16 origin.want_carbons = state == "enable" and stanza.tags[1].attr.xmlns; | |
17 return origin.send(st.reply(stanza)); | |
18 end | |
19 module:hook("iq-set/self/"..xmlns_carbons..":disable", toggle_carbons); | |
20 module:hook("iq-set/self/"..xmlns_carbons..":enable", toggle_carbons); | |
21 | |
22 local function message_handler(event, c2s) | |
23 local origin, stanza = event.origin, event.stanza; | |
24 local orig_type = stanza.attr.type; | |
25 local orig_from = stanza.attr.from; | |
26 local orig_to = stanza.attr.to; | |
27 | |
28 if not (orig_type == nil | |
29 or orig_type == "normal" | |
30 or orig_type == "chat") then | |
31 return -- No carbons for messages of type error or headline | |
32 end | |
33 | |
34 -- Stanza sent by a local client | |
35 local bare_jid = jid_bare(orig_from); | |
36 local target_session = origin; | |
37 local top_priority = false; | |
38 local user_sessions = bare_sessions[bare_jid]; | |
39 | |
40 -- Stanza about to be delivered to a local client | |
41 if not c2s then | |
42 bare_jid = jid_bare(orig_to); | |
43 target_session = full_sessions[orig_to]; | |
44 user_sessions = bare_sessions[bare_jid]; | |
45 if not target_session and user_sessions then | |
46 -- The top resources will already receive this message per normal routing rules, | |
47 -- so we are going to skip them in order to avoid sending duplicated messages. | |
48 local top_resources = user_sessions.top_resources; | |
49 top_priority = top_resources and top_resources[1].priority | |
50 end | |
51 end | |
52 | |
53 if not user_sessions then | |
54 module:log("debug", "Skip carbons for offline user"); | |
55 return -- No use in sending carbons to an offline user | |
56 end | |
57 | |
58 if stanza:get_child("private", xmlns_carbons) then | |
59 if not c2s then | |
60 stanza:maptags(function(tag) | |
61 if not ( tag.attr.xmlns == xmlns_carbons and tag.name == "private" ) then | |
62 return tag; | |
63 end | |
64 end); | |
65 end | |
66 module:log("debug", "Message tagged private, ignoring"); | |
67 return | |
68 elseif stanza:get_child("no-copy", "urn:xmpp:hints") then | |
69 module:log("debug", "Message has no-copy hint, ignoring"); | |
70 return | |
71 elseif stanza:get_child("x", "http://jabber.org/protocol/muc#user") then | |
72 module:log("debug", "MUC PM, ignoring"); | |
73 return | |
74 end | |
75 | |
76 -- Create the carbon copy and wrap it as per the Stanza Forwarding XEP | |
77 local copy = st.clone(stanza); | |
78 copy.attr.xmlns = "jabber:client"; | |
79 local carbon = st.message{ from = bare_jid, type = orig_type, } | |
80 :tag(c2s and "sent" or "received", { xmlns = xmlns_carbons }) | |
81 :tag("forwarded", { xmlns = xmlns_forward }) | |
82 :add_child(copy):reset(); | |
83 | |
84 user_sessions = user_sessions and user_sessions.sessions; | |
85 for _, session in pairs(user_sessions) do | |
86 -- Carbons are sent to resources that have enabled it | |
87 if session.want_carbons | |
88 -- but not the resource that sent the message, or the one that it's directed to | |
89 and session ~= target_session | |
90 -- and isn't among the top resources that would receive the message per standard routing rules | |
91 and (c2s or session.priority ~= top_priority) then | |
92 carbon.attr.to = session.full_jid; | |
93 module:log("debug", "Sending carbon to %s", session.full_jid); | |
94 session.send(carbon); | |
95 end | |
96 end | |
97 end | |
98 | |
99 local function c2s_message_handler(event) | |
100 return message_handler(event, true) | |
101 end | |
102 | |
103 -- Stanzas sent by local clients | |
104 module:hook("pre-message/host", c2s_message_handler, 1); | |
105 module:hook("pre-message/bare", c2s_message_handler, 1); | |
106 module:hook("pre-message/full", c2s_message_handler, 1); | |
107 -- Stanzas to local clients | |
108 module:hook("message/bare", message_handler, 1); | |
109 module:hook("message/full", message_handler, 1); | |
110 | |
111 module:add_feature(xmlns_carbons); |