Software `/` code `/` prosody

Annotate

plugins/mod_storage_xep0227.lua @ 11961:542a9a503073

util.async: Add sleep() method with configurable scheduling backend No scheduler set by default, so it will error (we plan to initialize it in util.startup). We wanted to avoid a hard dependency on util.timer (which in turn depends on network backends, etc.), and we didn't add timer.sleep() because we didn't want to add a hard dependency on util.async for things that don't need it.

author	Matthew Wild <mwild1@gmail.com>
date	Mon, 29 Nov 2021 14:11:24 +0000
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

542a9a503073

plugins/mod_storage_xep0227.lua

Software / code / prosody

Annotate

plugins/mod_storage_xep0227.lua @ 11961:542a9a503073

Software `/` code `/` prosody