mod_admin_telnet: Add server:memory() command to view details of Prosody's memory...
[prosody.git] / plugins / mod_disco.lua
index f403ccb2288cf0f5b1730b0873cf10c16177da6a..72c9a34c85ca87ec954e2e1c0dec6ebf45f4ad1c 100644 (file)
@@ -6,7 +6,7 @@
 -- COPYING file in the source package for more information.
 --
 
-local componentmanager_get_children = require "core.componentmanager".get_children;
+local get_children = require "core.hostmanager".get_children;
 local is_contact_subscribed = require "core.rostermanager".is_contact_subscribed;
 local jid_split = require "util.jid".split;
 local jid_bare = require "util.jid".bare;
@@ -32,35 +32,75 @@ do -- validate disco_items
        end
 end
 
-module:add_identity("server", "im", "Prosody"); -- FIXME should be in the non-existing mod_router
+module:add_identity("server", "im", module:get_option_string("name", "Prosody")); -- FIXME should be in the non-existing mod_router
 module:add_feature("http://jabber.org/protocol/disco#info");
 module:add_feature("http://jabber.org/protocol/disco#items");
 
--- Handle disco requests to the server
-
-local function build_server_disco_info(stanza)
+-- Generate and cache disco result and caps hash
+local _cached_server_disco_info, _cached_server_caps_feature, _cached_server_caps_hash;
+local function build_server_disco_info()
+       local query = st.stanza("query", { xmlns = "http://jabber.org/protocol/disco#info" });
        local done = {};
        for _,identity in ipairs(module:get_host_items("identity")) do
                local identity_s = identity.category.."\0"..identity.type;
                if not done[identity_s] then
-                       stanza:tag("identity", identity):up();
+                       query:tag("identity", identity):up();
                        done[identity_s] = true;
                end
        end
        for _,feature in ipairs(module:get_host_items("feature")) do
                if not done[feature] then
-                       stanza:tag("feature", {var=feature}):up();
+                       query:tag("feature", {var=feature}):up();
                        done[feature] = true;
                end
        end
+       for _,extension in ipairs(module:get_host_items("extension")) do
+               if not done[extension] then
+                       query:add_child(extension);
+                       done[extension] = true;
+               end
+       end
+       _cached_server_disco_info = query;
+       _cached_server_caps_hash = calculate_hash(query);
+       _cached_server_caps_feature = st.stanza("c", {
+               xmlns = "http://jabber.org/protocol/caps";
+               hash = "sha-1";
+               node = "http://prosody.im";
+               ver = _cached_server_caps_hash;
+       });
+end
+local function clear_disco_cache()
+       _cached_server_disco_info, _cached_server_caps_feature, _cached_server_caps_hash = nil, nil, nil;
+end
+local function get_server_disco_info()
+       if not _cached_server_disco_info then build_server_disco_info(); end
+       return _cached_server_disco_info;
+end
+local function get_server_caps_feature()
+       if not _cached_server_caps_feature then build_server_disco_info(); end
+       return _cached_server_caps_feature;
+end
+local function get_server_caps_hash()
+       if not _cached_server_caps_hash then build_server_disco_info(); end
+       return _cached_server_caps_hash;
 end
+
+module:hook("item-added/identity", clear_disco_cache);
+module:hook("item-added/feature", clear_disco_cache);
+module:hook("item-added/extension", clear_disco_cache);
+module:hook("item-removed/identity", clear_disco_cache);
+module:hook("item-removed/feature", clear_disco_cache);
+module:hook("item-removed/extension", clear_disco_cache);
+
+-- Handle disco requests to the server
 module:hook("iq/host/http://jabber.org/protocol/disco#info:query", function(event)
        local origin, stanza = event.origin, event.stanza;
        if stanza.attr.type ~= "get" then return; end
        local node = stanza.tags[1].attr.node;
-       if node and node ~= "" then return; end -- TODO fire event?
-       local reply = st.reply(stanza):query("http://jabber.org/protocol/disco#info");
-       build_server_disco_info(reply);
+       if node and node ~= "" and node ~= "http://prosody.im#"..get_server_caps_hash() then return; end -- TODO fire event?
+       local reply_query = get_server_disco_info();
+       reply_query.node = node;
+       local reply = st.reply(stanza):add_child(reply_query);
        origin.send(reply);
        return true;
 end);
@@ -71,8 +111,8 @@ module:hook("iq/host/http://jabber.org/protocol/disco#items:query", function(eve
        if node and node ~= "" then return; end -- TODO fire event?
 
        local reply = st.reply(stanza):query("http://jabber.org/protocol/disco#items");
-       for jid in pairs(componentmanager_get_children(module.host)) do
-               reply:tag("item", {jid = jid}):up();
+       for jid, name in pairs(get_children(module.host)) do
+               reply:tag("item", {jid = jid, name = name~=true and name or nil}):up();
        end
        for _, item in ipairs(disco_items) do
                reply:tag("item", {jid=item[1], name=item[2]}):up();
@@ -81,26 +121,10 @@ module:hook("iq/host/http://jabber.org/protocol/disco#items:query", function(eve
        return true;
 end);
 
--- Server caps hash calculation
-local caps_hash_feature;
-
-local function recalculate_server_caps()
-       local caps_hash = calculate_hash(st.stanza());
-       caps_hash_feature = st.stanza("c", {
-               xmlns = "http://jabber.org/protocol/caps";
-               hash = "sha-1";
-               node = "http://prosody.im";
-               ver = caps_hash;
-       });
-end
-recalculate_server_caps();
-
-module:hook("item-added/identity", recalculate_server_caps);
-module:hook("item-added/feature", recalculate_server_caps);
-
+-- Handle caps stream feature
 module:hook("stream-features", function (event)
-       if caps_hash_feature then
-               event.features:add_child(caps_hash_feature);
+       if event.origin.type == "c2s" then
+               event.features:add_child(get_server_caps_feature());
        end
 end);
 
@@ -114,7 +138,7 @@ module:hook("iq/bare/http://jabber.org/protocol/disco#info:query", function(even
        if not stanza.attr.to or is_contact_subscribed(username, module.host, jid_bare(stanza.attr.from)) then
                local reply = st.reply(stanza):tag('query', {xmlns='http://jabber.org/protocol/disco#info'});
                if not reply.attr.from then reply.attr.from = origin.username.."@"..origin.host; end -- COMPAT To satisfy Psi when querying own account
-               module:fire_event("account-disco-info", { session = origin, stanza = reply });
+               module:fire_event("account-disco-info", { origin = origin, stanza = reply });
                origin.send(reply);
                return true;
        end
@@ -128,7 +152,7 @@ module:hook("iq/bare/http://jabber.org/protocol/disco#items:query", function(eve
        if not stanza.attr.to or is_contact_subscribed(username, module.host, jid_bare(stanza.attr.from)) then
                local reply = st.reply(stanza):tag('query', {xmlns='http://jabber.org/protocol/disco#items'});
                if not reply.attr.from then reply.attr.from = origin.username.."@"..origin.host; end -- COMPAT To satisfy Psi when querying own account
-               module:fire_event("account-disco-items", { session = origin, stanza = reply });
+               module:fire_event("account-disco-items", { origin = origin, stanza = reply });
                origin.send(reply);
                return true;
        end