Software / code / prosody-modules
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 | 194:58c37562c67d |
| children | 196: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;
