changeset 195:4d9ed6374a1f

mod_archive: XEP-0059: Result Set Management - Limiting the Number of Items; Paging Forwards Through a Result Set; Getting the Item Count
author shinysky<shinysky1986(AT)gmail.com>
date Tue, 06 Jul 2010 00:18:12 +0800
parents 58c37562c67d
children a1c2677257da
files mod_archive/mod_archive.lua
diffstat 1 files changed, 30 insertions(+), 2 deletions(-) [+]
line wrap: on
line diff
--- a/mod_archive/mod_archive.lua	Mon Jul 05 11:12:36 2010 +0200
+++ b/mod_archive/mod_archive.lua	Tue Jul 06 00:18:12 2010 +0800
@@ -18,6 +18,7 @@
 module:add_feature("urn:xmpp:archive:manage");
 module:add_feature("urn:xmpp:archive:manual");
 module:add_feature("urn:xmpp:archive:pref");
+module:add_feature("http://jabber.org/protocol/rsm");
 
 ------------------------------------------------------------
 -- Utils
@@ -73,6 +74,10 @@
     return save;
 end 
 
+local function gen_uid(c)
+    return c.attr["start"] .. c.attr["with"];
+end
+
 ------------------------------------------------------------
 -- Preferences
 ------------------------------------------------------------
@@ -351,14 +356,37 @@
     end
     local reply = st.reply(stanza):tag('list', {xmlns='urn:xmpp:archive'});
     if table.getn(resset) > 0 then
-        local max = tonumber(elem.tags[1].tags[1]:get_text());
+        local max = elem.tags[1]:child_with_name("max");
+        if max then
+            max = tonumber(max:get_text());
+        else max = 100; end
+        local after = elem.tags[1]:child_with_name("after");
+        -- local before = elem.tags[1]:child_with_name("before");
+        -- local index = elem.tags[1]:child_with_name("index");
+        if after then after = after:get_text(); end
+        local found = false;
+        local first, last = nil, nil;
         -- Assuming result set is sorted.
         for i, c in ipairs(resset) do
-            if i <= max then
+            if after and not found then
+                if gen_uid(c) == after then
+                    found = true;
+                end
+            elseif max > 0 then
+                if not first then first = i; end
+                last = i;
                 local chat = st.stanza('chat', c.attr);
                 reply:add_child(chat);
+                max = max - 1;
             else break; end
         end
+        local set = st.stanza('set', {xmlns='http://jabber.org/protocol/rsm'});
+        if first then
+            set:tag('first', {index=first-1}):text(gen_uid(resset[first])):up()
+               :tag('last'):text(gen_uid(resset[last])):up();
+        end
+        set:tag('count'):text(tostring(table.getn(resset))):up();
+        reply:add_child(set);
     end
     origin.send(reply);
     return true;