Comparison

mod_storage_xep0227: Add API to iterate all stores of a user

comparison

equal deleted inserted replaced

-:e77c938ed92b
+:326f5466ddc7
 		instance._get_user_xml = assert(options.get_user_xml);
 	end
 	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));
 	if not f then
 		module:log("warn", "Unable to load XML file for <%s>: %s", "store listing", err);
 		return stores;
 	end
 	module:log("info", "Loaded %s", path);
 	local s = f:read("*a");
 	f:close();
-	local xml = parse_xml_real(s);
+	local user_xml = parse_xml_real(s);
-	for _, handler_funcs in pairs(handlers) do
+	return get_store_names_from_xml(self, user_xml);
-		if handler_funcs._stores then
-			stores:include(handler_funcs._stores(self, xml));
-		end
-	end
-	return stores;
 end
 function driver:stores(username)
 	local store_dir = prosody.paths.data;
 	end
 	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);