plugins/muc/members_only.lib.lua

   1 -- Prosody IM
   2 -- Copyright (C) 2008-2010 Matthew Wild
   3 -- Copyright (C) 2008-2010 Waqas Hussain
   4 -- Copyright (C) 2014 Daurnimator
   5 --
   6 -- This project is MIT/X11 licensed. Please see the
   7 -- COPYING file in the source package for more information.
   8 --
   9
  10 local st = require "util.stanza";
  11
  12 local muc_util = module:require "muc/util";
  13 local valid_affiliations = muc_util.valid_affiliations;
  14
  15 local function get_members_only(room)
  16         return room._data.members_only;
  17 end
  18
  19 local function set_members_only(room, members_only)
  20         members_only = members_only and true or nil;
  21         if room._data.members_only == members_only then return false; end
  22         room._data.members_only = members_only;
  23         if members_only then
  24                 --[[
  25                 If as a result of a change in the room configuration the room type is
  26                 changed to members-only but there are non-members in the room,
  27                 the service MUST remove any non-members from the room and include a
  28                 status code of 322 in the presence unavailable stanzas sent to those users
  29                 as well as any remaining occupants.
  30                 ]]
  31                 local occupants_changed = {};
  32                 for _, occupant in room:each_occupant() do
  33                         local affiliation = room:get_affiliation(occupant.bare_jid);
  34                         if valid_affiliations[affiliation or "none"] <= valid_affiliations.none then
  35                                 occupant.role = nil;
  36                                 room:save_occupant(occupant);
  37                                 occupants_changed[occupant] = true;
  38                         end
  39                 end
  40                 local x = st.stanza("x", {xmlns = "http://jabber.org/protocol/muc#user"})
  41                         :tag("status", {code="322"}):up();
  42                 for occupant in pairs(occupants_changed) do
  43                         room:publicise_occupant_status(occupant, x);
  44                         module:fire_event("muc-occupant-left", {room = room; nick = occupant.nick; occupant = occupant;});
  45                 end
  46         end
  47         if room.save then room:save(true); end
  48         return true;
  49 end
  50
  51 module:hook("muc-disco#info", function(event)
  52         event.reply:tag("feature", {var = get_members_only(event.room) and "muc_membersonly" or "muc_open"}):up();
  53 end);
  54
  55 module:hook("muc-config-form", function(event)
  56         table.insert(event.form, {
  57                 name = "muc#roomconfig_membersonly";
  58                 type = "boolean";
  59                 label = "Make Room Members-Only?";
  60                 value = get_members_only(event.room);
  61         });
  62 end);
  63
  64 module:hook("muc-config-submitted/muc#roomconfig_membersonly", function(event)
  65         if set_members_only(event.room, event.value) then
  66                 event.status_codes["104"] = true;
  67         end
  68 end);
  69
  70 -- No affiliation => role of "none"
  71 module:hook("muc-get-default-role", function(event)
  72         if not event.affiliation and get_members_only(event.room) then
  73                 return false;
  74         end
  75 end);
  76
  77 -- registration required for entering members-only room
  78 module:hook("muc-occupant-pre-join", function(event)
  79         local room = event.room;
  80         if get_members_only(room) then
  81                 local stanza = event.stanza;
  82                 local affiliation = room:get_affiliation(stanza.attr.from);
  83                 if valid_affiliations[affiliation or "none"] <= valid_affiliations.none then
  84                         local reply = st.error_reply(stanza, "auth", "registration-required"):up();
  85                         reply.tags[1].attr.code = "407";
  86                         event.origin.send(reply:tag("x", {xmlns = "http://jabber.org/protocol/muc"}));
  87                         return true;
  88                 end
  89         end
  90 end, -5);
  91
  92 -- Invitation privileges in members-only rooms SHOULD be restricted to room admins;
  93 -- if a member without privileges to edit the member list attempts to invite another user
  94 -- the service SHOULD return a <forbidden/> error to the occupant
  95 module:hook("muc-pre-invite", function(event)
  96         local room = event.room;
  97         if get_members_only(room) then
  98                 local stanza = event.stanza;
  99                 local affiliation = room:get_affiliation(stanza.attr.from);
 100                 if valid_affiliations[affiliation or "none"] < valid_affiliations.admin then
 101                         event.origin.send(st.error_reply(stanza, "auth", "forbidden"));
 102                         return true;
 103                 end
 104         end
 105 end);
 106
 107 -- When an invite is sent; add an affiliation for the invitee
 108 module:hook("muc-invite", function(event)
 109         local room = event.room;
 110         if get_members_only(room) then
 111                 local stanza = event.stanza;
 112                 local invitee = stanza.attr.to;
 113                 local affiliation = room:get_affiliation(invitee);
 114                 if valid_affiliations[affiliation or "none"] <= valid_affiliations.none then
 115                         local from = stanza:get_child("x", "http://jabber.org/protocol/muc#user")
 116                                 :get_child("invite").attr.from;
 117                         module:log("debug", "%s invited %s into members only room %s, granting membership",
 118                                 from, invitee, room.jid);
 119                         -- This might fail; ignore for now
 120                         room:set_affiliation(from, invitee, "member", "Invited by " .. from);
 121                 end
 122         end
 123 end);
 124
 125 return {
 126         get = get_members_only;
 127         set = set_members_only;
 128 };