From 58e93ec781de9f2859694cbb59c3dbfa468e2ba2 Mon Sep 17 00:00:00 2001 From: Kim Alvefur Date: Fri, 4 Nov 2016 13:48:21 +0100 Subject: mod_mam: Import from prosody-modules --- plugins/mod_mam/fallback_archive.lib.lua | 84 ++++++++++++++++++++++++++++++++ 1 file changed, 84 insertions(+) create mode 100644 plugins/mod_mam/fallback_archive.lib.lua (limited to 'plugins/mod_mam/fallback_archive.lib.lua') diff --git a/plugins/mod_mam/fallback_archive.lib.lua b/plugins/mod_mam/fallback_archive.lib.lua new file mode 100644 index 00000000..92155bef --- /dev/null +++ b/plugins/mod_mam/fallback_archive.lib.lua @@ -0,0 +1,84 @@ +-- luacheck: ignore 212/self + +local uuid = require "util.uuid".generate; +local store = module:shared("archive"); +local archive_store = { _provided_by = "mam"; name = "fallback"; }; + +function archive_store:append(username, key, value, when, with) + local archive = store[username]; + if not archive then + archive = { [0] = 0 }; + store[username] = archive; + end + local index = (archive[0] or #archive)+1; + local item = { key = key, when = when, with = with, value = value }; + if not key or archive[key] then + key = uuid(); + item.key = key; + end + archive[index] = item; + archive[key] = index; + archive[0] = index; + return key; +end + +function archive_store:find(username, query) + local archive = store[username] or {}; + local start, stop, step = 1, archive[0] or #archive, 1; + local qstart, qend, qwith = -math.huge, math.huge; + local limit; + + if query then + if query.reverse then + start, stop, step = stop, start, -1; + if query.before and archive[query.before] then + start = archive[query.before] - 1; + end + elseif query.after and archive[query.after] then + start = archive[query.after] + 1; + end + qwith = query.with; + limit = query.limit; + qstart = query.start or qstart; + qend = query["end"] or qend; + end + + return function () + if limit and limit <= 0 then return end + for i = start, stop, step do + local item = archive[i]; + if (not qwith or qwith == item.with) and item.when >= qstart and item.when <= qend then + if limit then limit = limit - 1; end + start = i + step; -- Start on next item + return item.key, item.value, item.when, item.with; + end + end + end +end + +function archive_store:delete(username, query) + if not query or next(query) == nil then + -- no specifics, delete everything + store[username] = nil; + return true; + end + local archive = store[username]; + if not archive then return true; end -- no messages, nothing to delete + + local start, stop, step = 1, archive[0] or #archive, 1; + local qstart = query.start or -math.huge; + local qend = query["end"] or math.huge; + local qwith = query.with; + store[username] = nil; + for i = 1, #archive do + local item = archive[i]; + local when, with = item.when, item.when; + -- Add things that don't match the query + if not ((not qwith or qwith == item.with) and item.when >= qstart and item.when <= qend) then + self:append(username, item.key, item.value, when, with); + end + end + return true; +end + +return archive_store; -- cgit v1.2.3 From 70b00611ce793bf4a2fb557a5fa5342babbdabb1 Mon Sep 17 00:00:00 2001 From: Kim Alvefur Date: Mon, 23 Jan 2017 19:18:15 +0100 Subject: mod_mam: Normalize copyright headers --- plugins/mod_mam/fallback_archive.lib.lua | 8 ++++++++ 1 file changed, 8 insertions(+) (limited to 'plugins/mod_mam/fallback_archive.lib.lua') diff --git a/plugins/mod_mam/fallback_archive.lib.lua b/plugins/mod_mam/fallback_archive.lib.lua index 92155bef..975f99b3 100644 --- a/plugins/mod_mam/fallback_archive.lib.lua +++ b/plugins/mod_mam/fallback_archive.lib.lua @@ -1,3 +1,11 @@ +-- Prosody IM +-- Copyright (C) 2008-2017 Matthew Wild +-- Copyright (C) 2008-2017 Waqas Hussain +-- Copyright (C) 2011-2017 Kim Alvefur +-- +-- This project is MIT/X11 licensed. Please see the +-- COPYING file in the source package for more information. +-- -- luacheck: ignore 212/self local uuid = require "util.uuid".generate; -- cgit v1.2.3 From 9f5d98c28311f7fa1b5dd65fcba560d3d8349ac4 Mon Sep 17 00:00:00 2001 From: Kim Alvefur Date: Mon, 23 Jan 2017 19:31:03 +0100 Subject: mod_mam: Remove unused variables [luacheck] --- plugins/mod_mam/fallback_archive.lib.lua | 1 - 1 file changed, 1 deletion(-) (limited to 'plugins/mod_mam/fallback_archive.lib.lua') diff --git a/plugins/mod_mam/fallback_archive.lib.lua b/plugins/mod_mam/fallback_archive.lib.lua index 975f99b3..71e65274 100644 --- a/plugins/mod_mam/fallback_archive.lib.lua +++ b/plugins/mod_mam/fallback_archive.lib.lua @@ -73,7 +73,6 @@ function archive_store:delete(username, query) local archive = store[username]; if not archive then return true; end -- no messages, nothing to delete - local start, stop, step = 1, archive[0] or #archive, 1; local qstart = query.start or -math.huge; local qend = query["end"] or math.huge; local qwith = query.with; -- cgit v1.2.3