|
462
|
1 local st = require "util.stanza"; |
|
|
2 local jid_bare = require "util.jid".bare; |
|
|
3 local jid_split = require "util.jid".split; |
|
|
4 local xmlns_carbons = "urn:xmpp:carbons:1"; |
|
|
5 local xmlns_forward = "urn:xmpp:forward:0"; |
|
|
6 local host_sessions = hosts[module.host].sessions; |
|
|
7 |
|
|
8 -- TODO merge message handlers into one somehow |
|
|
9 |
|
|
10 module:hook("iq/self/"..xmlns_carbons..":enable", function(event) |
|
|
11 local origin, stanza = event.origin, event.stanza; |
|
|
12 if stanza.attr.type == "set" then |
|
|
13 module:log("debug", "%s enabled carbons", origin.full_jid); |
|
|
14 origin.want_carbons = true; |
|
|
15 origin.send(st.reply(stanza)); |
|
|
16 return true |
|
|
17 end |
|
|
18 end); |
|
|
19 |
|
|
20 module:hook("iq/self/"..xmlns_carbons..":disable", function(event) |
|
|
21 local origin, stanza = event.origin, event.stanza; |
|
|
22 if stanza.attr.type == "set" then |
|
|
23 module:log("debug", "%s disabled carbons", origin.full_jid); |
|
|
24 origin.want_carbons = nil; |
|
|
25 origin.send(st.reply(stanza)); |
|
|
26 return true |
|
|
27 end |
|
|
28 end); |
|
|
29 |
|
|
30 function c2s_message_handler(event) |
|
|
31 local origin, stanza = event.origin, event.stanza; |
|
|
32 local orig_type = stanza.attr.type; |
|
|
33 local orig_to = stanza.attr.to; |
|
|
34 |
|
|
35 if not (orig_type == nil |
|
|
36 or orig_type == "normal" |
|
|
37 or orig_type == "chat") then |
|
|
38 return |
|
|
39 end |
|
|
40 |
|
|
41 local bare_jid, user_sessions; |
|
|
42 if origin.type == "s2s" then |
|
|
43 bare_jid = jid_bare(stanza.attr.from); |
|
|
44 user_sessions = host_sessions[jid_split(orig_to)]; |
|
|
45 else |
|
|
46 bare_jid = (origin.username.."@"..origin.host) |
|
|
47 user_sessions = host_sessions[origin.username]; |
|
|
48 end |
|
|
49 |
|
|
50 if not stanza:get_child("private", xmlns_carbons) |
|
|
51 and not stanza:get_child("forwarded", xmlns_forward) then |
|
|
52 user_sessions = user_sessions and user_sessions.sessions; |
|
|
53 for resource, session in pairs(user_sessions) do |
|
|
54 local full_jid = bare_jid .. "/" .. resource; |
|
|
55 if session ~= origin and session.want_carbons then |
|
|
56 local msg = st.clone(stanza); |
|
|
57 msg.attr.xmlns = msg.attr.xmlns or "jabber:client"; |
|
|
58 local fwd = st.message{ |
|
|
59 from = bare_jid, |
|
|
60 to = full_jid, |
|
|
61 type = orig_type, |
|
|
62 } |
|
|
63 :tag("forwarded", { xmlns = xmlns_forward }) |
|
|
64 :tag("received", { xmlns = xmlns_carbons }):up() |
|
|
65 :add_child(msg); |
|
|
66 core_route_stanza(origin, fwd); |
|
|
67 end |
|
|
68 end |
|
|
69 end |
|
|
70 end |
|
|
71 |
|
|
72 function s2c_message_handler(event) |
|
|
73 local origin, stanza = event.origin, event.stanza; |
|
|
74 local orig_type = stanza.attr.type; |
|
|
75 local orig_to = stanza.attr.to; |
|
|
76 |
|
|
77 if not (orig_type == nil |
|
|
78 or orig_type == "normal" |
|
|
79 or orig_type == "chat") then |
|
|
80 return |
|
|
81 end |
|
|
82 |
|
|
83 local full_jid, bare_jid = orig_to, jid_bare(orig_to); |
|
|
84 local username, hostname, resource = jid_split(full_jid); |
|
|
85 local user_sessions = username and host_sessions[username]; |
|
|
86 if not user_sessions or hostname ~= module.host then |
|
|
87 return |
|
|
88 end |
|
|
89 |
|
|
90 local no_carbon_to = {}; |
|
|
91 if resource then |
|
|
92 no_carbon_to[resource] = true; |
|
|
93 else |
|
|
94 local top_resources = user_sessions.top_resources; |
|
|
95 for i=1,top_resources do |
|
|
96 no_carbon_to[top_resources[i]] = true; |
|
|
97 end |
|
|
98 end |
|
|
99 |
|
|
100 if not stanza:get_child("private", xmlns_carbons) |
|
|
101 and not stanza:get_child("forwarded", xmlns_forward) then |
|
|
102 user_sessions = user_sessions and user_sessions.sessions; |
|
|
103 for resource, session in pairs(user_sessions) do |
|
|
104 local full_jid = bare_jid .. "/" .. resource; |
|
|
105 if not no_carbon_to[resource] and session.want_carbons then |
|
|
106 local msg = st.clone(stanza); |
|
|
107 msg.attr.xmlns = msg.attr.xmlns or "jabber:client"; |
|
|
108 local fwd = st.message{ |
|
|
109 from = bare_jid, |
|
|
110 to = full_jid, |
|
|
111 type = orig_type, |
|
|
112 } |
|
|
113 :tag("forwarded", { xmlns = xmlns_forward }) |
|
|
114 :tag("received", { xmlns = xmlns_carbons }):up() |
|
|
115 :add_child(msg); |
|
|
116 core_route_stanza(origin, fwd); |
|
|
117 end |
|
|
118 end |
|
|
119 end |
|
|
120 end |
|
|
121 |
|
|
122 -- Stanzas sent by local clients |
|
|
123 module:hook("pre-message/bare", c2s_message_handler, 1); |
|
|
124 module:hook("pre-message/full", c2s_message_handler, 1); |
|
|
125 -- Stanszas to local clients |
|
|
126 module:hook("message/bare", s2c_message_handler, 1); -- this will suck |
|
|
127 module:hook("message/full", s2c_message_handler, 1); |
|
|
128 |
|
|
129 module:add_feature(xmlns_carbons); |
