mod_storage_internal, datamanager: Add support for iterating over users with data in a store
--- a/plugins/mod_storage_internal.lua Wed Sep 19 23:29:25 2012 +0200
+++ b/plugins/mod_storage_internal.lua Fri Sep 21 17:23:08 2012 +0200
@@ -5,8 +5,8 @@
local driver = {};
local driver_mt = { __index = driver };
-function driver:open(store)
- return setmetatable({ store = store }, driver_mt);
+function driver:open(store, typ)
+ return setmetatable({ store = store, type = typ }, driver_mt);
end
function driver:get(user)
return datamanager.load(user, host, self.store);
@@ -20,6 +20,10 @@
return datamanager.stores(username, host);
end
+function driver:users()
+ return datamanager.users(host, self.store, self.type);
+end
+
function driver:purge(user)
return datamanager.purge(user, host);
end
--- a/util/datamanager.lua Wed Sep 19 23:29:25 2012 +0200
+++ b/util/datamanager.lua Fri Sep 21 17:23:08 2012 +0200
@@ -282,6 +282,25 @@
list = "list";
}
+function users(host, store, typ)
+ typ = type_map[typ or "keyval"];
+ local store_dir = format("%s/%s/%s", data_path, encode(host), encode(store));
+
+ local mode, err = lfs.attributes(store_dir, "mode");
+ if not mode then
+ return function() log("debug", err or (store_dir .. " does not exist")) end
+ end
+ local next, state = lfs.dir(store_dir);
+ return function(state)
+ for node in next, state do
+ local file, ext = node:match("^(.*)%.([dalist]+)$");
+ if file and ext == typ then
+ return decode(file);
+ end
+ end
+ end, state;
+end
+
function stores(username, host, typ)
typ = type_map[typ or "keyval"];
local store_dir = format("%s/%s/", data_path, encode(host));