Mercurial > lxmppd-hg-xmllex / file comparison
comparison: plugins/mod_groups.lua
plugins/mod_groups.lua
- changeset 1394
- 2ebed659b958
- parent 1388
- 546caa44620c
- child 1522
- 569d58d21612
equal
deleted
inserted
replaced
| 1393:576143941a76 | 1394:2ebed659b958 |
|---|---|
| 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 |
