plugins/mod_groups.lua

   1
   2 local groups = { default = {} };
   3 local members = { [false] = {} };
   4
   5 local groups_file;
   6
   7 local jid, datamanager = require "util.jid", require "util.datamanager";
   8 local jid_bare, jid_prep = jid.bare, jid.prep;
   9
  10 local module_host = module:get_host();
  11
  12 function inject_roster_contacts(username, host, roster)
  13         module:log("warn", "Injecting group members to roster");
  14         local bare_jid = username.."@"..host;
  15         if not members[bare_jid] then return; end -- Not a member of any groups
  16
  17         local function import_jids_to_roster(group_name)
  18                 for jid in pairs(groups[group_name]) do
  19                         -- Add them to roster
  20                         --module:log("debug", "processing jid %s in group %s", tostring(jid), tostring(group_name));
  21                         if jid ~= bare_jid then
  22                                 if not roster[jid] then roster[jid] = {}; end
  23                                 roster[jid].subscription = "both";
  24                                 if not roster[jid].groups then
  25                                         roster[jid].groups = { [group_name] = true };
  26                                 end
  27                                 roster[jid].groups[group_name] = true;
  28                                 roster[jid].persist = false;
  29                         end
  30                 end
  31         end
  32
  33         -- Find groups this JID is a member of
  34         for _, group_name in ipairs(members[bare_jid]) do
  35                 import_jids_to_roster(group_name);
  36         end
  37
  38         -- Import public groups
  39         for _, group_name in ipairs(members[false]) do
  40                 import_jids_to_roster(group_name);
  41         end
  42 end
  43
  44 function remove_virtual_contacts(username, host, datastore, data)
  45         if host == module_host and datastore == "roster" then
  46                 local new_roster = {};
  47                 for jid, contact in pairs(data) do
  48                         if contact.persist ~= false then
  49                                 new_roster[jid] = contact;
  50                         end
  51                 end
  52                 return username, host, datastore, new_roster;
  53         end
  54
  55         return username, host, datastore, data;
  56 end
  57
  58 function module.load()
  59         groups_file = config.get(module:get_host(), "core", "groups_file");
  60         if not groups_file then return; end
  61
  62         module:hook("roster-load", inject_roster_contacts);
  63         datamanager.add_callback(remove_virtual_contacts);
  64
  65         groups = { default = {} };
  66         members = { [false] = {} };
  67         local curr_group = "default";
  68         for line in io.lines(groups_file) do
  69                 if line:match("^%s*%[.-%]%s*$") then
  70                         curr_group = line:match("^%s*%[(.-)%]%s*$");
  71                         if curr_group:match("^%+") then
  72                                 curr_group = curr_group:gsub("^%+", "");
  73                                 members[false][#members[false]+1] = curr_group; -- Is a public group
  74                         end
  75                         module:log("debug", "New group: %s", tostring(curr_group));
  76                         groups[curr_group] = groups[curr_group] or {};
  77                 else
  78                         -- Add JID
  79                         local jid = jid_prep(line);
  80                         if jid then
  81                                 module:log("debug", "New member of %s: %s", tostring(curr_group), tostring(jid));
  82                                 groups[curr_group][jid] = true;
  83                                 members[jid] = members[jid] or {};
  84                                 members[jid][#members[jid]+1] = curr_group;
  85                         end
  86                 end
  87         end
  88         module:log("info", "Groups loaded successfully");
  89 end
  90
  91 function module.unload()
  92         datamanager.remove_callback(remove_virtual_contacts);
  93 end