Fri, 20 Apr 2012 01:39:31 +0200
plugins.archive: Move namespaces to the top
291
b78137f5118e
plugins.archive: Add a header comment
Kim Alvefur <zash@zash.se>
parents:
275
diff
changeset
|
1 | -- This implements XEP-0313: Message Archive Management |
b78137f5118e
plugins.archive: Add a header comment
Kim Alvefur <zash@zash.se>
parents:
275
diff
changeset
|
2 | -- http://xmpp.org/extensions/xep-0313.html |
b78137f5118e
plugins.archive: Add a header comment
Kim Alvefur <zash@zash.se>
parents:
275
diff
changeset
|
3 | -- (ie not XEP-0136) |
b78137f5118e
plugins.archive: Add a header comment
Kim Alvefur <zash@zash.se>
parents:
275
diff
changeset
|
4 | |
292
b674f5ca85d4
plugins.archive: Add local verse
Kim Alvefur <zash@zash.se>
parents:
291
diff
changeset
|
5 | local verse = require "verse"; |
293
65fb9ae79014
plugins.archive: Import util.stanza directly and use it
Kim Alvefur <zash@zash.se>
parents:
292
diff
changeset
|
6 | local st = require "util.stanza"; |
273
c5b7a4c717a6
plugins.archive: Experimental implementation of the Message Archive Management ProtoXEP
Kim Alvefur <zash@zash.se>
parents:
diff
changeset
|
7 | local xmlns_mam = "urn:xmpp:mam:tmp" |
295
5f7203bf1dae
plugins.archive: Move namespaces to the top
Kim Alvefur <zash@zash.se>
parents:
294
diff
changeset
|
8 | local xmlns_forward = "urn:xmpp:forward:0"; |
5f7203bf1dae
plugins.archive: Move namespaces to the top
Kim Alvefur <zash@zash.se>
parents:
294
diff
changeset
|
9 | local xmlns_delay = "urn:xmpp:delay"; |
273
c5b7a4c717a6
plugins.archive: Experimental implementation of the Message Archive Management ProtoXEP
Kim Alvefur <zash@zash.se>
parents:
diff
changeset
|
10 | local uuid = require "util.uuid".generate; |
c5b7a4c717a6
plugins.archive: Experimental implementation of the Message Archive Management ProtoXEP
Kim Alvefur <zash@zash.se>
parents:
diff
changeset
|
11 | |
c5b7a4c717a6
plugins.archive: Experimental implementation of the Message Archive Management ProtoXEP
Kim Alvefur <zash@zash.se>
parents:
diff
changeset
|
12 | function verse.plugins.archive(stream) |
c5b7a4c717a6
plugins.archive: Experimental implementation of the Message Archive Management ProtoXEP
Kim Alvefur <zash@zash.se>
parents:
diff
changeset
|
13 | function stream:query_archive(where, query_params, callback) |
c5b7a4c717a6
plugins.archive: Experimental implementation of the Message Archive Management ProtoXEP
Kim Alvefur <zash@zash.se>
parents:
diff
changeset
|
14 | local queryid = uuid(); |
293
65fb9ae79014
plugins.archive: Import util.stanza directly and use it
Kim Alvefur <zash@zash.se>
parents:
292
diff
changeset
|
15 | local query_st = st.iq{ type="get", to = where } |
273
c5b7a4c717a6
plugins.archive: Experimental implementation of the Message Archive Management ProtoXEP
Kim Alvefur <zash@zash.se>
parents:
diff
changeset
|
16 | :tag("query", { xmlns = xmlns_mam, queryid = queryid }); |
c5b7a4c717a6
plugins.archive: Experimental implementation of the Message Archive Management ProtoXEP
Kim Alvefur <zash@zash.se>
parents:
diff
changeset
|
17 | |
c5b7a4c717a6
plugins.archive: Experimental implementation of the Message Archive Management ProtoXEP
Kim Alvefur <zash@zash.se>
parents:
diff
changeset
|
18 | local params = { "with", "start", "end" }; |
c5b7a4c717a6
plugins.archive: Experimental implementation of the Message Archive Management ProtoXEP
Kim Alvefur <zash@zash.se>
parents:
diff
changeset
|
19 | local query_params = query_params or {}; |
c5b7a4c717a6
plugins.archive: Experimental implementation of the Message Archive Management ProtoXEP
Kim Alvefur <zash@zash.se>
parents:
diff
changeset
|
20 | for i=1,#params do |
c5b7a4c717a6
plugins.archive: Experimental implementation of the Message Archive Management ProtoXEP
Kim Alvefur <zash@zash.se>
parents:
diff
changeset
|
21 | local k = params[i]; |
c5b7a4c717a6
plugins.archive: Experimental implementation of the Message Archive Management ProtoXEP
Kim Alvefur <zash@zash.se>
parents:
diff
changeset
|
22 | if query_params[k] then |
275
c077b20ad59f
plugins.archive: Add missing :up()
Kim Alvefur <zash@zash.se>
parents:
273
diff
changeset
|
23 | query_st:tag(k):text(query_params[k]):up(); |
273
c5b7a4c717a6
plugins.archive: Experimental implementation of the Message Archive Management ProtoXEP
Kim Alvefur <zash@zash.se>
parents:
diff
changeset
|
24 | end |
c5b7a4c717a6
plugins.archive: Experimental implementation of the Message Archive Management ProtoXEP
Kim Alvefur <zash@zash.se>
parents:
diff
changeset
|
25 | end |
c5b7a4c717a6
plugins.archive: Experimental implementation of the Message Archive Management ProtoXEP
Kim Alvefur <zash@zash.se>
parents:
diff
changeset
|
26 | |
c5b7a4c717a6
plugins.archive: Experimental implementation of the Message Archive Management ProtoXEP
Kim Alvefur <zash@zash.se>
parents:
diff
changeset
|
27 | local results = {}; |
c5b7a4c717a6
plugins.archive: Experimental implementation of the Message Archive Management ProtoXEP
Kim Alvefur <zash@zash.se>
parents:
diff
changeset
|
28 | local function handle_archived_message(message) |
c5b7a4c717a6
plugins.archive: Experimental implementation of the Message Archive Management ProtoXEP
Kim Alvefur <zash@zash.se>
parents:
diff
changeset
|
29 | local result_tag = message:get_child("result", xmlns_mam); |
c5b7a4c717a6
plugins.archive: Experimental implementation of the Message Archive Management ProtoXEP
Kim Alvefur <zash@zash.se>
parents:
diff
changeset
|
30 | if result_tag and result_tag.attr.queryid == queryid then |
295
5f7203bf1dae
plugins.archive: Move namespaces to the top
Kim Alvefur <zash@zash.se>
parents:
294
diff
changeset
|
31 | local forwarded = message:get_child("forwarded", xmlns_forward); |
273
c5b7a4c717a6
plugins.archive: Experimental implementation of the Message Archive Management ProtoXEP
Kim Alvefur <zash@zash.se>
parents:
diff
changeset
|
32 | |
295
5f7203bf1dae
plugins.archive: Move namespaces to the top
Kim Alvefur <zash@zash.se>
parents:
294
diff
changeset
|
33 | local delay = forwarded:get_child("delay", xmlns_delay); |
273
c5b7a4c717a6
plugins.archive: Experimental implementation of the Message Archive Management ProtoXEP
Kim Alvefur <zash@zash.se>
parents:
diff
changeset
|
34 | local stamp = delay and delay.attr.stamp or nil; |
c5b7a4c717a6
plugins.archive: Experimental implementation of the Message Archive Management ProtoXEP
Kim Alvefur <zash@zash.se>
parents:
diff
changeset
|
35 | |
c5b7a4c717a6
plugins.archive: Experimental implementation of the Message Archive Management ProtoXEP
Kim Alvefur <zash@zash.se>
parents:
diff
changeset
|
36 | local message = forwarded:get_child("message", "jabber:client") |
c5b7a4c717a6
plugins.archive: Experimental implementation of the Message Archive Management ProtoXEP
Kim Alvefur <zash@zash.se>
parents:
diff
changeset
|
37 | |
c5b7a4c717a6
plugins.archive: Experimental implementation of the Message Archive Management ProtoXEP
Kim Alvefur <zash@zash.se>
parents:
diff
changeset
|
38 | results[#results+1] = { stamp = stamp, message = message }; |
c5b7a4c717a6
plugins.archive: Experimental implementation of the Message Archive Management ProtoXEP
Kim Alvefur <zash@zash.se>
parents:
diff
changeset
|
39 | return true |
c5b7a4c717a6
plugins.archive: Experimental implementation of the Message Archive Management ProtoXEP
Kim Alvefur <zash@zash.se>
parents:
diff
changeset
|
40 | end |
c5b7a4c717a6
plugins.archive: Experimental implementation of the Message Archive Management ProtoXEP
Kim Alvefur <zash@zash.se>
parents:
diff
changeset
|
41 | end |
c5b7a4c717a6
plugins.archive: Experimental implementation of the Message Archive Management ProtoXEP
Kim Alvefur <zash@zash.se>
parents:
diff
changeset
|
42 | |
c5b7a4c717a6
plugins.archive: Experimental implementation of the Message Archive Management ProtoXEP
Kim Alvefur <zash@zash.se>
parents:
diff
changeset
|
43 | self:hook("message", handle_archived_message, 1); |
c5b7a4c717a6
plugins.archive: Experimental implementation of the Message Archive Management ProtoXEP
Kim Alvefur <zash@zash.se>
parents:
diff
changeset
|
44 | self:send_iq(query_st, function(reply) |
c5b7a4c717a6
plugins.archive: Experimental implementation of the Message Archive Management ProtoXEP
Kim Alvefur <zash@zash.se>
parents:
diff
changeset
|
45 | self:unhook("message", handle_archived_message); |
c5b7a4c717a6
plugins.archive: Experimental implementation of the Message Archive Management ProtoXEP
Kim Alvefur <zash@zash.se>
parents:
diff
changeset
|
46 | callback(reply.attr.type == "result" and #results, results); |
c5b7a4c717a6
plugins.archive: Experimental implementation of the Message Archive Management ProtoXEP
Kim Alvefur <zash@zash.se>
parents:
diff
changeset
|
47 | return true |
c5b7a4c717a6
plugins.archive: Experimental implementation of the Message Archive Management ProtoXEP
Kim Alvefur <zash@zash.se>
parents:
diff
changeset
|
48 | end); |
c5b7a4c717a6
plugins.archive: Experimental implementation of the Message Archive Management ProtoXEP
Kim Alvefur <zash@zash.se>
parents:
diff
changeset
|
49 | end |
c5b7a4c717a6
plugins.archive: Experimental implementation of the Message Archive Management ProtoXEP
Kim Alvefur <zash@zash.se>
parents:
diff
changeset
|
50 | |
294
ac039aa3a4ef
plugins.archive: Implement archiving preferences
Kim Alvefur <zash@zash.se>
parents:
293
diff
changeset
|
51 | local default_attrs = { |
ac039aa3a4ef
plugins.archive: Implement archiving preferences
Kim Alvefur <zash@zash.se>
parents:
293
diff
changeset
|
52 | always = true, [true] = "always", |
ac039aa3a4ef
plugins.archive: Implement archiving preferences
Kim Alvefur <zash@zash.se>
parents:
293
diff
changeset
|
53 | never = false, [false] = "never", |
ac039aa3a4ef
plugins.archive: Implement archiving preferences
Kim Alvefur <zash@zash.se>
parents:
293
diff
changeset
|
54 | roster = "roster", |
ac039aa3a4ef
plugins.archive: Implement archiving preferences
Kim Alvefur <zash@zash.se>
parents:
293
diff
changeset
|
55 | } |
ac039aa3a4ef
plugins.archive: Implement archiving preferences
Kim Alvefur <zash@zash.se>
parents:
293
diff
changeset
|
56 | |
ac039aa3a4ef
plugins.archive: Implement archiving preferences
Kim Alvefur <zash@zash.se>
parents:
293
diff
changeset
|
57 | local function prefs_decode(stanza) -- from XML |
ac039aa3a4ef
plugins.archive: Implement archiving preferences
Kim Alvefur <zash@zash.se>
parents:
293
diff
changeset
|
58 | local prefs = {}; |
ac039aa3a4ef
plugins.archive: Implement archiving preferences
Kim Alvefur <zash@zash.se>
parents:
293
diff
changeset
|
59 | local default = stanza.attr.default; |
ac039aa3a4ef
plugins.archive: Implement archiving preferences
Kim Alvefur <zash@zash.se>
parents:
293
diff
changeset
|
60 | |
ac039aa3a4ef
plugins.archive: Implement archiving preferences
Kim Alvefur <zash@zash.se>
parents:
293
diff
changeset
|
61 | if default then |
ac039aa3a4ef
plugins.archive: Implement archiving preferences
Kim Alvefur <zash@zash.se>
parents:
293
diff
changeset
|
62 | prefs[false] = default_attrs[default]; |
ac039aa3a4ef
plugins.archive: Implement archiving preferences
Kim Alvefur <zash@zash.se>
parents:
293
diff
changeset
|
63 | end |
ac039aa3a4ef
plugins.archive: Implement archiving preferences
Kim Alvefur <zash@zash.se>
parents:
293
diff
changeset
|
64 | |
ac039aa3a4ef
plugins.archive: Implement archiving preferences
Kim Alvefur <zash@zash.se>
parents:
293
diff
changeset
|
65 | local always = stanza:get_child("always"); |
ac039aa3a4ef
plugins.archive: Implement archiving preferences
Kim Alvefur <zash@zash.se>
parents:
293
diff
changeset
|
66 | if always then |
ac039aa3a4ef
plugins.archive: Implement archiving preferences
Kim Alvefur <zash@zash.se>
parents:
293
diff
changeset
|
67 | for rule in always:childtags("jid") do |
ac039aa3a4ef
plugins.archive: Implement archiving preferences
Kim Alvefur <zash@zash.se>
parents:
293
diff
changeset
|
68 | local jid = rule:get_text(); |
ac039aa3a4ef
plugins.archive: Implement archiving preferences
Kim Alvefur <zash@zash.se>
parents:
293
diff
changeset
|
69 | prefs[jid] = true; |
ac039aa3a4ef
plugins.archive: Implement archiving preferences
Kim Alvefur <zash@zash.se>
parents:
293
diff
changeset
|
70 | end |
ac039aa3a4ef
plugins.archive: Implement archiving preferences
Kim Alvefur <zash@zash.se>
parents:
293
diff
changeset
|
71 | end |
ac039aa3a4ef
plugins.archive: Implement archiving preferences
Kim Alvefur <zash@zash.se>
parents:
293
diff
changeset
|
72 | |
ac039aa3a4ef
plugins.archive: Implement archiving preferences
Kim Alvefur <zash@zash.se>
parents:
293
diff
changeset
|
73 | local never = stanza:get_child("never"); |
ac039aa3a4ef
plugins.archive: Implement archiving preferences
Kim Alvefur <zash@zash.se>
parents:
293
diff
changeset
|
74 | if never then |
ac039aa3a4ef
plugins.archive: Implement archiving preferences
Kim Alvefur <zash@zash.se>
parents:
293
diff
changeset
|
75 | for rule in never:childtags("jid") do |
ac039aa3a4ef
plugins.archive: Implement archiving preferences
Kim Alvefur <zash@zash.se>
parents:
293
diff
changeset
|
76 | local jid = rule:get_text(); |
ac039aa3a4ef
plugins.archive: Implement archiving preferences
Kim Alvefur <zash@zash.se>
parents:
293
diff
changeset
|
77 | prefs[jid] = false; |
ac039aa3a4ef
plugins.archive: Implement archiving preferences
Kim Alvefur <zash@zash.se>
parents:
293
diff
changeset
|
78 | end |
ac039aa3a4ef
plugins.archive: Implement archiving preferences
Kim Alvefur <zash@zash.se>
parents:
293
diff
changeset
|
79 | end |
ac039aa3a4ef
plugins.archive: Implement archiving preferences
Kim Alvefur <zash@zash.se>
parents:
293
diff
changeset
|
80 | return prefs; |
ac039aa3a4ef
plugins.archive: Implement archiving preferences
Kim Alvefur <zash@zash.se>
parents:
293
diff
changeset
|
81 | end |
ac039aa3a4ef
plugins.archive: Implement archiving preferences
Kim Alvefur <zash@zash.se>
parents:
293
diff
changeset
|
82 | |
ac039aa3a4ef
plugins.archive: Implement archiving preferences
Kim Alvefur <zash@zash.se>
parents:
293
diff
changeset
|
83 | local function prefs_encode(prefs) -- into XML |
ac039aa3a4ef
plugins.archive: Implement archiving preferences
Kim Alvefur <zash@zash.se>
parents:
293
diff
changeset
|
84 | local default |
ac039aa3a4ef
plugins.archive: Implement archiving preferences
Kim Alvefur <zash@zash.se>
parents:
293
diff
changeset
|
85 | default, prefs[false] = prefs[false], nil; |
ac039aa3a4ef
plugins.archive: Implement archiving preferences
Kim Alvefur <zash@zash.se>
parents:
293
diff
changeset
|
86 | if default ~= nil then |
ac039aa3a4ef
plugins.archive: Implement archiving preferences
Kim Alvefur <zash@zash.se>
parents:
293
diff
changeset
|
87 | default = default_attrs[default]; |
ac039aa3a4ef
plugins.archive: Implement archiving preferences
Kim Alvefur <zash@zash.se>
parents:
293
diff
changeset
|
88 | end |
ac039aa3a4ef
plugins.archive: Implement archiving preferences
Kim Alvefur <zash@zash.se>
parents:
293
diff
changeset
|
89 | local reply = st.stanza("prefs", { xmlns = xmlns_mam, default = default }) |
ac039aa3a4ef
plugins.archive: Implement archiving preferences
Kim Alvefur <zash@zash.se>
parents:
293
diff
changeset
|
90 | local always = st.stanza("always"); |
ac039aa3a4ef
plugins.archive: Implement archiving preferences
Kim Alvefur <zash@zash.se>
parents:
293
diff
changeset
|
91 | local never = st.stanza("never"); |
ac039aa3a4ef
plugins.archive: Implement archiving preferences
Kim Alvefur <zash@zash.se>
parents:
293
diff
changeset
|
92 | for k,v in pairs(prefs) do |
ac039aa3a4ef
plugins.archive: Implement archiving preferences
Kim Alvefur <zash@zash.se>
parents:
293
diff
changeset
|
93 | (v and always or never):tag("jid"):text(k):up(); |
ac039aa3a4ef
plugins.archive: Implement archiving preferences
Kim Alvefur <zash@zash.se>
parents:
293
diff
changeset
|
94 | end |
ac039aa3a4ef
plugins.archive: Implement archiving preferences
Kim Alvefur <zash@zash.se>
parents:
293
diff
changeset
|
95 | return reply:add_child(always):add_child(never); |
ac039aa3a4ef
plugins.archive: Implement archiving preferences
Kim Alvefur <zash@zash.se>
parents:
293
diff
changeset
|
96 | end |
ac039aa3a4ef
plugins.archive: Implement archiving preferences
Kim Alvefur <zash@zash.se>
parents:
293
diff
changeset
|
97 | |
ac039aa3a4ef
plugins.archive: Implement archiving preferences
Kim Alvefur <zash@zash.se>
parents:
293
diff
changeset
|
98 | function stream:archive_prefs_get(callback) |
ac039aa3a4ef
plugins.archive: Implement archiving preferences
Kim Alvefur <zash@zash.se>
parents:
293
diff
changeset
|
99 | self:send_iq(st.iq{ type="get" }:tag("prefs", { xmlns = xmlns_mam }), |
ac039aa3a4ef
plugins.archive: Implement archiving preferences
Kim Alvefur <zash@zash.se>
parents:
293
diff
changeset
|
100 | function(result) |
ac039aa3a4ef
plugins.archive: Implement archiving preferences
Kim Alvefur <zash@zash.se>
parents:
293
diff
changeset
|
101 | if result and result.attr.type == "result" and result.tags[1] then |
ac039aa3a4ef
plugins.archive: Implement archiving preferences
Kim Alvefur <zash@zash.se>
parents:
293
diff
changeset
|
102 | local prefs = prefs_decode(result.tags[1]); |
ac039aa3a4ef
plugins.archive: Implement archiving preferences
Kim Alvefur <zash@zash.se>
parents:
293
diff
changeset
|
103 | callback(prefs, result); |
ac039aa3a4ef
plugins.archive: Implement archiving preferences
Kim Alvefur <zash@zash.se>
parents:
293
diff
changeset
|
104 | else |
ac039aa3a4ef
plugins.archive: Implement archiving preferences
Kim Alvefur <zash@zash.se>
parents:
293
diff
changeset
|
105 | callback(nil, result); |
ac039aa3a4ef
plugins.archive: Implement archiving preferences
Kim Alvefur <zash@zash.se>
parents:
293
diff
changeset
|
106 | end |
ac039aa3a4ef
plugins.archive: Implement archiving preferences
Kim Alvefur <zash@zash.se>
parents:
293
diff
changeset
|
107 | end); |
ac039aa3a4ef
plugins.archive: Implement archiving preferences
Kim Alvefur <zash@zash.se>
parents:
293
diff
changeset
|
108 | end |
ac039aa3a4ef
plugins.archive: Implement archiving preferences
Kim Alvefur <zash@zash.se>
parents:
293
diff
changeset
|
109 | |
ac039aa3a4ef
plugins.archive: Implement archiving preferences
Kim Alvefur <zash@zash.se>
parents:
293
diff
changeset
|
110 | function stream:archive_prefs_set(prefs, callback) |
ac039aa3a4ef
plugins.archive: Implement archiving preferences
Kim Alvefur <zash@zash.se>
parents:
293
diff
changeset
|
111 | self:send_iq(st.iq{ type="set" }:add_child(prefs_encode(prefs)), callback); |
ac039aa3a4ef
plugins.archive: Implement archiving preferences
Kim Alvefur <zash@zash.se>
parents:
293
diff
changeset
|
112 | end |
273
c5b7a4c717a6
plugins.archive: Experimental implementation of the Message Archive Management ProtoXEP
Kim Alvefur <zash@zash.se>
parents:
diff
changeset
|
113 | end |