Software / code / prosody-modules
Annotate
mod_carbons/mod_carbons.lua @ 463:7d6a05f94941
mod_carbons: Fix top_resources loop and correctly stamp sent messages (thanks xnyhps)
| author | Kim Alvefur <zash@zash.se> |
|---|---|
| date | Sun, 30 Oct 2011 01:41:33 +0200 |
| parent | 462:f28a3f260fc2 |
| child | 480:0cef5be669de |
| rev | line source |
|---|---|
| 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 }) | |
|
463
7d6a05f94941
mod_carbons: Fix top_resources loop and correctly stamp sent messages (thanks xnyhps)
Kim Alvefur <zash@zash.se>
parents:
462
diff
changeset
|
64 :tag("sent", { xmlns = xmlns_carbons }):up() |
| 462 | 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; | |
|
463
7d6a05f94941
mod_carbons: Fix top_resources loop and correctly stamp sent messages (thanks xnyhps)
Kim Alvefur <zash@zash.se>
parents:
462
diff
changeset
|
95 for i, session in ipairs(top_resources) do |
|
7d6a05f94941
mod_carbons: Fix top_resources loop and correctly stamp sent messages (thanks xnyhps)
Kim Alvefur <zash@zash.se>
parents:
462
diff
changeset
|
96 no_carbon_to[session.resource] = true; |
|
7d6a05f94941
mod_carbons: Fix top_resources loop and correctly stamp sent messages (thanks xnyhps)
Kim Alvefur <zash@zash.se>
parents:
462
diff
changeset
|
97 module:log("debug", "not going to send to /%s", resource); |
| 462 | 98 end |
| 99 end | |
| 100 | |
| 101 if not stanza:get_child("private", xmlns_carbons) | |
| 102 and not stanza:get_child("forwarded", xmlns_forward) then | |
| 103 user_sessions = user_sessions and user_sessions.sessions; | |
| 104 for resource, session in pairs(user_sessions) do | |
| 105 local full_jid = bare_jid .. "/" .. resource; | |
| 106 if not no_carbon_to[resource] and session.want_carbons then | |
| 107 local msg = st.clone(stanza); | |
| 108 msg.attr.xmlns = msg.attr.xmlns or "jabber:client"; | |
| 109 local fwd = st.message{ | |
| 110 from = bare_jid, | |
| 111 to = full_jid, | |
| 112 type = orig_type, | |
| 113 } | |
| 114 :tag("forwarded", { xmlns = xmlns_forward }) | |
| 115 :tag("received", { xmlns = xmlns_carbons }):up() | |
| 116 :add_child(msg); | |
| 117 core_route_stanza(origin, fwd); | |
| 118 end | |
| 119 end | |
| 120 end | |
| 121 end | |
| 122 | |
| 123 -- Stanzas sent by local clients | |
| 124 module:hook("pre-message/bare", c2s_message_handler, 1); | |
| 125 module:hook("pre-message/full", c2s_message_handler, 1); | |
| 126 -- Stanszas to local clients | |
|
463
7d6a05f94941
mod_carbons: Fix top_resources loop and correctly stamp sent messages (thanks xnyhps)
Kim Alvefur <zash@zash.se>
parents:
462
diff
changeset
|
127 module:hook("message/bare", s2c_message_handler, 1); |
| 462 | 128 module:hook("message/full", s2c_message_handler, 1); |
| 129 | |
| 130 module:add_feature(xmlns_carbons); |
