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         return true;
  48 end
  49
  50 module:hook("muc-disco#info", function(event)
  51         event.reply:tag("feature", {var = get_members_only(event.room) and "muc_membersonly" or "muc_open"}):up();
  52 end);
  53
  54 module:hook("muc-config-form", function(event)
  55         table.insert(event.form, {
  56                 name = "muc#roomconfig_membersonly";
  57                 type = "boolean";
  58                 label = "Make Room Members-Only?";
  59                 value = get_members_only(event.room);
  60         });
  61 end);
  62
  63 module:hook("muc-config-submitted/muc#roomconfig_membersonly", function(event)
  64         if set_members_only(event.room, event.value) then
  65                 event.status_codes["104"] = true;
  66         end
  67 end);
  68
  69 -- No affiliation => role of "none"
  70 module:hook("muc-get-default-role", function(event)
  71         if not event.affiliation and get_members_only(event.room) then
  72                 return false;
  73         end
  74 end);
  75
  76 -- registration required for entering members-only room
  77 module:hook("muc-occupant-pre-join", function(event)
  78         local room = event.room;
  79         if get_members_only(room) then
  80                 local stanza = event.stanza;
  81                 local affiliation = room:get_affiliation(stanza.attr.from);
  82                 if valid_affiliations[affiliation or "none"] <= valid_affiliations.none then
  83                         local reply = st.error_reply(stanza, "auth", "registration-required"):up();
  84                         reply.tags[1].attr.code = "407";
  85                         event.origin.send(reply:tag("x", {xmlns = "http://jabber.org/protocol/muc"}));
  86                         return true;
  87                 end
  88         end
  89 end, -5);
  90
  91 -- Invitation privileges in members-only rooms SHOULD be restricted to room admins;
  92 -- if a member without privileges to edit the member list attempts to invite another user
  93 -- the service SHOULD return a <forbidden/> error to the occupant
  94 module:hook("muc-pre-invite", function(event)
  95         local room = event.room;
  96         if get_members_only(room) then
  97                 local stanza = event.stanza;
  98                 local affiliation = room:get_affiliation(stanza.attr.from);
  99                 if valid_affiliations[affiliation or "none"] < valid_affiliations.admin then
 100                         event.origin.send(st.error_reply(stanza, "auth", "forbidden"));
 101                         return true;
 102                 end
 103         end
 104 end);
 105
 106 -- When an invite is sent; add an affiliation for the invitee
 107 module:hook("muc-invite", function(event)
 108         local room = event.room;
 109         if get_members_only(room) then
 110                 local stanza = event.stanza;
 111                 local invitee = stanza.attr.to;
 112                 local affiliation = room:get_affiliation(invitee);
 113                 if valid_affiliations[affiliation or "none"] <= valid_affiliations.none then
 114                         local from = stanza:get_child("x", "http://jabber.org/protocol/muc#user")
 115                                 :get_child("invite").attr.from;
 116                         module:log("debug", "%s invited %s into members only room %s, granting membership",
 117                                 from, invitee, room.jid);
 118                         -- This might fail; ignore for now
 119                         room:set_affiliation(from, invitee, "member", "Invited by " .. from);
 120                         room:save();
 121                 end
 122         end
 123 end);
 124
 125 return {
 126         get = get_members_only;
 127         set = set_members_only;
 128 };