mod_storage_xep0227: Add API to iterate all stores of a user

1 files changed, 23 insertions, 7 deletions

diff --git a/plugins/mod_storage_xep0227.lua b/plugins/mod_storage_xep0227.lua
index 45cfaa77..7b039f81 100644
--- a/plugins/mod_storage_xep0227.lua
+++ b/plugins/mod_storage_xep0227.lua
@@ -688,6 +688,18 @@ function driver:open_xep0227(datastore, typ, options)
 	return instance;
 end
 
+local function get_store_names_from_xml(self, user_xml)
+	local stores = set.new();
+	for handler_name, handler_funcs in pairs(handlers) do
+		if handler_funcs._stores then
+			stores:include(handler_funcs._stores(self, user_xml));
+		else
+			stores:include(handler_name);
+		end
+	end
+	return stores;
+end
+
 local function get_store_names(self, path)
 	local stores = set.new();
 	local f, err = io_open(paths.join(prosody.paths.data, path));
@@ -698,13 +710,8 @@ local function get_store_names(self, path)
 	module:log("info", "Loaded %s", path);
 	local s = f:read("*a");
 	f:close();
-	local xml = parse_xml_real(s);
-	for _, handler_funcs in pairs(handlers) do
-		if handler_funcs._stores then
-			stores:include(handler_funcs._stores(self, xml));
-		end
-	end
-	return stores;
+	local user_xml = parse_xml_real(s);
+	return get_store_names_from_xml(self, user_xml);
 end
 
 function driver:stores(username)
@@ -733,4 +740,13 @@ function driver:stores(username)
 	return store_names:items();
 end
 
+function driver:xep0227_user_stores(username, host)
+	local user_xml = self:_get_user_xml(username, host);
+	if not user_xml then
+		return nil;
+	end
+	local store_names = get_store_names_from_xml(username, host);
+	return store_names:items();
+end
+
 module:provides("storage", driver);