mirrors/prosody
1
0
Fork 0
mirror of https://github.com/bjc/prosody.git synced 2025-04-04 13:47:41 +03:00
Code Issues Projects Releases Packages Wiki Activity Actions

usermanager: Add create_user_with_role() method to atomically set initial role

Browse source
This commit is contained in:
Matthew Wild 2023-06-29 15:31:46 +01:00
parent b6e5085b6a
commit 641f070a9d
1 changed files with 26 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

26
core/usermanager.lua
View file

@ -205,6 +205,31 @@ local function set_user_role(user, host, role_name)
return role, err;
end
local function create_user_with_role(username, password, host, role)
local ok, err = create_user(username, nil, host);
if not ok then return ok, err; end
local role_ok, role_err = set_user_role(username, host, role);
if not role_ok then
delete_user(username, host);
return nil, "Failed to assign role: "..role_err;
end
if password then
local pw_ok, pw_err = set_password(username, password, host);
if not pw_ok then
return nil, "Failed to set password: "..pw_err;
end
local enable_ok, enable_err = enable_user(username, host);
if not enable_ok and enable_err ~= "method not implemented" then
return enable_ok, "Failed to enable account: "..enable_err;
end
end
return true;
end
local function user_can_assume_role(user, host, role_name)
if host and not hosts[host] then return false; end
if type(user) ~= "string" then return false; end
@ -308,6 +333,7 @@ return {
get_account_info = get_account_info;
user_exists = user_exists;
create_user = create_user;
create_user_with_role = create_user_with_role;
delete_user = delete_user;
user_is_enabled = user_is_enabled;
enable_user = enable_user;