prosodyctl: remove unused one-letter loop variable [luacheck]

[prosody.git] / plugins / mod_blocklist.lua

diff --git a/plugins/mod_blocklist.lua b/plugins/mod_blocklist.lua
index c5d0a6bbf1a24bcc7ff92b877ddd4db2c23d9f3b..8efbfd965295ab0ad250692831178e6112b1a9e2 100644 (file)
--- a/plugins/mod_blocklist.lua
+++ b/plugins/mod_blocklist.lua
@@ -1,7 +1,7 @@
 -- Prosody IM
 -- Copyright (C) 2009-2010 Matthew Wild
 -- Copyright (C) 2009-2010 Waqas Hussain
--- Copyright (C) 2014 Kim Alvefur
+-- Copyright (C) 2014-2015 Kim Alvefur
 --
 -- This project is MIT/X11 licensed. Please see the
 -- COPYING file in the source package for more information.
@@ -10,7 +10,11 @@
 --
 
 local user_exists = require"core.usermanager".user_exists;
-local is_contact_subscribed = require"core.rostermanager".is_contact_subscribed;
+local rostermanager = require"core.rostermanager";
+local is_contact_subscribed = rostermanager.is_contact_subscribed;
+local is_contact_pending_in = rostermanager.is_contact_pending_in;
+local load_roster = rostermanager.load_roster;
+local save_roster = rostermanager.save_roster;
 local st = require"util.stanza";
 local st_error_reply = st.error_reply;
 local jid_prep = require"util.jid".prep;
@@ -117,6 +121,7 @@ local function edit_blocklist(event)
        local origin, stanza = event.origin, event.stanza;
        local username = origin.username;
        local action = stanza.tags[1]; -- "block" or "unblock"
+       local is_blocking = action.name == "block" or nil; -- nil if unblocking
        local new = {}; -- JIDs to block depending or unblock on action
 
        -- XEP-0191 sayeth:
@@ -124,7 +129,11 @@ local function edit_blocklist(event)
        -- > server MUST send unavailable presence information to the contact (but
        -- > only if the contact is allowed to receive presence notifications [...]
        -- So contacts we need to do that for are added to the set below.
-       local send_unavailable = {};
+       local send_unavailable = is_blocking and {};
+
+       -- Because blocking someone currently also blocks the ability to reject
+       -- subscription requests, we'll preemptively reject such
+       local remove_pending = is_blocking and {};
 
        for item in action:childtags("item") do
                local jid = jid_prep(item.attr.jid);
@@ -134,13 +143,15 @@ local function edit_blocklist(event)
                end
                item.attr.jid = jid; -- echo back prepped
                new[jid] = true;
-               if is_contact_subscribed(username, module.host, jid) then
-                       send_unavailable[jid] = true;
+               if is_blocking then
+                       if is_contact_subscribed(username, module.host, jid) then
+                               send_unavailable[jid] = true;
+                       elseif is_contact_pending_in(username, module.host, jid) then
+                               remove_pending[jid] = true;
+                       end
                end
        end
 
-       local is_blocking = action.name == "block" or nil; -- nil if unblocking
-
        if is_blocking and not next(new) then
                -- <block/> element does not contain at least one <item/> child element
                origin.send(st_error_reply(stanza, "modify", "bad-request"));
@@ -180,6 +191,15 @@ local function edit_blocklist(event)
                                end
                        end
                end
+
+               if next(remove_pending) then
+                       local roster = load_roster(username, module.host);
+                       for jid in pairs(remove_pending) do
+                               roster[false].pending[jid] = nil;
+                       end
+                       save_roster(username, module.host, roster);
+                       -- Not much we can do about save failing here
+               end
        end
 
        local blocklist_push = st.iq({ type = "set", id = "blocklist-push" })
@@ -201,7 +221,7 @@ module:hook("iq-set/self/urn:xmpp:blocking:unblock", edit_blocklist);
 -- Cache invalidation, solved!
 module:hook_global("user-deleted", function (event)
        if event.host == module.host then
-               cache:set(event.username, nil);
+               cache2:set(event.username, nil);
                cache[event.username] = nil;
        end
 end);