mod_storage_sql: Return connection from connect even if already connected (thanks...
[prosody.git] / plugins / mod_admin_adhoc.lua
index 37930c013d4beb5ab21a76e57651f012b2e68367..50493abeead7f8b652ae9c6fd1aec87e87d3e64f 100644 (file)
@@ -1,4 +1,4 @@
--- Copyright (C) 2009-2010 Florian Zeitz
+-- Copyright (C) 2009-2011 Florian Zeitz
 --
 -- This file is MIT/X11 licensed. Please see the
 -- COPYING file in the source package for more information.
@@ -10,7 +10,8 @@ local prosody = _G.prosody;
 local hosts = prosody.hosts;
 local t_concat = table.concat;
 
-require "util.iterators";
+local iterators = require "util.iterators";
+local keys, values = iterators.keys, iterators.values;
 local usermanager_user_exists = require "core.usermanager".user_exists;
 local usermanager_create_user = require "core.usermanager".create_user;
 local usermanager_get_password = require "core.usermanager".get_password;
@@ -22,9 +23,19 @@ local timer_add_task = require "util.timer".add_task;
 local dataforms_new = require "util.dataforms".new;
 local array = require "util.array";
 local modulemanager = require "modulemanager";
+local core_post_stanza = prosody.core_post_stanza;
 
+module:depends("adhoc");
 local adhoc_new = module:require "adhoc".new;
 
+local function generate_error_message(errors)
+       local errmsg = {};
+       for name, err in pairs(errors) do
+               errmsg[#errmsg + 1] = name .. ": " .. err;
+       end
+       return { status = "completed", error = { message = t_concat(errmsg, "\n") } };
+end
+
 function add_user_command_handler(self, data, state)
        local add_user_layout = dataforms_new{
                title = "Adding a User";
@@ -40,9 +51,9 @@ function add_user_command_handler(self, data, state)
                if data.action == "cancel" then
                        return { status = "canceled" };
                end
-               local fields = add_user_layout:data(data.form);
-               if not fields.accountjid then
-                       return { status = "completed", error = { message = "You need to specify a JID." } };
+               local fields, err = add_user_layout:data(data.form);
+               if err then
+                       return generate_error_message(err);
                end
                local username, host, resource = jid.split(fields.accountjid);
                if data.to ~= host then
@@ -53,15 +64,14 @@ function add_user_command_handler(self, data, state)
                                return { status = "completed", error = { message = "Account already exists" } };
                        else
                                if usermanager_create_user(username, fields.password, host) then
-                                       module:log("info", "Created new account " .. username.."@"..host);
+                                       module:log("info", "Created new account %s@%s", username, host);
                                        return { status = "completed", info = "Account successfully created" };
                                else
                                        return { status = "completed", error = { message = "Failed to write data to disk" } };
                                end
                        end
                else
-                       module:log("debug", (fields.accountjid or "<nil>") .. " " .. (fields.password or "<nil>") .. " "
-                               .. (fields["password-verify"] or "<nil>"));
+                       module:log("debug", "Invalid data, password mismatch or empty username while creating account for %s", fields.accountjid or "<nil>");
                        return { status = "completed", error = { message = "Invalid data.\nPassword mismatch, or empty username" } };
                end
        else
@@ -83,9 +93,9 @@ function change_user_password_command_handler(self, data, state)
                if data.action == "cancel" then
                        return { status = "canceled" };
                end
-               local fields = change_user_password_layout:data(data.form);
-               if not fields.accountjid or fields.accountjid == "" or not fields.password then
-                       return { status = "completed", error = { message = "Please specify username and password" } };
+               local fields, err = change_user_password_layout:data(data.form);
+               if err then
+                       return generate_error_message(err);
                end
                local username, host, resource = jid.split(fields.accountjid);
                if data.to ~= host then
@@ -101,6 +111,16 @@ function change_user_password_command_handler(self, data, state)
        end
 end
 
+function config_reload_handler(self, data, state)
+       local ok, err = prosody.reload_config();
+       if ok then
+               return { status = "completed", info = "Configuration reloaded (modules may need to be reloaded for this to have an effect)" };
+       else
+               return { status = "completed", error = { message = "Failed to reload config: " .. tostring(err) } };
+       end
+end
+
+
 function delete_user_command_handler(self, data, state)
        local delete_user_layout = dataforms_new{
                title = "Deleting a User";
@@ -114,16 +134,19 @@ function delete_user_command_handler(self, data, state)
                if data.action == "cancel" then
                        return { status = "canceled" };
                end
-               local fields = delete_user_layout:data(data.form);
+               local fields, err = delete_user_layout:data(data.form);
+               if err then
+                       return generate_error_message(err);
+               end
                local failed = {};
                local succeeded = {};
                for _, aJID in ipairs(fields.accountjids) do
                        local username, host, resource = jid.split(aJID);
                        if (host == data.to) and  usermanager_user_exists(username, host) and disconnect_user(aJID) and usermanager_create_user(username, nil, host) then
-                               module:log("debug", "User " .. aJID .. " has been deleted");
+                               module:log("debug", "User %s has been deleted", aJID);
                                succeeded[#succeeded+1] = aJID;
                        else
-                               module:log("debug", "Tried to delete non-existant user "..aJID);
+                               module:log("debug", "Tried to delete non-existant user %s", aJID);
                                failed[#failed+1] = aJID;
                        end
                end
@@ -142,7 +165,7 @@ function disconnect_user(match_jid)
        local sessions = host.sessions[node] and host.sessions[node].sessions;
        for resource, session in pairs(sessions or {}) do
                if not givenResource or (resource == givenResource) then
-                       module:log("debug", "Disconnecting "..node.."@"..hostname.."/"..resource);
+                       module:log("debug", "Disconnecting %s@%s/%s", node, hostname, resource);
                        session:close();
                end
        end
@@ -163,7 +186,10 @@ function end_user_session_handler(self, data, state)
                        return { status = "canceled" };
                end
 
-               local fields = end_user_session_layout:data(data.form);
+               local fields, err = end_user_session_layout:data(data.form);
+               if err then
+                       return generate_error_message(err);
+               end
                local failed = {};
                local succeeded = {};
                for _, aJID in ipairs(fields.accountjids) do
@@ -211,9 +237,9 @@ function get_user_password_handler(self, data, state)
                if data.action == "cancel" then
                        return { status = "canceled" };
                end
-               local fields = get_user_password_layout:data(data.form);
-               if not fields.accountjid then
-                       return { status = "completed", error = { message = "Please specify a JID." } };
+               local fields, err = get_user_password_layout:data(data.form);
+               if err then
+                       return generate_error_message(err);
                end
                local user, host, resource = jid.split(fields.accountjid);
                local accountjid = "";
@@ -249,10 +275,10 @@ function get_user_roster_handler(self, data, state)
                        return { status = "canceled" };
                end
 
-               local fields = get_user_roster_layout:data(data.form);
+               local fields, err = get_user_roster_layout:data(data.form);
 
-               if not fields.accountjid then
-                       return { status = "completed", error = { message = "Please specify a JID" } };
+               if err then
+                       return generate_error_message(err);
                end
 
                local user, host, resource = jid.split(fields.accountjid);
@@ -311,10 +337,10 @@ function get_user_stats_handler(self, data, state)
                        return { status = "canceled" };
                end
 
-               local fields = get_user_stats_layout:data(data.form);
+               local fields, err = get_user_stats_layout:data(data.form);
 
-               if not fields.accountjid then
-                       return { status = "completed", error = { message = "Please specify a JID." } };
+               if err then
+                       return generate_error_message(err);
                end
 
                local user, host, resource = jid.split(fields.accountjid);
@@ -364,7 +390,11 @@ function get_online_users_command_handler(self, data, state)
                        return { status = "canceled" };
                end
 
-               local fields = get_online_users_layout:data(data.form);
+               local fields, err = get_online_users_layout:data(data.form);
+
+               if err then
+                       return generate_error_message(err);
+               end
 
                local max_items = nil
                if fields.max_items ~= "all" then
@@ -424,11 +454,9 @@ function load_module_handler(self, data, state)
                if data.action == "cancel" then
                        return { status = "canceled" };
                end
-               local fields = layout:data(data.form);
-               if (not fields.module) or (fields.module == "") then
-                       return { status = "completed", error = {
-                               message = "Please specify a module."
-                       } };
+               local fields, err = layout:data(data.form);
+               if err then
+                       return generate_error_message(err);
                end
                if modulemanager.is_loaded(data.to, fields.module) then
                        return { status = "completed", info = "Module already loaded" };
@@ -441,40 +469,41 @@ function load_module_handler(self, data, state)
                        '". Error was: "'..tostring(err or "<unspecified>")..'"' } };
                end
        else
-               local modules = array.collect(keys(hosts[data.to].modules)):sort();
                return { status = "executing", form = layout }, "executing";
        end
 end
 
--- TODO: Allow reloading multiple modules (depends on list-multi)
 function reload_modules_handler(self, data, state)
        local layout = dataforms_new {
-               title = "Reload module";
-               instructions = "Select the module to be reloaded";
+               title = "Reload modules";
+               instructions = "Select the modules to be reloaded";
 
                { name = "FORM_TYPE", type = "hidden", value = "http://prosody.im/protocol/modules#reload" };
-               { name = "module", type = "list-single", required = true, label = "Module to be reloaded:"};
+               { name = "modules", type = "list-multi", required = true, label = "Modules to be reloaded:"};
        };
        if state then
                if data.action == "cancel" then
                        return { status = "canceled" };
                end
-               local fields = layout:data(data.form);
-               if (not fields.module) or (fields.module == "") then
-                       return { status = "completed", error = {
-                               message = "Please specify a module. (This means your client misbehaved, as this field is required)"
-                       } };
+               local fields, err = layout:data(data.form);
+               if err then
+                       return generate_error_message(err);
                end
-               local ok, err = modulemanager.reload(data.to, fields.module);
-               if ok then
-                       return { status = "completed", info = 'Module "'..fields.module..'" successfully reloaded on host "'..data.to..'".' };
-               else
-                       return { status = "completed", error = { message = 'Failed to reload module "'..fields.module..'" on host "'..data.to..
-                       '". Error was: "'..tostring(err)..'"' } };
+               local ok_list, err_list = {}, {};
+               for _, module in ipairs(fields.modules) do
+                       local ok, err = modulemanager.reload(data.to, module);
+                       if ok then
+                               ok_list[#ok_list + 1] = module;
+                       else
+                               err_list[#err_list + 1] = module .. "(Error: " .. tostring(err) .. ")";
+                       end
                end
+               local info = (#ok_list > 0 and ("The following modules were successfully reloaded on host "..data.to..":\n"..t_concat(ok_list, "\n")) or "")..
+                       (#err_list > 0 and ("Failed to reload the following modules on host "..data.to..":\n"..t_concat(err_list, "\n")) or "");
+               return { status = "completed", info = info };
        else
                local modules = array.collect(keys(hosts[data.to].modules)):sort();
-               return { status = "executing", form = { layout = layout; values = { module = modules } } }, "executing";
+               return { status = "executing", form = { layout = layout; values = { modules = modules } } }, "executing";
        end
 end
 
@@ -522,7 +551,11 @@ function shut_down_service_handler(self, data, state)
                        return { status = "canceled" };
                end
 
-               local fields = shut_down_service_layout:data(data.form);
+               local fields, err = shut_down_service_layout:data(data.form);
+
+               if err then
+                       return generate_error_message(err);
+               end
 
                if fields.announcement and #fields.announcement > 0 then
                        local message = st.message({type = "headline"}, fields.announcement):up()
@@ -536,66 +569,68 @@ function shut_down_service_handler(self, data, state)
        else
                return { status = "executing", form = shut_down_service_layout }, "executing";
        end
-
-       return true;
 end
 
--- TODO: Allow unloading multiple modules (depends on list-multi)
 function unload_modules_handler(self, data, state)
        local layout = dataforms_new {
-               title = "Unload module";
-               instructions = "Select the module to be unloaded";
+               title = "Unload modules";
+               instructions = "Select the modules to be unloaded";
 
                { name = "FORM_TYPE", type = "hidden", value = "http://prosody.im/protocol/modules#unload" };
-               { name = "module", type = "list-single", required = true, label = "Module to be unloaded:"};
+               { name = "modules", type = "list-multi", required = true, label = "Modules to be unloaded:"};
        };
        if state then
                if data.action == "cancel" then
                        return { status = "canceled" };
                end
-               local fields = layout:data(data.form);
-               if (not fields.module) or (fields.module == "") then
-                       return { status = "completed", error = {
-                               message = "Please specify a module. (This means your client misbehaved, as this field is required)"
-                       } };
+               local fields, err = layout:data(data.form);
+               if err then
+                       return generate_error_message(err);
                end
-               local ok, err = modulemanager.unload(data.to, fields.module);
-               if ok then
-                       return { status = "completed", info = 'Module "'..fields.module..'" successfully unloaded on host "'..data.to..'".' };
-               else
-                       return { status = "completed", error = { message = 'Failed to unload module "'..fields.module..'" on host "'..data.to..
-                       '". Error was: "'..tostring(err)..'"' } };
+               local ok_list, err_list = {}, {};
+               for _, module in ipairs(fields.modules) do
+                       local ok, err = modulemanager.unload(data.to, module);
+                       if ok then
+                               ok_list[#ok_list + 1] = module;
+                       else
+                               err_list[#err_list + 1] = module .. "(Error: " .. tostring(err) .. ")";
+                       end
                end
+               local info = (#ok_list > 0 and ("The following modules were successfully unloaded on host "..data.to..":\n"..t_concat(ok_list, "\n")) or "")..
+                       (#err_list > 0 and ("Failed to unload the following modules on host "..data.to..":\n"..t_concat(err_list, "\n")) or "");
+               return { status = "completed", info = info };
        else
                local modules = array.collect(keys(hosts[data.to].modules)):sort();
-               return { status = "executing", form = { layout = layout; values = { module = modules } } }, "executing";
+               return { status = "executing", form = { layout = layout; values = { modules = modules } } }, "executing";
        end
 end
 
 local add_user_desc = adhoc_new("Add User", "http://jabber.org/protocol/admin#add-user", add_user_command_handler, "admin");
 local change_user_password_desc = adhoc_new("Change User Password", "http://jabber.org/protocol/admin#change-user-password", change_user_password_command_handler, "admin");
+local config_reload_desc = adhoc_new("Reload configuration", "http://prosody.im/protocol/config#reload", config_reload_handler, "global_admin");
 local delete_user_desc = adhoc_new("Delete User", "http://jabber.org/protocol/admin#delete-user", delete_user_command_handler, "admin");
 local end_user_session_desc = adhoc_new("End User Session", "http://jabber.org/protocol/admin#end-user-session", end_user_session_handler, "admin");
 local get_user_password_desc = adhoc_new("Get User Password", "http://jabber.org/protocol/admin#get-user-password", get_user_password_handler, "admin");
 local get_user_roster_desc = adhoc_new("Get User Roster","http://jabber.org/protocol/admin#get-user-roster", get_user_roster_handler, "admin");
 local get_user_stats_desc = adhoc_new("Get User Statistics","http://jabber.org/protocol/admin#user-stats", get_user_stats_handler, "admin");
-local get_online_users_desc = adhoc_new("Get List of Online Users", "http://jabber.org/protocol/admin#get-online-users", get_online_users_command_handler, "admin"); 
+local get_online_users_desc = adhoc_new("Get List of Online Users", "http://jabber.org/protocol/admin#get-online-users", get_online_users_command_handler, "admin");
 local list_modules_desc = adhoc_new("List loaded modules", "http://prosody.im/protocol/modules#list", list_modules_handler, "admin");
 local load_module_desc = adhoc_new("Load module", "http://prosody.im/protocol/modules#load", load_module_handler, "admin");
-local reload_modules_desc = adhoc_new("Reload module", "http://prosody.im/protocol/modules#reload", reload_modules_handler, "admin");
-local shut_down_service_desc = adhoc_new("Shut Down Service", "http://jabber.org/protocol/admin#shutdown", shut_down_service_handler, "admin");
-local unload_modules_desc = adhoc_new("Unload module", "http://prosody.im/protocol/modules#unload", unload_modules_handler, "admin");
-
-module:add_item("adhoc", add_user_desc);
-module:add_item("adhoc", change_user_password_desc);
-module:add_item("adhoc", delete_user_desc);
-module:add_item("adhoc", end_user_session_desc);
-module:add_item("adhoc", get_user_password_desc);
-module:add_item("adhoc", get_user_roster_desc);
-module:add_item("adhoc", get_user_stats_desc);
-module:add_item("adhoc", get_online_users_desc);
-module:add_item("adhoc", list_modules_desc);
-module:add_item("adhoc", load_module_desc);
-module:add_item("adhoc", reload_modules_desc);
-module:add_item("adhoc", shut_down_service_desc);
-module:add_item("adhoc", unload_modules_desc);
+local reload_modules_desc = adhoc_new("Reload modules", "http://prosody.im/protocol/modules#reload", reload_modules_handler, "admin");
+local shut_down_service_desc = adhoc_new("Shut Down Service", "http://jabber.org/protocol/admin#shutdown", shut_down_service_handler, "global_admin");
+local unload_modules_desc = adhoc_new("Unload modules", "http://prosody.im/protocol/modules#unload", unload_modules_handler, "admin");
+
+module:provides("adhoc", add_user_desc);
+module:provides("adhoc", change_user_password_desc);
+module:provides("adhoc", config_reload_desc);
+module:provides("adhoc", delete_user_desc);
+module:provides("adhoc", end_user_session_desc);
+module:provides("adhoc", get_user_password_desc);
+module:provides("adhoc", get_user_roster_desc);
+module:provides("adhoc", get_user_stats_desc);
+module:provides("adhoc", get_online_users_desc);
+module:provides("adhoc", list_modules_desc);
+module:provides("adhoc", load_module_desc);
+module:provides("adhoc", reload_modules_desc);
+module:provides("adhoc", shut_down_service_desc);
+module:provides("adhoc", unload_modules_desc);