mirror of
https://github.com/bjc/prosody.git
synced 2025-04-04 13:47:41 +03:00
298 lines
8.9 KiB
Lua
298 lines
8.9 KiB
Lua
-- Prosody IM
|
|
-- Copyright (C) 2008-2010 Matthew Wild
|
|
-- Copyright (C) 2008-2010 Waqas Hussain
|
|
--
|
|
-- This project is MIT/X11 licensed. Please see the
|
|
-- COPYING file in the source package for more information.
|
|
--
|
|
|
|
|
|
|
|
|
|
local log = require "util.logger".init("rostermanager");
|
|
|
|
local setmetatable = setmetatable;
|
|
local format = string.format;
|
|
local loadfile, setfenv, pcall = loadfile, setfenv, pcall;
|
|
local pairs, ipairs = pairs, ipairs;
|
|
local tostring = tostring;
|
|
|
|
local hosts = hosts;
|
|
local bare_sessions = bare_sessions;
|
|
|
|
local datamanager = require "util.datamanager"
|
|
local st = require "util.stanza";
|
|
|
|
module "rostermanager"
|
|
|
|
function add_to_roster(session, jid, item)
|
|
if session.roster then
|
|
local old_item = session.roster[jid];
|
|
session.roster[jid] = item;
|
|
if save_roster(session.username, session.host) then
|
|
return true;
|
|
else
|
|
session.roster[jid] = old_item;
|
|
return nil, "wait", "internal-server-error", "Unable to save roster";
|
|
end
|
|
else
|
|
return nil, "auth", "not-authorized", "Session's roster not loaded";
|
|
end
|
|
end
|
|
|
|
function remove_from_roster(session, jid)
|
|
if session.roster then
|
|
local old_item = session.roster[jid];
|
|
session.roster[jid] = nil;
|
|
if save_roster(session.username, session.host) then
|
|
return true;
|
|
else
|
|
session.roster[jid] = old_item;
|
|
return nil, "wait", "internal-server-error", "Unable to save roster";
|
|
end
|
|
else
|
|
return nil, "auth", "not-authorized", "Session's roster not loaded";
|
|
end
|
|
end
|
|
|
|
function roster_push(username, host, jid)
|
|
local roster = jid and jid ~= "pending" and hosts[host] and hosts[host].sessions[username] and hosts[host].sessions[username].roster;
|
|
if roster then
|
|
local item = hosts[host].sessions[username].roster[jid];
|
|
local stanza = st.iq({type="set"});
|
|
stanza:tag("query", {xmlns = "jabber:iq:roster", ver = tostring(roster[false].version or "1") });
|
|
if item then
|
|
stanza:tag("item", {jid = jid, subscription = item.subscription, name = item.name, ask = item.ask});
|
|
for group in pairs(item.groups) do
|
|
stanza:tag("group"):text(group):up();
|
|
end
|
|
else
|
|
stanza:tag("item", {jid = jid, subscription = "remove"});
|
|
end
|
|
stanza:up(); -- move out from item
|
|
stanza:up(); -- move out from stanza
|
|
-- stanza ready
|
|
for _, session in pairs(hosts[host].sessions[username].sessions) do
|
|
if session.interested then
|
|
-- FIXME do we need to set stanza.attr.to?
|
|
session.send(stanza);
|
|
end
|
|
end
|
|
end
|
|
end
|
|
|
|
function load_roster(username, host)
|
|
local jid = username.."@"..host;
|
|
log("debug", "load_roster: asked for: "..jid);
|
|
local user = bare_sessions[jid];
|
|
local roster;
|
|
if user then
|
|
roster = user.roster;
|
|
if roster then return roster; end
|
|
log("debug", "load_roster: loading for new user: "..username.."@"..host);
|
|
else -- Attempt to load roster for non-loaded user
|
|
log("debug", "load_roster: loading for offline user: "..username.."@"..host);
|
|
end
|
|
roster = datamanager.load(username, host, "roster") or {};
|
|
if user then user.roster = roster; end
|
|
if not roster[false] then roster[false] = { }; end
|
|
if roster[jid] then
|
|
roster[jid] = nil;
|
|
log("warn", "roster for "..jid.." has a self-contact");
|
|
end
|
|
hosts[host].events.fire_event("roster-load", username, host, roster);
|
|
return roster;
|
|
end
|
|
|
|
function save_roster(username, host, roster)
|
|
log("debug", "save_roster: saving roster for "..username.."@"..host);
|
|
if not roster then
|
|
roster = hosts[host] and hosts[host].sessions[username] and hosts[host].sessions[username].roster;
|
|
--if not roster then
|
|
-- --roster = load_roster(username, host);
|
|
-- return true; -- roster unchanged, no reason to save
|
|
--end
|
|
end
|
|
if roster then
|
|
if not roster[false] then roster[false] = {}; end
|
|
roster[false].version = (roster[false].version or 0) + 1;
|
|
return datamanager.store(username, host, "roster", roster);
|
|
end
|
|
log("warn", "save_roster: user had no roster to save");
|
|
return nil;
|
|
end
|
|
|
|
function process_inbound_subscription_approval(username, host, jid)
|
|
local roster = load_roster(username, host);
|
|
local item = roster[jid];
|
|
if item and item.ask then
|
|
if item.subscription == "none" then
|
|
item.subscription = "to";
|
|
else -- subscription == from
|
|
item.subscription = "both";
|
|
end
|
|
item.ask = nil;
|
|
return save_roster(username, host, roster);
|
|
end
|
|
end
|
|
|
|
function process_inbound_subscription_cancellation(username, host, jid)
|
|
local roster = load_roster(username, host);
|
|
local item = roster[jid];
|
|
local changed = nil;
|
|
if is_contact_pending_out(username, host, jid) then
|
|
item.ask = nil;
|
|
changed = true;
|
|
end
|
|
if item then
|
|
if item.subscription == "to" then
|
|
item.subscription = "none";
|
|
changed = true;
|
|
elseif item.subscription == "both" then
|
|
item.subscription = "from";
|
|
changed = true;
|
|
end
|
|
end
|
|
if changed then
|
|
return save_roster(username, host, roster);
|
|
end
|
|
end
|
|
|
|
function process_inbound_unsubscribe(username, host, jid)
|
|
local roster = load_roster(username, host);
|
|
local item = roster[jid];
|
|
local changed = nil;
|
|
if is_contact_pending_in(username, host, jid) then
|
|
roster.pending[jid] = nil; -- TODO maybe delete roster.pending if empty?
|
|
changed = true;
|
|
end
|
|
if item then
|
|
if item.subscription == "from" then
|
|
item.subscription = "none";
|
|
changed = true;
|
|
elseif item.subscription == "both" then
|
|
item.subscription = "to";
|
|
changed = true;
|
|
end
|
|
end
|
|
if changed then
|
|
return save_roster(username, host, roster);
|
|
end
|
|
end
|
|
|
|
function is_contact_subscribed(username, host, jid)
|
|
local roster = load_roster(username, host);
|
|
local item = roster[jid];
|
|
return item and (item.subscription == "from" or item.subscription == "both");
|
|
end
|
|
|
|
function is_contact_pending_in(username, host, jid)
|
|
local roster = load_roster(username, host);
|
|
return roster.pending and roster.pending[jid];
|
|
end
|
|
function set_contact_pending_in(username, host, jid, pending)
|
|
local roster = load_roster(username, host);
|
|
local item = roster[jid];
|
|
if item and (item.subscription == "from" or item.subscription == "both") then
|
|
return; -- false
|
|
end
|
|
if not roster.pending then roster.pending = {}; end
|
|
roster.pending[jid] = true;
|
|
return save_roster(username, host, roster);
|
|
end
|
|
function is_contact_pending_out(username, host, jid)
|
|
local roster = load_roster(username, host);
|
|
local item = roster[jid];
|
|
return item and item.ask;
|
|
end
|
|
function set_contact_pending_out(username, host, jid) -- subscribe
|
|
local roster = load_roster(username, host);
|
|
local item = roster[jid];
|
|
if item and (item.ask or item.subscription == "to" or item.subscription == "both") then
|
|
return true;
|
|
end
|
|
if not item then
|
|
item = {subscription = "none", groups = {}};
|
|
roster[jid] = item;
|
|
end
|
|
item.ask = "subscribe";
|
|
log("debug", "set_contact_pending_out: saving roster; set "..username.."@"..host..".roster["..jid.."].ask=subscribe");
|
|
return save_roster(username, host, roster);
|
|
end
|
|
function unsubscribe(username, host, jid)
|
|
local roster = load_roster(username, host);
|
|
local item = roster[jid];
|
|
if not item then return false; end
|
|
if (item.subscription == "from" or item.subscription == "none") and not item.ask then
|
|
return true;
|
|
end
|
|
item.ask = nil;
|
|
if item.subscription == "both" then
|
|
item.subscription = "from";
|
|
elseif item.subscription == "to" then
|
|
item.subscription = "none";
|
|
end
|
|
return save_roster(username, host, roster);
|
|
end
|
|
function subscribed(username, host, jid)
|
|
if is_contact_pending_in(username, host, jid) then
|
|
local roster = load_roster(username, host);
|
|
local item = roster[jid];
|
|
if not item then -- FIXME should roster item be auto-created?
|
|
item = {subscription = "none", groups = {}};
|
|
roster[jid] = item;
|
|
end
|
|
if item.subscription == "none" then
|
|
item.subscription = "from";
|
|
else -- subscription == to
|
|
item.subscription = "both";
|
|
end
|
|
roster.pending[jid] = nil;
|
|
-- TODO maybe remove roster.pending if empty
|
|
return save_roster(username, host, roster);
|
|
end -- TODO else implement optional feature pre-approval (ask = subscribed)
|
|
end
|
|
function unsubscribed(username, host, jid)
|
|
local roster = load_roster(username, host);
|
|
local item = roster[jid];
|
|
local pending = is_contact_pending_in(username, host, jid);
|
|
local changed = nil;
|
|
if is_contact_pending_in(username, host, jid) then
|
|
roster.pending[jid] = nil; -- TODO maybe delete roster.pending if empty?
|
|
changed = true;
|
|
end
|
|
if item then
|
|
if item.subscription == "from" then
|
|
item.subscription = "none";
|
|
changed = true;
|
|
elseif item.subscription == "both" then
|
|
item.subscription = "to";
|
|
changed = true;
|
|
end
|
|
end
|
|
if changed then
|
|
return save_roster(username, host, roster);
|
|
end
|
|
end
|
|
|
|
function process_outbound_subscription_request(username, host, jid)
|
|
local roster = load_roster(username, host);
|
|
local item = roster[jid];
|
|
if item and (item.subscription == "none" or item.subscription == "from") then
|
|
item.ask = "subscribe";
|
|
return save_roster(username, host, roster);
|
|
end
|
|
end
|
|
|
|
--[[function process_outbound_subscription_approval(username, host, jid)
|
|
local roster = load_roster(username, host);
|
|
local item = roster[jid];
|
|
if item and (item.subscription == "none" or item.subscription == "from" then
|
|
item.ask = "subscribe";
|
|
return save_roster(username, host, roster);
|
|
end
|
|
end]]
|
|
|
|
|
|
|
|
return _M;
|