aboutsummaryrefslogtreecommitdiffstats
path: root/plugins
diff options
context:
space:
mode:
Diffstat (limited to 'plugins')
-rw-r--r--plugins/mod_offline.lua28
-rw-r--r--plugins/mod_storage_internal.lua136
2 files changed, 135 insertions, 29 deletions
diff --git a/plugins/mod_offline.lua b/plugins/mod_offline.lua
index 08ab8490..fe4df573 100644
--- a/plugins/mod_offline.lua
+++ b/plugins/mod_offline.lua
@@ -7,29 +7,24 @@
--
-local datamanager = require "util.datamanager";
-local st = require "util.stanza";
local datetime = require "util.datetime";
-local ipairs = ipairs;
local jid_split = require "util.jid".split;
+local offline_messages = module:open_store("offline", "archive");
+
module:add_feature("msgoffline");
module:hook("message/offline/handle", function(event)
local origin, stanza = event.origin, event.stanza;
local to = stanza.attr.to;
- local node, host;
+ local node;
if to then
- node, host = jid_split(to)
+ node = jid_split(to)
else
- node, host = origin.username, origin.host;
+ node = origin.username;
end
- stanza.attr.stamp, stanza.attr.stamp_legacy = datetime.datetime(), datetime.legacy();
- local result = datamanager.list_append(node, host, "offline", st.preserialize(stanza));
- stanza.attr.stamp, stanza.attr.stamp_legacy = nil, nil;
-
- return result;
+ return offline_messages:append(node, nil, stanza);
end, -1);
module:hook("message/offline/broadcast", function(event)
@@ -37,15 +32,12 @@ module:hook("message/offline/broadcast", function(event)
local node, host = origin.username, origin.host;
- local data = datamanager.list_load(node, host, "offline");
+ local data = offline_messages:find(node);
if not data then return true; end
- for _, stanza in ipairs(data) do
- stanza = st.deserialize(stanza);
- stanza:tag("delay", {xmlns = "urn:xmpp:delay", from = host, stamp = stanza.attr.stamp}):up(); -- XEP-0203
- stanza:tag("x", {xmlns = "jabber:x:delay", from = host, stamp = stanza.attr.stamp_legacy}):up(); -- XEP-0091 (deprecated)
- stanza.attr.stamp, stanza.attr.stamp_legacy = nil, nil;
+ for _, stanza, when in data do
+ stanza:tag("delay", {xmlns = "urn:xmpp:delay", from = host, stamp = datetime.datetime(when)}):up(); -- XEP-0203
origin.send(stanza);
end
- datamanager.list_store(node, host, "offline", nil);
+ offline_messages:delete(node);
return true;
end, -1);
diff --git a/plugins/mod_storage_internal.lua b/plugins/mod_storage_internal.lua
index ade4f0a6..09760838 100644
--- a/plugins/mod_storage_internal.lua
+++ b/plugins/mod_storage_internal.lua
@@ -1,34 +1,148 @@
local datamanager = require "core.storagemanager".olddm;
+local array = require "util.array";
+local datetime = require "util.datetime";
+local st = require "util.stanza";
+local now = require "util.time".now;
+local id = require "util.id".medium;
local host = module.host;
local driver = {};
-local driver_mt = { __index = driver };
function driver:open(store, typ)
- if typ and typ ~= "keyval" then
+ local mt = self[typ or "keyval"]
+ if not mt then
return nil, "unsupported-store";
end
- return setmetatable({ store = store, type = typ }, driver_mt);
+ return setmetatable({ store = store, type = typ }, mt);
end
-function driver:get(user)
+
+function driver:stores(username) -- luacheck: ignore 212/self
+ return datamanager.stores(username, host);
+end
+
+function driver:purge(user) -- luacheck: ignore 212/self
+ return datamanager.purge(user, host);
+end
+
+local keyval = { };
+driver.keyval = { __index = keyval };
+
+function keyval:get(user)
return datamanager.load(user, host, self.store);
end
-function driver:set(user, data)
+function keyval:set(user, data)
return datamanager.store(user, host, self.store, data);
end
-function driver:stores(username)
- return datamanager.stores(username, host);
+function keyval:users()
+ return datamanager.users(host, self.store, self.type);
end
-function driver:users()
- return datamanager.users(host, self.store, self.type);
+local archive = {};
+driver.archive = { __index = archive };
+
+function archive:append(username, key, value, when, with)
+ key = key or id();
+ when = when or now();
+ if not st.is_stanza(value) then
+ return nil, "unsupported-datatype";
+ end
+ value = st.preserialize(st.clone(value));
+ value.key = key;
+ value.when = when;
+ value.with = with;
+ value.attr.stamp = datetime.datetime(when);
+ value.attr.stamp_legacy = datetime.legacy(when);
+ local ok, err = datamanager.list_append(username, host, self.store, value);
+ if not ok then return ok, err; end
+ return key;
end
-function driver:purge(user)
- return datamanager.purge(user, host);
+function archive:find(username, query)
+ local items, err = datamanager.list_load(username, host, self.store);
+ if not items then return items, err; end
+ local count = #items;
+ local i = 0;
+ if query then
+ items = array(items);
+ if query.with then
+ items:filter(function (item)
+ return item.with == query.with;
+ end);
+ end
+ if query.start then
+ items:filter(function (item)
+ return item.when >= query.start;
+ end);
+ end
+ if query["end"] then
+ items:filter(function (item)
+ return item.when <= query["end"];
+ end);
+ end
+ count = #items;
+ if query.reverse then
+ items:reverse();
+ if query.before then
+ for j = 1, count do
+ if (items[j].key or tostring(j)) == query.before then
+ i = j;
+ break;
+ end
+ end
+ end
+ elseif query.after then
+ for j = 1, count do
+ if (items[j].key or tostring(j)) == query.after then
+ i = j;
+ break;
+ end
+ end
+ end
+ if query.limit and #items - i > query.limit then
+ items[i+query.limit+1] = nil;
+ end
+ end
+ return function ()
+ i = i + 1;
+ local item = items[i];
+ if not item then return; end
+ local key = item.key or tostring(i);
+ local when = item.when or datetime.parse(item.attr.stamp);
+ local with = item.with;
+ item.key, item.when, item.with = nil, nil, nil;
+ item.attr.stamp = nil;
+ item.attr.stamp_legacy = nil;
+ item = st.deserialize(item);
+ return key, item, when, with;
+ end, count;
+end
+
+function archive:dates(username)
+ local items, err = datamanager.list_load(username, host, self.store);
+ if not items then return items, err; end
+ return array(items):pluck("when"):map(datetime.date):unique();
+end
+
+function archive:delete(username, query)
+ if not query or next(query) == nil then
+ return datamanager.list_store(username, host, self.store, nil);
+ end
+ for k in pairs(query) do
+ if k ~= "end" then return nil, "unsupported-query-field"; end
+ end
+ local items, err = datamanager.list_load(username, host, self.store);
+ if not items then return items, err; end
+ items = array(items);
+ items:filter(function (item)
+ return item.when > query["end"];
+ end);
+ local count = #items;
+ local ok, err = datamanager.list_store(username, host, self.store, items);
+ if not ok then return ok, err; end
+ return count;
end
module:provides("storage", driver);