35 errmsg[#errmsg + 1] = name .. ": " .. err; |
39 errmsg[#errmsg + 1] = name .. ": " .. err; |
36 end |
40 end |
37 return { status = "completed", error = { message = t_concat(errmsg, "\n") } }; |
41 return { status = "completed", error = { message = t_concat(errmsg, "\n") } }; |
38 end |
42 end |
39 |
43 |
40 function add_user_command_handler(self, data, state) |
44 -- Adding a new user |
41 local add_user_layout = dataforms_new{ |
45 local add_user_layout = dataforms_new{ |
42 title = "Adding a User"; |
46 title = "Adding a User"; |
43 instructions = "Fill out this form to add a user."; |
47 instructions = "Fill out this form to add a user."; |
44 |
48 |
45 { name = "FORM_TYPE", type = "hidden", value = "http://jabber.org/protocol/admin" }; |
49 { name = "FORM_TYPE", type = "hidden", value = "http://jabber.org/protocol/admin" }; |
46 { name = "accountjid", type = "jid-single", required = true, label = "The Jabber ID for the account to be added" }; |
50 { name = "accountjid", type = "jid-single", required = true, label = "The Jabber ID for the account to be added" }; |
47 { name = "password", type = "text-private", label = "The password for this account" }; |
51 { name = "password", type = "text-private", label = "The password for this account" }; |
48 { name = "password-verify", type = "text-private", label = "Retype password" }; |
52 { name = "password-verify", type = "text-private", label = "Retype password" }; |
49 }; |
53 }; |
50 |
54 |
51 if state then |
55 local add_user_command_handler = adhoc_simple(add_user_layout, function(fields, err) |
52 if data.action == "cancel" then |
56 if err then |
53 return { status = "canceled" }; |
57 return generate_error_message(err); |
54 end |
58 end |
55 local fields, err = add_user_layout:data(data.form); |
59 local username, host, resource = jid.split(fields.accountjid); |
56 if err then |
60 if module_host ~= host then |
57 return generate_error_message(err); |
61 return { status = "completed", error = { message = "Trying to add a user on " .. host .. " but command was sent to " .. module_host}}; |
58 end |
62 end |
59 local username, host, resource = jid.split(fields.accountjid); |
63 if (fields["password"] == fields["password-verify"]) and username and host then |
60 if data.to ~= host then |
64 if usermanager_user_exists(username, host) then |
61 return { status = "completed", error = { message = "Trying to add a user on " .. host .. " but command was sent to " .. data.to}}; |
65 return { status = "completed", error = { message = "Account already exists" } }; |
62 end |
66 else |
63 if (fields["password"] == fields["password-verify"]) and username and host then |
67 if usermanager_create_user(username, fields.password, host) then |
64 if usermanager_user_exists(username, host) then |
68 module:log("info", "Created new account %s@%s", username, host); |
65 return { status = "completed", error = { message = "Account already exists" } }; |
69 return { status = "completed", info = "Account successfully created" }; |
66 else |
70 else |
67 if usermanager_create_user(username, fields.password, host) then |
71 return { status = "completed", error = { message = "Failed to write data to disk" } }; |
68 module:log("info", "Created new account %s@%s", username, host); |
|
69 return { status = "completed", info = "Account successfully created" }; |
|
70 else |
|
71 return { status = "completed", error = { message = "Failed to write data to disk" } }; |
|
72 end |
|
73 end |
72 end |
74 else |
73 end |
75 module:log("debug", "Invalid data, password mismatch or empty username while creating account for %s", fields.accountjid or "<nil>"); |
74 else |
76 return { status = "completed", error = { message = "Invalid data.\nPassword mismatch, or empty username" } }; |
75 module:log("debug", "Invalid data, password mismatch or empty username while creating account for %s", fields.accountjid or "<nil>"); |
77 end |
76 return { status = "completed", error = { message = "Invalid data.\nPassword mismatch, or empty username" } }; |
78 else |
77 end |
79 return { status = "executing", actions = {"next", "complete", default = "complete"}, form = add_user_layout }, "executing"; |
78 end); |
80 end |
79 |
81 end |
80 -- Changing a user's password |
82 |
81 local change_user_password_layout = dataforms_new{ |
83 function change_user_password_command_handler(self, data, state) |
82 title = "Changing a User Password"; |
84 local change_user_password_layout = dataforms_new{ |
83 instructions = "Fill out this form to change a user's password."; |
85 title = "Changing a User Password"; |
84 |
86 instructions = "Fill out this form to change a user's password."; |
85 { name = "FORM_TYPE", type = "hidden", value = "http://jabber.org/protocol/admin" }; |
87 |
86 { name = "accountjid", type = "jid-single", required = true, label = "The Jabber ID for this account" }; |
88 { name = "FORM_TYPE", type = "hidden", value = "http://jabber.org/protocol/admin" }; |
87 { name = "password", type = "text-private", required = true, label = "The password for this account" }; |
89 { name = "accountjid", type = "jid-single", required = true, label = "The Jabber ID for this account" }; |
88 }; |
90 { name = "password", type = "text-private", required = true, label = "The password for this account" }; |
89 |
91 }; |
90 local change_user_password_command_handler = adhoc_simple(change_user_password_layout, function(fields, err) |
92 |
91 if err then |
93 if state then |
92 return generate_error_message(err); |
94 if data.action == "cancel" then |
93 end |
95 return { status = "canceled" }; |
94 local username, host, resource = jid.split(fields.accountjid); |
96 end |
95 if module_host ~= host then |
97 local fields, err = change_user_password_layout:data(data.form); |
96 return { status = "completed", error = { message = "Trying to change the password of a user on " .. host .. " but command was sent to " .. module_host}}; |
98 if err then |
97 end |
99 return generate_error_message(err); |
98 if usermanager_user_exists(username, host) and usermanager_set_password(username, fields.password, host) then |
100 end |
99 return { status = "completed", info = "Password successfully changed" }; |
101 local username, host, resource = jid.split(fields.accountjid); |
100 else |
102 if data.to ~= host then |
101 return { status = "completed", error = { message = "User does not exist" } }; |
103 return { status = "completed", error = { message = "Trying to change the password of a user on " .. host .. " but command was sent to " .. data.to}}; |
102 end |
104 end |
103 end); |
105 if usermanager_user_exists(username, host) and usermanager_set_password(username, fields.password, host) then |
104 |
106 return { status = "completed", info = "Password successfully changed" }; |
105 -- Reloading the config |
107 else |
106 local function config_reload_handler(self, data, state) |
108 return { status = "completed", error = { message = "User does not exist" } }; |
|
109 end |
|
110 else |
|
111 return { status = "executing", actions = {"next", "complete", default = "complete"}, form = change_user_password_layout }, "executing"; |
|
112 end |
|
113 end |
|
114 |
|
115 function config_reload_handler(self, data, state) |
|
116 local ok, err = prosody.reload_config(); |
107 local ok, err = prosody.reload_config(); |
117 if ok then |
108 if ok then |
118 return { status = "completed", info = "Configuration reloaded (modules may need to be reloaded for this to have an effect)" }; |
109 return { status = "completed", info = "Configuration reloaded (modules may need to be reloaded for this to have an effect)" }; |
119 else |
110 else |
120 return { status = "completed", error = { message = "Failed to reload config: " .. tostring(err) } }; |
111 return { status = "completed", error = { message = "Failed to reload config: " .. tostring(err) } }; |
121 end |
112 end |
122 end |
113 end |
123 |
114 |
124 |
115 -- Deleting a user's account |
125 function delete_user_command_handler(self, data, state) |
116 local delete_user_layout = dataforms_new{ |
126 local delete_user_layout = dataforms_new{ |
117 title = "Deleting a User"; |
127 title = "Deleting a User"; |
118 instructions = "Fill out this form to delete a user."; |
128 instructions = "Fill out this form to delete a user."; |
119 |
129 |
120 { name = "FORM_TYPE", type = "hidden", value = "http://jabber.org/protocol/admin" }; |
130 { name = "FORM_TYPE", type = "hidden", value = "http://jabber.org/protocol/admin" }; |
121 { name = "accountjids", type = "jid-multi", label = "The Jabber ID(s) to delete" }; |
131 { name = "accountjids", type = "jid-multi", label = "The Jabber ID(s) to delete" }; |
122 }; |
132 }; |
123 |
133 |
124 local delete_user_command_handler = adhoc_simple(delete_user_layout, function(fields, err) |
134 if state then |
125 if err then |
135 if data.action == "cancel" then |
126 return generate_error_message(err); |
136 return { status = "canceled" }; |
127 end |
137 end |
128 local failed = {}; |
138 local fields, err = delete_user_layout:data(data.form); |
129 local succeeded = {}; |
139 if err then |
130 for _, aJID in ipairs(fields.accountjids) do |
140 return generate_error_message(err); |
131 local username, host, resource = jid.split(aJID); |
141 end |
132 if (host == module_host) and usermanager_user_exists(username, host) and usermanager_delete_user(username, host) then |
142 local failed = {}; |
133 module:log("debug", "User %s has been deleted", aJID); |
143 local succeeded = {}; |
134 succeeded[#succeeded+1] = aJID; |
144 for _, aJID in ipairs(fields.accountjids) do |
135 else |
145 local username, host, resource = jid.split(aJID); |
136 module:log("debug", "Tried to delete non-existant user %s", aJID); |
146 if (host == data.to) and usermanager_user_exists(username, host) and usermanager_delete_user(username, host) then |
137 failed[#failed+1] = aJID; |
147 module:log("debug", "User %s has been deleted", aJID); |
138 end |
148 succeeded[#succeeded+1] = aJID; |
139 end |
149 else |
140 return {status = "completed", info = (#succeeded ~= 0 and |
150 module:log("debug", "Tried to delete non-existant user %s", aJID); |
141 "The following accounts were successfully deleted:\n"..t_concat(succeeded, "\n").."\n" or "").. |
151 failed[#failed+1] = aJID; |
142 (#failed ~= 0 and |
152 end |
143 "The following accounts could not be deleted:\n"..t_concat(failed, "\n") or "") }; |
153 end |
144 end); |
154 return {status = "completed", info = (#succeeded ~= 0 and |
145 |
155 "The following accounts were successfully deleted:\n"..t_concat(succeeded, "\n").."\n" or "").. |
146 -- Ending a user's session |
156 (#failed ~= 0 and |
147 local function disconnect_user(match_jid) |
157 "The following accounts could not be deleted:\n"..t_concat(failed, "\n") or "") }; |
|
158 else |
|
159 return { status = "executing", actions = {"next", "complete", default = "complete"}, form = delete_user_layout }, "executing"; |
|
160 end |
|
161 end |
|
162 |
|
163 function disconnect_user(match_jid) |
|
164 local node, hostname, givenResource = jid.split(match_jid); |
148 local node, hostname, givenResource = jid.split(match_jid); |
165 local host = hosts[hostname]; |
149 local host = hosts[hostname]; |
166 local sessions = host.sessions[node] and host.sessions[node].sessions; |
150 local sessions = host.sessions[node] and host.sessions[node].sessions; |
167 for resource, session in pairs(sessions or {}) do |
151 for resource, session in pairs(sessions or {}) do |
168 if not givenResource or (resource == givenResource) then |
152 if not givenResource or (resource == givenResource) then |
171 end |
155 end |
172 end |
156 end |
173 return true; |
157 return true; |
174 end |
158 end |
175 |
159 |
176 function end_user_session_handler(self, data, state) |
160 local end_user_session_layout = dataforms_new{ |
177 local end_user_session_layout = dataforms_new{ |
161 title = "Ending a User Session"; |
178 title = "Ending a User Session"; |
162 instructions = "Fill out this form to end a user's session."; |
179 instructions = "Fill out this form to end a user's session."; |
163 |
180 |
164 { name = "FORM_TYPE", type = "hidden", value = "http://jabber.org/protocol/admin" }; |
181 { name = "FORM_TYPE", type = "hidden", value = "http://jabber.org/protocol/admin" }; |
165 { name = "accountjids", type = "jid-multi", label = "The Jabber ID(s) for which to end sessions" }; |
182 { name = "accountjids", type = "jid-multi", label = "The Jabber ID(s) for which to end sessions" }; |
166 }; |
183 }; |
167 |
184 |
168 local end_user_session_handler = adhoc_simple(end_user_session_layout, function(fields, err) |
185 if state then |
169 if err then |
186 if data.action == "cancel" then |
170 return generate_error_message(err); |
187 return { status = "canceled" }; |
171 end |
188 end |
172 local failed = {}; |
189 |
173 local succeeded = {}; |
190 local fields, err = end_user_session_layout:data(data.form); |
174 for _, aJID in ipairs(fields.accountjids) do |
191 if err then |
175 local username, host, resource = jid.split(aJID); |
192 return generate_error_message(err); |
176 if (host == module_host) and usermanager_user_exists(username, host) and disconnect_user(aJID) then |
193 end |
177 succeeded[#succeeded+1] = aJID; |
194 local failed = {}; |
178 else |
195 local succeeded = {}; |
179 failed[#failed+1] = aJID; |
196 for _, aJID in ipairs(fields.accountjids) do |
180 end |
197 local username, host, resource = jid.split(aJID); |
181 end |
198 if (host == data.to) and usermanager_user_exists(username, host) and disconnect_user(aJID) then |
182 return {status = "completed", info = (#succeeded ~= 0 and |
199 succeeded[#succeeded+1] = aJID; |
183 "The following accounts were successfully disconnected:\n"..t_concat(succeeded, "\n").."\n" or "").. |
|
184 (#failed ~= 0 and |
|
185 "The following accounts could not be disconnected:\n"..t_concat(failed, "\n") or "") }; |
|
186 end); |
|
187 |
|
188 -- Getting a user's password |
|
189 local get_user_password_layout = dataforms_new{ |
|
190 title = "Getting User's Password"; |
|
191 instructions = "Fill out this form to get a user's password."; |
|
192 |
|
193 { name = "FORM_TYPE", type = "hidden", value = "http://jabber.org/protocol/admin" }; |
|
194 { name = "accountjid", type = "jid-single", required = true, label = "The Jabber ID for which to retrieve the password" }; |
|
195 }; |
|
196 |
|
197 local get_user_password_result_layout = dataforms_new{ |
|
198 { name = "FORM_TYPE", type = "hidden", value = "http://jabber.org/protocol/admin" }; |
|
199 { name = "accountjid", type = "jid-single", label = "JID" }; |
|
200 { name = "password", type = "text-single", label = "Password" }; |
|
201 }; |
|
202 |
|
203 local get_user_password_handler = adhoc_simple(get_user_password_layout, function(fields, err) |
|
204 if err then |
|
205 return generate_error_message(err); |
|
206 end |
|
207 local user, host, resource = jid.split(fields.accountjid); |
|
208 local accountjid = ""; |
|
209 local password = ""; |
|
210 if host ~= module_host then |
|
211 return { status = "completed", error = { message = "Tried to get password for a user on " .. host .. " but command was sent to " .. module_host } }; |
|
212 elseif usermanager_user_exists(user, host) then |
|
213 accountjid = fields.accountjid; |
|
214 password = usermanager_get_password(user, host); |
|
215 else |
|
216 return { status = "completed", error = { message = "User does not exist" } }; |
|
217 end |
|
218 return { status = "completed", result = { layout = get_user_password_result_layout, values = {accountjid = accountjid, password = password} } }; |
|
219 end); |
|
220 |
|
221 -- Getting a user's roster |
|
222 local get_user_roster_layout = dataforms_new{ |
|
223 { name = "FORM_TYPE", type = "hidden", value = "http://jabber.org/protocol/admin" }; |
|
224 { name = "accountjid", type = "jid-single", required = true, label = "The Jabber ID for which to retrieve the roster" }; |
|
225 }; |
|
226 |
|
227 local get_user_roster_result_layout = dataforms_new{ |
|
228 { name = "FORM_TYPE", type = "hidden", value = "http://jabber.org/protocol/admin" }; |
|
229 { name = "accountjid", type = "jid-single", label = "This is the roster for" }; |
|
230 { name = "roster", type = "text-multi", label = "Roster XML" }; |
|
231 }; |
|
232 |
|
233 local get_user_roster_handler = adhoc_simple(get_user_roster_layout, function(fields, err) |
|
234 if err then |
|
235 return generate_error_message(err); |
|
236 end |
|
237 |
|
238 local user, host, resource = jid.split(fields.accountjid); |
|
239 if host ~= module_host then |
|
240 return { status = "completed", error = { message = "Tried to get roster for a user on " .. host .. " but command was sent to " .. module_host } }; |
|
241 elseif not usermanager_user_exists(user, host) then |
|
242 return { status = "completed", error = { message = "User does not exist" } }; |
|
243 end |
|
244 local roster = rm_load_roster(user, host); |
|
245 |
|
246 local query = st.stanza("query", { xmlns = "jabber:iq:roster" }); |
|
247 for jid in pairs(roster) do |
|
248 if jid ~= "pending" and jid then |
|
249 query:tag("item", { |
|
250 jid = jid, |
|
251 subscription = roster[jid].subscription, |
|
252 ask = roster[jid].ask, |
|
253 name = roster[jid].name, |
|
254 }); |
|
255 for group in pairs(roster[jid].groups) do |
|
256 query:tag("group"):text(group):up(); |
|
257 end |
|
258 query:up(); |
|
259 end |
|
260 end |
|
261 |
|
262 local query_text = tostring(query):gsub("><", ">\n<"); |
|
263 |
|
264 local result = get_user_roster_result_layout:form({ accountjid = user.."@"..host, roster = query_text }, "result"); |
|
265 result:add_child(query); |
|
266 return { status = "completed", other = result }; |
|
267 end); |
|
268 |
|
269 -- Getting user statistics |
|
270 local get_user_stats_layout = dataforms_new{ |
|
271 title = "Get User Statistics"; |
|
272 instructions = "Fill out this form to gather user statistics."; |
|
273 |
|
274 { name = "FORM_TYPE", type = "hidden", value = "http://jabber.org/protocol/admin" }; |
|
275 { name = "accountjid", type = "jid-single", required = true, label = "The Jabber ID for statistics" }; |
|
276 }; |
|
277 |
|
278 local get_user_stats_result_layout = dataforms_new{ |
|
279 { name = "FORM_TYPE", type = "hidden", value = "http://jabber.org/protocol/admin" }; |
|
280 { name = "ipaddresses", type = "text-multi", label = "IP Addresses" }; |
|
281 { name = "rostersize", type = "text-single", label = "Roster size" }; |
|
282 { name = "onlineresources", type = "text-multi", label = "Online Resources" }; |
|
283 }; |
|
284 |
|
285 local get_user_stats_handler = adhoc_simple(get_user_stats_layout, function(fields, err) |
|
286 if err then |
|
287 return generate_error_message(err); |
|
288 end |
|
289 |
|
290 local user, host, resource = jid.split(fields.accountjid); |
|
291 if host ~= module_host then |
|
292 return { status = "completed", error = { message = "Tried to get stats for a user on " .. host .. " but command was sent to " .. module_host } }; |
|
293 elseif not usermanager_user_exists(user, host) then |
|
294 return { status = "completed", error = { message = "User does not exist" } }; |
|
295 end |
|
296 local roster = rm_load_roster(user, host); |
|
297 local rostersize = 0; |
|
298 local IPs = ""; |
|
299 local resources = ""; |
|
300 for jid in pairs(roster) do |
|
301 if jid ~= "pending" and jid then |
|
302 rostersize = rostersize + 1; |
|
303 end |
|
304 end |
|
305 for resource, session in pairs((hosts[host].sessions[user] and hosts[host].sessions[user].sessions) or {}) do |
|
306 resources = resources .. "\n" .. resource; |
|
307 IPs = IPs .. "\n" .. session.ip; |
|
308 end |
|
309 return { status = "completed", result = {layout = get_user_stats_result_layout, values = {ipaddresses = IPs, rostersize = tostring(rostersize), |
|
310 onlineresources = resources}} }; |
|
311 end); |
|
312 |
|
313 -- Getting a list of online users |
|
314 local get_online_users_layout = dataforms_new{ |
|
315 title = "Getting List of Online Users"; |
|
316 instructions = "How many users should be returned at most?"; |
|
317 |
|
318 { name = "FORM_TYPE", type = "hidden", value = "http://jabber.org/protocol/admin" }; |
|
319 { name = "max_items", type = "list-single", label = "Maximum number of users", |
|
320 value = { "25", "50", "75", "100", "150", "200", "all" } }; |
|
321 { name = "details", type = "boolean", label = "Show details" }; |
|
322 }; |
|
323 |
|
324 local get_online_users_result_layout = dataforms_new{ |
|
325 { name = "FORM_TYPE", type = "hidden", value = "http://jabber.org/protocol/admin" }; |
|
326 { name = "onlineuserjids", type = "text-multi", label = "The list of all online users" }; |
|
327 }; |
|
328 |
|
329 local get_online_users_command_handler = adhoc_simple(get_online_users_layout, function(fields, err) |
|
330 if err then |
|
331 return generate_error_message(err); |
|
332 end |
|
333 |
|
334 local max_items = nil |
|
335 if fields.max_items ~= "all" then |
|
336 max_items = tonumber(fields.max_items); |
|
337 end |
|
338 local count = 0; |
|
339 local users = {}; |
|
340 for username, user in pairs(hosts[module_host].sessions or {}) do |
|
341 if (max_items ~= nil) and (count >= max_items) then |
|
342 break; |
|
343 end |
|
344 users[#users+1] = username.."@"..module_host; |
|
345 count = count + 1; |
|
346 if fields.details then |
|
347 for resource, session in pairs(user.sessions or {}) do |
|
348 local status, priority = "unavailable", tostring(session.priority or "-"); |
|
349 if session.presence then |
|
350 status = session.presence:child_with_name("show"); |
|
351 if status then |
|
352 status = status:get_text() or "[invalid!]"; |
|
353 else |
|
354 status = "available"; |
|
355 end |
|
356 end |
|
357 users[#users+1] = " - "..resource..": "..status.."("..priority..")"; |
|
358 end |
|
359 end |
|
360 end |
|
361 return { status = "completed", result = {layout = get_online_users_result_layout, values = {onlineuserjids=t_concat(users, "\n")}} }; |
|
362 end); |
|
363 |
|
364 -- Getting a list of loaded modules |
|
365 local list_modules_result = dataforms_new { |
|
366 title = "List of loaded modules"; |
|
367 |
|
368 { name = "FORM_TYPE", type = "hidden", value = "http://prosody.im/protocol/modules#list" }; |
|
369 { name = "modules", type = "text-multi", label = "The following modules are loaded:" }; |
|
370 }; |
|
371 |
|
372 local function list_modules_handler(self, data, state) |
|
373 local modules = array.collect(keys(hosts[module_host].modules)):sort():concat("\n"); |
|
374 return { status = "completed", result = { layout = list_modules_result; values = { modules = modules } } }; |
|
375 end |
|
376 |
|
377 -- Loading a module |
|
378 local load_module_layout = dataforms_new { |
|
379 title = "Load module"; |
|
380 instructions = "Specify the module to be loaded"; |
|
381 |
|
382 { name = "FORM_TYPE", type = "hidden", value = "http://prosody.im/protocol/modules#load" }; |
|
383 { name = "module", type = "text-single", required = true, label = "Module to be loaded:"}; |
|
384 }; |
|
385 |
|
386 local load_module_handler = adhoc_simple(load_module_layout, function(fields, err) |
|
387 if err then |
|
388 return generate_error_message(err); |
|
389 end |
|
390 if modulemanager.is_loaded(module_host, fields.module) then |
|
391 return { status = "completed", info = "Module already loaded" }; |
|
392 end |
|
393 local ok, err = modulemanager.load(module_host, fields.module); |
|
394 if ok then |
|
395 return { status = "completed", info = 'Module "'..fields.module..'" successfully loaded on host "'..module_host..'".' }; |
|
396 else |
|
397 return { status = "completed", error = { message = 'Failed to load module "'..fields.module..'" on host "'..module_host.. |
|
398 '". Error was: "'..tostring(err or "<unspecified>")..'"' } }; |
|
399 end |
|
400 end); |
|
401 |
|
402 -- Globally loading a module |
|
403 local globally_load_module_layout = dataforms_new { |
|
404 title = "Globally load module"; |
|
405 instructions = "Specify the module to be loaded on all hosts"; |
|
406 |
|
407 { name = "FORM_TYPE", type = "hidden", value = "http://prosody.im/protocol/modules#global-load" }; |
|
408 { name = "module", type = "text-single", required = true, label = "Module to globally load:"}; |
|
409 }; |
|
410 |
|
411 local globally_load_module_handler = adhoc_simple(globally_load_module_layout, function(fields, err) |
|
412 local ok_list, err_list = {}, {}; |
|
413 |
|
414 if err then |
|
415 return generate_error_message(err); |
|
416 end |
|
417 |
|
418 local ok, err = modulemanager.load(module_host, fields.module); |
|
419 if ok then |
|
420 ok_list[#ok_list + 1] = module_host; |
|
421 else |
|
422 err_list[#err_list + 1] = module_host .. " (Error: " .. tostring(err) .. ")"; |
|
423 end |
|
424 |
|
425 -- Is this a global module? |
|
426 if modulemanager.is_loaded("*", fields.module) and not modulemanager.is_loaded(module_host, fields.module) then |
|
427 return { status = "completed", info = 'Global module '..fields.module..' loaded.' }; |
|
428 end |
|
429 |
|
430 -- This is either a shared or "normal" module, load it on all other hosts |
|
431 for host_name, host in pairs(hosts) do |
|
432 if host_name ~= module_host and host.type == "local" then |
|
433 local ok, err = modulemanager.load(host_name, fields.module); |
|
434 if ok then |
|
435 ok_list[#ok_list + 1] = host_name; |
200 else |
436 else |
201 failed[#failed+1] = aJID; |
437 err_list[#err_list + 1] = host_name .. " (Error: " .. tostring(err) .. ")"; |
202 end |
438 end |
203 end |
439 end |
204 return {status = "completed", info = (#succeeded ~= 0 and |
440 end |
205 "The following accounts were successfully disconnected:\n"..t_concat(succeeded, "\n").."\n" or "").. |
441 |
206 (#failed ~= 0 and |
442 local info = (#ok_list > 0 and ("The module "..fields.module.." was successfully loaded onto the hosts:\n"..t_concat(ok_list, "\n")) or "") |
207 "The following accounts could not be disconnected:\n"..t_concat(failed, "\n") or "") }; |
443 .. ((#ok_list > 0 and #err_list > 0) and "\n" or "") .. |
208 else |
444 (#err_list > 0 and ("Failed to load the module "..fields.module.." onto the hosts:\n"..t_concat(err_list, "\n")) or ""); |
209 return { status = "executing", actions = {"next", "complete", default = "complete"}, form = end_user_session_layout }, "executing"; |
445 return { status = "completed", info = info }; |
210 end |
446 end); |
211 end |
447 |
212 |
448 -- Reloading modules |
213 function get_user_password_handler(self, data, state) |
449 local reload_modules_layout = dataforms_new { |
214 local get_user_password_layout = dataforms_new{ |
450 title = "Reload modules"; |
215 title = "Getting User's Password"; |
451 instructions = "Select the modules to be reloaded"; |
216 instructions = "Fill out this form to get a user's password."; |
452 |
217 |
453 { name = "FORM_TYPE", type = "hidden", value = "http://prosody.im/protocol/modules#reload" }; |
218 { name = "FORM_TYPE", type = "hidden", value = "http://jabber.org/protocol/admin" }; |
454 { name = "modules", type = "list-multi", required = true, label = "Modules to be reloaded:"}; |
219 { name = "accountjid", type = "jid-single", required = true, label = "The Jabber ID for which to retrieve the password" }; |
455 }; |
220 }; |
456 |
221 |
457 local reload_modules_handler = adhoc_initial(reload_modules_layout, function() |
222 local get_user_password_result_layout = dataforms_new{ |
458 return { modules = array.collect(keys(hosts[module_host].modules)):sort() }; |
223 { name = "FORM_TYPE", type = "hidden", value = "http://jabber.org/protocol/admin" }; |
459 end, function(fields, err) |
224 { name = "accountjid", type = "jid-single", label = "JID" }; |
460 if err then |
225 { name = "password", type = "text-single", label = "Password" }; |
461 return generate_error_message(err); |
226 }; |
462 end |
227 |
463 local ok_list, err_list = {}, {}; |
228 if state then |
464 for _, module in ipairs(fields.modules) do |
229 if data.action == "cancel" then |
465 local ok, err = modulemanager.reload(module_host, module); |
230 return { status = "canceled" }; |
466 if ok then |
231 end |
467 ok_list[#ok_list + 1] = module; |
232 local fields, err = get_user_password_layout:data(data.form); |
|
233 if err then |
|
234 return generate_error_message(err); |
|
235 end |
|
236 local user, host, resource = jid.split(fields.accountjid); |
|
237 local accountjid = ""; |
|
238 local password = ""; |
|
239 if host ~= data.to then |
|
240 return { status = "completed", error = { message = "Tried to get password for a user on " .. host .. " but command was sent to " .. data.to } }; |
|
241 elseif usermanager_user_exists(user, host) then |
|
242 accountjid = fields.accountjid; |
|
243 password = usermanager_get_password(user, host); |
|
244 else |
468 else |
245 return { status = "completed", error = { message = "User does not exist" } }; |
469 err_list[#err_list + 1] = module .. "(Error: " .. tostring(err) .. ")"; |
246 end |
470 end |
247 return { status = "completed", result = { layout = get_user_password_result_layout, values = {accountjid = accountjid, password = password} } }; |
471 end |
248 else |
472 local info = (#ok_list > 0 and ("The following modules were successfully reloaded on host "..module_host..":\n"..t_concat(ok_list, "\n")) or "") |
249 return { status = "executing", actions = {"next", "complete", default = "complete"}, form = get_user_password_layout }, "executing"; |
473 .. ((#ok_list > 0 and #err_list > 0) and "\n" or "") .. |
250 end |
474 (#err_list > 0 and ("Failed to reload the following modules on host "..module_host..":\n"..t_concat(err_list, "\n")) or ""); |
251 end |
475 return { status = "completed", info = info }; |
252 |
476 end); |
253 function get_user_roster_handler(self, data, state) |
477 |
254 local get_user_roster_layout = dataforms_new{ |
478 -- Globally reloading a module |
255 { name = "FORM_TYPE", type = "hidden", value = "http://jabber.org/protocol/admin" }; |
479 local globally_reload_module_layout = dataforms_new { |
256 { name = "accountjid", type = "jid-single", required = true, label = "The Jabber ID for which to retrieve the roster" }; |
480 title = "Globally reload module"; |
257 }; |
481 instructions = "Specify the module to reload on all hosts"; |
258 |
482 |
259 local get_user_roster_result_layout = dataforms_new{ |
483 { name = "FORM_TYPE", type = "hidden", value = "http://prosody.im/protocol/modules#global-reload" }; |
260 { name = "FORM_TYPE", type = "hidden", value = "http://jabber.org/protocol/admin" }; |
484 { name = "module", type = "list-single", required = true, label = "Module to globally reload:"}; |
261 { name = "accountjid", type = "jid-single", label = "This is the roster for" }; |
485 }; |
262 { name = "roster", type = "text-multi", label = "Roster XML" }; |
486 |
263 }; |
487 local globally_reload_module_handler = adhoc_initial(globally_reload_module_layout, function() |
264 |
488 local loaded_modules = array(keys(modulemanager.get_modules("*"))); |
265 if state then |
489 for _, host in pairs(hosts) do |
266 if data.action == "cancel" then |
490 loaded_modules:append(array(keys(host.modules))); |
267 return { status = "canceled" }; |
491 end |
268 end |
492 loaded_modules = array(keys(set.new(loaded_modules):items())):sort(); |
269 |
493 return { module = loaded_modules }; |
270 local fields, err = get_user_roster_layout:data(data.form); |
494 end, function(fields, err) |
271 |
495 local is_global = false; |
272 if err then |
496 |
273 return generate_error_message(err); |
497 if err then |
274 end |
498 return generate_error_message(err); |
275 |
499 end |
276 local user, host, resource = jid.split(fields.accountjid); |
500 |
277 if host ~= data.to then |
501 if modulemanager.is_loaded("*", fields.module) then |
278 return { status = "completed", error = { message = "Tried to get roster for a user on " .. host .. " but command was sent to " .. data.to } }; |
502 local ok, err = modulemanager.reload("*", fields.module); |
279 elseif not usermanager_user_exists(user, host) then |
503 if not ok then |
280 return { status = "completed", error = { message = "User does not exist" } }; |
504 return { status = "completed", info = 'Global module '..fields.module..' failed to reload: '..err }; |
281 end |
505 end |
282 local roster = rm_load_roster(user, host); |
506 is_global = true; |
283 |
507 end |
284 local query = st.stanza("query", { xmlns = "jabber:iq:roster" }); |
508 |
285 for jid in pairs(roster) do |
509 local ok_list, err_list = {}, {}; |
286 if jid ~= "pending" and jid then |
510 for host_name, host in pairs(hosts) do |
287 query:tag("item", { |
511 if modulemanager.is_loaded(host_name, fields.module) then |
288 jid = jid, |
512 local ok, err = modulemanager.reload(host_name, fields.module); |
289 subscription = roster[jid].subscription, |
513 if ok then |
290 ask = roster[jid].ask, |
514 ok_list[#ok_list + 1] = host_name; |
291 name = roster[jid].name, |
515 else |
292 }); |
516 err_list[#err_list + 1] = host_name .. " (Error: " .. tostring(err) .. ")"; |
293 for group in pairs(roster[jid].groups) do |
|
294 query:tag("group"):text(group):up(); |
|
295 end |
|
296 query:up(); |
|
297 end |
517 end |
298 end |
518 end |
299 |
519 end |
300 local query_text = tostring(query):gsub("><", ">\n<"); |
520 |
301 |
521 if #ok_list == 0 and #err_list == 0 then |
302 local result = get_user_roster_result_layout:form({ accountjid = user.."@"..host, roster = query_text }, "result"); |
522 if is_global then |
303 result:add_child(query); |
523 return { status = "completed", info = 'Successfully reloaded global module '..fields.module }; |
304 return { status = "completed", other = result }; |
|
305 else |
|
306 return { status = "executing", actions = {"next", "complete", default = "complete"}, form = get_user_roster_layout }, "executing"; |
|
307 end |
|
308 end |
|
309 |
|
310 function get_user_stats_handler(self, data, state) |
|
311 local get_user_stats_layout = dataforms_new{ |
|
312 title = "Get User Statistics"; |
|
313 instructions = "Fill out this form to gather user statistics."; |
|
314 |
|
315 { name = "FORM_TYPE", type = "hidden", value = "http://jabber.org/protocol/admin" }; |
|
316 { name = "accountjid", type = "jid-single", required = true, label = "The Jabber ID for statistics" }; |
|
317 }; |
|
318 |
|
319 local get_user_stats_result_layout = dataforms_new{ |
|
320 { name = "FORM_TYPE", type = "hidden", value = "http://jabber.org/protocol/admin" }; |
|
321 { name = "ipaddresses", type = "text-multi", label = "IP Addresses" }; |
|
322 { name = "rostersize", type = "text-single", label = "Roster size" }; |
|
323 { name = "onlineresources", type = "text-multi", label = "Online Resources" }; |
|
324 }; |
|
325 |
|
326 if state then |
|
327 if data.action == "cancel" then |
|
328 return { status = "canceled" }; |
|
329 end |
|
330 |
|
331 local fields, err = get_user_stats_layout:data(data.form); |
|
332 |
|
333 if err then |
|
334 return generate_error_message(err); |
|
335 end |
|
336 |
|
337 local user, host, resource = jid.split(fields.accountjid); |
|
338 if host ~= data.to then |
|
339 return { status = "completed", error = { message = "Tried to get stats for a user on " .. host .. " but command was sent to " .. data.to } }; |
|
340 elseif not usermanager_user_exists(user, host) then |
|
341 return { status = "completed", error = { message = "User does not exist" } }; |
|
342 end |
|
343 local roster = rm_load_roster(user, host); |
|
344 local rostersize = 0; |
|
345 local IPs = ""; |
|
346 local resources = ""; |
|
347 for jid in pairs(roster) do |
|
348 if jid ~= "pending" and jid then |
|
349 rostersize = rostersize + 1; |
|
350 end |
|
351 end |
|
352 for resource, session in pairs((hosts[host].sessions[user] and hosts[host].sessions[user].sessions) or {}) do |
|
353 resources = resources .. "\n" .. resource; |
|
354 IPs = IPs .. "\n" .. session.ip; |
|
355 end |
|
356 return { status = "completed", result = {layout = get_user_stats_result_layout, values = {ipaddresses = IPs, rostersize = tostring(rostersize), |
|
357 onlineresources = resources}} }; |
|
358 else |
|
359 return { status = "executing", actions = {"next", "complete", default = "complete"}, form = get_user_stats_layout }, "executing"; |
|
360 end |
|
361 end |
|
362 |
|
363 function get_online_users_command_handler(self, data, state) |
|
364 local get_online_users_layout = dataforms_new{ |
|
365 title = "Getting List of Online Users"; |
|
366 instructions = "How many users should be returned at most?"; |
|
367 |
|
368 { name = "FORM_TYPE", type = "hidden", value = "http://jabber.org/protocol/admin" }; |
|
369 { name = "max_items", type = "list-single", label = "Maximum number of users", |
|
370 value = { "25", "50", "75", "100", "150", "200", "all" } }; |
|
371 { name = "details", type = "boolean", label = "Show details" }; |
|
372 }; |
|
373 |
|
374 local get_online_users_result_layout = dataforms_new{ |
|
375 { name = "FORM_TYPE", type = "hidden", value = "http://jabber.org/protocol/admin" }; |
|
376 { name = "onlineuserjids", type = "text-multi", label = "The list of all online users" }; |
|
377 }; |
|
378 |
|
379 if state then |
|
380 if data.action == "cancel" then |
|
381 return { status = "canceled" }; |
|
382 end |
|
383 |
|
384 local fields, err = get_online_users_layout:data(data.form); |
|
385 |
|
386 if err then |
|
387 return generate_error_message(err); |
|
388 end |
|
389 |
|
390 local max_items = nil |
|
391 if fields.max_items ~= "all" then |
|
392 max_items = tonumber(fields.max_items); |
|
393 end |
|
394 local count = 0; |
|
395 local users = {}; |
|
396 for username, user in pairs(hosts[data.to].sessions or {}) do |
|
397 if (max_items ~= nil) and (count >= max_items) then |
|
398 break; |
|
399 end |
|
400 users[#users+1] = username.."@"..data.to; |
|
401 count = count + 1; |
|
402 if fields.details then |
|
403 for resource, session in pairs(user.sessions or {}) do |
|
404 local status, priority = "unavailable", tostring(session.priority or "-"); |
|
405 if session.presence then |
|
406 status = session.presence:child_with_name("show"); |
|
407 if status then |
|
408 status = status:get_text() or "[invalid!]"; |
|
409 else |
|
410 status = "available"; |
|
411 end |
|
412 end |
|
413 users[#users+1] = " - "..resource..": "..status.."("..priority..")"; |
|
414 end |
|
415 end |
|
416 end |
|
417 return { status = "completed", result = {layout = get_online_users_result_layout, values = {onlineuserjids=t_concat(users, "\n")}} }; |
|
418 else |
|
419 return { status = "executing", actions = {"next", "complete", default = "complete"}, form = get_online_users_layout }, "executing"; |
|
420 end |
|
421 end |
|
422 |
|
423 function list_modules_handler(self, data, state) |
|
424 local result = dataforms_new { |
|
425 title = "List of loaded modules"; |
|
426 |
|
427 { name = "FORM_TYPE", type = "hidden", value = "http://prosody.im/protocol/modules#list" }; |
|
428 { name = "modules", type = "text-multi", label = "The following modules are loaded:" }; |
|
429 }; |
|
430 |
|
431 local modules = array.collect(keys(hosts[data.to].modules)):sort():concat("\n"); |
|
432 |
|
433 return { status = "completed", result = { layout = result; values = { modules = modules } } }; |
|
434 end |
|
435 |
|
436 function load_module_handler(self, data, state) |
|
437 local layout = dataforms_new { |
|
438 title = "Load module"; |
|
439 instructions = "Specify the module to be loaded"; |
|
440 |
|
441 { name = "FORM_TYPE", type = "hidden", value = "http://prosody.im/protocol/modules#load" }; |
|
442 { name = "module", type = "text-single", required = true, label = "Module to be loaded:"}; |
|
443 }; |
|
444 if state then |
|
445 if data.action == "cancel" then |
|
446 return { status = "canceled" }; |
|
447 end |
|
448 local fields, err = layout:data(data.form); |
|
449 if err then |
|
450 return generate_error_message(err); |
|
451 end |
|
452 if modulemanager.is_loaded(data.to, fields.module) then |
|
453 return { status = "completed", info = "Module already loaded" }; |
|
454 end |
|
455 local ok, err = modulemanager.load(data.to, fields.module); |
|
456 if ok then |
|
457 return { status = "completed", info = 'Module "'..fields.module..'" successfully loaded on host "'..data.to..'".' }; |
|
458 else |
524 else |
459 return { status = "completed", error = { message = 'Failed to load module "'..fields.module..'" on host "'..data.to.. |
525 return { status = "completed", info = 'Module '..fields.module..' not loaded on any host.' }; |
460 '". Error was: "'..tostring(err or "<unspecified>")..'"' } }; |
526 end |
461 end |
527 end |
462 else |
528 |
463 return { status = "executing", actions = {"next", "complete", default = "complete"}, form = layout }, "executing"; |
529 local info = (#ok_list > 0 and ("The module "..fields.module.." was successfully reloaded on the hosts:\n"..t_concat(ok_list, "\n")) or "") |
464 end |
530 .. ((#ok_list > 0 and #err_list > 0) and "\n" or "") .. |
465 end |
531 (#err_list > 0 and ("Failed to reload the module "..fields.module.." on the hosts:\n"..t_concat(err_list, "\n")) or ""); |
466 |
532 return { status = "completed", info = info }; |
467 local function globally_load_module_handler(self, data, state) |
533 end); |
468 local layout = dataforms_new { |
534 |
469 title = "Globally load module"; |
535 local function send_to_online(message, server) |
470 instructions = "Specify the module to be loaded on all hosts"; |
|
471 |
|
472 { name = "FORM_TYPE", type = "hidden", value = "http://prosody.im/protocol/modules#global-load" }; |
|
473 { name = "module", type = "text-single", required = true, label = "Module to globally load:"}; |
|
474 }; |
|
475 if state then |
|
476 local ok_list, err_list = {}, {}; |
|
477 |
|
478 if data.action == "cancel" then |
|
479 return { status = "canceled" }; |
|
480 end |
|
481 |
|
482 local fields, err = layout:data(data.form); |
|
483 if err then |
|
484 return generate_error_message(err); |
|
485 end |
|
486 |
|
487 local ok, err = modulemanager.load(data.to, fields.module); |
|
488 if ok then |
|
489 ok_list[#ok_list + 1] = data.to; |
|
490 else |
|
491 err_list[#err_list + 1] = data.to .. " (Error: " .. tostring(err) .. ")"; |
|
492 end |
|
493 |
|
494 -- Is this a global module? |
|
495 if modulemanager.is_loaded("*", fields.module) and not modulemanager.is_loaded(data.to, fields.module) then |
|
496 return { status = "completed", info = 'Global module '..fields.module..' loaded.' }; |
|
497 end |
|
498 |
|
499 -- This is either a shared or "normal" module, load it on all other hosts |
|
500 for host_name, host in pairs(hosts) do |
|
501 if host_name ~= data.to and host.type == "local" then |
|
502 local ok, err = modulemanager.load(host_name, fields.module); |
|
503 if ok then |
|
504 ok_list[#ok_list + 1] = host_name; |
|
505 else |
|
506 err_list[#err_list + 1] = host_name .. " (Error: " .. tostring(err) .. ")"; |
|
507 end |
|
508 end |
|
509 end |
|
510 |
|
511 local info = (#ok_list > 0 and ("The module "..fields.module.." was successfully loaded onto the hosts:\n"..t_concat(ok_list, "\n")) or "") |
|
512 .. ((#ok_list > 0 and #err_list > 0) and "\n" or "") .. |
|
513 (#err_list > 0 and ("Failed to load the module "..fields.module.." onto the hosts:\n"..t_concat(err_list, "\n")) or ""); |
|
514 return { status = "completed", info = info }; |
|
515 else |
|
516 return { status = "executing", actions = {"next", "complete", default = "complete"}, form = layout }, "executing"; |
|
517 end |
|
518 end |
|
519 |
|
520 function reload_modules_handler(self, data, state) |
|
521 local layout = dataforms_new { |
|
522 title = "Reload modules"; |
|
523 instructions = "Select the modules to be reloaded"; |
|
524 |
|
525 { name = "FORM_TYPE", type = "hidden", value = "http://prosody.im/protocol/modules#reload" }; |
|
526 { name = "modules", type = "list-multi", required = true, label = "Modules to be reloaded:"}; |
|
527 }; |
|
528 if state then |
|
529 if data.action == "cancel" then |
|
530 return { status = "canceled" }; |
|
531 end |
|
532 local fields, err = layout:data(data.form); |
|
533 if err then |
|
534 return generate_error_message(err); |
|
535 end |
|
536 local ok_list, err_list = {}, {}; |
|
537 for _, module in ipairs(fields.modules) do |
|
538 local ok, err = modulemanager.reload(data.to, module); |
|
539 if ok then |
|
540 ok_list[#ok_list + 1] = module; |
|
541 else |
|
542 err_list[#err_list + 1] = module .. "(Error: " .. tostring(err) .. ")"; |
|
543 end |
|
544 end |
|
545 local info = (#ok_list > 0 and ("The following modules were successfully reloaded on host "..data.to..":\n"..t_concat(ok_list, "\n")) or "") |
|
546 .. ((#ok_list > 0 and #err_list > 0) and "\n" or "") .. |
|
547 (#err_list > 0 and ("Failed to reload the following modules on host "..data.to..":\n"..t_concat(err_list, "\n")) or ""); |
|
548 return { status = "completed", info = info }; |
|
549 else |
|
550 local modules = array.collect(keys(hosts[data.to].modules)):sort(); |
|
551 return { status = "executing", actions = {"next", "complete", default = "complete"}, form = { layout = layout; values = { modules = modules } } }, "executing"; |
|
552 end |
|
553 end |
|
554 |
|
555 local function globally_reload_module_handler(self, data, state) |
|
556 local layout = dataforms_new { |
|
557 title = "Globally reload module"; |
|
558 instructions = "Specify the module to reload on all hosts"; |
|
559 |
|
560 { name = "FORM_TYPE", type = "hidden", value = "http://prosody.im/protocol/modules#global-reload" }; |
|
561 { name = "module", type = "list-single", required = true, label = "Module to globally reload:"}; |
|
562 }; |
|
563 if state then |
|
564 if data.action == "cancel" then |
|
565 return { status = "canceled" }; |
|
566 end |
|
567 |
|
568 local is_global = false; |
|
569 local fields, err = layout:data(data.form); |
|
570 if err then |
|
571 return generate_error_message(err); |
|
572 end |
|
573 |
|
574 if modulemanager.is_loaded("*", fields.module) then |
|
575 local ok, err = modulemanager.reload("*", fields.module); |
|
576 if not ok then |
|
577 return { status = "completed", info = 'Global module '..fields.module..' failed to reload: '..err }; |
|
578 end |
|
579 is_global = true; |
|
580 end |
|
581 |
|
582 local ok_list, err_list = {}, {}; |
|
583 for host_name, host in pairs(hosts) do |
|
584 if modulemanager.is_loaded(host_name, fields.module) then |
|
585 local ok, err = modulemanager.reload(host_name, fields.module); |
|
586 if ok then |
|
587 ok_list[#ok_list + 1] = host_name; |
|
588 else |
|
589 err_list[#err_list + 1] = host_name .. " (Error: " .. tostring(err) .. ")"; |
|
590 end |
|
591 end |
|
592 end |
|
593 |
|
594 if #ok_list == 0 and #err_list == 0 then |
|
595 if is_global then |
|
596 return { status = "completed", info = 'Successfully reloaded global module '..fields.module }; |
|
597 else |
|
598 return { status = "completed", info = 'Module '..fields.module..' not loaded on any host.' }; |
|
599 end |
|
600 end |
|
601 |
|
602 local info = (#ok_list > 0 and ("The module "..fields.module.." was successfully reloaded on the hosts:\n"..t_concat(ok_list, "\n")) or "") |
|
603 .. ((#ok_list > 0 and #err_list > 0) and "\n" or "") .. |
|
604 (#err_list > 0 and ("Failed to reload the module "..fields.module.." on the hosts:\n"..t_concat(err_list, "\n")) or ""); |
|
605 return { status = "completed", info = info }; |
|
606 else |
|
607 local loaded_modules = array(keys(modulemanager.get_modules("*"))); |
|
608 for _, host in pairs(hosts) do |
|
609 loaded_modules:append(array(keys(host.modules))); |
|
610 end |
|
611 loaded_modules = array(keys(set.new(loaded_modules):items())):sort(); |
|
612 return { status = "executing", actions = {"next", "complete", default = "complete"}, form = { layout = layout, values = { module = loaded_modules } } }, "executing"; |
|
613 end |
|
614 end |
|
615 |
|
616 function send_to_online(message, server) |
|
617 if server then |
536 if server then |
618 sessions = { [server] = hosts[server] }; |
537 sessions = { [server] = hosts[server] }; |
619 else |
538 else |
620 sessions = hosts; |
539 sessions = hosts; |
621 end |
540 end |
631 end |
550 end |
632 |
551 |
633 return c; |
552 return c; |
634 end |
553 end |
635 |
554 |
636 function shut_down_service_handler(self, data, state) |
555 -- Shutting down the service |
637 local shut_down_service_layout = dataforms_new{ |
556 local shut_down_service_layout = dataforms_new{ |
638 title = "Shutting Down the Service"; |
557 title = "Shutting Down the Service"; |
639 instructions = "Fill out this form to shut down the service."; |
558 instructions = "Fill out this form to shut down the service."; |
640 |
559 |
641 { name = "FORM_TYPE", type = "hidden", value = "http://jabber.org/protocol/admin" }; |
560 { name = "FORM_TYPE", type = "hidden", value = "http://jabber.org/protocol/admin" }; |
642 { name = "delay", type = "list-single", label = "Time delay before shutting down", |
561 { name = "delay", type = "list-single", label = "Time delay before shutting down", |
643 value = { {label = "30 seconds", value = "30"}, |
562 value = { {label = "30 seconds", value = "30"}, |
644 {label = "60 seconds", value = "60"}, |
563 {label = "60 seconds", value = "60"}, |
645 {label = "90 seconds", value = "90"}, |
564 {label = "90 seconds", value = "90"}, |
646 {label = "2 minutes", value = "120"}, |
565 {label = "2 minutes", value = "120"}, |
647 {label = "3 minutes", value = "180"}, |
566 {label = "3 minutes", value = "180"}, |
648 {label = "4 minutes", value = "240"}, |
567 {label = "4 minutes", value = "240"}, |
649 {label = "5 minutes", value = "300"}, |
568 {label = "5 minutes", value = "300"}, |
650 }; |
|
651 }; |
569 }; |
652 { name = "announcement", type = "text-multi", label = "Announcement" }; |
|
653 }; |
570 }; |
654 |
571 { name = "announcement", type = "text-multi", label = "Announcement" }; |
655 if state then |
572 }; |
656 if data.action == "cancel" then |
573 |
657 return { status = "canceled" }; |
574 local shut_down_service_handler = adhoc_simple(shut_down_service_layout, function(fields, err) |
658 end |
575 if err then |
659 |
576 return generate_error_message(err); |
660 local fields, err = shut_down_service_layout:data(data.form); |
577 end |
661 |
578 |
662 if err then |
579 if fields.announcement and #fields.announcement > 0 then |
663 return generate_error_message(err); |
580 local message = st.message({type = "headline"}, fields.announcement):up() |
664 end |
581 :tag("subject"):text("Server is shutting down"); |
665 |
582 send_to_online(message); |
666 if fields.announcement and #fields.announcement > 0 then |
583 end |
667 local message = st.message({type = "headline"}, fields.announcement):up() |
584 |
668 :tag("subject"):text("Server is shutting down"); |
585 timer_add_task(tonumber(fields.delay or "5"), function(time) prosody.shutdown("Shutdown by adhoc command") end); |
669 send_to_online(message); |
586 |
670 end |
587 return { status = "completed", info = "Server is about to shut down" }; |
671 |
588 end); |
672 timer_add_task(tonumber(fields.delay or "5"), function(time) prosody.shutdown("Shutdown by adhoc command") end); |
589 |
673 |
590 -- Unloading modules |
674 return { status = "completed", info = "Server is about to shut down" }; |
591 local unload_modules_layout = dataforms_new { |
675 else |
592 title = "Unload modules"; |
676 return { status = "executing", actions = {"next", "complete", default = "complete"}, form = shut_down_service_layout }, "executing"; |
593 instructions = "Select the modules to be unloaded"; |
677 end |
594 |
678 end |
595 { name = "FORM_TYPE", type = "hidden", value = "http://prosody.im/protocol/modules#unload" }; |
679 |
596 { name = "modules", type = "list-multi", required = true, label = "Modules to be unloaded:"}; |
680 function unload_modules_handler(self, data, state) |
597 }; |
681 local layout = dataforms_new { |
598 |
682 title = "Unload modules"; |
599 local unload_modules_handler = adhoc_initial(unload_modules_layout, function() |
683 instructions = "Select the modules to be unloaded"; |
600 return { modules = array.collect(keys(hosts[module_host].modules)):sort() }; |
684 |
601 end, function(fields, err) |
685 { name = "FORM_TYPE", type = "hidden", value = "http://prosody.im/protocol/modules#unload" }; |
602 if err then |
686 { name = "modules", type = "list-multi", required = true, label = "Modules to be unloaded:"}; |
603 return generate_error_message(err); |
687 }; |
604 end |
688 if state then |
605 local ok_list, err_list = {}, {}; |
689 if data.action == "cancel" then |
606 for _, module in ipairs(fields.modules) do |
690 return { status = "canceled" }; |
607 local ok, err = modulemanager.unload(module_host, module); |
691 end |
608 if ok then |
692 local fields, err = layout:data(data.form); |
609 ok_list[#ok_list + 1] = module; |
693 if err then |
610 else |
694 return generate_error_message(err); |
611 err_list[#err_list + 1] = module .. "(Error: " .. tostring(err) .. ")"; |
695 end |
612 end |
696 local ok_list, err_list = {}, {}; |
613 end |
697 for _, module in ipairs(fields.modules) do |
614 local info = (#ok_list > 0 and ("The following modules were successfully unloaded on host "..module_host..":\n"..t_concat(ok_list, "\n")) or "") |
698 local ok, err = modulemanager.unload(data.to, module); |
615 .. ((#ok_list > 0 and #err_list > 0) and "\n" or "") .. |
|
616 (#err_list > 0 and ("Failed to unload the following modules on host "..module_host..":\n"..t_concat(err_list, "\n")) or ""); |
|
617 return { status = "completed", info = info }; |
|
618 end); |
|
619 |
|
620 -- Globally unloading a module |
|
621 local globally_unload_module_layout = dataforms_new { |
|
622 title = "Globally unload module"; |
|
623 instructions = "Specify a module to unload on all hosts"; |
|
624 |
|
625 { name = "FORM_TYPE", type = "hidden", value = "http://prosody.im/protocol/modules#global-unload" }; |
|
626 { name = "module", type = "list-single", required = true, label = "Module to globally unload:"}; |
|
627 }; |
|
628 |
|
629 local globally_unload_module_handler = adhoc_initial(globally_unload_module_layout, function() |
|
630 local loaded_modules = array(keys(modulemanager.get_modules("*"))); |
|
631 for _, host in pairs(hosts) do |
|
632 loaded_modules:append(array(keys(host.modules))); |
|
633 end |
|
634 loaded_modules = array(keys(set.new(loaded_modules):items())):sort(); |
|
635 return { module = loaded_modules }; |
|
636 end, function(fields, err) |
|
637 local is_global = false; |
|
638 if err then |
|
639 return generate_error_message(err); |
|
640 end |
|
641 |
|
642 if modulemanager.is_loaded("*", fields.module) then |
|
643 local ok, err = modulemanager.unload("*", fields.module); |
|
644 if not ok then |
|
645 return { status = "completed", info = 'Global module '..fields.module..' failed to unload: '..err }; |
|
646 end |
|
647 is_global = true; |
|
648 end |
|
649 |
|
650 local ok_list, err_list = {}, {}; |
|
651 for host_name, host in pairs(hosts) do |
|
652 if modulemanager.is_loaded(host_name, fields.module) then |
|
653 local ok, err = modulemanager.unload(host_name, fields.module); |
699 if ok then |
654 if ok then |
700 ok_list[#ok_list + 1] = module; |
655 ok_list[#ok_list + 1] = host_name; |
701 else |
656 else |
702 err_list[#err_list + 1] = module .. "(Error: " .. tostring(err) .. ")"; |
657 err_list[#err_list + 1] = host_name .. " (Error: " .. tostring(err) .. ")"; |
703 end |
658 end |
704 end |
659 end |
705 local info = (#ok_list > 0 and ("The following modules were successfully unloaded on host "..data.to..":\n"..t_concat(ok_list, "\n")) or "") |
660 end |
706 .. ((#ok_list > 0 and #err_list > 0) and "\n" or "") .. |
661 |
707 (#err_list > 0 and ("Failed to unload the following modules on host "..data.to..":\n"..t_concat(err_list, "\n")) or ""); |
662 if #ok_list == 0 and #err_list == 0 then |
708 return { status = "completed", info = info }; |
663 if is_global then |
709 else |
664 return { status = "completed", info = 'Successfully unloaded global module '..fields.module }; |
710 local modules = array.collect(keys(hosts[data.to].modules)):sort(); |
|
711 return { status = "executing", actions = {"next", "complete", default = "complete"}, form = { layout = layout; values = { modules = modules } } }, "executing"; |
|
712 end |
|
713 end |
|
714 |
|
715 local function globally_unload_module_handler(self, data, state) |
|
716 local layout = dataforms_new { |
|
717 title = "Globally unload module"; |
|
718 instructions = "Specify a module to unload on all hosts"; |
|
719 |
|
720 { name = "FORM_TYPE", type = "hidden", value = "http://prosody.im/protocol/modules#global-unload" }; |
|
721 { name = "module", type = "list-single", required = true, label = "Module to globally unload:"}; |
|
722 }; |
|
723 if state then |
|
724 if data.action == "cancel" then |
|
725 return { status = "canceled" }; |
|
726 end |
|
727 |
|
728 local is_global = false; |
|
729 local fields, err = layout:data(data.form); |
|
730 if err then |
|
731 return generate_error_message(err); |
|
732 end |
|
733 |
|
734 if modulemanager.is_loaded("*", fields.module) then |
|
735 local ok, err = modulemanager.unload("*", fields.module); |
|
736 if not ok then |
|
737 return { status = "completed", info = 'Global module '..fields.module..' failed to unload: '..err }; |
|
738 end |
|
739 is_global = true; |
|
740 end |
|
741 |
|
742 local ok_list, err_list = {}, {}; |
|
743 for host_name, host in pairs(hosts) do |
|
744 if modulemanager.is_loaded(host_name, fields.module) then |
|
745 local ok, err = modulemanager.unload(host_name, fields.module); |
|
746 if ok then |
|
747 ok_list[#ok_list + 1] = host_name; |
|
748 else |
|
749 err_list[#err_list + 1] = host_name .. " (Error: " .. tostring(err) .. ")"; |
|
750 end |
|
751 end |
|
752 end |
|
753 |
|
754 if #ok_list == 0 and #err_list == 0 then |
|
755 if is_global then |
|
756 return { status = "completed", info = 'Successfully unloaded global module '..fields.module }; |
|
757 else |
|
758 return { status = "completed", info = 'Module '..fields.module..' not loaded on any host.' }; |
|
759 end |
|
760 end |
|
761 |
|
762 local info = (#ok_list > 0 and ("The module "..fields.module.." was successfully unloaded on the hosts:\n"..t_concat(ok_list, "\n")) or "") |
|
763 .. ((#ok_list > 0 and #err_list > 0) and "\n" or "") .. |
|
764 (#err_list > 0 and ("Failed to unload the module "..fields.module.." on the hosts:\n"..t_concat(err_list, "\n")) or ""); |
|
765 return { status = "completed", info = info }; |
|
766 else |
|
767 local loaded_modules = array(keys(modulemanager.get_modules("*"))); |
|
768 for _, host in pairs(hosts) do |
|
769 loaded_modules:append(array(keys(host.modules))); |
|
770 end |
|
771 loaded_modules = array(keys(set.new(loaded_modules):items())):sort(); |
|
772 return { status = "executing", actions = {"next", "complete", default = "complete"}, form = { layout = layout, values = { module = loaded_modules } } }, "executing"; |
|
773 end |
|
774 end |
|
775 |
|
776 |
|
777 function activate_host_handler(self, data, state) |
|
778 local layout = dataforms_new { |
|
779 title = "Activate host"; |
|
780 instructions = ""; |
|
781 |
|
782 { name = "FORM_TYPE", type = "hidden", value = "http://prosody.im/protocol/hosts#activate" }; |
|
783 { name = "host", type = "text-single", required = true, label = "Host:"}; |
|
784 }; |
|
785 if state then |
|
786 if data.action == "cancel" then |
|
787 return { status = "canceled" }; |
|
788 end |
|
789 local fields, err = layout:data(data.form); |
|
790 if err then |
|
791 return generate_error_message(err); |
|
792 end |
|
793 local ok, err = hostmanager_activate(fields.host); |
|
794 |
|
795 if ok then |
|
796 return { status = "completed", info = fields.host .. " activated" }; |
|
797 else |
665 else |
798 return { status = "canceled", error = err } |
666 return { status = "completed", info = 'Module '..fields.module..' not loaded on any host.' }; |
799 end |
667 end |
800 else |
668 end |
801 return { status = "executing", actions = {"next", "complete", default = "complete"}, form = { layout = layout } }, "executing"; |
669 |
802 end |
670 local info = (#ok_list > 0 and ("The module "..fields.module.." was successfully unloaded on the hosts:\n"..t_concat(ok_list, "\n")) or "") |
803 end |
671 .. ((#ok_list > 0 and #err_list > 0) and "\n" or "") .. |
804 |
672 (#err_list > 0 and ("Failed to unload the module "..fields.module.." on the hosts:\n"..t_concat(err_list, "\n")) or ""); |
805 function deactivate_host_handler(self, data, state) |
673 return { status = "completed", info = info }; |
806 local layout = dataforms_new { |
674 end); |
807 title = "Deactivate host"; |
675 |
808 instructions = ""; |
676 -- Activating a host |
809 |
677 local activate_host_layout = dataforms_new { |
810 { name = "FORM_TYPE", type = "hidden", value = "http://prosody.im/protocol/hosts#activate" }; |
678 title = "Activate host"; |
811 { name = "host", type = "text-single", required = true, label = "Host:"}; |
679 instructions = ""; |
812 }; |
680 |
813 if state then |
681 { name = "FORM_TYPE", type = "hidden", value = "http://prosody.im/protocol/hosts#activate" }; |
814 if data.action == "cancel" then |
682 { name = "host", type = "text-single", required = true, label = "Host:"}; |
815 return { status = "canceled" }; |
683 }; |
816 end |
684 |
817 local fields, err = layout:data(data.form); |
685 local activate_host_handler = adhoc_simple(activate_host_layout, function(fields, err) |
818 if err then |
686 if err then |
819 return generate_error_message(err); |
687 return generate_error_message(err); |
820 end |
688 end |
821 local ok, err = hostmanager_deactivate(fields.host); |
689 local ok, err = hostmanager_activate(fields.host); |
822 |
690 |
823 if ok then |
691 if ok then |
824 return { status = "completed", info = fields.host .. " deactivated" }; |
692 return { status = "completed", info = fields.host .. " activated" }; |
825 else |
693 else |
826 return { status = "canceled", error = err } |
694 return { status = "canceled", error = err } |
827 end |
695 end |
828 else |
696 end); |
829 return { status = "executing", actions = {"next", "complete", default = "complete"}, form = { layout = layout } }, "executing"; |
697 |
830 end |
698 -- Deactivating a host |
831 end |
699 local deactivate_host_layout = dataforms_new { |
|
700 title = "Deactivate host"; |
|
701 instructions = ""; |
|
702 |
|
703 { name = "FORM_TYPE", type = "hidden", value = "http://prosody.im/protocol/hosts#activate" }; |
|
704 { name = "host", type = "text-single", required = true, label = "Host:"}; |
|
705 }; |
|
706 |
|
707 local deactivate_host_handler = adhoc_simple(deactivate_host_layout, function(fields, err) |
|
708 if err then |
|
709 return generate_error_message(err); |
|
710 end |
|
711 local ok, err = hostmanager_deactivate(fields.host); |
|
712 |
|
713 if ok then |
|
714 return { status = "completed", info = fields.host .. " deactivated" }; |
|
715 else |
|
716 return { status = "canceled", error = err } |
|
717 end |
|
718 end); |
832 |
719 |
833 |
720 |
834 local add_user_desc = adhoc_new("Add User", "http://jabber.org/protocol/admin#add-user", add_user_command_handler, "admin"); |
721 local add_user_desc = adhoc_new("Add User", "http://jabber.org/protocol/admin#add-user", add_user_command_handler, "admin"); |
835 local change_user_password_desc = adhoc_new("Change User Password", "http://jabber.org/protocol/admin#change-user-password", change_user_password_command_handler, "admin"); |
722 local change_user_password_desc = adhoc_new("Change User Password", "http://jabber.org/protocol/admin#change-user-password", change_user_password_command_handler, "admin"); |
836 local config_reload_desc = adhoc_new("Reload configuration", "http://prosody.im/protocol/config#reload", config_reload_handler, "global_admin"); |
723 local config_reload_desc = adhoc_new("Reload configuration", "http://prosody.im/protocol/config#reload", config_reload_handler, "global_admin"); |