net.server_event: Add setquitting() function to be compatible with net.server_select

[prosody.git] / prosodyctl

diff --git a/prosodyctl b/prosodyctl
index 2557d8652b1b8986db2ea09081a12281427e3e77..575948a1745833eb7ef6d9443bd37469efa486ca 100755 (executable)
--- a/prosodyctl
+++ b/prosodyctl
@@ -1,5 +1,5 @@
 #!/usr/bin/env lua
--- Prosody IM v0.4
+-- Prosody IM
 -- Copyright (C) 2008-2009 Matthew Wild
 -- Copyright (C) 2008-2009 Waqas Hussain
 -- 
@@ -32,7 +32,6 @@ end
 -- Required to be able to find packages installed with luarocks
 pcall(require, "luarocks.require")
 
-
 config = require "core.configmanager"
 
 do
@@ -66,19 +65,28 @@ require "util.datamanager".set_data_path(data_path);
 
 -- Switch away from root and into the prosody user --
 local switched_user, current_uid;
+
+local want_pposix_version = "0.3.1";
 local ok, pposix = pcall(require, "util.pposix");
+
 if ok and pposix then
+       if pposix._VERSION ~= want_pposix_version then print(string.format("Unknown version (%s) of binary pposix module, expected %s", tostring(pposix._VERSION), want_pposix_version)); return; end
        current_uid = pposix.getuid();
        if current_uid == 0 then
                -- We haz root!
                local desired_user = config.get("*", "core", "prosody_user") or "prosody";
-               local ok, err = pposix.setuid(desired_user);
+               local desired_group = config.get("*", "core", "prosody_group") or desired_user;
+               local ok, err = pposix.setgid(desired_group);
                if ok then
-                       -- Yay!
-                       switched_user = true;
-               else
+                       ok, err = pposix.setuid(desired_user);
+                       if ok then
+                               -- Yay!
+                               switched_user = true;
+                       end
+               end
+               if not switched_user then
                        -- Boo!
-                       print("Warning: Couldn't switch to Prosody user '"..tostring(desired_user).."': "..tostring(err));
+                       print("Warning: Couldn't switch to Prosody user/group '"..tostring(desired_user).."'/'"..tostring(desired_group).."': "..tostring(err));
                end
        end
 else
@@ -93,15 +101,20 @@ local error_messages = setmetatable({
                ["no-password"] = "No password was supplied";
                ["no-such-user"] = "The given user does not exist on the server";
                ["unable-to-save-data"] = "Unable to store, perhaps you don't have permission?";
-               ["no-pidfile"] = "There is no pidfile option in the configuration file, see http://prosody.im/doc/prosodyctl#pidfile for help";
+               ["no-pidfile"] = "There is no 'pidfile' option in the configuration file, see http://prosody.im/doc/prosodyctl#pidfile for help";
                ["no-such-method"] = "This module has no commands";
                ["not-running"] = "Prosody is not running";
                }, { __index = function (t,k) return "Error: "..(tostring(k):gsub("%-", " "):gsub("^.", string.upper)); end });
 
+local events = require "util.events".new();
+
 hosts = {};
+prosody = { hosts = hosts, events = events };
 
-require "core.hostmanager"
-require "core.eventmanager".fire_event("server-starting");
+for hostname, config in pairs(config.getconfig()) do
+       hosts[hostname] = { events = events };
+end
+       
 require "core.modulemanager"
 
 require "util.prosodyctl"
@@ -125,17 +138,21 @@ end
 
 local function getchar(n)
        os.execute("stty raw -echo");
-       local char = io.read(n or 1);
+       local ok, char = pcall(io.read, n or 1);
        os.execute("stty sane");
-       return char;
+       if ok then
+               return char;
+       end
 end
        
 local function getpass()
        os.execute("stty -echo");
-       local pass = io.read("*l");
+       local ok, pass = pcall(io.read, "*l");
        os.execute("stty sane");
        io.write("\n");
-       return pass;
+       if ok then
+               return pass;
+       end
 end
 
 function show_yesno(prompt)
@@ -154,6 +171,10 @@ local function read_password()
        while true do
                io.write("Enter new password: ");
                password = getpass();
+               if not password then
+                       show_message("No password - cancelled");
+                       return;
+               end
                io.write("Retype new password: ");
                if getpass() ~= password then
                        if not show_yesno [=[Passwords did not match, try again? [Y/n]]=] then
@@ -441,6 +462,32 @@ function commands.unregister(arg)
        return 1;
 end
 
+local http_errors = {
+       [404] = "Plugin not found, did you type the address correctly?"
+       };
+
+function commands.addplugin(arg)
+       local url = arg[1];
+       if url:match("^http://") then
+               local http = require "socket.http";
+               show_message("Fetching...");
+               local code, err = http.request(url);
+               if not code or not tostring(err):match("^[23]") then
+                       show_message("Failed: "..(http_errors[err] or ("HTTP error "..err)));
+                       return 1;
+               end
+               if url:match("%.lua$") then
+                       local ok, err = datamanager.store(url:match("/mod_([^/]+)$"), "*", "plugins", {code});
+                       if not ok then
+                               show_message("Failed to save to data store: "..err);
+                               return 1;
+                       end
+               end
+               show_message("Saved. Don't forget to load the module using the config file or admin console!");
+       else
+               show_message("Sorry, I don't understand how to fetch plugins from there.");
+       end
+end
 
 ---------------------