mirror of https://github.com/bjc/prosody
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
106 lines
2.9 KiB
106 lines
2.9 KiB
|
|
local mainlog = log;
|
|
local function log(type, message)
|
|
mainlog(type, "rostermanager", message);
|
|
end
|
|
|
|
local setmetatable = setmetatable;
|
|
local format = string.format;
|
|
local loadfile, setfenv, pcall = loadfile, setfenv, pcall;
|
|
|
|
local hosts = hosts;
|
|
|
|
require "util.datamanager"
|
|
|
|
local datamanager = datamanager;
|
|
local st = require "util.stanza";
|
|
local send = require "core.sessionmanager".send_to_session;
|
|
|
|
module "rostermanager"
|
|
|
|
--[[function getroster(username, host)
|
|
return {
|
|
["mattj@localhost"] = true,
|
|
["tobias@getjabber.ath.cx"] = true,
|
|
["waqas@getjabber.ath.cx"] = true,
|
|
["thorns@getjabber.ath.cx"] = true,
|
|
["idw@getjabber.ath.cx"] = true,
|
|
}
|
|
--return datamanager.load(username, host, "roster") or {};
|
|
end]]
|
|
|
|
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)
|
|
if hosts[host] and hosts[host].sessions[username] and hosts[host].sessions[username].roster then
|
|
local item = hosts[host].sessions[username].roster[jid];
|
|
local stanza = st.iq({type="set"});
|
|
stanza:tag("query", {xmlns = "jabber:iq:roster"});
|
|
if item then
|
|
stanza:tag("item", {jid = jid, subscription = item.subscription, name = item.name});
|
|
else
|
|
stanza:tag("item", {jid = jid, subscription = "remove"});
|
|
end
|
|
for group in item.groups do
|
|
stanza:tag("group"):text(group):up();
|
|
end
|
|
stanza:up();
|
|
stanza:up();
|
|
-- stanza ready
|
|
for _, session in ipairs(hosts[host].sessions[username].sessions) do
|
|
if session.full_jid then
|
|
-- FIXME do we need to set stanza.attr.to?
|
|
send(session, stanza);
|
|
end
|
|
end
|
|
end
|
|
end
|
|
|
|
function load_roster(username, host)
|
|
if hosts[host] and hosts[host].sessions[username] then
|
|
local roster = hosts[host].sessions[username].roster;
|
|
if not roster then
|
|
roster = datamanager.load(username, host, "roster") or {};
|
|
hosts[host].sessions[username].roster = roster;
|
|
end
|
|
return roster;
|
|
end
|
|
-- Attempt to load roster for non-loaded user
|
|
end
|
|
|
|
function save_roster(username, host)
|
|
if hosts[host] and hosts[host].sessions[username] and hosts[host].sessions[username].roster then
|
|
return datamanager.save(username, host, "roster", hosts[host].sessions[username].roster);
|
|
end
|
|
return nil;
|
|
end
|
|
|
|
return _M; |