Mercurial > prosody-modules
changeset 1566:9158882dd9a1
mod_storage_muc_log: Provides an archive API to mod_muc_log data
author | Kim Alvefur <zash@zash.se> |
---|---|
date | Mon, 10 Nov 2014 14:41:01 +0100 (2014-11-10) |
parents | f9cb09d451c7 |
children | 585bb8ac11bb |
files | mod_storage_muc_log/mod_storage_muc_log.lua |
diffstat | 1 files changed, 154 insertions(+), 0 deletions(-) [+] |
line wrap: on
line diff
--- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/mod_storage_muc_log/mod_storage_muc_log.lua Mon Nov 10 14:41:01 2014 +0100 @@ -0,0 +1,154 @@ + +local datamanager = require"core.storagemanager".olddm; +local xml_parse = require"util.xml".parse; +local data_load, data_store = datamanager.load, datamanager.store; +local datastore = "muc_log"; +local datetime = require"util.datetime" +local lfs = require"lfs"; +local noop = function () end; +local os_date = os.date; + +local timef, datef = "!%X", "!%y%m%d"; +local host = module.host; + +local driver = {}; +local driver_mt = { __index = driver }; + +do + -- Sanity check + -- Fun fact: 09:00 and 21:00 en_HK are both "09:00:00 UTC" + local t = os_date("!*t"); + t.hour = 9; + local am = os_date(timef, os.time(t)); + t.hour = 21; + local pm = os_date(timef, os.time(t)); + if am == pm then + module:log("warn", "Timestamps in AM and PM are identical in your locale, expect timestamps to be wrong"); + end +end + +local function parse_silly(date, time) + local year, month, day = date:match("^(%d%d)(%d%d)(%d%d)"); + year = "20"..year; + -- year = (year < "70" and "20" or "19") .. year; + local hour, min, sec = time:match("(%d%d)%D+(%d%d)%D+(%d%d)"); + if hour == "12" and time:find("[Aa][Mm]") then + hour = "00"; + elseif hour < "12" and time:find("[Pp][Mm]") then + hour = tostring(tonumber(hour) % 12 + 12); + end + return datetime.parse(("%s-%s-%sT%s:%s:%sZ"):format(year, month, day, hour or "00", min or "00", sec or "00")); +end + +local function st_with(tag) + local with = tag.attr.type; + return with and tag.name .. "<" .. with or tag.name; +end + +function driver:append(node, key, when, with, stanza) + local today = os_date(datef, when); + local now = os_date(timef, when); + local data = data_load(node, host, datastore .. "/" .. today) or {}; + data[#data + 1] = "<stanza time=\"".. now .. "\">" .. tostring(stanza) .. "</stanza>\n"; + datamanager.getpath(node, host, datastore, nil, true); -- create the datastore dir + local ok, err = data_store(node, host, datastore .. "/" .. today, data); + if not ok then + return ok, err; + end + return today .. "_" .. #data; +end + +function driver:find(node, query) + local path = datamanager.getpath(node, host, datastore):match("(.*)/"); + + local ok, iter, state, var = pcall(lfs.dir, path); + if not ok then + module:log("warn", iter); + return nil, iter; + end + + local dates, i = {}, 1; + for dir in iter, state, var do + if lfs.attributes(datamanager.getpath(node, host, datastore .. "/" .. dir), "mode") == "file" then + dates[i], i = dir, i+1; + end + end + if dates[1] == nil then return noop, 0; end + table.sort(dates); + + return coroutine.wrap(function () + local query = query; + local dates = dates; + local start_date = query and query.start and os_date(datef, query.start) or dates[1]; + local end_date = query and query["end"] and os_date(datef, query["end"]) or dates[#dates]; + local start_time = query and query.start and os_date(timef, query.start) or dates[1]; + local end_time = query and query["end"] and os_date(timef, query["end"]) or dates[#dates]; + local query_with = query and query.with; + local query_limit = query and query.limit; + local seek_once = query and query.after; + + local today, time, data, err, item; + local inner_start, inner_stop, inner_step; + local outer_start, outer_stop, outer_step = 1, #dates, 1; + if query and query.reverse then + outer_start, outer_stop, outer_step = outer_stop, outer_start, -outer_step; + seek_once = query.before; + if seek_once then + end_date = seek_once:match"^(%d+)_%d"; + end + elseif seek_once then + start_date = seek_once:match"^(%d+)_%d"; + end + local matches = 0; + for i = outer_start, outer_stop, outer_step do + today = dates[i]; + if today >= start_date and today <= end_date then + data, err = data_load(node, host, datastore .. "/" .. today); + if data then + inner_start, inner_stop, inner_step = 1, #data, 1; + if query and query.reverse then + inner_start, inner_stop, inner_step = inner_stop, inner_start, -inner_step; + end + if seek_once then + inner_start = tonumber(seek_once:match("_(%d+)$")); + inner_start = inner_start + (query and query.reverse and -1 or 1); + seek_once = nil; + end + for i = inner_start, inner_stop, inner_step do + item, err = data[i]; + if item then + item, err = xml_parse(item); + end + if item then + time = item.attr.time; + item = item.tags[1]; + if (today >= start_date or time >= start_time) and + (today <= end_date or time <= end_time) and + (not query_with or query_with == st_with(item)) and + item:get_child_text("alreadyJoined") ~= "true" then + matches = matches + 1; + coroutine.yield(today.."_"..i, item, parse_silly(today, time)); + if query_limit and matches >= query_limit then + return; + end + end + elseif err then + module:log("warn", err); + end + end + elseif err then + module:log("warn", err); + end + end + end + end); +end + +function open(_, store, typ) + if typ ~= "archive" then + return nil, "unsupported-store"; + end + return setmetatable({ store = store, type = typ }, driver_mt); +end + +module:provides "storage";