Fri, 17 Dec 2010 17:28:05 +0100
util.stanza: Iterate on childtags instead of all childs.
1523
841d61be198f
Remove version number from copyright headers
Matthew Wild <mwild1@gmail.com>
parents:
894
diff
changeset
|
1 | -- Prosody IM |
2923
b7049746bd29
Update copyright headers for 2010
Matthew Wild <mwild1@gmail.com>
parents:
1523
diff
changeset
|
2 | -- Copyright (C) 2008-2010 Matthew Wild |
b7049746bd29
Update copyright headers for 2010
Matthew Wild <mwild1@gmail.com>
parents:
1523
diff
changeset
|
3 | -- Copyright (C) 2008-2010 Waqas Hussain |
519
cccd610a0ef9
Insert copyright/license headers
Matthew Wild <mwild1@gmail.com>
parents:
438
diff
changeset
|
4 | -- |
758 | 5 | -- This project is MIT/X11 licensed. Please see the |
6 | -- COPYING file in the source package for more information. | |
519
cccd610a0ef9
Insert copyright/license headers
Matthew Wild <mwild1@gmail.com>
parents:
438
diff
changeset
|
7 | -- |
cccd610a0ef9
Insert copyright/license headers
Matthew Wild <mwild1@gmail.com>
parents:
438
diff
changeset
|
8 | |
cccd610a0ef9
Insert copyright/license headers
Matthew Wild <mwild1@gmail.com>
parents:
438
diff
changeset
|
9 | |
185
a67c88ce1c6a
Added support for XEP-0049: Private XML Storage (mod_private)
Waqas Hussain <waqas20@gmail.com>
parents:
diff
changeset
|
10 | local st = require "util.stanza" |
a67c88ce1c6a
Added support for XEP-0049: Private XML Storage (mod_private)
Waqas Hussain <waqas20@gmail.com>
parents:
diff
changeset
|
11 | |
a67c88ce1c6a
Added support for XEP-0049: Private XML Storage (mod_private)
Waqas Hussain <waqas20@gmail.com>
parents:
diff
changeset
|
12 | local jid_split = require "util.jid".split; |
a67c88ce1c6a
Added support for XEP-0049: Private XML Storage (mod_private)
Waqas Hussain <waqas20@gmail.com>
parents:
diff
changeset
|
13 | local datamanager = require "util.datamanager" |
a67c88ce1c6a
Added support for XEP-0049: Private XML Storage (mod_private)
Waqas Hussain <waqas20@gmail.com>
parents:
diff
changeset
|
14 | |
541
3521e0851c9e
Change modules to use the new add_feature module API method.
Waqas Hussain <waqas20@gmail.com>
parents:
519
diff
changeset
|
15 | module:add_feature("jabber:iq:private"); |
421
63be85693710
Modules now sending disco replies
Waqas Hussain <waqas20@gmail.com>
parents:
310
diff
changeset
|
16 | |
3522
4646b5b039ca
mod_private: Updated to use the new events API. Smaller, more robust.
Waqas Hussain <waqas20@gmail.com>
parents:
3114
diff
changeset
|
17 | module:hook("iq/self/jabber:iq:private:query", function(event) |
4646b5b039ca
mod_private: Updated to use the new events API. Smaller, more robust.
Waqas Hussain <waqas20@gmail.com>
parents:
3114
diff
changeset
|
18 | local origin, stanza = event.origin, event.stanza; |
4646b5b039ca
mod_private: Updated to use the new events API. Smaller, more robust.
Waqas Hussain <waqas20@gmail.com>
parents:
3114
diff
changeset
|
19 | local type = stanza.attr.type; |
4646b5b039ca
mod_private: Updated to use the new events API. Smaller, more robust.
Waqas Hussain <waqas20@gmail.com>
parents:
3114
diff
changeset
|
20 | local query = stanza.tags[1]; |
4646b5b039ca
mod_private: Updated to use the new events API. Smaller, more robust.
Waqas Hussain <waqas20@gmail.com>
parents:
3114
diff
changeset
|
21 | if #query.tags == 1 then |
4646b5b039ca
mod_private: Updated to use the new events API. Smaller, more robust.
Waqas Hussain <waqas20@gmail.com>
parents:
3114
diff
changeset
|
22 | local tag = query.tags[1]; |
4646b5b039ca
mod_private: Updated to use the new events API. Smaller, more robust.
Waqas Hussain <waqas20@gmail.com>
parents:
3114
diff
changeset
|
23 | local key = tag.name..":"..tag.attr.xmlns; |
4646b5b039ca
mod_private: Updated to use the new events API. Smaller, more robust.
Waqas Hussain <waqas20@gmail.com>
parents:
3114
diff
changeset
|
24 | local data, err = datamanager.load(origin.username, origin.host, "private"); |
4646b5b039ca
mod_private: Updated to use the new events API. Smaller, more robust.
Waqas Hussain <waqas20@gmail.com>
parents:
3114
diff
changeset
|
25 | if err then |
4646b5b039ca
mod_private: Updated to use the new events API. Smaller, more robust.
Waqas Hussain <waqas20@gmail.com>
parents:
3114
diff
changeset
|
26 | origin.send(st.error_reply(stanza, "wait", "internal-server-error")); |
4646b5b039ca
mod_private: Updated to use the new events API. Smaller, more robust.
Waqas Hussain <waqas20@gmail.com>
parents:
3114
diff
changeset
|
27 | return true; |
4646b5b039ca
mod_private: Updated to use the new events API. Smaller, more robust.
Waqas Hussain <waqas20@gmail.com>
parents:
3114
diff
changeset
|
28 | end |
4646b5b039ca
mod_private: Updated to use the new events API. Smaller, more robust.
Waqas Hussain <waqas20@gmail.com>
parents:
3114
diff
changeset
|
29 | if stanza.attr.type == "get" then |
4646b5b039ca
mod_private: Updated to use the new events API. Smaller, more robust.
Waqas Hussain <waqas20@gmail.com>
parents:
3114
diff
changeset
|
30 | if data and data[key] then |
4646b5b039ca
mod_private: Updated to use the new events API. Smaller, more robust.
Waqas Hussain <waqas20@gmail.com>
parents:
3114
diff
changeset
|
31 | origin.send(st.reply(stanza):tag("query", {xmlns = "jabber:iq:private"}):add_child(st.deserialize(data[key]))); |
185
a67c88ce1c6a
Added support for XEP-0049: Private XML Storage (mod_private)
Waqas Hussain <waqas20@gmail.com>
parents:
diff
changeset
|
32 | else |
3522
4646b5b039ca
mod_private: Updated to use the new events API. Smaller, more robust.
Waqas Hussain <waqas20@gmail.com>
parents:
3114
diff
changeset
|
33 | origin.send(st.reply(stanza):add_child(stanza.tags[1])); |
4646b5b039ca
mod_private: Updated to use the new events API. Smaller, more robust.
Waqas Hussain <waqas20@gmail.com>
parents:
3114
diff
changeset
|
34 | end |
4646b5b039ca
mod_private: Updated to use the new events API. Smaller, more robust.
Waqas Hussain <waqas20@gmail.com>
parents:
3114
diff
changeset
|
35 | else -- set |
4646b5b039ca
mod_private: Updated to use the new events API. Smaller, more robust.
Waqas Hussain <waqas20@gmail.com>
parents:
3114
diff
changeset
|
36 | if not data then data = {}; end; |
4646b5b039ca
mod_private: Updated to use the new events API. Smaller, more robust.
Waqas Hussain <waqas20@gmail.com>
parents:
3114
diff
changeset
|
37 | if #tag == 0 then |
4646b5b039ca
mod_private: Updated to use the new events API. Smaller, more robust.
Waqas Hussain <waqas20@gmail.com>
parents:
3114
diff
changeset
|
38 | data[key] = nil; |
4646b5b039ca
mod_private: Updated to use the new events API. Smaller, more robust.
Waqas Hussain <waqas20@gmail.com>
parents:
3114
diff
changeset
|
39 | else |
4646b5b039ca
mod_private: Updated to use the new events API. Smaller, more robust.
Waqas Hussain <waqas20@gmail.com>
parents:
3114
diff
changeset
|
40 | data[key] = st.preserialize(tag); |
4646b5b039ca
mod_private: Updated to use the new events API. Smaller, more robust.
Waqas Hussain <waqas20@gmail.com>
parents:
3114
diff
changeset
|
41 | end |
4646b5b039ca
mod_private: Updated to use the new events API. Smaller, more robust.
Waqas Hussain <waqas20@gmail.com>
parents:
3114
diff
changeset
|
42 | -- TODO delete datastore if empty |
4646b5b039ca
mod_private: Updated to use the new events API. Smaller, more robust.
Waqas Hussain <waqas20@gmail.com>
parents:
3114
diff
changeset
|
43 | if datamanager.store(origin.username, origin.host, "private", data) then |
4646b5b039ca
mod_private: Updated to use the new events API. Smaller, more robust.
Waqas Hussain <waqas20@gmail.com>
parents:
3114
diff
changeset
|
44 | origin.send(st.reply(stanza)); |
4646b5b039ca
mod_private: Updated to use the new events API. Smaller, more robust.
Waqas Hussain <waqas20@gmail.com>
parents:
3114
diff
changeset
|
45 | else |
4646b5b039ca
mod_private: Updated to use the new events API. Smaller, more robust.
Waqas Hussain <waqas20@gmail.com>
parents:
3114
diff
changeset
|
46 | origin.send(st.error_reply(stanza, "wait", "internal-server-error")); |
185
a67c88ce1c6a
Added support for XEP-0049: Private XML Storage (mod_private)
Waqas Hussain <waqas20@gmail.com>
parents:
diff
changeset
|
47 | end |
a67c88ce1c6a
Added support for XEP-0049: Private XML Storage (mod_private)
Waqas Hussain <waqas20@gmail.com>
parents:
diff
changeset
|
48 | end |
3522
4646b5b039ca
mod_private: Updated to use the new events API. Smaller, more robust.
Waqas Hussain <waqas20@gmail.com>
parents:
3114
diff
changeset
|
49 | else |
4646b5b039ca
mod_private: Updated to use the new events API. Smaller, more robust.
Waqas Hussain <waqas20@gmail.com>
parents:
3114
diff
changeset
|
50 | origin.send(st.error_reply(stanza, "modify", "bad-format")); |
4646b5b039ca
mod_private: Updated to use the new events API. Smaller, more robust.
Waqas Hussain <waqas20@gmail.com>
parents:
3114
diff
changeset
|
51 | end |
4646b5b039ca
mod_private: Updated to use the new events API. Smaller, more robust.
Waqas Hussain <waqas20@gmail.com>
parents:
3114
diff
changeset
|
52 | return true; |
4646b5b039ca
mod_private: Updated to use the new events API. Smaller, more robust.
Waqas Hussain <waqas20@gmail.com>
parents:
3114
diff
changeset
|
53 | end); |