| 1 | -- Prosody IM |
| 2 | -- Copyright (C) 2008-2010 Matthew Wild |
| 3 | -- Copyright (C) 2008-2010 Waqas Hussain |
| 4 | -- |
| 5 | -- This project is MIT/X11 licensed. Please see the |
| 6 | -- COPYING file in the source package for more information. |
| 7 | -- |
| 8 | |
| 9 | |
| 10 | local st = require "util.stanza" |
| 11 | |
| 12 | local jid_split = require "util.jid".split; |
| 13 | local datamanager = require "util.datamanager" |
| 14 | |
| 15 | module:add_feature("jabber:iq:private"); |
| 16 | |
| 17 | module:hook("iq/self/jabber:iq:private:query", function(event) |
| 18 | local origin, stanza = event.origin, event.stanza; |
| 19 | local type = stanza.attr.type; |
| 20 | local query = stanza.tags[1]; |
| 21 | if #query.tags == 1 then |
| 22 | local tag = query.tags[1]; |
| 23 | local key = tag.name..":"..tag.attr.xmlns; |
| 24 | local data, err = datamanager.load(origin.username, origin.host, "private"); |
| 25 | if err then |
| 26 | origin.send(st.error_reply(stanza, "wait", "internal-server-error")); |
| 27 | return true; |
| 28 | end |
| 29 | if stanza.attr.type == "get" then |
| 30 | if data and data[key] then |
| 31 | origin.send(st.reply(stanza):tag("query", {xmlns = "jabber:iq:private"}):add_child(st.deserialize(data[key]))); |
| 32 | else |
| 33 | origin.send(st.reply(stanza):add_child(stanza.tags[1])); |
| 34 | end |
| 35 | else -- set |
| 36 | if not data then data = {}; end; |
| 37 | if #tag == 0 then |
| 38 | data[key] = nil; |
| 39 | else |
| 40 | data[key] = st.preserialize(tag); |
| 41 | end |
| 42 | -- TODO delete datastore if empty |
| 43 | if datamanager.store(origin.username, origin.host, "private", data) then |
| 44 | origin.send(st.reply(stanza)); |
| 45 | else |
| 46 | origin.send(st.error_reply(stanza, "wait", "internal-server-error")); |
| 47 | end |
| 48 | end |
| 49 | else |
| 50 | origin.send(st.error_reply(stanza, "modify", "bad-format")); |
| 51 | end |
| 52 | return true; |
| 53 | end); |