authz: Add method for retrieving all roles

Some of the OAuth stuff highlights a small need to retrieve a list of
roles somehow. Handy if you ever need a role selector in adhoc or
something.

Unless there's some O(n) thing we were avoiding?

2 files changed, 11 insertions, 0 deletions

diff --git a/core/usermanager.lua b/core/usermanager.lua
index 3afb5037..5fe8022f 100644
--- a/core/usermanager.lua
+++ b/core/usermanager.lua
@@ -53,6 +53,7 @@ local fallback_authz_provider = {
 	get_users_with_role = function (role_name) end;
 	add_default_permission = function (role_name, action, policy) end;
 	get_role_by_name = function (role_name) end;
+	get_all_roles = function () end;
 };
 
 local provider_mt = { __index = new_null_provider() };
@@ -293,6 +294,11 @@ local function get_role_by_name(role_name, host)
 	return hosts[host].authz.get_role_by_name(role_name);
 end
 
+local function get_all_roles(host)
+	if host and not hosts[host] then return false; end
+	return hosts[host].authz.get_all_roles();
+end
+
 return {
 	new_null_provider = new_null_provider;
 	initialize_host = initialize_host;
@@ -320,6 +326,7 @@ return {
 	set_jid_role = set_jid_role;
 	get_jids_with_role = get_jids_with_role;
 	get_role_by_name = get_role_by_name;
+	get_all_roles = get_all_roles;
 
 	-- Deprecated
 	is_admin = is_admin;
diff --git a/plugins/mod_authz_internal.lua b/plugins/mod_authz_internal.lua
index c2895613..7e01e689 100644
--- a/plugins/mod_authz_internal.lua
+++ b/plugins/mod_authz_internal.lua
@@ -280,6 +280,10 @@ function get_role_by_name(role_name)
 	return assert(role_registry[role_name], role_name);
 end
 
+function get_all_roles()
+	return role_registry;
+end
+
 -- COMPAT: Migrate from 0.12 role storage
 local function do_migration(migrate_host)
 	local old_role_store = assert(module:context(migrate_host):open_store("roles"));