mod_uptime: Fix bad uptime if module is loaded at startup

[prosody.git] / plugins / mod_pep.lua

diff --git a/plugins/mod_pep.lua b/plugins/mod_pep.lua
index 3d0b6ad847953e1157e8399feb0dfa51d4f5cc95..8cc4aedfb6004a5c85417802c8608273b8a2fc8a 100644 (file)
--- a/plugins/mod_pep.lua
+++ b/plugins/mod_pep.lua
@@ -1,3 +1,11 @@
+-- Prosody IM
+-- Copyright (C) 2008-2009 Matthew Wild
+-- Copyright (C) 2008-2009 Waqas Hussain
+-- 
+-- This project is MIT/X11 licensed. Please see the
+-- COPYING file in the source package for more information.
+--
+
 
 local jid_bare = require "util.jid".bare;
 local jid_split = require "util.jid".split;
@@ -33,8 +41,10 @@ local function publish(session, node, item)
        -- store for the future
        local user_data = data[bare];
        if disable then
-               if user_data then user_data[node] = nil; end
-               if not next(user_data) then data[bare] = nil; end
+               if user_data then
+                       user_data[node] = nil;
+                       if not next(user_data) then data[bare] = nil; end
+               end
        else
                if not user_data then user_data = {}; data[bare] = user_data; end
                user_data[node] = stanza;
@@ -48,6 +58,19 @@ local function publish(session, node, item)
                end
        end
 end
+local function publish_all(user, recipient, session)
+       local d = data[user];
+       local notify = recipients[user] and recipients[user][recipient];
+       if d and notify then
+               for node in pairs(notify) do
+                       local message = d[node];
+                       if message then
+                               message.attr.to = recipient;
+                               session.send(message);
+                       end
+               end
+       end
+end
 
 local function get_caps_hash_from_presence(stanza, current)
        local t = stanza.attr.type;
@@ -85,11 +108,16 @@ module:hook("presence/bare", function(event)
                        if recipients[user] then recipients[user][recipient] = nil; end
                else
                        recipients[user] = recipients[user] or {};
-                       recipients[user][recipient] = hash;
-                       origin.send(
-                               st.stanza("iq", {from=stanza.attr.to, to=stanza.attr.from, id="disco", type="get"})
-                                       :query("http://jabber.org/protocol/disco#info")
-                       );
+                       if hash_map[hash] then
+                               recipients[user][recipient] = hash_map[hash];
+                               publish_all(user, recipient, origin);
+                       else
+                               recipients[user][recipient] = hash;
+                               origin.send(
+                                       st.stanza("iq", {from=stanza.attr.to, to=stanza.attr.from, id="disco", type="get"})
+                                               :query("http://jabber.org/protocol/disco#info")
+                               );
+                       end
                end
        end
 end, 10);
@@ -173,22 +201,14 @@ module:hook("iq/bare/disco", function(event)
                        local notify = {};
                        for _, feature in pairs(disco.tags) do
                                if feature.name == "feature" and feature.attr.var then
-                                       local nfeature = feature.attr.var:match("^(.*)+notify$");
+                                       local nfeature = feature.attr.var:match("^(.*)%+notify$");
                                        if nfeature then notify[nfeature] = true; end
                                end
                        end
                        hash_map[ver] = notify; -- update hash map
                        recipients[user][contact] = notify; -- set recipient's data to calculated data
                        -- send messages to recipient
-                       local d = data[user];
-                       if d then
-                               for node, message in pairs(notify) do
-                                       if d[node] then
-                                               message.attr.to = stanza.attr.from;
-                                               session.send(message);
-                                       end
-                               end
-                       end
+                       publish_all(user, contact, session);
                end
        end
 end);