diff mod_storage_xmlarchive/mod_storage_xmlarchive.lua @ 3898:e9e19b9a6a55

mod_storage_xmlarchive: Add a new API similar to map-stores This can be used to replace messages with tombstones
author Kim Alvefur <zash@zash.se>
date Sat, 22 Feb 2020 21:12:29 +0100
parents bb18a1f5e9d7
children a03abb4bb6d7
line wrap: on
line diff
--- a/mod_storage_xmlarchive/mod_storage_xmlarchive.lua	Sat Feb 22 21:11:31 2020 +0100
+++ b/mod_storage_xmlarchive/mod_storage_xmlarchive.lua	Sat Feb 22 21:12:29 2020 +0100
@@ -11,6 +11,7 @@
 local st = require"util.stanza";
 local dt = require"util.datetime";
 local new_stream = require "util.xmppstream".new;
+local xml = require "util.xml";
 local empty = {};
 
 if not dm.append_raw then
@@ -59,6 +60,76 @@
 	return id;
 end
 
+function archive:get(username, id)
+	local dates = self:dates(username) or empty;
+	local day_idx, item_idx, items = self:_get_idx(username, id, dates);
+	if not day_idx then
+		return nil, "item-not-found";
+	end
+	module:log("debug", ":_get_idx(%q, %q) --> %q, %q", username, id, dates[day_idx], items[item_idx]);
+	local day = dates[day_idx];
+	local item = items[item_idx];
+	module:log("debug", "item = %q", item);
+	local filename = dm.getpath(username.."@"..day, self.host, self.store, "xml");
+	local xmlfile, ferr = io.open(filename, "r");
+	if not xmlfile then return nil, ferr; end
+	local p,err = xmlfile:seek("set", item.offset);
+	if p ~= item.offset or err ~= nil then return nil, err; end
+	local data = xmlfile:read(item.length);
+	local parsed, perr = xml.parse(data);
+	if not parsed then return nil, perr; end
+	return parsed, dt.parse(item.when), item.with;
+end
+
+local overwrite = module:get_option("xmlarchive_overwrite", false);
+
+function archive:set(username, id, data, new_when, new_with)
+	if not is_stanza(data) then
+		module:log("error", "Attempt to store non-stanza object, traceback: %s", debug.traceback());
+		return nil, "unsupported-datatype";
+	end
+
+	username = username or "@";
+	data = tostring(data) .. "\n";
+
+	local dates = self:dates(username) or empty;
+	local day_idx, item_idx, items = self:_get_idx(username, id, dates);
+	if not day_idx then
+		return nil, "item-not-found";
+	end
+	local day = dates[day_idx];
+	local item = items[item_idx];
+
+	local filename = dm.getpath(username.."@"..day, self.host, self.store, "xml");
+
+	local replaced, err = dm.append_raw(username.."@"..day, self.host, self.store, "xml", data);
+	if not replaced then return nil, err; end
+	local new_offset = err;
+
+	-- default yes or no?
+	if overwrite then
+		local xmlfile, ferr = io.open(filename, "r+");
+		if not xmlfile then return nil, ferr; end
+		local p,err = xmlfile:seek("set", item.offset);
+		if p ~= item.offset or err ~= nil then return nil, err; end
+		local _,err = xmlfile:write((" "):rep(item.length));
+		if err ~= nil then return nil, err; end
+		local _,err = xmlfile:close();
+		if err ~= nil then return nil, err; end
+	end
+
+	items[item_idx] = {
+		id = id,
+		when = new_when or item.when,
+		with = new_with or item.with,
+		offset = new_offset,
+		length = #data,
+		replaces = item,
+	};
+	local ok, err = dm.list_store(username.."@"..day, self.host, self.store, items);
+	return ok, err;
+end
+
 function archive:_get_idx(username, id, dates)
 	module:log("debug", "Looking for item with id %q", id);
 	dates = dates or self:dates(username) or empty;
@@ -329,7 +400,6 @@
 	if arg[1] == "convert" and (arg[2] == "to" or arg[2] == "from") and arg[4] then
 		local convert;
 		if arg[2] == "to" then
-			local xml = require "util.xml";
 			function convert(user, host, store)
 				local dates, err = archive.dates({ host = host, store = store }, user);
 				if not dates then assert(not err, err); return end