-- Prosody IM

-- Copyright (C) 2008-2010 Matthew Wild

-- Copyright (C) 2008-2010 Waqas Hussain

-- 

-- This project is MIT/X11 licensed. Please see the

-- COPYING file in the source package for more information.

--

local config = require "core.configmanager";

local encodings = require "util.encodings";

local stringprep = encodings.stringprep;

local usermanager = require "core.usermanager";

local signal = require "util.signal";

local set = require "util.set";

local lfs = require "lfs";

local nodeprep, nameprep = stringprep.nodeprep, stringprep.nameprep;

local io, os = io, os;

local tostring, tonumber = tostring, tonumber;

local CFG_SOURCEDIR = _G.CFG_SOURCEDIR;

local prosody = prosody;

module "prosodyctl"

function adduser(params)

	local user, host, password = nodeprep(params.user), nameprep(params.host), params.password;

	if not user then

		return false, "invalid-username";

	elseif not host then

		return false, "invalid-hostname";

	end

	local provider = prosody.hosts[host].users;

	if not(provider) or provider.name == "null" then

		usermanager.initialize_host(host);

	end

	local ok = usermanager.create_user(user, password, host);

	if not ok then

		return false, "unable-to-save-data";

	end

	return true;

end

function user_exists(params)

	local user, host, password = nodeprep(params.user), nameprep(params.host), params.password;

	local provider = prosody.hosts[host].users;

	if not(provider) or provider.name == "null" then

		usermanager.initialize_host(host);

	end

	return usermanager.user_exists(user, host);

end

function passwd(params)

	if not _M.user_exists(params) then

		return false, "no-such-user";

	end

	return _M.adduser(params);

end

function deluser(params)

	if not _M.user_exists(params) then

		return false, "no-such-user";

	end

	params.password = nil;

	return _M.adduser(params);

end

function getpid()

	local pidfile = config.get("*", "core", "pidfile");

	if not pidfile then

		return false, "no-pidfile";

	end

	local modules_enabled = set.new(config.get("*", "core", "modules_enabled"));

	if not modules_enabled:contains("posix") then

		return false, "no-posix";

	end

	local file, err = io.open(pidfile, "r+");

	if not file then

		return false, "pidfile-read-failed", err;

	end

	local locked, err = lfs.lock(file, "w");

	if locked then

		file:close();

		return false, "pidfile-not-locked";

	end

	local pid = tonumber(file:read("*a"));

	file:close();

	if not pid then

		return false, "invalid-pid";

	end

	return true, pid;

end

function isrunning()

	local ok, pid, err = _M.getpid();

	if not ok then

		if pid == "pidfile-read-failed" or pid == "pidfile-not-locked" then

			-- Report as not running, since we can't open the pidfile

			-- (it probably doesn't exist)

			return true, false;

		end

		return ok, pid;

	end

	return true, signal.kill(pid, 0) == 0;

end

function start()

	local ok, ret = _M.isrunning();

	if not ok then

		return ok, ret;

	end

	if ret then

		return false, "already-running";

	end

	if not CFG_SOURCEDIR then

		os.execute("./prosody");

	else

		os.execute(CFG_SOURCEDIR.."/../../bin/prosody");

	end

	return true;

end

function stop()

	local ok, ret = _M.isrunning();

	if not ok then

		return ok, ret;

	end

	if not ret then

		return false, "not-running";

	end

	local ok, pid = _M.getpid()

	if not ok then return false, pid; end

	signal.kill(pid, signal.SIGTERM);

	return true;

end