mod_carbons: Initial commit.
local st = require "util.stanza";
local jid_bare = require "util.jid".bare;
local jid_split = require "util.jid".split;
local xmlns_carbons = "urn:xmpp:carbons:1";
local xmlns_forward = "urn:xmpp:forward:0";
local host_sessions = hosts[module.host].sessions;
-- TODO merge message handlers into one somehow
module:hook("iq/self/"..xmlns_carbons..":enable", function(event)
local origin, stanza = event.origin, event.stanza;
if stanza.attr.type == "set" then
module:log("debug", "%s enabled carbons", origin.full_jid);
origin.want_carbons = true;
origin.send(st.reply(stanza));
return true
end
end);
module:hook("iq/self/"..xmlns_carbons..":disable", function(event)
local origin, stanza = event.origin, event.stanza;
if stanza.attr.type == "set" then
module:log("debug", "%s disabled carbons", origin.full_jid);
origin.want_carbons = nil;
origin.send(st.reply(stanza));
return true
end
end);
function c2s_message_handler(event)
local origin, stanza = event.origin, event.stanza;
local orig_type = stanza.attr.type;
local orig_to = stanza.attr.to;
if not (orig_type == nil
or orig_type == "normal"
or orig_type == "chat") then
return
end
local bare_jid, user_sessions;
if origin.type == "s2s" then
bare_jid = jid_bare(stanza.attr.from);
user_sessions = host_sessions[jid_split(orig_to)];
else
bare_jid = (origin.username.."@"..origin.host)
user_sessions = host_sessions[origin.username];
end
if not stanza:get_child("private", xmlns_carbons)
and not stanza:get_child("forwarded", xmlns_forward) then
user_sessions = user_sessions and user_sessions.sessions;
for resource, session in pairs(user_sessions) do
local full_jid = bare_jid .. "/" .. resource;
if session ~= origin and session.want_carbons then
local msg = st.clone(stanza);
msg.attr.xmlns = msg.attr.xmlns or "jabber:client";
local fwd = st.message{
from = bare_jid,
to = full_jid,
type = orig_type,
}
:tag("forwarded", { xmlns = xmlns_forward })
:tag("received", { xmlns = xmlns_carbons }):up()
:add_child(msg);
core_route_stanza(origin, fwd);
end
end
end
end
function s2c_message_handler(event)
local origin, stanza = event.origin, event.stanza;
local orig_type = stanza.attr.type;
local orig_to = stanza.attr.to;
if not (orig_type == nil
or orig_type == "normal"
or orig_type == "chat") then
return
end
local full_jid, bare_jid = orig_to, jid_bare(orig_to);
local username, hostname, resource = jid_split(full_jid);
local user_sessions = username and host_sessions[username];
if not user_sessions or hostname ~= module.host then
return
end
local no_carbon_to = {};
if resource then
no_carbon_to[resource] = true;
else
local top_resources = user_sessions.top_resources;
for i=1,top_resources do
no_carbon_to[top_resources[i]] = true;
end
end
if not stanza:get_child("private", xmlns_carbons)
and not stanza:get_child("forwarded", xmlns_forward) then
user_sessions = user_sessions and user_sessions.sessions;
for resource, session in pairs(user_sessions) do
local full_jid = bare_jid .. "/" .. resource;
if not no_carbon_to[resource] and session.want_carbons then
local msg = st.clone(stanza);
msg.attr.xmlns = msg.attr.xmlns or "jabber:client";
local fwd = st.message{
from = bare_jid,
to = full_jid,
type = orig_type,
}
:tag("forwarded", { xmlns = xmlns_forward })
:tag("received", { xmlns = xmlns_carbons }):up()
:add_child(msg);
core_route_stanza(origin, fwd);
end
end
end
end
-- Stanzas sent by local clients
module:hook("pre-message/bare", c2s_message_handler, 1);
module:hook("pre-message/full", c2s_message_handler, 1);
-- Stanszas to local clients
module:hook("message/bare", s2c_message_handler, 1); -- this will suck
module:hook("message/full", s2c_message_handler, 1);
module:add_feature(xmlns_carbons);