Software /
code /
prosody-modules
Comparison
mod_vcard_command/mod_vcard_command.lua @ 3025:5b8983e100da
mod_vcard_command: a simplistic way to get and set vcard data for registered users
Use "prosodyctl get user@example.com" to print the XML to stdout, and
"prosodyctl set user@example.com /file/name" to set the vcard info from
an xml file.
author | Stefan `Sec` Zehl <sec@42.org> |
---|---|
date | Tue, 08 Oct 2013 16:38:40 +0000 |
child | 3026:b2c9b832612b |
comparison
equal
deleted
inserted
replaced
3024:f54c80404ad3 | 3025:5b8983e100da |
---|---|
1 ----------------------------------------------------------- | |
2 -- mod_vcard_command: Manage vcards through prosodyctl | |
3 -- version 0.02 | |
4 ----------------------------------------------------------- | |
5 -- Copyright (C) 2013 Stefan `Sec` Zehl | |
6 -- | |
7 -- This project is MIT/X11 licensed. Please see the | |
8 -- COPYING file in the source package for more information. | |
9 ----------------------------------------------------------- | |
10 | |
11 if not rawget(_G, "prosodyctl") then | |
12 module:log("error", "Do not load this module in Prosody"); | |
13 module.host = "*"; | |
14 return; | |
15 end | |
16 | |
17 | |
18 -- Workaround for lack of util.startup... | |
19 _G.bare_sessions = _G.bare_sessions or {}; | |
20 | |
21 local storagemanager = require "core.storagemanager"; | |
22 local datamanager = require "util.datamanager"; | |
23 local xml = require "util.xml"; | |
24 local jid = require "util.jid"; | |
25 local warn = prosodyctl.show_warning; | |
26 local st = require "util.stanza" | |
27 -- local vcards = module:open_store("vcard"); | |
28 | |
29 -- Print anything - including nested tables | |
30 function table_print (tt, indent, done) | |
31 done = done or {} | |
32 indent = indent or 0 | |
33 if type(tt) == "table" then | |
34 for key, value in pairs (tt) do | |
35 io.write(string.rep (" ", indent)) -- indent it | |
36 if type (value) == "table" and not done [value] then | |
37 done [value] = true | |
38 io.write(string.format("[%s] => table\n", tostring (key))); | |
39 io.write(string.rep (" ", indent+4)) -- indent it | |
40 io.write("(\n"); | |
41 table_print (value, indent + 7, done) | |
42 io.write(string.rep (" ", indent+4)) -- indent it | |
43 io.write(")\n"); | |
44 else | |
45 io.write(string.format("[%s] => %s\n", | |
46 tostring (key), tostring(value))) | |
47 end | |
48 end | |
49 else | |
50 io.write(tt .. "\n") | |
51 end | |
52 end | |
53 | |
54 -- Make a *one-way* subscription. User will see when contact is online, | |
55 -- contact will not see when user is online. | |
56 function vcard_get(user_jid) | |
57 local user_username, user_host = jid.split(user_jid); | |
58 if not hosts[user_host] then | |
59 warn("The host '%s' is not configured for this server.", user_host); | |
60 return; | |
61 end | |
62 storagemanager.initialize_host(user_host); | |
63 local vCard; | |
64 vCard = st.deserialize(datamanager.load(user_username, user_host, "vcard")); | |
65 if vCard then | |
66 print(vCard); | |
67 else | |
68 warn("The user '%s' has no vCard configured.",user_jid); | |
69 end | |
70 end | |
71 | |
72 function vcard_set(user_jid, file) | |
73 local user_username, user_host = jid.split(user_jid); | |
74 if not hosts[user_host] then | |
75 warn("The host '%s' is not configured for this server.", user_host); | |
76 return; | |
77 end | |
78 storagemanager.initialize_host(user_host); | |
79 local f = io.input(file); | |
80 local xmldata=io.read("*all"); | |
81 io.close(f); | |
82 | |
83 local vCard=st.preserialize(xml.parse(xmldata)); | |
84 | |
85 if vCard then | |
86 datamanager.store(user_username, user_host, "vcard", vCard); | |
87 else | |
88 warn("Could not parse the file."); | |
89 end | |
90 end | |
91 | |
92 function vcard_delete(user_jid) | |
93 local user_username, user_host = jid.split(user_jid); | |
94 if not hosts[user_host] then | |
95 warn("The host '%s' is not configured for this server.", user_host); | |
96 return; | |
97 end | |
98 storagemanager.initialize_host(user_host); | |
99 datamanager.store(user_username, user_host, "vcard", nil); | |
100 end | |
101 | |
102 function module.command(arg) | |
103 local command = arg[1]; | |
104 if not command then | |
105 warn("Valid subcommands: get | set | delete "); | |
106 return 0; | |
107 end | |
108 table.remove(arg, 1); | |
109 if command == "get" then | |
110 vcard_get(arg[1]); | |
111 return 0; | |
112 elseif command == "set" then | |
113 vcard_set(arg[1], arg[2]); | |
114 return 0; | |
115 elseif command == "delete" then | |
116 vcard_delete(arg[1]); | |
117 return 0; | |
118 else | |
119 warn("Unknown command: %s", command); | |
120 return 1; | |
121 end | |
122 return 0; | |
123 end |