Software `/` code `/` prosody

Annotate

plugins/mod_storage_xep0227.lua @ 11928:16cf863b36c0

mod_csi_simple: Skip initiating flush in all but inactive state Both in the flushing and active states the right thing to do is skip directly to returning the data. Also in any unknown state, like if the filter is somehow left behind on module unload.

author	Kim Alvefur <zash@zash.se>
date	Sat, 20 Nov 2021 23:54:43 +0100
parent	11840:5e9e75c277a2
child	12082:e87563fefd85

rev	line source
3414 9a1f6239b63c storage/mod_xep0227: Initial commit. Waqas Hussain <waqas20@gmail.com> parents: diff changeset	1
9a1f6239b63c storage/mod_xep0227: Initial commit. Waqas Hussain <waqas20@gmail.com> parents: diff changeset	2 local ipairs, pairs = ipairs, pairs;
9a1f6239b63c storage/mod_xep0227: Initial commit. Waqas Hussain <waqas20@gmail.com> parents: diff changeset	3 local setmetatable = setmetatable;
9a1f6239b63c storage/mod_xep0227: Initial commit. Waqas Hussain <waqas20@gmail.com> parents: diff changeset	4 local tostring = tostring;
11789 f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	5 local next, unpack = next, table.unpack or unpack; --luacheck: ignore 113/unpack
3414 9a1f6239b63c storage/mod_xep0227: Initial commit. Waqas Hussain <waqas20@gmail.com> parents: diff changeset	6 local t_remove = table.remove;
9a1f6239b63c storage/mod_xep0227: Initial commit. Waqas Hussain <waqas20@gmail.com> parents: diff changeset	7 local os_remove = os.remove;
9a1f6239b63c storage/mod_xep0227: Initial commit. Waqas Hussain <waqas20@gmail.com> parents: diff changeset	8 local io_open = io.open;
11789 f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	9 local jid_bare = require "util.jid".bare;
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	10 local jid_prep = require "util.jid".prep;
3414 9a1f6239b63c storage/mod_xep0227: Initial commit. Waqas Hussain <waqas20@gmail.com> parents: diff changeset	11
11789 f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	12 local array = require "util.array";
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	13 local base64 = require "util.encodings".base64;
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	14 local dt = require "util.datetime";
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	15 local hex = require "util.hex";
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	16 local it = require "util.iterators";
6697 3d27f5855f4b mod_storage_xep0227: Use configured storage path Kim Alvefur <zash@zash.se> parents: 6696 diff changeset	17 local paths = require"util.paths";
11789 f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	18 local set = require "util.set";
3414 9a1f6239b63c storage/mod_xep0227: Initial commit. Waqas Hussain <waqas20@gmail.com> parents: diff changeset	19 local st = require "util.stanza";
5219 060d7fc6caf4 plugins/storage/mod_xep0227: Use util.xml. Waqas Hussain <waqas20@gmail.com> parents: 5121 diff changeset	20 local parse_xml_real = require "util.xml".parse;
3414 9a1f6239b63c storage/mod_xep0227: Initial commit. Waqas Hussain <waqas20@gmail.com> parents: diff changeset	21
11789 f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	22 local lfs = require "lfs";
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	23
3414 9a1f6239b63c storage/mod_xep0227: Initial commit. Waqas Hussain <waqas20@gmail.com> parents: diff changeset	24 local function getXml(user, host)
9a1f6239b63c storage/mod_xep0227: Initial commit. Waqas Hussain <waqas20@gmail.com> parents: diff changeset	25 local jid = user.."@"..host;
6697 3d27f5855f4b mod_storage_xep0227: Use configured storage path Kim Alvefur <zash@zash.se> parents: 6696 diff changeset	26 local path = paths.join(prosody.paths.data, jid..".xml");
11789 f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	27 local f, err = io_open(path);
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	28 if not f then
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	29 module:log("debug", "Unable to load XML file for <%s>: %s", jid, err);
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	30 return;
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	31 end
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	32 module:log("debug", "Loaded %s", path);
3414 9a1f6239b63c storage/mod_xep0227: Initial commit. Waqas Hussain <waqas20@gmail.com> parents: diff changeset	33 local s = f:read("*a");
6698 95a8aeca1fc9 mod_storage_xep0227: Close file handle after reading Kim Alvefur <zash@zash.se> parents: 6697 diff changeset	34 f:close();
3414 9a1f6239b63c storage/mod_xep0227: Initial commit. Waqas Hussain <waqas20@gmail.com> parents: diff changeset	35 return parse_xml_real(s);
9a1f6239b63c storage/mod_xep0227: Initial commit. Waqas Hussain <waqas20@gmail.com> parents: diff changeset	36 end
9a1f6239b63c storage/mod_xep0227: Initial commit. Waqas Hussain <waqas20@gmail.com> parents: diff changeset	37 local function setXml(user, host, xml)
9a1f6239b63c storage/mod_xep0227: Initial commit. Waqas Hussain <waqas20@gmail.com> parents: diff changeset	38 local jid = user.."@"..host;
6697 3d27f5855f4b mod_storage_xep0227: Use configured storage path Kim Alvefur <zash@zash.se> parents: 6696 diff changeset	39 local path = paths.join(prosody.paths.data, jid..".xml");
6700 0103dc8fa179 mod_storage_xep0227: Return error from io.open if unable to open file for writing Kim Alvefur <zash@zash.se> parents: 6699 diff changeset	40 local f, err = io_open(path, "w");
0103dc8fa179 mod_storage_xep0227: Return error from io.open if unable to open file for writing Kim Alvefur <zash@zash.se> parents: 6699 diff changeset	41 if not f then return f, err; end
3414 9a1f6239b63c storage/mod_xep0227: Initial commit. Waqas Hussain <waqas20@gmail.com> parents: diff changeset	42 if xml then
9a1f6239b63c storage/mod_xep0227: Initial commit. Waqas Hussain <waqas20@gmail.com> parents: diff changeset	43 local s = tostring(xml);
9a1f6239b63c storage/mod_xep0227: Initial commit. Waqas Hussain <waqas20@gmail.com> parents: diff changeset	44 f:write(s);
9a1f6239b63c storage/mod_xep0227: Initial commit. Waqas Hussain <waqas20@gmail.com> parents: diff changeset	45 f:close();
9a1f6239b63c storage/mod_xep0227: Initial commit. Waqas Hussain <waqas20@gmail.com> parents: diff changeset	46 return true;
9a1f6239b63c storage/mod_xep0227: Initial commit. Waqas Hussain <waqas20@gmail.com> parents: diff changeset	47 else
6699 ccdd0b615106 mod_storage_xep0227: Open file for writing even if removing so os.remove has a file to delete Kim Alvefur <zash@zash.se> parents: 6698 diff changeset	48 f:close();
3414 9a1f6239b63c storage/mod_xep0227: Initial commit. Waqas Hussain <waqas20@gmail.com> parents: diff changeset	49 return os_remove(path);
9a1f6239b63c storage/mod_xep0227: Initial commit. Waqas Hussain <waqas20@gmail.com> parents: diff changeset	50 end
9a1f6239b63c storage/mod_xep0227: Initial commit. Waqas Hussain <waqas20@gmail.com> parents: diff changeset	51 end
9a1f6239b63c storage/mod_xep0227: Initial commit. Waqas Hussain <waqas20@gmail.com> parents: diff changeset	52 local function getUserElement(xml)
9a1f6239b63c storage/mod_xep0227: Initial commit. Waqas Hussain <waqas20@gmail.com> parents: diff changeset	53 if xml and xml.name == "server-data" then
9a1f6239b63c storage/mod_xep0227: Initial commit. Waqas Hussain <waqas20@gmail.com> parents: diff changeset	54 local host = xml.tags[1];
9a1f6239b63c storage/mod_xep0227: Initial commit. Waqas Hussain <waqas20@gmail.com> parents: diff changeset	55 if host and host.name == "host" then
9a1f6239b63c storage/mod_xep0227: Initial commit. Waqas Hussain <waqas20@gmail.com> parents: diff changeset	56 local user = host.tags[1];
9a1f6239b63c storage/mod_xep0227: Initial commit. Waqas Hussain <waqas20@gmail.com> parents: diff changeset	57 if user and user.name == "user" then
9a1f6239b63c storage/mod_xep0227: Initial commit. Waqas Hussain <waqas20@gmail.com> parents: diff changeset	58 return user;
9a1f6239b63c storage/mod_xep0227: Initial commit. Waqas Hussain <waqas20@gmail.com> parents: diff changeset	59 end
9a1f6239b63c storage/mod_xep0227: Initial commit. Waqas Hussain <waqas20@gmail.com> parents: diff changeset	60 end
9a1f6239b63c storage/mod_xep0227: Initial commit. Waqas Hussain <waqas20@gmail.com> parents: diff changeset	61 end
11789 f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	62 module:log("warn", "Unable to find user element");
3414 9a1f6239b63c storage/mod_xep0227: Initial commit. Waqas Hussain <waqas20@gmail.com> parents: diff changeset	63 end
9a1f6239b63c storage/mod_xep0227: Initial commit. Waqas Hussain <waqas20@gmail.com> parents: diff changeset	64 local function createOuterXml(user, host)
6702 6a5c6c95cf78 mod_storage_xep0227: Use the registered namespace Kim Alvefur <zash@zash.se> parents: 6701 diff changeset	65 return st.stanza("server-data", {xmlns='urn:xmpp:pie:0'})
3414 9a1f6239b63c storage/mod_xep0227: Initial commit. Waqas Hussain <waqas20@gmail.com> parents: diff changeset	66 :tag("host", {jid=host})
9a1f6239b63c storage/mod_xep0227: Initial commit. Waqas Hussain <waqas20@gmail.com> parents: diff changeset	67 :tag("user", {name = user});
9a1f6239b63c storage/mod_xep0227: Initial commit. Waqas Hussain <waqas20@gmail.com> parents: diff changeset	68 end
11789 f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	69 local function removeFromArray(arr, value)
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	70 for i,item in ipairs(arr) do
3414 9a1f6239b63c storage/mod_xep0227: Initial commit. Waqas Hussain <waqas20@gmail.com> parents: diff changeset	71 if item == value then
11789 f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	72 t_remove(arr, i);
3414 9a1f6239b63c storage/mod_xep0227: Initial commit. Waqas Hussain <waqas20@gmail.com> parents: diff changeset	73 return;
9a1f6239b63c storage/mod_xep0227: Initial commit. Waqas Hussain <waqas20@gmail.com> parents: diff changeset	74 end
9a1f6239b63c storage/mod_xep0227: Initial commit. Waqas Hussain <waqas20@gmail.com> parents: diff changeset	75 end
9a1f6239b63c storage/mod_xep0227: Initial commit. Waqas Hussain <waqas20@gmail.com> parents: diff changeset	76 end
9a1f6239b63c storage/mod_xep0227: Initial commit. Waqas Hussain <waqas20@gmail.com> parents: diff changeset	77 local function removeStanzaChild(s, child)
9a1f6239b63c storage/mod_xep0227: Initial commit. Waqas Hussain <waqas20@gmail.com> parents: diff changeset	78 removeFromArray(s.tags, child);
9a1f6239b63c storage/mod_xep0227: Initial commit. Waqas Hussain <waqas20@gmail.com> parents: diff changeset	79 removeFromArray(s, child);
9a1f6239b63c storage/mod_xep0227: Initial commit. Waqas Hussain <waqas20@gmail.com> parents: diff changeset	80 end
9a1f6239b63c storage/mod_xep0227: Initial commit. Waqas Hussain <waqas20@gmail.com> parents: diff changeset	81
11789 f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	82 local function hex_to_base64(s)
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	83 return base64.encode(hex.from(s));
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	84 end
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	85
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	86 local function base64_to_hex(s)
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	87 return base64.encode(hex.from(s));
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	88 end
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	89
3414 9a1f6239b63c storage/mod_xep0227: Initial commit. Waqas Hussain <waqas20@gmail.com> parents: diff changeset	90 local handlers = {};
9a1f6239b63c storage/mod_xep0227: Initial commit. Waqas Hussain <waqas20@gmail.com> parents: diff changeset	91
11789 f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	92 -- In order to support custom account properties
6703 353a7d4dfdc2 mod_storage_xep0227: Store data from mod_auth_internal_hashed in a private namespace Kim Alvefur <zash@zash.se> parents: 6702 diff changeset	93 local extended = "http://prosody.im/protocol/extended-xep0227\1";
353a7d4dfdc2 mod_storage_xep0227: Store data from mod_auth_internal_hashed in a private namespace Kim Alvefur <zash@zash.se> parents: 6702 diff changeset	94
11789 f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	95 local scram_hash_name = module:get_option_string("password_hash", "SHA-1");
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	96 local scram_properties = set.new({ "server_key", "stored_key", "iteration_count", "salt" });
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	97
3414 9a1f6239b63c storage/mod_xep0227: Initial commit. Waqas Hussain <waqas20@gmail.com> parents: diff changeset	98 handlers.accounts = {
9a1f6239b63c storage/mod_xep0227: Initial commit. Waqas Hussain <waqas20@gmail.com> parents: diff changeset	99 get = function(self, user)
6704 03afecdf2e3c mod_storage_xep0227: Silence luacheck warnings Kim Alvefur <zash@zash.se> parents: 6703 diff changeset	100 user = getUserElement(getXml(user, self.host));
11789 f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	101 local scram_credentials = user and user:get_child_with_attr(
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	102 "scram-credentials", "urn:xmpp:pie:0#scram",
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	103 "mechanism", "SCRAM-"..scram_hash_name
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	104 );
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	105 if scram_credentials then
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	106 return {
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	107 iteration_count = tonumber(scram_credentials:get_child_text("iter-count"));
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	108 server_key = base64_to_hex(scram_credentials:get_child_text("server-key"));
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	109 stored_key = base64_to_hex(scram_credentials:get_child_text("stored-key"));
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	110 salt = base64.decode(scram_credentials:get_child_text("salt"));
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	111 };
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	112 elseif user and user.attr.password then
3414 9a1f6239b63c storage/mod_xep0227: Initial commit. Waqas Hussain <waqas20@gmail.com> parents: diff changeset	113 return { password = user.attr.password };
6703 353a7d4dfdc2 mod_storage_xep0227: Store data from mod_auth_internal_hashed in a private namespace Kim Alvefur <zash@zash.se> parents: 6702 diff changeset	114 elseif user then
353a7d4dfdc2 mod_storage_xep0227: Store data from mod_auth_internal_hashed in a private namespace Kim Alvefur <zash@zash.se> parents: 6702 diff changeset	115 local data = {};
353a7d4dfdc2 mod_storage_xep0227: Store data from mod_auth_internal_hashed in a private namespace Kim Alvefur <zash@zash.se> parents: 6702 diff changeset	116 for k, v in pairs(user.attr) do
353a7d4dfdc2 mod_storage_xep0227: Store data from mod_auth_internal_hashed in a private namespace Kim Alvefur <zash@zash.se> parents: 6702 diff changeset	117 if k:sub(1, #extended) == extended then
353a7d4dfdc2 mod_storage_xep0227: Store data from mod_auth_internal_hashed in a private namespace Kim Alvefur <zash@zash.se> parents: 6702 diff changeset	118 data[k:sub(#extended+1)] = v;
353a7d4dfdc2 mod_storage_xep0227: Store data from mod_auth_internal_hashed in a private namespace Kim Alvefur <zash@zash.se> parents: 6702 diff changeset	119 end
353a7d4dfdc2 mod_storage_xep0227: Store data from mod_auth_internal_hashed in a private namespace Kim Alvefur <zash@zash.se> parents: 6702 diff changeset	120 end
353a7d4dfdc2 mod_storage_xep0227: Store data from mod_auth_internal_hashed in a private namespace Kim Alvefur <zash@zash.se> parents: 6702 diff changeset	121 return data;
3414 9a1f6239b63c storage/mod_xep0227: Initial commit. Waqas Hussain <waqas20@gmail.com> parents: diff changeset	122 end
9a1f6239b63c storage/mod_xep0227: Initial commit. Waqas Hussain <waqas20@gmail.com> parents: diff changeset	123 end;
9a1f6239b63c storage/mod_xep0227: Initial commit. Waqas Hussain <waqas20@gmail.com> parents: diff changeset	124 set = function(self, user, data)
11789 f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	125 if not data then
3414 9a1f6239b63c storage/mod_xep0227: Initial commit. Waqas Hussain <waqas20@gmail.com> parents: diff changeset	126 return setXml(user, self.host, nil);
9a1f6239b63c storage/mod_xep0227: Initial commit. Waqas Hussain <waqas20@gmail.com> parents: diff changeset	127 end
11789 f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	128
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	129 local xml = getXml(user, self.host);
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	130 if not xml then xml = createOuterXml(user, self.host); end
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	131 local usere = getUserElement(xml);
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	132
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	133 local account_properties = set.new(it.to_array(it.keys(data)));
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	134
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	135 -- Include SCRAM credentials if known
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	136 if account_properties:contains_set(scram_properties) then
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	137 local scram_el = st.stanza("scram-credentials", { xmlns = "urn:xmpp:pie:0#scram", mechanism = "SCRAM-"..scram_hash_name })
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	138 :text_tag("server-key", hex_to_base64(data.server_key))
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	139 :text_tag("stored-key", hex_to_base64(data.stored_key))
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	140 :text_tag("iter-count", ("%d"):format(data.iteration_count))
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	141 :text_tag("salt", base64.encode(data.salt));
11840 5e9e75c277a2 mod_storage_xep0227: Add scram-credentials to user element rather than server Matthew Wild <mwild1@gmail.com> parents: 11789 diff changeset	142 usere:add_child(scram_el);
11789 f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	143 account_properties:exclude(scram_properties);
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	144 end
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	145
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	146 -- Include the password if present
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	147 if account_properties:contains("password") then
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	148 usere.attr.password = data.password;
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	149 account_properties:remove("password");
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	150 end
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	151
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	152 -- Preserve remaining properties as namespaced attributes
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	153 for property in account_properties do
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	154 usere.attr[extended..property] = data[property];
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	155 end
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	156
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	157 return setXml(user, self.host, xml);
3414 9a1f6239b63c storage/mod_xep0227: Initial commit. Waqas Hussain <waqas20@gmail.com> parents: diff changeset	158 end;
9a1f6239b63c storage/mod_xep0227: Initial commit. Waqas Hussain <waqas20@gmail.com> parents: diff changeset	159 };
9a1f6239b63c storage/mod_xep0227: Initial commit. Waqas Hussain <waqas20@gmail.com> parents: diff changeset	160 handlers.vcard = {
9a1f6239b63c storage/mod_xep0227: Initial commit. Waqas Hussain <waqas20@gmail.com> parents: diff changeset	161 get = function(self, user)
6704 03afecdf2e3c mod_storage_xep0227: Silence luacheck warnings Kim Alvefur <zash@zash.se> parents: 6703 diff changeset	162 user = getUserElement(getXml(user, self.host));
3414 9a1f6239b63c storage/mod_xep0227: Initial commit. Waqas Hussain <waqas20@gmail.com> parents: diff changeset	163 if user then
9a1f6239b63c storage/mod_xep0227: Initial commit. Waqas Hussain <waqas20@gmail.com> parents: diff changeset	164 local vcard = user:get_child("vCard", 'vcard-temp');
9a1f6239b63c storage/mod_xep0227: Initial commit. Waqas Hussain <waqas20@gmail.com> parents: diff changeset	165 if vcard then
9a1f6239b63c storage/mod_xep0227: Initial commit. Waqas Hussain <waqas20@gmail.com> parents: diff changeset	166 return st.preserialize(vcard);
9a1f6239b63c storage/mod_xep0227: Initial commit. Waqas Hussain <waqas20@gmail.com> parents: diff changeset	167 end
9a1f6239b63c storage/mod_xep0227: Initial commit. Waqas Hussain <waqas20@gmail.com> parents: diff changeset	168 end
9a1f6239b63c storage/mod_xep0227: Initial commit. Waqas Hussain <waqas20@gmail.com> parents: diff changeset	169 end;
9a1f6239b63c storage/mod_xep0227: Initial commit. Waqas Hussain <waqas20@gmail.com> parents: diff changeset	170 set = function(self, user, data)
9a1f6239b63c storage/mod_xep0227: Initial commit. Waqas Hussain <waqas20@gmail.com> parents: diff changeset	171 local xml = getXml(user, self.host);
9a1f6239b63c storage/mod_xep0227: Initial commit. Waqas Hussain <waqas20@gmail.com> parents: diff changeset	172 local usere = xml and getUserElement(xml);
9a1f6239b63c storage/mod_xep0227: Initial commit. Waqas Hussain <waqas20@gmail.com> parents: diff changeset	173 if usere then
9a1f6239b63c storage/mod_xep0227: Initial commit. Waqas Hussain <waqas20@gmail.com> parents: diff changeset	174 local vcard = usere:get_child("vCard", 'vcard-temp');
9a1f6239b63c storage/mod_xep0227: Initial commit. Waqas Hussain <waqas20@gmail.com> parents: diff changeset	175 if vcard then
9a1f6239b63c storage/mod_xep0227: Initial commit. Waqas Hussain <waqas20@gmail.com> parents: diff changeset	176 removeStanzaChild(usere, vcard);
9a1f6239b63c storage/mod_xep0227: Initial commit. Waqas Hussain <waqas20@gmail.com> parents: diff changeset	177 elseif not data then
9a1f6239b63c storage/mod_xep0227: Initial commit. Waqas Hussain <waqas20@gmail.com> parents: diff changeset	178 return true;
9a1f6239b63c storage/mod_xep0227: Initial commit. Waqas Hussain <waqas20@gmail.com> parents: diff changeset	179 end
9a1f6239b63c storage/mod_xep0227: Initial commit. Waqas Hussain <waqas20@gmail.com> parents: diff changeset	180 if data then
9a1f6239b63c storage/mod_xep0227: Initial commit. Waqas Hussain <waqas20@gmail.com> parents: diff changeset	181 vcard = st.deserialize(data);
9a1f6239b63c storage/mod_xep0227: Initial commit. Waqas Hussain <waqas20@gmail.com> parents: diff changeset	182 usere:add_child(vcard);
9a1f6239b63c storage/mod_xep0227: Initial commit. Waqas Hussain <waqas20@gmail.com> parents: diff changeset	183 end
9a1f6239b63c storage/mod_xep0227: Initial commit. Waqas Hussain <waqas20@gmail.com> parents: diff changeset	184 return setXml(user, self.host, xml);
9a1f6239b63c storage/mod_xep0227: Initial commit. Waqas Hussain <waqas20@gmail.com> parents: diff changeset	185 end
9a1f6239b63c storage/mod_xep0227: Initial commit. Waqas Hussain <waqas20@gmail.com> parents: diff changeset	186 return true;
9a1f6239b63c storage/mod_xep0227: Initial commit. Waqas Hussain <waqas20@gmail.com> parents: diff changeset	187 end;
9a1f6239b63c storage/mod_xep0227: Initial commit. Waqas Hussain <waqas20@gmail.com> parents: diff changeset	188 };
9a1f6239b63c storage/mod_xep0227: Initial commit. Waqas Hussain <waqas20@gmail.com> parents: diff changeset	189 handlers.private = {
9a1f6239b63c storage/mod_xep0227: Initial commit. Waqas Hussain <waqas20@gmail.com> parents: diff changeset	190 get = function(self, user)
6704 03afecdf2e3c mod_storage_xep0227: Silence luacheck warnings Kim Alvefur <zash@zash.se> parents: 6703 diff changeset	191 user = getUserElement(getXml(user, self.host));
3414 9a1f6239b63c storage/mod_xep0227: Initial commit. Waqas Hussain <waqas20@gmail.com> parents: diff changeset	192 if user then
9a1f6239b63c storage/mod_xep0227: Initial commit. Waqas Hussain <waqas20@gmail.com> parents: diff changeset	193 local private = user:get_child("query", "jabber:iq:private");
9a1f6239b63c storage/mod_xep0227: Initial commit. Waqas Hussain <waqas20@gmail.com> parents: diff changeset	194 if private then
9a1f6239b63c storage/mod_xep0227: Initial commit. Waqas Hussain <waqas20@gmail.com> parents: diff changeset	195 local r = {};
9a1f6239b63c storage/mod_xep0227: Initial commit. Waqas Hussain <waqas20@gmail.com> parents: diff changeset	196 for _, tag in ipairs(private.tags) do
9a1f6239b63c storage/mod_xep0227: Initial commit. Waqas Hussain <waqas20@gmail.com> parents: diff changeset	197 r[tag.name..":"..tag.attr.xmlns] = st.preserialize(tag);
9a1f6239b63c storage/mod_xep0227: Initial commit. Waqas Hussain <waqas20@gmail.com> parents: diff changeset	198 end
9a1f6239b63c storage/mod_xep0227: Initial commit. Waqas Hussain <waqas20@gmail.com> parents: diff changeset	199 return r;
9a1f6239b63c storage/mod_xep0227: Initial commit. Waqas Hussain <waqas20@gmail.com> parents: diff changeset	200 end
9a1f6239b63c storage/mod_xep0227: Initial commit. Waqas Hussain <waqas20@gmail.com> parents: diff changeset	201 end
9a1f6239b63c storage/mod_xep0227: Initial commit. Waqas Hussain <waqas20@gmail.com> parents: diff changeset	202 end;
9a1f6239b63c storage/mod_xep0227: Initial commit. Waqas Hussain <waqas20@gmail.com> parents: diff changeset	203 set = function(self, user, data)
9a1f6239b63c storage/mod_xep0227: Initial commit. Waqas Hussain <waqas20@gmail.com> parents: diff changeset	204 local xml = getXml(user, self.host);
9a1f6239b63c storage/mod_xep0227: Initial commit. Waqas Hussain <waqas20@gmail.com> parents: diff changeset	205 local usere = xml and getUserElement(xml);
9a1f6239b63c storage/mod_xep0227: Initial commit. Waqas Hussain <waqas20@gmail.com> parents: diff changeset	206 if usere then
9a1f6239b63c storage/mod_xep0227: Initial commit. Waqas Hussain <waqas20@gmail.com> parents: diff changeset	207 local private = usere:get_child("query", 'jabber:iq:private');
9a1f6239b63c storage/mod_xep0227: Initial commit. Waqas Hussain <waqas20@gmail.com> parents: diff changeset	208 if private then removeStanzaChild(usere, private); end
9a1f6239b63c storage/mod_xep0227: Initial commit. Waqas Hussain <waqas20@gmail.com> parents: diff changeset	209 if data and next(data) ~= nil then
9a1f6239b63c storage/mod_xep0227: Initial commit. Waqas Hussain <waqas20@gmail.com> parents: diff changeset	210 private = st.stanza("query", {xmlns='jabber:iq:private'});
9a1f6239b63c storage/mod_xep0227: Initial commit. Waqas Hussain <waqas20@gmail.com> parents: diff changeset	211 for _,tag in pairs(data) do
9a1f6239b63c storage/mod_xep0227: Initial commit. Waqas Hussain <waqas20@gmail.com> parents: diff changeset	212 private:add_child(st.deserialize(tag));
9a1f6239b63c storage/mod_xep0227: Initial commit. Waqas Hussain <waqas20@gmail.com> parents: diff changeset	213 end
9a1f6239b63c storage/mod_xep0227: Initial commit. Waqas Hussain <waqas20@gmail.com> parents: diff changeset	214 usere:add_child(private);
9a1f6239b63c storage/mod_xep0227: Initial commit. Waqas Hussain <waqas20@gmail.com> parents: diff changeset	215 end
9a1f6239b63c storage/mod_xep0227: Initial commit. Waqas Hussain <waqas20@gmail.com> parents: diff changeset	216 return setXml(user, self.host, xml);
9a1f6239b63c storage/mod_xep0227: Initial commit. Waqas Hussain <waqas20@gmail.com> parents: diff changeset	217 end
9a1f6239b63c storage/mod_xep0227: Initial commit. Waqas Hussain <waqas20@gmail.com> parents: diff changeset	218 return true;
9a1f6239b63c storage/mod_xep0227: Initial commit. Waqas Hussain <waqas20@gmail.com> parents: diff changeset	219 end;
9a1f6239b63c storage/mod_xep0227: Initial commit. Waqas Hussain <waqas20@gmail.com> parents: diff changeset	220 };
9a1f6239b63c storage/mod_xep0227: Initial commit. Waqas Hussain <waqas20@gmail.com> parents: diff changeset	221
8351 cc05b6366576 mod_storage_xep0227: Add roster storage (fixes #1023) Kim Alvefur <zash@zash.se> parents: 8350 diff changeset	222 handlers.roster = {
cc05b6366576 mod_storage_xep0227: Add roster storage (fixes #1023) Kim Alvefur <zash@zash.se> parents: 8350 diff changeset	223 get = function(self, user)
cc05b6366576 mod_storage_xep0227: Add roster storage (fixes #1023) Kim Alvefur <zash@zash.se> parents: 8350 diff changeset	224 user = getUserElement(getXml(user, self.host));
cc05b6366576 mod_storage_xep0227: Add roster storage (fixes #1023) Kim Alvefur <zash@zash.se> parents: 8350 diff changeset	225 if user then
cc05b6366576 mod_storage_xep0227: Add roster storage (fixes #1023) Kim Alvefur <zash@zash.se> parents: 8350 diff changeset	226 local roster = user:get_child("query", "jabber:iq:roster");
cc05b6366576 mod_storage_xep0227: Add roster storage (fixes #1023) Kim Alvefur <zash@zash.se> parents: 8350 diff changeset	227 if roster then
cc05b6366576 mod_storage_xep0227: Add roster storage (fixes #1023) Kim Alvefur <zash@zash.se> parents: 8350 diff changeset	228 local r = {
cc05b6366576 mod_storage_xep0227: Add roster storage (fixes #1023) Kim Alvefur <zash@zash.se> parents: 8350 diff changeset	229 [false] = {
cc05b6366576 mod_storage_xep0227: Add roster storage (fixes #1023) Kim Alvefur <zash@zash.se> parents: 8350 diff changeset	230 version = roster.attr.version;
cc05b6366576 mod_storage_xep0227: Add roster storage (fixes #1023) Kim Alvefur <zash@zash.se> parents: 8350 diff changeset	231 pending = {};
cc05b6366576 mod_storage_xep0227: Add roster storage (fixes #1023) Kim Alvefur <zash@zash.se> parents: 8350 diff changeset	232 }
cc05b6366576 mod_storage_xep0227: Add roster storage (fixes #1023) Kim Alvefur <zash@zash.se> parents: 8350 diff changeset	233 };
cc05b6366576 mod_storage_xep0227: Add roster storage (fixes #1023) Kim Alvefur <zash@zash.se> parents: 8350 diff changeset	234 for item in roster:childtags("item") do
cc05b6366576 mod_storage_xep0227: Add roster storage (fixes #1023) Kim Alvefur <zash@zash.se> parents: 8350 diff changeset	235 r[item.attr.jid] = {
cc05b6366576 mod_storage_xep0227: Add roster storage (fixes #1023) Kim Alvefur <zash@zash.se> parents: 8350 diff changeset	236 jid = item.attr.jid,
cc05b6366576 mod_storage_xep0227: Add roster storage (fixes #1023) Kim Alvefur <zash@zash.se> parents: 8350 diff changeset	237 subscription = item.attr.subscription,
cc05b6366576 mod_storage_xep0227: Add roster storage (fixes #1023) Kim Alvefur <zash@zash.se> parents: 8350 diff changeset	238 ask = item.attr.ask,
cc05b6366576 mod_storage_xep0227: Add roster storage (fixes #1023) Kim Alvefur <zash@zash.se> parents: 8350 diff changeset	239 name = item.attr.name,
cc05b6366576 mod_storage_xep0227: Add roster storage (fixes #1023) Kim Alvefur <zash@zash.se> parents: 8350 diff changeset	240 groups = {};
cc05b6366576 mod_storage_xep0227: Add roster storage (fixes #1023) Kim Alvefur <zash@zash.se> parents: 8350 diff changeset	241 };
cc05b6366576 mod_storage_xep0227: Add roster storage (fixes #1023) Kim Alvefur <zash@zash.se> parents: 8350 diff changeset	242 for group in item:childtags("group") do
cc05b6366576 mod_storage_xep0227: Add roster storage (fixes #1023) Kim Alvefur <zash@zash.se> parents: 8350 diff changeset	243 r[item.attr.jid].groups[group:get_text()] = true;
cc05b6366576 mod_storage_xep0227: Add roster storage (fixes #1023) Kim Alvefur <zash@zash.se> parents: 8350 diff changeset	244 end
cc05b6366576 mod_storage_xep0227: Add roster storage (fixes #1023) Kim Alvefur <zash@zash.se> parents: 8350 diff changeset	245 for pending in user:childtags("presence", "jabber:client") do
cc05b6366576 mod_storage_xep0227: Add roster storage (fixes #1023) Kim Alvefur <zash@zash.se> parents: 8350 diff changeset	246 r[false].pending[pending.attr.from] = true;
cc05b6366576 mod_storage_xep0227: Add roster storage (fixes #1023) Kim Alvefur <zash@zash.se> parents: 8350 diff changeset	247 end
cc05b6366576 mod_storage_xep0227: Add roster storage (fixes #1023) Kim Alvefur <zash@zash.se> parents: 8350 diff changeset	248 end
cc05b6366576 mod_storage_xep0227: Add roster storage (fixes #1023) Kim Alvefur <zash@zash.se> parents: 8350 diff changeset	249 return r;
cc05b6366576 mod_storage_xep0227: Add roster storage (fixes #1023) Kim Alvefur <zash@zash.se> parents: 8350 diff changeset	250 end
cc05b6366576 mod_storage_xep0227: Add roster storage (fixes #1023) Kim Alvefur <zash@zash.se> parents: 8350 diff changeset	251 end
cc05b6366576 mod_storage_xep0227: Add roster storage (fixes #1023) Kim Alvefur <zash@zash.se> parents: 8350 diff changeset	252 end;
cc05b6366576 mod_storage_xep0227: Add roster storage (fixes #1023) Kim Alvefur <zash@zash.se> parents: 8350 diff changeset	253 set = function(self, user, data)
cc05b6366576 mod_storage_xep0227: Add roster storage (fixes #1023) Kim Alvefur <zash@zash.se> parents: 8350 diff changeset	254 local xml = getXml(user, self.host);
cc05b6366576 mod_storage_xep0227: Add roster storage (fixes #1023) Kim Alvefur <zash@zash.se> parents: 8350 diff changeset	255 local usere = xml and getUserElement(xml);
cc05b6366576 mod_storage_xep0227: Add roster storage (fixes #1023) Kim Alvefur <zash@zash.se> parents: 8350 diff changeset	256 if usere then
cc05b6366576 mod_storage_xep0227: Add roster storage (fixes #1023) Kim Alvefur <zash@zash.se> parents: 8350 diff changeset	257 local roster = usere:get_child("query", 'jabber:iq:roster');
cc05b6366576 mod_storage_xep0227: Add roster storage (fixes #1023) Kim Alvefur <zash@zash.se> parents: 8350 diff changeset	258 if roster then removeStanzaChild(usere, roster); end
cc05b6366576 mod_storage_xep0227: Add roster storage (fixes #1023) Kim Alvefur <zash@zash.se> parents: 8350 diff changeset	259 usere:maptags(function (tag)
cc05b6366576 mod_storage_xep0227: Add roster storage (fixes #1023) Kim Alvefur <zash@zash.se> parents: 8350 diff changeset	260 if tag.attr.xmlns == "jabber:client" and tag.name == "presence" and tag.attr.type == "subscribe" then
cc05b6366576 mod_storage_xep0227: Add roster storage (fixes #1023) Kim Alvefur <zash@zash.se> parents: 8350 diff changeset	261 return nil;
cc05b6366576 mod_storage_xep0227: Add roster storage (fixes #1023) Kim Alvefur <zash@zash.se> parents: 8350 diff changeset	262 end
cc05b6366576 mod_storage_xep0227: Add roster storage (fixes #1023) Kim Alvefur <zash@zash.se> parents: 8350 diff changeset	263 return tag;
cc05b6366576 mod_storage_xep0227: Add roster storage (fixes #1023) Kim Alvefur <zash@zash.se> parents: 8350 diff changeset	264 end);
cc05b6366576 mod_storage_xep0227: Add roster storage (fixes #1023) Kim Alvefur <zash@zash.se> parents: 8350 diff changeset	265 if data and next(data) ~= nil then
cc05b6366576 mod_storage_xep0227: Add roster storage (fixes #1023) Kim Alvefur <zash@zash.se> parents: 8350 diff changeset	266 roster = st.stanza("query", {xmlns='jabber:iq:roster'});
cc05b6366576 mod_storage_xep0227: Add roster storage (fixes #1023) Kim Alvefur <zash@zash.se> parents: 8350 diff changeset	267 usere:add_child(roster);
cc05b6366576 mod_storage_xep0227: Add roster storage (fixes #1023) Kim Alvefur <zash@zash.se> parents: 8350 diff changeset	268 for jid, item in pairs(data) do
cc05b6366576 mod_storage_xep0227: Add roster storage (fixes #1023) Kim Alvefur <zash@zash.se> parents: 8350 diff changeset	269 if jid then
cc05b6366576 mod_storage_xep0227: Add roster storage (fixes #1023) Kim Alvefur <zash@zash.se> parents: 8350 diff changeset	270 roster:tag("item", {
cc05b6366576 mod_storage_xep0227: Add roster storage (fixes #1023) Kim Alvefur <zash@zash.se> parents: 8350 diff changeset	271 jid = jid,
cc05b6366576 mod_storage_xep0227: Add roster storage (fixes #1023) Kim Alvefur <zash@zash.se> parents: 8350 diff changeset	272 subscription = item.subscription,
cc05b6366576 mod_storage_xep0227: Add roster storage (fixes #1023) Kim Alvefur <zash@zash.se> parents: 8350 diff changeset	273 ask = item.ask,
cc05b6366576 mod_storage_xep0227: Add roster storage (fixes #1023) Kim Alvefur <zash@zash.se> parents: 8350 diff changeset	274 name = item.name,
cc05b6366576 mod_storage_xep0227: Add roster storage (fixes #1023) Kim Alvefur <zash@zash.se> parents: 8350 diff changeset	275 });
cc05b6366576 mod_storage_xep0227: Add roster storage (fixes #1023) Kim Alvefur <zash@zash.se> parents: 8350 diff changeset	276 for group in pairs(item.groups) do
cc05b6366576 mod_storage_xep0227: Add roster storage (fixes #1023) Kim Alvefur <zash@zash.se> parents: 8350 diff changeset	277 roster:tag("group"):text(group):up();
cc05b6366576 mod_storage_xep0227: Add roster storage (fixes #1023) Kim Alvefur <zash@zash.se> parents: 8350 diff changeset	278 end
cc05b6366576 mod_storage_xep0227: Add roster storage (fixes #1023) Kim Alvefur <zash@zash.se> parents: 8350 diff changeset	279 roster:up(); -- move out from item
cc05b6366576 mod_storage_xep0227: Add roster storage (fixes #1023) Kim Alvefur <zash@zash.se> parents: 8350 diff changeset	280 else
cc05b6366576 mod_storage_xep0227: Add roster storage (fixes #1023) Kim Alvefur <zash@zash.se> parents: 8350 diff changeset	281 roster.attr.version = item.version;
cc05b6366576 mod_storage_xep0227: Add roster storage (fixes #1023) Kim Alvefur <zash@zash.se> parents: 8350 diff changeset	282 for pending_jid in pairs(item.pending) do
cc05b6366576 mod_storage_xep0227: Add roster storage (fixes #1023) Kim Alvefur <zash@zash.se> parents: 8350 diff changeset	283 usere:add_child(st.presence({ from = pending_jid, type = "subscribe" }));
cc05b6366576 mod_storage_xep0227: Add roster storage (fixes #1023) Kim Alvefur <zash@zash.se> parents: 8350 diff changeset	284 end
cc05b6366576 mod_storage_xep0227: Add roster storage (fixes #1023) Kim Alvefur <zash@zash.se> parents: 8350 diff changeset	285 end
cc05b6366576 mod_storage_xep0227: Add roster storage (fixes #1023) Kim Alvefur <zash@zash.se> parents: 8350 diff changeset	286 end
cc05b6366576 mod_storage_xep0227: Add roster storage (fixes #1023) Kim Alvefur <zash@zash.se> parents: 8350 diff changeset	287 end
cc05b6366576 mod_storage_xep0227: Add roster storage (fixes #1023) Kim Alvefur <zash@zash.se> parents: 8350 diff changeset	288 return setXml(user, self.host, xml);
cc05b6366576 mod_storage_xep0227: Add roster storage (fixes #1023) Kim Alvefur <zash@zash.se> parents: 8350 diff changeset	289 end
cc05b6366576 mod_storage_xep0227: Add roster storage (fixes #1023) Kim Alvefur <zash@zash.se> parents: 8350 diff changeset	290 return true;
cc05b6366576 mod_storage_xep0227: Add roster storage (fixes #1023) Kim Alvefur <zash@zash.se> parents: 8350 diff changeset	291 end;
cc05b6366576 mod_storage_xep0227: Add roster storage (fixes #1023) Kim Alvefur <zash@zash.se> parents: 8350 diff changeset	292 };
cc05b6366576 mod_storage_xep0227: Add roster storage (fixes #1023) Kim Alvefur <zash@zash.se> parents: 8350 diff changeset	293
11789 f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	294 -- PEP node configuration/etc. (not items)
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	295 local xmlns_pubsub_owner = "http://jabber.org/protocol/pubsub#owner";
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	296 local lib_pubsub = module:require "pubsub";
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	297 handlers.pep = {
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	298 get = function (self, user)
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	299 local xml = getXml(user, self.host);
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	300 local user_el = xml and getUserElement(xml);
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	301 if not user_el then
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	302 return nil;
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	303 end
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	304 local nodes = {
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	305 --[[
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	306 [node_name] = {
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	307 name = node_name;
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	308 config = {};
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	309 affiliations = {};
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	310 subscribers = {};
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	311 };
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	312 ]]
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	313 };
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	314 local owner_el = user_el:get_child("pubsub", xmlns_pubsub_owner);
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	315 for node_el in owner_el:childtags() do
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	316 local node_name = node_el.attr.node;
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	317 local node = nodes[node_name];
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	318 if not node then
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	319 node = {
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	320 name = node_name;
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	321 config = {};
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	322 affiliations = {};
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	323 subscribers = {};
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	324 };
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	325 nodes[node_name] = node;
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	326 end
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	327 if node_el.name == "configure" then
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	328 local form = node_el:get_child("x", "jabber:x:data");
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	329 if form then
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	330 node.config = lib_pubsub.node_config_form:data(form);
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	331 end
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	332 elseif node_el.name == "affiliations" then
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	333 for affiliation_el in node_el:childtags("affiliation") do
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	334 local aff_jid = jid_prep(affiliation_el.attr.jid);
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	335 local aff_value = affiliation_el.attr.affiliation;
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	336 if aff_jid and aff_value then
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	337 node.affiliations[aff_jid] = aff_value;
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	338 end
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	339 end
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	340 elseif node_el.name == "subscriptions" then
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	341 for subscription_el in node_el:childtags("subscription") do
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	342 local sub_jid = jid_prep(subscription_el.attr.jid);
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	343 local sub_state = subscription_el.attr.subscription;
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	344 if sub_jid and sub_state == "subscribed" then
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	345 local options;
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	346 local subscription_options_el = subscription_el:get_child("options");
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	347 if subscription_options_el then
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	348 local options_form = subscription_options_el:get_child("x", "jabber:x:data");
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	349 if options_form then
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	350 options = lib_pubsub.subscription_options_form:data(options_form);
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	351 end
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	352 end
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	353 node.subscribers[sub_jid] = options or true;
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	354 end
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	355 end
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	356 else
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	357 module:log("warn", "Ignoring unknown pubsub element: %s", node_el.name);
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	358 end
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	359 end
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	360 return nodes;
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	361 end;
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	362 set = function(self, user, data)
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	363 local xml = getXml(user, self.host);
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	364 local user_el = xml and getUserElement(xml);
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	365 if not user_el then
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	366 return true;
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	367 end
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	368 -- Remove existing data, if any
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	369 user_el:remove_children("pubsub", xmlns_pubsub_owner);
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	370
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	371 -- Generate new data
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	372 local owner_el = st.stanza("pubsub", { xmlns = xmlns_pubsub_owner });
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	373
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	374 for node_name, node_data in pairs(data) do
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	375 local configure_el = st.stanza("configure", { node = node_name })
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	376 :add_child(lib_pubsub.node_config_form:form(node_data.config, "submit"));
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	377 owner_el:add_child(configure_el);
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	378 if node_data.affiliations and next(node_data.affiliations) ~= nil then
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	379 local affiliations_el = st.stanza("affiliations", { node = node_name });
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	380 for aff_jid, aff_value in pairs(node_data.affiliations) do
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	381 affiliations_el:tag("affiliation", { jid = aff_jid, affiliation = aff_value }):up();
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	382 end
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	383 owner_el:add_child(affiliations_el);
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	384 end
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	385 if node_data.subscribers and next(node_data.subscribers) ~= nil then
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	386 local subscriptions_el = st.stanza("subscriptions", { node = node_name });
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	387 for sub_jid, sub_data in pairs(node_data.subscribers) do
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	388 local sub_el = st.stanza("subscription", { jid = sub_jid, subscribed = "subscribed" });
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	389 if sub_data ~= true then
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	390 local options_form = lib_pubsub.subscription_options_form:form(sub_data, "submit");
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	391 sub_el:tag("options"):add_child(options_form):up();
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	392 end
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	393 subscriptions_el:add_child(sub_el);
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	394 end
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	395 owner_el:add_child(subscriptions_el);
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	396 end
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	397 end
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	398
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	399 user_el:add_child(owner_el);
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	400
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	401 return setXml(user, self.host, xml);
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	402 end;
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	403 };
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	404
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	405 -- PEP items
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	406 local xmlns_pubsub = "http://jabber.org/protocol/pubsub";
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	407 handlers.pep_ = {
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	408 _stores = function (self, xml) --luacheck: ignore 212/self
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	409 local store_names = set.new();
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	410
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	411 local user_el = xml and getUserElement(xml);
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	412 if not user_el then
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	413 return store_names;
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	414 end
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	415
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	416 -- Locate existing pubsub element, if any
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	417 local pubsub_el = user_el:get_child("pubsub", xmlns_pubsub);
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	418 if not pubsub_el then
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	419 return store_names;
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	420 end
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	421
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	422 -- Find node items element, if any
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	423 for items_el in pubsub_el:childtags("items") do
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	424 store_names:add("pep_"..items_el.attr.node);
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	425 end
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	426 return store_names;
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	427 end;
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	428 find = function (self, user, query)
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	429 -- query keys: limit, reverse, key (id)
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	430
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	431 local xml = getXml(user, self.host);
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	432 local user_el = xml and getUserElement(xml);
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	433 if not user_el then
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	434 return nil, "no 227 user element found";
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	435 end
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	436
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	437 local node_name = self.datastore:match("^pep_(.+)$");
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	438
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	439 -- Locate existing pubsub element, if any
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	440 local pubsub_el = user_el:get_child("pubsub", xmlns_pubsub);
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	441 if not pubsub_el then
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	442 return nil;
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	443 end
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	444
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	445 -- Find node items element, if any
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	446 local node_items_el;
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	447 for items_el in pubsub_el:childtags("items") do
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	448 if items_el.attr.node == node_name then
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	449 node_items_el = items_el;
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	450 break;
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	451 end
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	452 end
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	453
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	454 if not node_items_el then
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	455 return nil;
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	456 end
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	457
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	458 local user_jid = user.."@"..self.host;
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	459
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	460 local results = {};
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	461 for item_el in node_items_el:childtags("item") do
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	462 if query and query.key then
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	463 if item_el.attr.id == query.key then
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	464 table.insert(results, { item_el.attr.id, item_el.tags[1], 0, user_jid });
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	465 break;
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	466 end
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	467 else
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	468 table.insert(results, { item_el.attr.id, item_el.tags[1], 0, user_jid });
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	469 end
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	470 if query and query.limit and #results >= query.limit then
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	471 break;
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	472 end
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	473 end
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	474 if query and query.reverse then
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	475 return array.reverse(results);
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	476 end
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	477 local i = 0;
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	478 return function ()
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	479 i = i + 1;
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	480 local v = results[i];
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	481 if v == nil then return nil; end
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	482 return unpack(v, 1, 4);
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	483 end;
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	484 end;
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	485 append = function (self, user, key, payload, when, with) --luacheck: ignore 212/when 212/with 212/key
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	486 local xml = getXml(user, self.host);
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	487 local user_el = xml and getUserElement(xml);
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	488 if not user_el then
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	489 return true;
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	490 end
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	491
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	492 local node_name = self.datastore:match("^pep_(.+)$");
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	493
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	494 -- Locate existing pubsub element, if any
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	495 local pubsub_el = user_el:get_child("pubsub", xmlns_pubsub);
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	496 if not pubsub_el then
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	497 pubsub_el = st.stanza("pubsub", { xmlns = xmlns_pubsub });
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	498 user_el:add_child(pubsub_el);
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	499 end
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	500
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	501 -- Find node items element, if any
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	502 local node_items_el;
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	503 for items_el in pubsub_el:childtags("items") do
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	504 if items_el.attr.node == node_name then
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	505 node_items_el = items_el;
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	506 break;
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	507 end
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	508 end
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	509
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	510 if not node_items_el then
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	511 -- Doesn't exist yet, create one
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	512 node_items_el = st.stanza("items", { node = node_name });
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	513 pubsub_el:add_child(node_items_el);
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	514 end
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	515
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	516 -- Append item to pubsub_el
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	517 local item_el = st.stanza("item", { id = key })
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	518 :add_child(payload);
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	519 node_items_el:add_child(item_el);
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	520
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	521 return setXml(user, self.host, xml);
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	522 end;
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	523 delete = function (self, user, query)
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	524 -- query keys: limit, reverse, key (id)
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	525
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	526 local xml = getXml(user, self.host);
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	527 local user_el = xml and getUserElement(xml);
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	528 if not user_el then
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	529 return nil, "no 227 user element found";
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	530 end
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	531
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	532 local node_name = self.datastore:match("^pep_(.+)$");
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	533
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	534 -- Locate existing pubsub element, if any
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	535 local pubsub_el = user_el:get_child("pubsub", xmlns_pubsub);
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	536 if not pubsub_el then
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	537 return nil;
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	538 end
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	539
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	540 -- Find node items element, if any
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	541 local node_items_el;
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	542 for items_el in pubsub_el:childtags("items") do
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	543 if items_el.attr.node == node_name then
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	544 node_items_el = items_el;
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	545 break;
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	546 end
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	547 end
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	548
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	549 if not node_items_el then
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	550 return nil;
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	551 end
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	552
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	553 local results = array();
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	554 for item_el in pubsub_el:childtags("item") do
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	555 if query and query.key then
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	556 if item_el.attr.id == query.key then
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	557 table.insert(results, item_el);
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	558 break;
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	559 end
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	560 else
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	561 table.insert(results, item_el);
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	562 end
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	563 if query and query.limit and #results >= query.limit then
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	564 break;
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	565 end
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	566 end
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	567 if query and query.truncate then
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	568 results:sub(-query.truncate);
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	569 end
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	570
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	571 -- Actually remove the matching items
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	572 local delete_keys = set.new(results:map(function (item) return item.attr.id; end));
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	573 pubsub_el:maptags(function (item_el)
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	574 if delete_keys:contains(item_el.attr.id) then
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	575 return nil;
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	576 end
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	577 return item_el;
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	578 end);
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	579 return setXml(user, self.host, xml);
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	580 end;
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	581 };
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	582
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	583 -- MAM archives
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	584 local xmlns_pie_mam = "urn:xmpp:pie:0#mam";
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	585 handlers.archive = {
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	586 find = function (self, user, query)
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	587 assert(query == nil, "XEP-0313 queries are not supported on XEP-0227 files");
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	588
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	589 local xml = getXml(user, self.host);
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	590 local user_el = xml and getUserElement(xml);
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	591 if not user_el then
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	592 return nil, "no 227 user element found";
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	593 end
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	594
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	595 -- Locate existing archive element, if any
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	596 local archive_el = user_el:get_child("archive", xmlns_pie_mam);
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	597 if not archive_el then
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	598 return nil;
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	599 end
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	600
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	601 local user_jid = user.."@"..self.host;
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	602
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	603
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	604 local f, s, result_el = archive_el:childtags("result", "urn:xmpp:mam:2");
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	605 return function ()
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	606 result_el = f(s, result_el);
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	607 if not result_el then return nil; end
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	608
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	609 local id = result_el.attr.id;
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	610 local item = result_el:find("{urn:xmpp:forward:0}forwarded/{jabber:client}message");
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	611 assert(item, "Invalid stanza in XEP-0227 archive");
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	612 local when = dt.parse(result_el:find("{urn:xmpp:forward:0}forwarded/{urn:xmpp:delay}delay@stamp"));
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	613 local to_bare, from_bare = jid_bare(item.attr.to), jid_bare(item.attr.from);
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	614 local with = to_bare == user_jid and from_bare or to_bare;
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	615 -- id, item, when, with
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	616 return id, item, when, with;
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	617 end;
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	618 end;
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	619 append = function (self, user, key, payload, when, with) --luacheck: ignore 212/when 212/with 212/key
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	620 local xml = getXml(user, self.host);
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	621 local user_el = xml and getUserElement(xml);
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	622 if not user_el then
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	623 return true;
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	624 end
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	625
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	626 -- Locate existing archive element, if any
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	627 local archive_el = user_el:get_child("archive", xmlns_pie_mam);
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	628 if not archive_el then
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	629 archive_el = st.stanza("archive", { xmlns = xmlns_pie_mam });
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	630 user_el:add_child(archive_el);
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	631 end
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	632
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	633 local item = st.clone(payload);
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	634 item.attr.xmlns = "jabber:client";
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	635
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	636 local result_el = st.stanza("result", { xmlns = "urn:xmpp:mam:2", id = key })
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	637 :tag("forwarded", { xmlns = "urn:xmpp:forward:0" })
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	638 :tag("delay", { xmlns = "urn:xmpp:delay", stamp = dt.datetime(when) }):up()
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	639 :add_child(item)
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	640 :up();
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	641
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	642 -- Append item to archive_el
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	643 archive_el:add_child(result_el);
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	644
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	645 return setXml(user, self.host, xml);
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	646 end;
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	647 };
8351 cc05b6366576 mod_storage_xep0227: Add roster storage (fixes #1023) Kim Alvefur <zash@zash.se> parents: 8350 diff changeset	648
3414 9a1f6239b63c storage/mod_xep0227: Initial commit. Waqas Hussain <waqas20@gmail.com> parents: diff changeset	649 -----------------------------
9a1f6239b63c storage/mod_xep0227: Initial commit. Waqas Hussain <waqas20@gmail.com> parents: diff changeset	650 local driver = {};
9a1f6239b63c storage/mod_xep0227: Initial commit. Waqas Hussain <waqas20@gmail.com> parents: diff changeset	651
11789 f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	652 local function users(self)
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	653 local file_patt = "^.*@"..(self.host:gsub("%p", "%%%1")).."%.xml$";
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	654
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	655 local f, s, filename = lfs.dir(prosody.paths.data);
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	656
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	657 return function ()
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	658 filename = f(s, filename);
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	659 while filename and not filename:match(file_patt) do
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	660 filename = f(s, filename);
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	661 end
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	662 if not filename then return nil; end
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	663 return filename:match("^[^@]+");
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	664 end;
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	665 end
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	666
8352 6ff50541d2a6 mod_storage_xep0227: Ignore unused 'self' argument [luacheck] Kim Alvefur <zash@zash.se> parents: 8351 diff changeset	667 function driver:open(datastore, typ) -- luacheck: ignore 212/self
11789 f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	668 if typ and typ ~= "keyval" and typ ~= "archive" then return nil, "unsupported-store"; end
3414 9a1f6239b63c storage/mod_xep0227: Initial commit. Waqas Hussain <waqas20@gmail.com> parents: diff changeset	669 local handler = handlers[datastore];
11789 f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	670 if not handler and datastore:match("^pep_") then
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	671 handler = handlers.pep_;
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	672 end
6701 88a25c364a14 mod_storage_xep0227: Update open method for current API Kim Alvefur <zash@zash.se> parents: 6700 diff changeset	673 if not handler then return nil, "unsupported-datastore"; end
11789 f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	674 local instance = setmetatable({ host = module.host; datastore = datastore; users = users; }, { __index = handler });
3414 9a1f6239b63c storage/mod_xep0227: Initial commit. Waqas Hussain <waqas20@gmail.com> parents: diff changeset	675 if instance.init then instance:init(); end
9a1f6239b63c storage/mod_xep0227: Initial commit. Waqas Hussain <waqas20@gmail.com> parents: diff changeset	676 return instance;
9a1f6239b63c storage/mod_xep0227: Initial commit. Waqas Hussain <waqas20@gmail.com> parents: diff changeset	677 end
9a1f6239b63c storage/mod_xep0227: Initial commit. Waqas Hussain <waqas20@gmail.com> parents: diff changeset	678
11789 f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	679 local function get_store_names(self, path)
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	680 local stores = set.new();
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	681 local f, err = io_open(paths.join(prosody.paths.data, path));
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	682 if not f then
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	683 module:log("warn", "Unable to load XML file for <%s>: %s", "store listing", err);
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	684 return stores;
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	685 end
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	686 module:log("info", "Loaded %s", path);
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	687 local s = f:read("*a");
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	688 f:close();
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	689 local xml = parse_xml_real(s);
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	690 for _, handler_funcs in pairs(handlers) do
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	691 if handler_funcs._stores then
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	692 stores:include(handler_funcs._stores(self, xml));
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	693 end
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	694 end
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	695 return stores;
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	696 end
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	697
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	698 function driver:stores(username)
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	699 local store_dir = prosody.paths.data;
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	700
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	701 local mode, err = lfs.attributes(store_dir, "mode");
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	702 if not mode then
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	703 return function() module:log("debug", "Could not iterate over stores in %s: %s", store_dir, err); end
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	704 end
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	705
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	706 local file_patt = "^.*@"..(module.host:gsub("%p", "%%%1")).."%.xml$";
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	707
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	708 local all_users = username == true;
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	709
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	710 local store_names = set.new();
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	711
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	712 for filename in lfs.dir(prosody.paths.data) do
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	713 if filename:match(file_patt) then
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	714 if all_users or filename == username.."@"..module.host..".xml" then
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	715 store_names:include(get_store_names(self, filename));
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	716 if not all_users then break; end
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	717 end
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	718 end
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	719 end
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	720
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	721 return store_names:items();
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	722 end
f3085620b6ff mod_storage_xep0227: Update for XEP-0227 r1.1: Support for SCRAM, MAM, PEP Matthew Wild <mwild1@gmail.com> parents: 8352 diff changeset	723
5121 b5a5643f8572 core.storagemanager, mod_storage_: "data-driver" -> "storage-provider", to allow using module:provides(). Waqas Hussain <waqas20@gmail.com>* parents: 3414 diff changeset	724 module:provides("storage", driver);

prosody

16cf863b36c0

plugins/mod_storage_xep0227.lua

Software / code / prosody

Annotate

plugins/mod_storage_xep0227.lua @ 11928:16cf863b36c0

Software `/` code `/` prosody