local st = require"util.stanza";
local jid_split = require "util.jid".split;
local jid_bare = require "util.jid".bare;
local is_contact_subscribed = require "core.rostermanager".is_contact_subscribed;
local full_sessions = prosody.full_sessions;
local function has_directed_presence(user, jid)
local session = full_sessions[user];
if session then
local directed = session.directed;
if directed then
return directed[jid];
end
end
end
function check_subscribed(event)
local stanza = event.stanza;
local to_user, to_host, to_resource = jid_split(stanza.attr.to);
local from_jid = jid_bare(stanza.attr.from);
if to_user and not has_directed_presence(stanza.attr.to, from_jid) and not is_contact_subscribed(to_user, to_host, from_jid) then
-- Allow all messages from your own jid
if from_jid == to_user.."@"..to_host then
return nil; -- Pass through
end
if to_resource and stanza.attr.type == "groupchat"
or stanza.name == "iq" and (stanza.attr.type == "result" or stanza.attr.type == "error") then
return nil; -- Pass through
end
if stanza.name == "iq" and ( stanza.attr.type == "get" or stanza.attr.type == "set" ) then
event.origin.send(st.error_reply(stanza, "cancel", "service-unavailable"));
end
return true; -- Drop stanza
end
end
module:hook("message/bare", check_subscribed, 200);
module:hook("message/full", check_subscribed, 200);
module:hook("iq/full", check_subscribed, 200);