prosody-modules: mod_mam/mod

annotate mod_mam/mod_mam.lua @ 3568:6b3181fe5617

mod_auth_token: Timezone fix for TOTP checking luatz.time() returns milliseconds since epoch which is in UTC time, so we don't need to convert to UTC with gmtime. By calling gmtime, TOTP validation was failing when this module wasn't running on machine set to UTC time.

author	JC Brand <jc@opkode.com>
date	Thu, 02 May 2019 11:07:27 +0200
parents	0d2d4d5bb5f5
children	e45a3d04367b

rev	line source
635 30be50d2537f mod_mam: Update header Kim Alvefur <zash@zash.se> parents: 634 diff changeset	1 -- XEP-0313: Message Archive Management for Prosody
2000 b5adfe72709b mod_mam: Improve error messages when failed to open storage Kim Alvefur <zash@zash.se> parents: 1957 diff changeset	2 -- Copyright (C) 2011-2016 Kim Alvefur
523 eff140d53b83 mod_mam: Add experimental implementation of the Message Archive Management ProtoXEP Kim Alvefur <zash@zash.se> parents: diff changeset	3 --
eff140d53b83 mod_mam: Add experimental implementation of the Message Archive Management ProtoXEP Kim Alvefur <zash@zash.se> parents: diff changeset	4 -- This file is MIT/X11 licensed.
eff140d53b83 mod_mam: Add experimental implementation of the Message Archive Management ProtoXEP Kim Alvefur <zash@zash.se> parents: diff changeset	5
2506 5941aac79f06 mod_mam: Add support for XEP-0313 v0.5 Kim Alvefur <zash@zash.se> parents: 2505 diff changeset	6 local xmlns_mam0 = "urn:xmpp:mam:0";
5941aac79f06 mod_mam: Add support for XEP-0313 v0.5 Kim Alvefur <zash@zash.se> parents: 2505 diff changeset	7 local xmlns_mam1 = "urn:xmpp:mam:1";
2516 248054199d0f mod_mam: Add support for XEP-0313 v0.6 Kim Alvefur <zash@zash.se> parents: 2515 diff changeset	8 local xmlns_mam2 = "urn:xmpp:mam:2";
558 66de25ffc8d9 mod_mam: Implement archiving preferences. Kim Alvefur <zash@zash.se> parents: 523 diff changeset	9 local xmlns_delay = "urn:xmpp:delay";
66de25ffc8d9 mod_mam: Implement archiving preferences. Kim Alvefur <zash@zash.se> parents: 523 diff changeset	10 local xmlns_forward = "urn:xmpp:forward:0";
2516 248054199d0f mod_mam: Add support for XEP-0313 v0.6 Kim Alvefur <zash@zash.se> parents: 2515 diff changeset	11 local xmlns_st_id = "urn:xmpp:sid:0";
558 66de25ffc8d9 mod_mam: Implement archiving preferences. Kim Alvefur <zash@zash.se> parents: 523 diff changeset	12
2293 144b74caa5ef mod_mam: Check that user exists (#731) Kim Alvefur <zash@zash.se> parents: 2270 diff changeset	13 local um = require "core.usermanager";
523 eff140d53b83 mod_mam: Add experimental implementation of the Message Archive Management ProtoXEP Kim Alvefur <zash@zash.se> parents: diff changeset	14 local st = require "util.stanza";
701 cc5805f83583 mod_mam: Implement support for Result Set Management in queries. Kim Alvefur <zash@zash.se> parents: 675 diff changeset	15 local rsm = module:require "rsm";
1399 7fb6b607afd6 mod_mam: Shuffle imports Kim Alvefur <zash@zash.se> parents: 1369 diff changeset	16 local get_prefs = module:require"mamprefs".get;
7fb6b607afd6 mod_mam: Shuffle imports Kim Alvefur <zash@zash.se> parents: 1369 diff changeset	17 local set_prefs = module:require"mamprefs".set;
7fb6b607afd6 mod_mam: Shuffle imports Kim Alvefur <zash@zash.se> parents: 1369 diff changeset	18 local prefs_to_stanza = module:require"mamprefsxml".tostanza;
7fb6b607afd6 mod_mam: Shuffle imports Kim Alvefur <zash@zash.se> parents: 1369 diff changeset	19 local prefs_from_stanza = module:require"mamprefsxml".fromstanza;
523 eff140d53b83 mod_mam: Add experimental implementation of the Message Archive Management ProtoXEP Kim Alvefur <zash@zash.se> parents: diff changeset	20 local jid_bare = require "util.jid".bare;
eff140d53b83 mod_mam: Add experimental implementation of the Message Archive Management ProtoXEP Kim Alvefur <zash@zash.se> parents: diff changeset	21 local jid_split = require "util.jid".split;
2516 248054199d0f mod_mam: Add support for XEP-0313 v0.6 Kim Alvefur <zash@zash.se> parents: 2515 diff changeset	22 local jid_prepped_split = require "util.jid".prepped_split;
1484 53a3a19d6093 mod_mam: Update to version 0.3 of XEP-0313 Kim Alvefur <zash@zash.se> parents: 1403 diff changeset	23 local dataform = require "util.dataforms".new;
558 66de25ffc8d9 mod_mam: Implement archiving preferences. Kim Alvefur <zash@zash.se> parents: 523 diff changeset	24 local host = module.host;
66de25ffc8d9 mod_mam: Implement archiving preferences. Kim Alvefur <zash@zash.se> parents: 523 diff changeset	25
66de25ffc8d9 mod_mam: Implement archiving preferences. Kim Alvefur <zash@zash.se> parents: 523 diff changeset	26 local rm_load_roster = require "core.rostermanager".load_roster;
66de25ffc8d9 mod_mam: Implement archiving preferences. Kim Alvefur <zash@zash.se> parents: 523 diff changeset	27
1116 2345a30dd8b4 mod_mam: Update to use 'archive' storage type. Note: this breaks support with 0.9 and older. Kim Alvefur <zash@zash.se> parents: 1114 diff changeset	28 local getmetatable = getmetatable;
2345a30dd8b4 mod_mam: Update to use 'archive' storage type. Note: this breaks support with 0.9 and older. Kim Alvefur <zash@zash.se> parents: 1114 diff changeset	29 local function is_stanza(x)
2345a30dd8b4 mod_mam: Update to use 'archive' storage type. Note: this breaks support with 0.9 and older. Kim Alvefur <zash@zash.se> parents: 1114 diff changeset	30 return getmetatable(x) == st.stanza_mt;
2345a30dd8b4 mod_mam: Update to use 'archive' storage type. Note: this breaks support with 0.9 and older. Kim Alvefur <zash@zash.se> parents: 1114 diff changeset	31 end
2345a30dd8b4 mod_mam: Update to use 'archive' storage type. Note: this breaks support with 0.9 and older. Kim Alvefur <zash@zash.se> parents: 1114 diff changeset	32
672 8ae5317ba032 mod_mam: local tostring and some comments Kim Alvefur <zash@zash.se> parents: 671 diff changeset	33 local tostring = tostring;
523 eff140d53b83 mod_mam: Add experimental implementation of the Message Archive Management ProtoXEP Kim Alvefur <zash@zash.se> parents: diff changeset	34 local time_now = os.time;
707 f987c7b79008 mod_mam: Fix typo Kim Alvefur <zash@zash.se> parents: 706 diff changeset	35 local m_min = math.min;
523 eff140d53b83 mod_mam: Add experimental implementation of the Message Archive Management ProtoXEP Kim Alvefur <zash@zash.se> parents: diff changeset	36 local timestamp, timestamp_parse = require "util.datetime".datetime, require "util.datetime".parse;
706 5c2b96c4dde6 mod_mam: Enforce a max number of items returned, with a default. Kim Alvefur <zash@zash.se> parents: 705 diff changeset	37 local default_max_items, max_max_items = 20, module:get_option_number("max_archive_query_results", 50);
2017 d44ac0756c46 mod_mam: Enable archiving by default, less of a privacy issue if messages expire Kim Alvefur <zash@zash.se> parents: 2016 diff changeset	38 local global_default_policy = module:get_option("default_archive_policy", true);
1587 3ac2b835c7b3 mod_mam: Make sure default_archive_policy is a boolean or "roster" (thanks souliane) Kim Alvefur <zash@zash.se> parents: 1484 diff changeset	39 if global_default_policy ~= "roster" then
3ac2b835c7b3 mod_mam: Make sure default_archive_policy is a boolean or "roster" (thanks souliane) Kim Alvefur <zash@zash.se> parents: 1484 diff changeset	40 global_default_policy = module:get_option_boolean("default_archive_policy", global_default_policy);
3ac2b835c7b3 mod_mam: Make sure default_archive_policy is a boolean or "roster" (thanks souliane) Kim Alvefur <zash@zash.se> parents: 1484 diff changeset	41 end
1116 2345a30dd8b4 mod_mam: Update to use 'archive' storage type. Note: this breaks support with 0.9 and older. Kim Alvefur <zash@zash.se> parents: 1114 diff changeset	42
2510 d81882aa0e1e mod_mam: Make the store configurable to ease move to 0.10 version Kim Alvefur <zash@zash.se> parents: 2506 diff changeset	43 local archive_store = module:get_option_string("archive_store", "archive2");
2662 2d5e0e2938d1 mod_mam: Remove extraneous assert when opening the store Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: 2644 diff changeset	44 local archive = module:open_store(archive_store, "archive");
2020 37b30f10fbba mod_mam: Bit of spacing between blocks of code Kim Alvefur <zash@zash.se> parents: 2019 diff changeset	45
2024 6f4dcc723a60 mod_mam: Use the fallback driver if either the null driver or a driver that does not implement the archive api is returned Kim Alvefur <zash@zash.se> parents: 2023 diff changeset	46 if archive.name == "null" or not archive.find then
2644 0f44d04d0d18 mod_mam: Ignore long log lines [luacheck] Kim Alvefur <zash@zash.se> parents: 2516 diff changeset	47 -- luacheck: ignore 631
2024 6f4dcc723a60 mod_mam: Use the fallback driver if either the null driver or a driver that does not implement the archive api is returned Kim Alvefur <zash@zash.se> parents: 2023 diff changeset	48 if not archive.find then
6f4dcc723a60 mod_mam: Use the fallback driver if either the null driver or a driver that does not implement the archive api is returned Kim Alvefur <zash@zash.se> parents: 2023 diff changeset	49 module:log("debug", "Attempt to open archive storage returned a valid driver but it does not seem to implement the storage API");
6f4dcc723a60 mod_mam: Use the fallback driver if either the null driver or a driver that does not implement the archive api is returned Kim Alvefur <zash@zash.se> parents: 2023 diff changeset	50 module:log("debug", "mod_%s does not support archiving", archive._provided_by or archive.name and "storage_"..archive.name.."(?)" or "<unknown>");
6f4dcc723a60 mod_mam: Use the fallback driver if either the null driver or a driver that does not implement the archive api is returned Kim Alvefur <zash@zash.se> parents: 2023 diff changeset	51 else
6f4dcc723a60 mod_mam: Use the fallback driver if either the null driver or a driver that does not implement the archive api is returned Kim Alvefur <zash@zash.se> parents: 2023 diff changeset	52 module:log("debug", "Attempt to open archive storage returned null driver");
6f4dcc723a60 mod_mam: Use the fallback driver if either the null driver or a driver that does not implement the archive api is returned Kim Alvefur <zash@zash.se> parents: 2023 diff changeset	53 end
2030 66156e4d5274 mod_mam: This was meant to be a debug message Kim Alvefur <zash@zash.se> parents: 2025 diff changeset	54 module:log("debug", "See https://prosody.im/doc/storage and https://prosody.im/doc/archiving for more information");
2023 98b4794b72e4 mod_mam: Include an in-memory fallback driver Kim Alvefur <zash@zash.se> parents: 2022 diff changeset	55 module:log("info", "Using in-memory fallback archive driver");
98b4794b72e4 mod_mam: Include an in-memory fallback driver Kim Alvefur <zash@zash.se> parents: 2022 diff changeset	56 archive = module:require "fallback_archive";
1185 30b681898c2d mod_mam: Log error message if unable to open archive storage Kim Alvefur <zash@zash.se> parents: 1152 diff changeset	57 end
558 66de25ffc8d9 mod_mam: Implement archiving preferences. Kim Alvefur <zash@zash.se> parents: 523 diff changeset	58
2724 9f17d0f28d77 mod_mam: Don't ask the storage backend to count all items when expiry is disabled Kim Alvefur <zash@zash.se> parents: 2711 diff changeset	59 local use_total = true;
9f17d0f28d77 mod_mam: Don't ask the storage backend to count all items when expiry is disabled Kim Alvefur <zash@zash.se> parents: 2711 diff changeset	60
2016 279885fd9728 mod_mam: Add support for trimming old archived messages Kim Alvefur <zash@zash.se> parents: 2001 diff changeset	61 local cleanup;
279885fd9728 mod_mam: Add support for trimming old archived messages Kim Alvefur <zash@zash.se> parents: 2001 diff changeset	62
2764 1872a9129c2f mod_mam: Use a FIFO queue for scheduling archive expiry Kim Alvefur <zash@zash.se> parents: 2762 diff changeset	63 local function schedule_cleanup(username)
1872a9129c2f mod_mam: Use a FIFO queue for scheduling archive expiry Kim Alvefur <zash@zash.se> parents: 2762 diff changeset	64 if cleanup and not cleanup[username] then
1872a9129c2f mod_mam: Use a FIFO queue for scheduling archive expiry Kim Alvefur <zash@zash.se> parents: 2762 diff changeset	65 table.insert(cleanup, username);
1872a9129c2f mod_mam: Use a FIFO queue for scheduling archive expiry Kim Alvefur <zash@zash.se> parents: 2762 diff changeset	66 cleanup[username] = true;
1872a9129c2f mod_mam: Use a FIFO queue for scheduling archive expiry Kim Alvefur <zash@zash.se> parents: 2762 diff changeset	67 end
1872a9129c2f mod_mam: Use a FIFO queue for scheduling archive expiry Kim Alvefur <zash@zash.se> parents: 2762 diff changeset	68 end
1872a9129c2f mod_mam: Use a FIFO queue for scheduling archive expiry Kim Alvefur <zash@zash.se> parents: 2762 diff changeset	69
558 66de25ffc8d9 mod_mam: Implement archiving preferences. Kim Alvefur <zash@zash.se> parents: 523 diff changeset	70 -- Handle prefs.
2506 5941aac79f06 mod_mam: Add support for XEP-0313 v0.5 Kim Alvefur <zash@zash.se> parents: 2505 diff changeset	71 local function handle_prefs(event)
523 eff140d53b83 mod_mam: Add experimental implementation of the Message Archive Management ProtoXEP Kim Alvefur <zash@zash.se> parents: diff changeset	72 local origin, stanza = event.origin, event.stanza;
2506 5941aac79f06 mod_mam: Add support for XEP-0313 v0.5 Kim Alvefur <zash@zash.se> parents: 2505 diff changeset	73 local xmlns_mam = stanza.tags[1].attr.xmlns;
558 66de25ffc8d9 mod_mam: Implement archiving preferences. Kim Alvefur <zash@zash.se> parents: 523 diff changeset	74 local user = origin.username;
523 eff140d53b83 mod_mam: Add experimental implementation of the Message Archive Management ProtoXEP Kim Alvefur <zash@zash.se> parents: diff changeset	75 if stanza.attr.type == "get" then
2506 5941aac79f06 mod_mam: Add support for XEP-0313 v0.5 Kim Alvefur <zash@zash.se> parents: 2505 diff changeset	76 local prefs = prefs_to_stanza(get_prefs(user), xmlns_mam);
1135 0d6ab5e4bc30 mod_mam: Break out routines for converting prefs between XML and our internal representation into a library Kim Alvefur <zash@zash.se> parents: 1116 diff changeset	77 local reply = st.reply(stanza):add_child(prefs);
1681 d20cfc5ba827 mod_mam: Always return true when a stanza event has been handled Kim Alvefur <zash@zash.se> parents: 1678 diff changeset	78 origin.send(reply);
523 eff140d53b83 mod_mam: Add experimental implementation of the Message Archive Management ProtoXEP Kim Alvefur <zash@zash.se> parents: diff changeset	79 else -- type == "set"
558 66de25ffc8d9 mod_mam: Implement archiving preferences. Kim Alvefur <zash@zash.se> parents: 523 diff changeset	80 local new_prefs = stanza:get_child("prefs", xmlns_mam);
1135 0d6ab5e4bc30 mod_mam: Break out routines for converting prefs between XML and our internal representation into a library Kim Alvefur <zash@zash.se> parents: 1116 diff changeset	81 local prefs = prefs_from_stanza(new_prefs);
558 66de25ffc8d9 mod_mam: Implement archiving preferences. Kim Alvefur <zash@zash.se> parents: 523 diff changeset	82 local ok, err = set_prefs(user, prefs);
66de25ffc8d9 mod_mam: Implement archiving preferences. Kim Alvefur <zash@zash.se> parents: 523 diff changeset	83 if not ok then
1681 d20cfc5ba827 mod_mam: Always return true when a stanza event has been handled Kim Alvefur <zash@zash.se> parents: 1678 diff changeset	84 origin.send(st.error_reply(stanza, "cancel", "internal-server-error", "Error storing preferences: "..tostring(err)));
d20cfc5ba827 mod_mam: Always return true when a stanza event has been handled Kim Alvefur <zash@zash.se> parents: 1678 diff changeset	85 else
d20cfc5ba827 mod_mam: Always return true when a stanza event has been handled Kim Alvefur <zash@zash.se> parents: 1678 diff changeset	86 origin.send(st.reply(stanza));
558 66de25ffc8d9 mod_mam: Implement archiving preferences. Kim Alvefur <zash@zash.se> parents: 523 diff changeset	87 end
523 eff140d53b83 mod_mam: Add experimental implementation of the Message Archive Management ProtoXEP Kim Alvefur <zash@zash.se> parents: diff changeset	88 end
1681 d20cfc5ba827 mod_mam: Always return true when a stanza event has been handled Kim Alvefur <zash@zash.se> parents: 1678 diff changeset	89 return true;
2506 5941aac79f06 mod_mam: Add support for XEP-0313 v0.5 Kim Alvefur <zash@zash.se> parents: 2505 diff changeset	90 end
5941aac79f06 mod_mam: Add support for XEP-0313 v0.5 Kim Alvefur <zash@zash.se> parents: 2505 diff changeset	91
5941aac79f06 mod_mam: Add support for XEP-0313 v0.5 Kim Alvefur <zash@zash.se> parents: 2505 diff changeset	92 module:hook("iq/self/"..xmlns_mam0..":prefs", handle_prefs);
5941aac79f06 mod_mam: Add support for XEP-0313 v0.5 Kim Alvefur <zash@zash.se> parents: 2505 diff changeset	93 module:hook("iq/self/"..xmlns_mam1..":prefs", handle_prefs);
2516 248054199d0f mod_mam: Add support for XEP-0313 v0.6 Kim Alvefur <zash@zash.se> parents: 2515 diff changeset	94 module:hook("iq/self/"..xmlns_mam2..":prefs", handle_prefs);
523 eff140d53b83 mod_mam: Add experimental implementation of the Message Archive Management ProtoXEP Kim Alvefur <zash@zash.se> parents: diff changeset	95
1484 53a3a19d6093 mod_mam: Update to version 0.3 of XEP-0313 Kim Alvefur <zash@zash.se> parents: 1403 diff changeset	96 local query_form = dataform {
2506 5941aac79f06 mod_mam: Add support for XEP-0313 v0.5 Kim Alvefur <zash@zash.se> parents: 2505 diff changeset	97 { name = "FORM_TYPE"; type = "hidden"; value = xmlns_mam0; };
1484 53a3a19d6093 mod_mam: Update to version 0.3 of XEP-0313 Kim Alvefur <zash@zash.se> parents: 1403 diff changeset	98 { name = "with"; type = "jid-single"; };
53a3a19d6093 mod_mam: Update to version 0.3 of XEP-0313 Kim Alvefur <zash@zash.se> parents: 1403 diff changeset	99 { name = "start"; type = "text-single" };
53a3a19d6093 mod_mam: Update to version 0.3 of XEP-0313 Kim Alvefur <zash@zash.se> parents: 1403 diff changeset	100 { name = "end"; type = "text-single"; };
53a3a19d6093 mod_mam: Update to version 0.3 of XEP-0313 Kim Alvefur <zash@zash.se> parents: 1403 diff changeset	101 };
53a3a19d6093 mod_mam: Update to version 0.3 of XEP-0313 Kim Alvefur <zash@zash.se> parents: 1403 diff changeset	102
53a3a19d6093 mod_mam: Update to version 0.3 of XEP-0313 Kim Alvefur <zash@zash.se> parents: 1403 diff changeset	103 -- Serve form
2506 5941aac79f06 mod_mam: Add support for XEP-0313 v0.5 Kim Alvefur <zash@zash.se> parents: 2505 diff changeset	104 local function handle_get_form(event)
1484 53a3a19d6093 mod_mam: Update to version 0.3 of XEP-0313 Kim Alvefur <zash@zash.se> parents: 1403 diff changeset	105 local origin, stanza = event.origin, event.stanza;
2506 5941aac79f06 mod_mam: Add support for XEP-0313 v0.5 Kim Alvefur <zash@zash.se> parents: 2505 diff changeset	106 local xmlns_mam = stanza.tags[1].attr.xmlns;
5941aac79f06 mod_mam: Add support for XEP-0313 v0.5 Kim Alvefur <zash@zash.se> parents: 2505 diff changeset	107 query_form[1].value = xmlns_mam;
2505 1398f3f76c0f mod_mam: Add missing wrapping <query> element when returning the query form Kim Alvefur <zash@zash.se> parents: 2293 diff changeset	108 origin.send(st.reply(stanza):query(xmlns_mam):add_child(query_form:form()));
1681 d20cfc5ba827 mod_mam: Always return true when a stanza event has been handled Kim Alvefur <zash@zash.se> parents: 1678 diff changeset	109 return true;
2506 5941aac79f06 mod_mam: Add support for XEP-0313 v0.5 Kim Alvefur <zash@zash.se> parents: 2505 diff changeset	110 end
5941aac79f06 mod_mam: Add support for XEP-0313 v0.5 Kim Alvefur <zash@zash.se> parents: 2505 diff changeset	111
5941aac79f06 mod_mam: Add support for XEP-0313 v0.5 Kim Alvefur <zash@zash.se> parents: 2505 diff changeset	112 module:hook("iq-get/self/"..xmlns_mam0..":query", handle_get_form);
5941aac79f06 mod_mam: Add support for XEP-0313 v0.5 Kim Alvefur <zash@zash.se> parents: 2505 diff changeset	113 module:hook("iq-get/self/"..xmlns_mam1..":query", handle_get_form);
2516 248054199d0f mod_mam: Add support for XEP-0313 v0.6 Kim Alvefur <zash@zash.se> parents: 2515 diff changeset	114 module:hook("iq-get/self/"..xmlns_mam2..":query", handle_get_form);
1484 53a3a19d6093 mod_mam: Update to version 0.3 of XEP-0313 Kim Alvefur <zash@zash.se> parents: 1403 diff changeset	115
1325 b21236b6b8d8 Backed out changeset 853a382c9bd6 Kim Alvefur <zash@zash.se> parents: 1324 diff changeset	116 -- Handle archive queries
2506 5941aac79f06 mod_mam: Add support for XEP-0313 v0.5 Kim Alvefur <zash@zash.se> parents: 2505 diff changeset	117 local function handle_mam_query(event)
1324 853a382c9bd6 mod_turncredentials: Advertise the XEP-0215 feature (thanks Gryffus) Kim Alvefur <zash@zash.se> parents: 1187 diff changeset	118 local origin, stanza = event.origin, event.stanza;
2506 5941aac79f06 mod_mam: Add support for XEP-0313 v0.5 Kim Alvefur <zash@zash.se> parents: 2505 diff changeset	119 local xmlns_mam = stanza.tags[1].attr.xmlns;
523 eff140d53b83 mod_mam: Add experimental implementation of the Message Archive Management ProtoXEP Kim Alvefur <zash@zash.se> parents: diff changeset	120 local query = stanza.tags[1];
1112 1dc07833355e mod_mam: Use more specific hook Kim Alvefur <zash@zash.se> parents: 1111 diff changeset	121 local qid = query.attr.queryid;
1dc07833355e mod_mam: Use more specific hook Kim Alvefur <zash@zash.se> parents: 1111 diff changeset	122
2764 1872a9129c2f mod_mam: Use a FIFO queue for scheduling archive expiry Kim Alvefur <zash@zash.se> parents: 2762 diff changeset	123 schedule_cleanup(origin.username);
2016 279885fd9728 mod_mam: Add support for trimming old archived messages Kim Alvefur <zash@zash.se> parents: 2001 diff changeset	124
1112 1dc07833355e mod_mam: Use more specific hook Kim Alvefur <zash@zash.se> parents: 1111 diff changeset	125 -- Search query parameters
1484 53a3a19d6093 mod_mam: Update to version 0.3 of XEP-0313 Kim Alvefur <zash@zash.se> parents: 1403 diff changeset	126 local qwith, qstart, qend;
53a3a19d6093 mod_mam: Update to version 0.3 of XEP-0313 Kim Alvefur <zash@zash.se> parents: 1403 diff changeset	127 local form = query:get_child("x", "jabber:x:data");
53a3a19d6093 mod_mam: Update to version 0.3 of XEP-0313 Kim Alvefur <zash@zash.se> parents: 1403 diff changeset	128 if form then
53a3a19d6093 mod_mam: Update to version 0.3 of XEP-0313 Kim Alvefur <zash@zash.se> parents: 1403 diff changeset	129 local err;
2506 5941aac79f06 mod_mam: Add support for XEP-0313 v0.5 Kim Alvefur <zash@zash.se> parents: 2505 diff changeset	130 query_form[1].value = xmlns_mam;
1484 53a3a19d6093 mod_mam: Update to version 0.3 of XEP-0313 Kim Alvefur <zash@zash.se> parents: 1403 diff changeset	131 form, err = query_form:data(form);
53a3a19d6093 mod_mam: Update to version 0.3 of XEP-0313 Kim Alvefur <zash@zash.se> parents: 1403 diff changeset	132 if err then
1698 55bc42c1d8c5 mod_mam: Semicolons! Kim Alvefur <zash@zash.se> parents: 1685 diff changeset	133 origin.send(st.error_reply(stanza, "modify", "bad-request", select(2, next(err))));
1681 d20cfc5ba827 mod_mam: Always return true when a stanza event has been handled Kim Alvefur <zash@zash.se> parents: 1678 diff changeset	134 return true;
1484 53a3a19d6093 mod_mam: Update to version 0.3 of XEP-0313 Kim Alvefur <zash@zash.se> parents: 1403 diff changeset	135 end
53a3a19d6093 mod_mam: Update to version 0.3 of XEP-0313 Kim Alvefur <zash@zash.se> parents: 1403 diff changeset	136 qwith, qstart, qend = form["with"], form["start"], form["end"];
53a3a19d6093 mod_mam: Update to version 0.3 of XEP-0313 Kim Alvefur <zash@zash.se> parents: 1403 diff changeset	137 qwith = qwith and jid_bare(qwith); -- dataforms does jidprep
53a3a19d6093 mod_mam: Update to version 0.3 of XEP-0313 Kim Alvefur <zash@zash.se> parents: 1403 diff changeset	138 end
523 eff140d53b83 mod_mam: Add experimental implementation of the Message Archive Management ProtoXEP Kim Alvefur <zash@zash.se> parents: diff changeset	139
1112 1dc07833355e mod_mam: Use more specific hook Kim Alvefur <zash@zash.se> parents: 1111 diff changeset	140 if qstart or qend then -- Validate timestamps
1698 55bc42c1d8c5 mod_mam: Semicolons! Kim Alvefur <zash@zash.se> parents: 1685 diff changeset	141 local vstart, vend = (qstart and timestamp_parse(qstart)), (qend and timestamp_parse(qend));
1112 1dc07833355e mod_mam: Use more specific hook Kim Alvefur <zash@zash.se> parents: 1111 diff changeset	142 if (qstart and not vstart) or (qend and not vend) then
1dc07833355e mod_mam: Use more specific hook Kim Alvefur <zash@zash.se> parents: 1111 diff changeset	143 origin.send(st.error_reply(stanza, "modify", "bad-request", "Invalid timestamp"))
1698 55bc42c1d8c5 mod_mam: Semicolons! Kim Alvefur <zash@zash.se> parents: 1685 diff changeset	144 return true;
1112 1dc07833355e mod_mam: Use more specific hook Kim Alvefur <zash@zash.se> parents: 1111 diff changeset	145 end
1dc07833355e mod_mam: Use more specific hook Kim Alvefur <zash@zash.se> parents: 1111 diff changeset	146 qstart, qend = vstart, vend;
1dc07833355e mod_mam: Use more specific hook Kim Alvefur <zash@zash.se> parents: 1111 diff changeset	147 end
558 66de25ffc8d9 mod_mam: Implement archiving preferences. Kim Alvefur <zash@zash.se> parents: 523 diff changeset	148
1484 53a3a19d6093 mod_mam: Update to version 0.3 of XEP-0313 Kim Alvefur <zash@zash.se> parents: 1403 diff changeset	149 module:log("debug", "Archive query, id %s with %s from %s until %s)",
53a3a19d6093 mod_mam: Update to version 0.3 of XEP-0313 Kim Alvefur <zash@zash.se> parents: 1403 diff changeset	150 tostring(qid), qwith or "anyone", qstart or "the dawn of time", qend or "now");
523 eff140d53b83 mod_mam: Add experimental implementation of the Message Archive Management ProtoXEP Kim Alvefur <zash@zash.se> parents: diff changeset	151
1112 1dc07833355e mod_mam: Use more specific hook Kim Alvefur <zash@zash.se> parents: 1111 diff changeset	152 -- RSM stuff
1116 2345a30dd8b4 mod_mam: Update to use 'archive' storage type. Note: this breaks support with 0.9 and older. Kim Alvefur <zash@zash.se> parents: 1114 diff changeset	153 local qset = rsm.get(query);
1112 1dc07833355e mod_mam: Use more specific hook Kim Alvefur <zash@zash.se> parents: 1111 diff changeset	154 local qmax = m_min(qset and qset.max or default_max_items, max_max_items);
1dc07833355e mod_mam: Use more specific hook Kim Alvefur <zash@zash.se> parents: 1111 diff changeset	155 local reverse = qset and qset.before or false;
1116 2345a30dd8b4 mod_mam: Update to use 'archive' storage type. Note: this breaks support with 0.9 and older. Kim Alvefur <zash@zash.se> parents: 1114 diff changeset	156 local before, after = qset and qset.before, qset and qset.after;
2345a30dd8b4 mod_mam: Update to use 'archive' storage type. Note: this breaks support with 0.9 and older. Kim Alvefur <zash@zash.se> parents: 1114 diff changeset	157 if type(before) ~= "string" then before = nil; end
2345a30dd8b4 mod_mam: Update to use 'archive' storage type. Note: this breaks support with 0.9 and older. Kim Alvefur <zash@zash.se> parents: 1114 diff changeset	158
2345a30dd8b4 mod_mam: Update to use 'archive' storage type. Note: this breaks support with 0.9 and older. Kim Alvefur <zash@zash.se> parents: 1114 diff changeset	159 -- Load all the data!
2345a30dd8b4 mod_mam: Update to use 'archive' storage type. Note: this breaks support with 0.9 and older. Kim Alvefur <zash@zash.se> parents: 1114 diff changeset	160 local data, err = archive:find(origin.username, {
2345a30dd8b4 mod_mam: Update to use 'archive' storage type. Note: this breaks support with 0.9 and older. Kim Alvefur <zash@zash.se> parents: 1114 diff changeset	161 start = qstart; ["end"] = qend; -- Time range
2345a30dd8b4 mod_mam: Update to use 'archive' storage type. Note: this breaks support with 0.9 and older. Kim Alvefur <zash@zash.se> parents: 1114 diff changeset	162 with = qwith;
1682 6b2122630b92 mod_mam: Support the mandatory 'complete' attribute by requesting one extra item from storage Kim Alvefur <zash@zash.se> parents: 1681 diff changeset	163 limit = qmax + 1;
1116 2345a30dd8b4 mod_mam: Update to use 'archive' storage type. Note: this breaks support with 0.9 and older. Kim Alvefur <zash@zash.se> parents: 1114 diff changeset	164 before = before; after = after;
2345a30dd8b4 mod_mam: Update to use 'archive' storage type. Note: this breaks support with 0.9 and older. Kim Alvefur <zash@zash.se> parents: 1114 diff changeset	165 reverse = reverse;
2725 210c3a7644cb mod_mam: Use correct variable Kim Alvefur <zash@zash.se> parents: 2724 diff changeset	166 total = use_total;
1116 2345a30dd8b4 mod_mam: Update to use 'archive' storage type. Note: this breaks support with 0.9 and older. Kim Alvefur <zash@zash.se> parents: 1114 diff changeset	167 });
558 66de25ffc8d9 mod_mam: Implement archiving preferences. Kim Alvefur <zash@zash.se> parents: 523 diff changeset	168
1116 2345a30dd8b4 mod_mam: Update to use 'archive' storage type. Note: this breaks support with 0.9 and older. Kim Alvefur <zash@zash.se> parents: 1114 diff changeset	169 if not data then
1681 d20cfc5ba827 mod_mam: Always return true when a stanza event has been handled Kim Alvefur <zash@zash.se> parents: 1678 diff changeset	170 origin.send(st.error_reply(stanza, "cancel", "internal-server-error", err));
d20cfc5ba827 mod_mam: Always return true when a stanza event has been handled Kim Alvefur <zash@zash.se> parents: 1678 diff changeset	171 return true;
1116 2345a30dd8b4 mod_mam: Update to use 'archive' storage type. Note: this breaks support with 0.9 and older. Kim Alvefur <zash@zash.se> parents: 1114 diff changeset	172 end
2019 7198c129657c mod_mam: Always convert 'total' to a number Kim Alvefur <zash@zash.se> parents: 2018 diff changeset	173 local total = tonumber(err);
1116 2345a30dd8b4 mod_mam: Update to use 'archive' storage type. Note: this breaks support with 0.9 and older. Kim Alvefur <zash@zash.se> parents: 1114 diff changeset	174
2506 5941aac79f06 mod_mam: Add support for XEP-0313 v0.5 Kim Alvefur <zash@zash.se> parents: 2505 diff changeset	175 if xmlns_mam == xmlns_mam0 then
5941aac79f06 mod_mam: Add support for XEP-0313 v0.5 Kim Alvefur <zash@zash.se> parents: 2505 diff changeset	176 origin.send(st.reply(stanza));
5941aac79f06 mod_mam: Add support for XEP-0313 v0.5 Kim Alvefur <zash@zash.se> parents: 2505 diff changeset	177 end
1403 6b3db167374a mod_mam: Mirror to and from attributes from iq on result messages Kim Alvefur <zash@zash.se> parents: 1400 diff changeset	178 local msg_reply_attr = { to = stanza.attr.from, from = stanza.attr.to };
6b3db167374a mod_mam: Mirror to and from attributes from iq on result messages Kim Alvefur <zash@zash.se> parents: 1400 diff changeset	179
1685 cd87a2eba8f2 mod_mam: Reverse the order of reversed queries back to chronological order Kim Alvefur <zash@zash.se> parents: 1684 diff changeset	180 local results = {};
cd87a2eba8f2 mod_mam: Reverse the order of reversed queries back to chronological order Kim Alvefur <zash@zash.se> parents: 1684 diff changeset	181
1116 2345a30dd8b4 mod_mam: Update to use 'archive' storage type. Note: this breaks support with 0.9 and older. Kim Alvefur <zash@zash.se> parents: 1114 diff changeset	182 -- Wrap it in stuff and deliver
1684 838150167871 mod_mam: Move variable into loop scope Kim Alvefur <zash@zash.se> parents: 1683 diff changeset	183 local first, last;
1682 6b2122630b92 mod_mam: Support the mandatory 'complete' attribute by requesting one extra item from storage Kim Alvefur <zash@zash.se> parents: 1681 diff changeset	184 local count = 0;
6b2122630b92 mod_mam: Support the mandatory 'complete' attribute by requesting one extra item from storage Kim Alvefur <zash@zash.se> parents: 1681 diff changeset	185 local complete = "true";
1116 2345a30dd8b4 mod_mam: Update to use 'archive' storage type. Note: this breaks support with 0.9 and older. Kim Alvefur <zash@zash.se> parents: 1114 diff changeset	186 for id, item, when in data do
1682 6b2122630b92 mod_mam: Support the mandatory 'complete' attribute by requesting one extra item from storage Kim Alvefur <zash@zash.se> parents: 1681 diff changeset	187 count = count + 1;
6b2122630b92 mod_mam: Support the mandatory 'complete' attribute by requesting one extra item from storage Kim Alvefur <zash@zash.se> parents: 1681 diff changeset	188 if count > qmax then
6b2122630b92 mod_mam: Support the mandatory 'complete' attribute by requesting one extra item from storage Kim Alvefur <zash@zash.se> parents: 1681 diff changeset	189 complete = nil;
6b2122630b92 mod_mam: Support the mandatory 'complete' attribute by requesting one extra item from storage Kim Alvefur <zash@zash.se> parents: 1681 diff changeset	190 break;
6b2122630b92 mod_mam: Support the mandatory 'complete' attribute by requesting one extra item from storage Kim Alvefur <zash@zash.se> parents: 1681 diff changeset	191 end
1684 838150167871 mod_mam: Move variable into loop scope Kim Alvefur <zash@zash.se> parents: 1683 diff changeset	192 local fwd_st = st.message(msg_reply_attr)
1116 2345a30dd8b4 mod_mam: Update to use 'archive' storage type. Note: this breaks support with 0.9 and older. Kim Alvefur <zash@zash.se> parents: 1114 diff changeset	193 :tag("result", { xmlns = xmlns_mam, queryid = qid, id = id })
2345a30dd8b4 mod_mam: Update to use 'archive' storage type. Note: this breaks support with 0.9 and older. Kim Alvefur <zash@zash.se> parents: 1114 diff changeset	194 :tag("forwarded", { xmlns = xmlns_forward })
2345a30dd8b4 mod_mam: Update to use 'archive' storage type. Note: this breaks support with 0.9 and older. Kim Alvefur <zash@zash.se> parents: 1114 diff changeset	195 :tag("delay", { xmlns = xmlns_delay, stamp = timestamp(when) }):up();
2345a30dd8b4 mod_mam: Update to use 'archive' storage type. Note: this breaks support with 0.9 and older. Kim Alvefur <zash@zash.se> parents: 1114 diff changeset	196
2345a30dd8b4 mod_mam: Update to use 'archive' storage type. Note: this breaks support with 0.9 and older. Kim Alvefur <zash@zash.se> parents: 1114 diff changeset	197 if not is_stanza(item) then
2345a30dd8b4 mod_mam: Update to use 'archive' storage type. Note: this breaks support with 0.9 and older. Kim Alvefur <zash@zash.se> parents: 1114 diff changeset	198 item = st.deserialize(item);
1112 1dc07833355e mod_mam: Use more specific hook Kim Alvefur <zash@zash.se> parents: 1111 diff changeset	199 end
1116 2345a30dd8b4 mod_mam: Update to use 'archive' storage type. Note: this breaks support with 0.9 and older. Kim Alvefur <zash@zash.se> parents: 1114 diff changeset	200 item.attr.xmlns = "jabber:client";
2345a30dd8b4 mod_mam: Update to use 'archive' storage type. Note: this breaks support with 0.9 and older. Kim Alvefur <zash@zash.se> parents: 1114 diff changeset	201 fwd_st:add_child(item);
701 cc5805f83583 mod_mam: Implement support for Result Set Management in queries. Kim Alvefur <zash@zash.se> parents: 675 diff changeset	202
1325 b21236b6b8d8 Backed out changeset 853a382c9bd6 Kim Alvefur <zash@zash.se> parents: 1324 diff changeset	203 if not first then first = id; end
b21236b6b8d8 Backed out changeset 853a382c9bd6 Kim Alvefur <zash@zash.se> parents: 1324 diff changeset	204 last = id;
1116 2345a30dd8b4 mod_mam: Update to use 'archive' storage type. Note: this breaks support with 0.9 and older. Kim Alvefur <zash@zash.se> parents: 1114 diff changeset	205
1685 cd87a2eba8f2 mod_mam: Reverse the order of reversed queries back to chronological order Kim Alvefur <zash@zash.se> parents: 1684 diff changeset	206 if reverse then
cd87a2eba8f2 mod_mam: Reverse the order of reversed queries back to chronological order Kim Alvefur <zash@zash.se> parents: 1684 diff changeset	207 results[count] = fwd_st;
cd87a2eba8f2 mod_mam: Reverse the order of reversed queries back to chronological order Kim Alvefur <zash@zash.se> parents: 1684 diff changeset	208 else
cd87a2eba8f2 mod_mam: Reverse the order of reversed queries back to chronological order Kim Alvefur <zash@zash.se> parents: 1684 diff changeset	209 origin.send(fwd_st);
cd87a2eba8f2 mod_mam: Reverse the order of reversed queries back to chronological order Kim Alvefur <zash@zash.se> parents: 1684 diff changeset	210 end
1112 1dc07833355e mod_mam: Use more specific hook Kim Alvefur <zash@zash.se> parents: 1111 diff changeset	211 end
2020 37b30f10fbba mod_mam: Bit of spacing between blocks of code Kim Alvefur <zash@zash.se> parents: 2019 diff changeset	212
1685 cd87a2eba8f2 mod_mam: Reverse the order of reversed queries back to chronological order Kim Alvefur <zash@zash.se> parents: 1684 diff changeset	213 if reverse then
cd87a2eba8f2 mod_mam: Reverse the order of reversed queries back to chronological order Kim Alvefur <zash@zash.se> parents: 1684 diff changeset	214 for i = #results, 1, -1 do
cd87a2eba8f2 mod_mam: Reverse the order of reversed queries back to chronological order Kim Alvefur <zash@zash.se> parents: 1684 diff changeset	215 origin.send(results[i]);
cd87a2eba8f2 mod_mam: Reverse the order of reversed queries back to chronological order Kim Alvefur <zash@zash.se> parents: 1684 diff changeset	216 end
2021 5fb917b86838 mod_mam: Merge identical conditional blocks Kim Alvefur <zash@zash.se> parents: 2020 diff changeset	217 first, last = last, first;
1685 cd87a2eba8f2 mod_mam: Reverse the order of reversed queries back to chronological order Kim Alvefur <zash@zash.se> parents: 1684 diff changeset	218 end
cd87a2eba8f2 mod_mam: Reverse the order of reversed queries back to chronological order Kim Alvefur <zash@zash.se> parents: 1684 diff changeset	219
1112 1dc07833355e mod_mam: Use more specific hook Kim Alvefur <zash@zash.se> parents: 1111 diff changeset	220 -- That's all folks!
1dc07833355e mod_mam: Use more specific hook Kim Alvefur <zash@zash.se> parents: 1111 diff changeset	221 module:log("debug", "Archive query %s completed", tostring(qid));
705 c9d0ba39a33b mod_mam: Move RSM pointer to last message into a MAM-namespaced child Kim Alvefur <zash@zash.se> parents: 702 diff changeset	222
2506 5941aac79f06 mod_mam: Add support for XEP-0313 v0.5 Kim Alvefur <zash@zash.se> parents: 2505 diff changeset	223 local fin;
5941aac79f06 mod_mam: Add support for XEP-0313 v0.5 Kim Alvefur <zash@zash.se> parents: 2505 diff changeset	224 if xmlns_mam == xmlns_mam0 then
5941aac79f06 mod_mam: Add support for XEP-0313 v0.5 Kim Alvefur <zash@zash.se> parents: 2505 diff changeset	225 fin = st.message(msg_reply_attr);
5941aac79f06 mod_mam: Add support for XEP-0313 v0.5 Kim Alvefur <zash@zash.se> parents: 2505 diff changeset	226 else
5941aac79f06 mod_mam: Add support for XEP-0313 v0.5 Kim Alvefur <zash@zash.se> parents: 2505 diff changeset	227 fin = st.reply(stanza);
5941aac79f06 mod_mam: Add support for XEP-0313 v0.5 Kim Alvefur <zash@zash.se> parents: 2505 diff changeset	228 end
5941aac79f06 mod_mam: Add support for XEP-0313 v0.5 Kim Alvefur <zash@zash.se> parents: 2505 diff changeset	229 do
5941aac79f06 mod_mam: Add support for XEP-0313 v0.5 Kim Alvefur <zash@zash.se> parents: 2505 diff changeset	230 fin:tag("fin", { xmlns = xmlns_mam, queryid = qid, complete = complete })
1484 53a3a19d6093 mod_mam: Update to version 0.3 of XEP-0313 Kim Alvefur <zash@zash.se> parents: 1403 diff changeset	231 :add_child(rsm.generate {
2506 5941aac79f06 mod_mam: Add support for XEP-0313 v0.5 Kim Alvefur <zash@zash.se> parents: 2505 diff changeset	232 first = first, last = last, count = total })
5941aac79f06 mod_mam: Add support for XEP-0313 v0.5 Kim Alvefur <zash@zash.se> parents: 2505 diff changeset	233 end
5941aac79f06 mod_mam: Add support for XEP-0313 v0.5 Kim Alvefur <zash@zash.se> parents: 2505 diff changeset	234 origin.send(fin);
1681 d20cfc5ba827 mod_mam: Always return true when a stanza event has been handled Kim Alvefur <zash@zash.se> parents: 1678 diff changeset	235 return true;
2506 5941aac79f06 mod_mam: Add support for XEP-0313 v0.5 Kim Alvefur <zash@zash.se> parents: 2505 diff changeset	236 end
5941aac79f06 mod_mam: Add support for XEP-0313 v0.5 Kim Alvefur <zash@zash.se> parents: 2505 diff changeset	237 module:hook("iq-set/self/"..xmlns_mam0..":query", handle_mam_query);
5941aac79f06 mod_mam: Add support for XEP-0313 v0.5 Kim Alvefur <zash@zash.se> parents: 2505 diff changeset	238 module:hook("iq-set/self/"..xmlns_mam1..":query", handle_mam_query);
2516 248054199d0f mod_mam: Add support for XEP-0313 v0.6 Kim Alvefur <zash@zash.se> parents: 2515 diff changeset	239 module:hook("iq-set/self/"..xmlns_mam2..":query", handle_mam_query);
523 eff140d53b83 mod_mam: Add experimental implementation of the Message Archive Management ProtoXEP Kim Alvefur <zash@zash.se> parents: diff changeset	240
558 66de25ffc8d9 mod_mam: Implement archiving preferences. Kim Alvefur <zash@zash.se> parents: 523 diff changeset	241 local function has_in_roster(user, who)
66de25ffc8d9 mod_mam: Implement archiving preferences. Kim Alvefur <zash@zash.se> parents: 523 diff changeset	242 local roster = rm_load_roster(user, host);
66de25ffc8d9 mod_mam: Implement archiving preferences. Kim Alvefur <zash@zash.se> parents: 523 diff changeset	243 module:log("debug", "%s has %s in roster? %s", user, who, roster[who] and "yes" or "no");
798 2b8ceb4d1a73 mod_mam: remove useless check Kim Alvefur <zash@zash.se> parents: 751 diff changeset	244 return roster[who];
558 66de25ffc8d9 mod_mam: Implement archiving preferences. Kim Alvefur <zash@zash.se> parents: 523 diff changeset	245 end
66de25ffc8d9 mod_mam: Implement archiving preferences. Kim Alvefur <zash@zash.se> parents: 523 diff changeset	246
66de25ffc8d9 mod_mam: Implement archiving preferences. Kim Alvefur <zash@zash.se> parents: 523 diff changeset	247 local function shall_store(user, who)
66de25ffc8d9 mod_mam: Implement archiving preferences. Kim Alvefur <zash@zash.se> parents: 523 diff changeset	248 -- TODO Cache this?
2293 144b74caa5ef mod_mam: Check that user exists (#731) Kim Alvefur <zash@zash.se> parents: 2270 diff changeset	249 if not um.user_exists(user, host) then
144b74caa5ef mod_mam: Check that user exists (#731) Kim Alvefur <zash@zash.se> parents: 2270 diff changeset	250 return false;
144b74caa5ef mod_mam: Check that user exists (#731) Kim Alvefur <zash@zash.se> parents: 2270 diff changeset	251 end
558 66de25ffc8d9 mod_mam: Implement archiving preferences. Kim Alvefur <zash@zash.se> parents: 523 diff changeset	252 local prefs = get_prefs(user);
66de25ffc8d9 mod_mam: Implement archiving preferences. Kim Alvefur <zash@zash.se> parents: 523 diff changeset	253 local rule = prefs[who];
1698 55bc42c1d8c5 mod_mam: Semicolons! Kim Alvefur <zash@zash.se> parents: 1685 diff changeset	254 module:log("debug", "%s's rule for %s is %s", user, who, tostring(rule));
558 66de25ffc8d9 mod_mam: Implement archiving preferences. Kim Alvefur <zash@zash.se> parents: 523 diff changeset	255 if rule ~= nil then
66de25ffc8d9 mod_mam: Implement archiving preferences. Kim Alvefur <zash@zash.se> parents: 523 diff changeset	256 return rule;
66de25ffc8d9 mod_mam: Implement archiving preferences. Kim Alvefur <zash@zash.se> parents: 523 diff changeset	257 end
1788 d2b82b90c413 mod_mam: Unindent 'else' block Kim Alvefur <zash@zash.se> parents: 1773 diff changeset	258 -- Below could be done by a metatable
d2b82b90c413 mod_mam: Unindent 'else' block Kim Alvefur <zash@zash.se> parents: 1773 diff changeset	259 local default = prefs[false];
d2b82b90c413 mod_mam: Unindent 'else' block Kim Alvefur <zash@zash.se> parents: 1773 diff changeset	260 module:log("debug", "%s's default rule is %s", user, tostring(default));
d2b82b90c413 mod_mam: Unindent 'else' block Kim Alvefur <zash@zash.se> parents: 1773 diff changeset	261 if default == nil then
d2b82b90c413 mod_mam: Unindent 'else' block Kim Alvefur <zash@zash.se> parents: 1773 diff changeset	262 default = global_default_policy;
d2b82b90c413 mod_mam: Unindent 'else' block Kim Alvefur <zash@zash.se> parents: 1773 diff changeset	263 module:log("debug", "Using global default rule, %s", tostring(default));
d2b82b90c413 mod_mam: Unindent 'else' block Kim Alvefur <zash@zash.se> parents: 1773 diff changeset	264 end
d2b82b90c413 mod_mam: Unindent 'else' block Kim Alvefur <zash@zash.se> parents: 1773 diff changeset	265 if default == "roster" then
d2b82b90c413 mod_mam: Unindent 'else' block Kim Alvefur <zash@zash.se> parents: 1773 diff changeset	266 return has_in_roster(user, who);
d2b82b90c413 mod_mam: Unindent 'else' block Kim Alvefur <zash@zash.se> parents: 1773 diff changeset	267 end
d2b82b90c413 mod_mam: Unindent 'else' block Kim Alvefur <zash@zash.se> parents: 1773 diff changeset	268 return default;
558 66de25ffc8d9 mod_mam: Implement archiving preferences. Kim Alvefur <zash@zash.se> parents: 523 diff changeset	269 end
66de25ffc8d9 mod_mam: Implement archiving preferences. Kim Alvefur <zash@zash.se> parents: 523 diff changeset	270
66de25ffc8d9 mod_mam: Implement archiving preferences. Kim Alvefur <zash@zash.se> parents: 523 diff changeset	271 -- Handle messages
523 eff140d53b83 mod_mam: Add experimental implementation of the Message Archive Management ProtoXEP Kim Alvefur <zash@zash.se> parents: diff changeset	272 local function message_handler(event, c2s)
eff140d53b83 mod_mam: Add experimental implementation of the Message Archive Management ProtoXEP Kim Alvefur <zash@zash.se> parents: diff changeset	273 local origin, stanza = event.origin, event.stanza;
2270 9c99200afd17 mod_mam: Log messages from local clients on the clients session Kim Alvefur <zash@zash.se> parents: 2049 diff changeset	274 local log = c2s and origin.log or module._log;
523 eff140d53b83 mod_mam: Add experimental implementation of the Message Archive Management ProtoXEP Kim Alvefur <zash@zash.se> parents: diff changeset	275 local orig_type = stanza.attr.type or "normal";
eff140d53b83 mod_mam: Add experimental implementation of the Message Archive Management ProtoXEP Kim Alvefur <zash@zash.se> parents: diff changeset	276 local orig_from = stanza.attr.from;
1149 d055c44a7f61 mod_mam: Clean up and explain the code that determines who's archive we put the message in Kim Alvefur <zash@zash.se> parents: 1135 diff changeset	277 local orig_to = stanza.attr.to or orig_from;
d055c44a7f61 mod_mam: Clean up and explain the code that determines who's archive we put the message in Kim Alvefur <zash@zash.se> parents: 1135 diff changeset	278 -- Stanza without 'to' are treated as if it was to their own bare jid
523 eff140d53b83 mod_mam: Add experimental implementation of the Message Archive Management ProtoXEP Kim Alvefur <zash@zash.se> parents: diff changeset	279
2516 248054199d0f mod_mam: Add support for XEP-0313 v0.6 Kim Alvefur <zash@zash.se> parents: 2515 diff changeset	280 -- Whos storage do we put it in?
248054199d0f mod_mam: Add support for XEP-0313 v0.6 Kim Alvefur <zash@zash.se> parents: 2515 diff changeset	281 local store_user = c2s and origin.username or jid_split(orig_to);
248054199d0f mod_mam: Add support for XEP-0313 v0.6 Kim Alvefur <zash@zash.se> parents: 2515 diff changeset	282 -- And who are they chatting with?
248054199d0f mod_mam: Add support for XEP-0313 v0.6 Kim Alvefur <zash@zash.se> parents: 2515 diff changeset	283 local with = jid_bare(c2s and orig_to or orig_from);
248054199d0f mod_mam: Add support for XEP-0313 v0.6 Kim Alvefur <zash@zash.se> parents: 2515 diff changeset	284
248054199d0f mod_mam: Add support for XEP-0313 v0.6 Kim Alvefur <zash@zash.se> parents: 2515 diff changeset	285 -- Filter out <stanza-id> that claim to be from us
2760 7c2416a1eb94 mod_mam: Clone stanzas before mutating (thanks waqas) (fixes #961) Kim Alvefur <zash@zash.se> parents: 2725 diff changeset	286 if stanza:get_child("stanza-id", xmlns_st_id) then
7c2416a1eb94 mod_mam: Clone stanzas before mutating (thanks waqas) (fixes #961) Kim Alvefur <zash@zash.se> parents: 2725 diff changeset	287 stanza = st.clone(stanza);
7c2416a1eb94 mod_mam: Clone stanzas before mutating (thanks waqas) (fixes #961) Kim Alvefur <zash@zash.se> parents: 2725 diff changeset	288 stanza:maptags(function (tag)
7c2416a1eb94 mod_mam: Clone stanzas before mutating (thanks waqas) (fixes #961) Kim Alvefur <zash@zash.se> parents: 2725 diff changeset	289 if tag.name == "stanza-id" and tag.attr.xmlns == xmlns_st_id then
7c2416a1eb94 mod_mam: Clone stanzas before mutating (thanks waqas) (fixes #961) Kim Alvefur <zash@zash.se> parents: 2725 diff changeset	290 local by_user, by_host, res = jid_prepped_split(tag.attr.by);
7c2416a1eb94 mod_mam: Clone stanzas before mutating (thanks waqas) (fixes #961) Kim Alvefur <zash@zash.se> parents: 2725 diff changeset	291 if not res and by_host == module.host and by_user == store_user then
7c2416a1eb94 mod_mam: Clone stanzas before mutating (thanks waqas) (fixes #961) Kim Alvefur <zash@zash.se> parents: 2725 diff changeset	292 return nil;
7c2416a1eb94 mod_mam: Clone stanzas before mutating (thanks waqas) (fixes #961) Kim Alvefur <zash@zash.se> parents: 2725 diff changeset	293 end
2516 248054199d0f mod_mam: Add support for XEP-0313 v0.6 Kim Alvefur <zash@zash.se> parents: 2515 diff changeset	294 end
2760 7c2416a1eb94 mod_mam: Clone stanzas before mutating (thanks waqas) (fixes #961) Kim Alvefur <zash@zash.se> parents: 2725 diff changeset	295 return tag;
7c2416a1eb94 mod_mam: Clone stanzas before mutating (thanks waqas) (fixes #961) Kim Alvefur <zash@zash.se> parents: 2725 diff changeset	296 end);
7c2416a1eb94 mod_mam: Clone stanzas before mutating (thanks waqas) (fixes #961) Kim Alvefur <zash@zash.se> parents: 2725 diff changeset	297 event.stanza = stanza;
7c2416a1eb94 mod_mam: Clone stanzas before mutating (thanks waqas) (fixes #961) Kim Alvefur <zash@zash.se> parents: 2725 diff changeset	298 end
2516 248054199d0f mod_mam: Add support for XEP-0313 v0.6 Kim Alvefur <zash@zash.se> parents: 2515 diff changeset	299
1790 4c2146f5bf39 mod_mam: Store chat messages and normal messages with a body Kim Alvefur <zash@zash.se> parents: 1788 diff changeset	300 -- We store chat messages or normal messages that have a body
1957 ca33cca2e028 mod_mam: Clarify condition presendence Kim Alvefur <zash@zash.se> parents: 1790 diff changeset	301 if not(orig_type == "chat" or (orig_type == "normal" and stanza:get_child("body")) ) then
2270 9c99200afd17 mod_mam: Log messages from local clients on the clients session Kim Alvefur <zash@zash.se> parents: 2049 diff changeset	302 log("debug", "Not archiving stanza: %s (type)", stanza:top_tag());
1790 4c2146f5bf39 mod_mam: Store chat messages and normal messages with a body Kim Alvefur <zash@zash.se> parents: 1788 diff changeset	303 return;
4c2146f5bf39 mod_mam: Store chat messages and normal messages with a body Kim Alvefur <zash@zash.se> parents: 1788 diff changeset	304 end
2512 bc7539ccbde1 mod_mam: Normalize whitespace Kim Alvefur <zash@zash.se> parents: 2511 diff changeset	305
1150 296820f18ba6 mod_mam: Add support for XEP-0334: Message Processing Hints Kim Alvefur <zash@zash.se> parents: 1149 diff changeset	306 -- or if hints suggest we shouldn't
2511 34a6ba5dfe4f mod_mam: Bring hints processing in line with 0.10 version Kim Alvefur <zash@zash.se> parents: 2510 diff changeset	307 if not stanza:get_child("store", "urn:xmpp:hints") then -- No hint telling us we should store
34a6ba5dfe4f mod_mam: Bring hints processing in line with 0.10 version Kim Alvefur <zash@zash.se> parents: 2510 diff changeset	308 if stanza:get_child("no-permanent-store", "urn:xmpp:hints")
34a6ba5dfe4f mod_mam: Bring hints processing in line with 0.10 version Kim Alvefur <zash@zash.se> parents: 2510 diff changeset	309 or stanza:get_child("no-store", "urn:xmpp:hints") then -- Hint telling us we should NOT store
34a6ba5dfe4f mod_mam: Bring hints processing in line with 0.10 version Kim Alvefur <zash@zash.se> parents: 2510 diff changeset	310 log("debug", "Not archiving stanza: %s (hint)", stanza:top_tag());
34a6ba5dfe4f mod_mam: Bring hints processing in line with 0.10 version Kim Alvefur <zash@zash.se> parents: 2510 diff changeset	311 return;
34a6ba5dfe4f mod_mam: Bring hints processing in line with 0.10 version Kim Alvefur <zash@zash.se> parents: 2510 diff changeset	312 end
523 eff140d53b83 mod_mam: Add experimental implementation of the Message Archive Management ProtoXEP Kim Alvefur <zash@zash.se> parents: diff changeset	313 end
eff140d53b83 mod_mam: Add experimental implementation of the Message Archive Management ProtoXEP Kim Alvefur <zash@zash.se> parents: diff changeset	314
1149 d055c44a7f61 mod_mam: Clean up and explain the code that determines who's archive we put the message in Kim Alvefur <zash@zash.se> parents: 1135 diff changeset	315 -- Check with the users preferences
d055c44a7f61 mod_mam: Clean up and explain the code that determines who's archive we put the message in Kim Alvefur <zash@zash.se> parents: 1135 diff changeset	316 if shall_store(store_user, with) then
2270 9c99200afd17 mod_mam: Log messages from local clients on the clients session Kim Alvefur <zash@zash.se> parents: 2049 diff changeset	317 log("debug", "Archiving stanza: %s", stanza:top_tag());
558 66de25ffc8d9 mod_mam: Implement archiving preferences. Kim Alvefur <zash@zash.se> parents: 523 diff changeset	318
66de25ffc8d9 mod_mam: Implement archiving preferences. Kim Alvefur <zash@zash.se> parents: 523 diff changeset	319 -- And stash it
2515 4cb549622862 mod_mam: Fix to expect archive id as first return value from archive:append Kim Alvefur <zash@zash.se> parents: 2513 diff changeset	320 local ok = archive:append(store_user, nil, stanza, time_now(), with);
1746 5734a6199938 mod_mam: Fire event on successful storage of message Kim Alvefur <zash@zash.se> parents: 1737 diff changeset	321 if ok then
2760 7c2416a1eb94 mod_mam: Clone stanzas before mutating (thanks waqas) (fixes #961) Kim Alvefur <zash@zash.se> parents: 2725 diff changeset	322 local clone_for_other_handlers = st.clone(stanza);
2515 4cb549622862 mod_mam: Fix to expect archive id as first return value from archive:append Kim Alvefur <zash@zash.se> parents: 2513 diff changeset	323 local id = ok;
2760 7c2416a1eb94 mod_mam: Clone stanzas before mutating (thanks waqas) (fixes #961) Kim Alvefur <zash@zash.se> parents: 2725 diff changeset	324 clone_for_other_handlers:tag("stanza-id", { xmlns = xmlns_st_id, by = store_user.."@"..host, id = id }):up();
7c2416a1eb94 mod_mam: Clone stanzas before mutating (thanks waqas) (fixes #961) Kim Alvefur <zash@zash.se> parents: 2725 diff changeset	325 event.stanza = clone_for_other_handlers;
2764 1872a9129c2f mod_mam: Use a FIFO queue for scheduling archive expiry Kim Alvefur <zash@zash.se> parents: 2762 diff changeset	326 schedule_cleanup(store_user);
2762 f2096c03428b mod_mam: Fix traceback due to borked transplant from 0.10 Kim Alvefur <zash@zash.se> parents: 2761 diff changeset	327 module:fire_event("archive-message-added", { origin = origin, stanza = stanza, for_user = store_user, id = id });
1746 5734a6199938 mod_mam: Fire event on successful storage of message Kim Alvefur <zash@zash.se> parents: 1737 diff changeset	328 end
558 66de25ffc8d9 mod_mam: Implement archiving preferences. Kim Alvefur <zash@zash.se> parents: 523 diff changeset	329 else
2270 9c99200afd17 mod_mam: Log messages from local clients on the clients session Kim Alvefur <zash@zash.se> parents: 2049 diff changeset	330 log("debug", "Not archiving stanza: %s (prefs)", stanza:top_tag());
558 66de25ffc8d9 mod_mam: Implement archiving preferences. Kim Alvefur <zash@zash.se> parents: 523 diff changeset	331 end
523 eff140d53b83 mod_mam: Add experimental implementation of the Message Archive Management ProtoXEP Kim Alvefur <zash@zash.se> parents: diff changeset	332 end
eff140d53b83 mod_mam: Add experimental implementation of the Message Archive Management ProtoXEP Kim Alvefur <zash@zash.se> parents: diff changeset	333
eff140d53b83 mod_mam: Add experimental implementation of the Message Archive Management ProtoXEP Kim Alvefur <zash@zash.se> parents: diff changeset	334 local function c2s_message_handler(event)
eff140d53b83 mod_mam: Add experimental implementation of the Message Archive Management ProtoXEP Kim Alvefur <zash@zash.se> parents: diff changeset	335 return message_handler(event, true);
eff140d53b83 mod_mam: Add experimental implementation of the Message Archive Management ProtoXEP Kim Alvefur <zash@zash.se> parents: diff changeset	336 end
eff140d53b83 mod_mam: Add experimental implementation of the Message Archive Management ProtoXEP Kim Alvefur <zash@zash.se> parents: diff changeset	337
2516 248054199d0f mod_mam: Add support for XEP-0313 v0.6 Kim Alvefur <zash@zash.se> parents: 2515 diff changeset	338 local function strip_stanza_id(event)
248054199d0f mod_mam: Add support for XEP-0313 v0.6 Kim Alvefur <zash@zash.se> parents: 2515 diff changeset	339 local strip_by = jid_bare(event.origin.full_jid);
2761 b13d07613b80 mod_mam: Clone stanza before stripping stanza id after carbons Kim Alvefur <zash@zash.se> parents: 2760 diff changeset	340 event.stanza = st.clone(event.stanza);
2516 248054199d0f mod_mam: Add support for XEP-0313 v0.6 Kim Alvefur <zash@zash.se> parents: 2515 diff changeset	341 event.stanza:maptags(function(tag)
248054199d0f mod_mam: Add support for XEP-0313 v0.6 Kim Alvefur <zash@zash.se> parents: 2515 diff changeset	342 if not ( tag.attr.xmlns == xmlns_st_id and tag.attr.by == strip_by ) then
248054199d0f mod_mam: Add support for XEP-0313 v0.6 Kim Alvefur <zash@zash.se> parents: 2515 diff changeset	343 return tag;
248054199d0f mod_mam: Add support for XEP-0313 v0.6 Kim Alvefur <zash@zash.se> parents: 2515 diff changeset	344 end
248054199d0f mod_mam: Add support for XEP-0313 v0.6 Kim Alvefur <zash@zash.se> parents: 2515 diff changeset	345 end);
248054199d0f mod_mam: Add support for XEP-0313 v0.6 Kim Alvefur <zash@zash.se> parents: 2515 diff changeset	346 end
248054199d0f mod_mam: Add support for XEP-0313 v0.6 Kim Alvefur <zash@zash.se> parents: 2515 diff changeset	347
248054199d0f mod_mam: Add support for XEP-0313 v0.6 Kim Alvefur <zash@zash.se> parents: 2515 diff changeset	348 module:hook("pre-message/bare", strip_stanza_id, 0.01);
248054199d0f mod_mam: Add support for XEP-0313 v0.6 Kim Alvefur <zash@zash.se> parents: 2515 diff changeset	349 module:hook("pre-message/full", strip_stanza_id, 0.01);
248054199d0f mod_mam: Add support for XEP-0313 v0.6 Kim Alvefur <zash@zash.se> parents: 2515 diff changeset	350
2016 279885fd9728 mod_mam: Add support for trimming old archived messages Kim Alvefur <zash@zash.se> parents: 2001 diff changeset	351 local cleanup_after = module:get_option_string("archive_expires_after", "1w");
2022 77b9c7e5fd63 mod_mam: Allow interval between archive cleanup to be changed Kim Alvefur <zash@zash.se> parents: 2021 diff changeset	352 local cleanup_interval = module:get_option_number("archive_cleanup_interval", 4 * 60 * 60);
2016 279885fd9728 mod_mam: Add support for trimming old archived messages Kim Alvefur <zash@zash.se> parents: 2001 diff changeset	353 if cleanup_after ~= "never" then
279885fd9728 mod_mam: Add support for trimming old archived messages Kim Alvefur <zash@zash.se> parents: 2001 diff changeset	354 local day = 86400;
279885fd9728 mod_mam: Add support for trimming old archived messages Kim Alvefur <zash@zash.se> parents: 2001 diff changeset	355 local multipliers = { d = day, w = day * 7, m = 31 * day, y = 365.2425 * day };
2049 50c188cf0ae3 mod_mam: Include 'w' (week) in pattern for "archive_expires_after" option (fixes #627) Kim Alvefur <zash@zash.se> parents: 2042 diff changeset	356 local n, m = cleanup_after:lower():match("(%d+)%s*([dwmy]?)");
2016 279885fd9728 mod_mam: Add support for trimming old archived messages Kim Alvefur <zash@zash.se> parents: 2001 diff changeset	357 if not n then
279885fd9728 mod_mam: Add support for trimming old archived messages Kim Alvefur <zash@zash.se> parents: 2001 diff changeset	358 module:log("error", "Could not parse archive_expires_after string %q", cleanup_after);
279885fd9728 mod_mam: Add support for trimming old archived messages Kim Alvefur <zash@zash.se> parents: 2001 diff changeset	359 return false;
279885fd9728 mod_mam: Add support for trimming old archived messages Kim Alvefur <zash@zash.se> parents: 2001 diff changeset	360 end
279885fd9728 mod_mam: Add support for trimming old archived messages Kim Alvefur <zash@zash.se> parents: 2001 diff changeset	361
279885fd9728 mod_mam: Add support for trimming old archived messages Kim Alvefur <zash@zash.se> parents: 2001 diff changeset	362 cleanup_after = tonumber(n) * ( multipliers[m] or 1 );
279885fd9728 mod_mam: Add support for trimming old archived messages Kim Alvefur <zash@zash.se> parents: 2001 diff changeset	363
2042 752eee154891 mod_mam: Add some debug logging Kim Alvefur <zash@zash.se> parents: 2039 diff changeset	364 module:log("debug", "archive_expires_after = %d -- in seconds", cleanup_after);
752eee154891 mod_mam: Add some debug logging Kim Alvefur <zash@zash.se> parents: 2039 diff changeset	365
2016 279885fd9728 mod_mam: Add support for trimming old archived messages Kim Alvefur <zash@zash.se> parents: 2001 diff changeset	366 if not archive.delete then
279885fd9728 mod_mam: Add support for trimming old archived messages Kim Alvefur <zash@zash.se> parents: 2001 diff changeset	367 module:log("error", "archive_expires_after set but mod_%s does not support deleting", archive._provided_by);
279885fd9728 mod_mam: Add support for trimming old archived messages Kim Alvefur <zash@zash.se> parents: 2001 diff changeset	368 return false;
279885fd9728 mod_mam: Add support for trimming old archived messages Kim Alvefur <zash@zash.se> parents: 2001 diff changeset	369 end
279885fd9728 mod_mam: Add support for trimming old archived messages Kim Alvefur <zash@zash.se> parents: 2001 diff changeset	370
2513 66a61f48e376 mod_mam: Add comments Kim Alvefur <zash@zash.se> parents: 2512 diff changeset	371 -- Set of known users to do message expiry for
66a61f48e376 mod_mam: Add comments Kim Alvefur <zash@zash.se> parents: 2512 diff changeset	372 -- Populated either below or when new messages are added
2016 279885fd9728 mod_mam: Add support for trimming old archived messages Kim Alvefur <zash@zash.se> parents: 2001 diff changeset	373 cleanup = {};
279885fd9728 mod_mam: Add support for trimming old archived messages Kim Alvefur <zash@zash.se> parents: 2001 diff changeset	374
2513 66a61f48e376 mod_mam: Add comments Kim Alvefur <zash@zash.se> parents: 2512 diff changeset	375 -- Iterating over users is not supported by all authentication modules
66a61f48e376 mod_mam: Add comments Kim Alvefur <zash@zash.se> parents: 2512 diff changeset	376 -- Catch and ignore error if not supported
2016 279885fd9728 mod_mam: Add support for trimming old archived messages Kim Alvefur <zash@zash.se> parents: 2001 diff changeset	377 pcall(function ()
2513 66a61f48e376 mod_mam: Add comments Kim Alvefur <zash@zash.se> parents: 2512 diff changeset	378 -- If this works, then we schedule cleanup for all known users on startup
2016 279885fd9728 mod_mam: Add support for trimming old archived messages Kim Alvefur <zash@zash.se> parents: 2001 diff changeset	379 for user in um.users(module.host) do
2764 1872a9129c2f mod_mam: Use a FIFO queue for scheduling archive expiry Kim Alvefur <zash@zash.se> parents: 2762 diff changeset	380 schedule_cleanup(user);
2016 279885fd9728 mod_mam: Add support for trimming old archived messages Kim Alvefur <zash@zash.se> parents: 2001 diff changeset	381 end
279885fd9728 mod_mam: Add support for trimming old archived messages Kim Alvefur <zash@zash.se> parents: 2001 diff changeset	382 end);
279885fd9728 mod_mam: Add support for trimming old archived messages Kim Alvefur <zash@zash.se> parents: 2001 diff changeset	383
2513 66a61f48e376 mod_mam: Add comments Kim Alvefur <zash@zash.se> parents: 2512 diff changeset	384 -- At odd intervals, delete old messages for one user
2039 464edd03099a mod_mam: Add a bit of randomness to cleanup intervals to keep some time between multiple instances Kim Alvefur <zash@zash.se> parents: 2033 diff changeset	385 module:add_timer(math.random(10, 60), function()
2764 1872a9129c2f mod_mam: Use a FIFO queue for scheduling archive expiry Kim Alvefur <zash@zash.se> parents: 2762 diff changeset	386 local user = table.remove(cleanup, 1);
2016 279885fd9728 mod_mam: Add support for trimming old archived messages Kim Alvefur <zash@zash.se> parents: 2001 diff changeset	387 if user then
279885fd9728 mod_mam: Add support for trimming old archived messages Kim Alvefur <zash@zash.se> parents: 2001 diff changeset	388 module:log("debug", "Removing old messages for user %q", user);
279885fd9728 mod_mam: Add support for trimming old archived messages Kim Alvefur <zash@zash.se> parents: 2001 diff changeset	389 local ok, err = archive:delete(user, { ["end"] = os.time() - cleanup_after; })
279885fd9728 mod_mam: Add support for trimming old archived messages Kim Alvefur <zash@zash.se> parents: 2001 diff changeset	390 if not ok then
279885fd9728 mod_mam: Add support for trimming old archived messages Kim Alvefur <zash@zash.se> parents: 2001 diff changeset	391 module:log("warn", "Could not expire archives for user %s: %s", user, err);
2042 752eee154891 mod_mam: Add some debug logging Kim Alvefur <zash@zash.se> parents: 2039 diff changeset	392 else
752eee154891 mod_mam: Add some debug logging Kim Alvefur <zash@zash.se> parents: 2039 diff changeset	393 -- :affected() is a recent addition for eg SQLite3 in LuaDBI
752eee154891 mod_mam: Add some debug logging Kim Alvefur <zash@zash.se> parents: 2039 diff changeset	394 pcall(function(stmt)
752eee154891 mod_mam: Add some debug logging Kim Alvefur <zash@zash.se> parents: 2039 diff changeset	395 module:log("debug", "Removed %d messages", stmt:affected());
752eee154891 mod_mam: Add some debug logging Kim Alvefur <zash@zash.se> parents: 2039 diff changeset	396 end, err);
2016 279885fd9728 mod_mam: Add support for trimming old archived messages Kim Alvefur <zash@zash.se> parents: 2001 diff changeset	397 end
2025 ae98bb884110 mod_mam: Fix Top Level Error from string[table] = nil instead of table[string] = nil Kim Alvefur <zash@zash.se> parents: 2024 diff changeset	398 cleanup[user] = nil;
2016 279885fd9728 mod_mam: Add support for trimming old archived messages Kim Alvefur <zash@zash.se> parents: 2001 diff changeset	399 end
2039 464edd03099a mod_mam: Add a bit of randomness to cleanup intervals to keep some time between multiple instances Kim Alvefur <zash@zash.se> parents: 2033 diff changeset	400 return math.random(cleanup_interval, cleanup_interval * 2);
2016 279885fd9728 mod_mam: Add support for trimming old archived messages Kim Alvefur <zash@zash.se> parents: 2001 diff changeset	401 end);
2724 9f17d0f28d77 mod_mam: Don't ask the storage backend to count all items when expiry is disabled Kim Alvefur <zash@zash.se> parents: 2711 diff changeset	402 else
9f17d0f28d77 mod_mam: Don't ask the storage backend to count all items when expiry is disabled Kim Alvefur <zash@zash.se> parents: 2711 diff changeset	403 -- Don't ask the backend to count the potentially unbounded number of items,
9f17d0f28d77 mod_mam: Don't ask the storage backend to count all items when expiry is disabled Kim Alvefur <zash@zash.se> parents: 2711 diff changeset	404 -- it'll get slow.
9f17d0f28d77 mod_mam: Don't ask the storage backend to count all items when expiry is disabled Kim Alvefur <zash@zash.se> parents: 2711 diff changeset	405 use_total = false;
2016 279885fd9728 mod_mam: Add support for trimming old archived messages Kim Alvefur <zash@zash.se> parents: 2001 diff changeset	406 end
279885fd9728 mod_mam: Add support for trimming old archived messages Kim Alvefur <zash@zash.se> parents: 2001 diff changeset	407
523 eff140d53b83 mod_mam: Add experimental implementation of the Message Archive Management ProtoXEP Kim Alvefur <zash@zash.se> parents: diff changeset	408 -- Stanzas sent by local clients
2914 0d2d4d5bb5f5 [mod_mam] Fix hook priority for local clients, too (see also commit 2af42a3af131) tmolitor <thilo@eightysoft.de> parents: 2768 diff changeset	409 local priority = 0.075
0d2d4d5bb5f5 [mod_mam] Fix hook priority for local clients, too (see also commit 2af42a3af131) tmolitor <thilo@eightysoft.de> parents: 2768 diff changeset	410 assert(priority < 0.1, "priority must be after mod_firewall");
0d2d4d5bb5f5 [mod_mam] Fix hook priority for local clients, too (see also commit 2af42a3af131) tmolitor <thilo@eightysoft.de> parents: 2768 diff changeset	411 assert(priority > 0.05, "priority must be before mod_carbons");
0d2d4d5bb5f5 [mod_mam] Fix hook priority for local clients, too (see also commit 2af42a3af131) tmolitor <thilo@eightysoft.de> parents: 2768 diff changeset	412 assert(priority > 0.01, "priority must be before strip_stanza_id");
0d2d4d5bb5f5 [mod_mam] Fix hook priority for local clients, too (see also commit 2af42a3af131) tmolitor <thilo@eightysoft.de> parents: 2768 diff changeset	413 module:hook("pre-message/bare", c2s_message_handler, priority);
0d2d4d5bb5f5 [mod_mam] Fix hook priority for local clients, too (see also commit 2af42a3af131) tmolitor <thilo@eightysoft.de> parents: 2768 diff changeset	414 module:hook("pre-message/full", c2s_message_handler, priority);
523 eff140d53b83 mod_mam: Add experimental implementation of the Message Archive Management ProtoXEP Kim Alvefur <zash@zash.se> parents: diff changeset	415 -- Stanszas to local clients
2914 0d2d4d5bb5f5 [mod_mam] Fix hook priority for local clients, too (see also commit 2af42a3af131) tmolitor <thilo@eightysoft.de> parents: 2768 diff changeset	416 priority = 0.075
2768 2af42a3af131 mod_mam: Squeeze in hooks with priority between sevral other modules (fixes #996) Kim Alvefur <zash@zash.se> parents: 2764 diff changeset	417 assert(priority > 0, "priority must be before mod_message");
2af42a3af131 mod_mam: Squeeze in hooks with priority between sevral other modules (fixes #996) Kim Alvefur <zash@zash.se> parents: 2764 diff changeset	418 assert(priority < 0.1, "priority must be after mod_firewall");
2af42a3af131 mod_mam: Squeeze in hooks with priority between sevral other modules (fixes #996) Kim Alvefur <zash@zash.se> parents: 2764 diff changeset	419 assert(priority > 0.05, "priority must be before mod_carbons");
2af42a3af131 mod_mam: Squeeze in hooks with priority between sevral other modules (fixes #996) Kim Alvefur <zash@zash.se> parents: 2764 diff changeset	420 module:hook("message/bare", message_handler, priority);
2af42a3af131 mod_mam: Squeeze in hooks with priority between sevral other modules (fixes #996) Kim Alvefur <zash@zash.se> parents: 2764 diff changeset	421 module:hook("message/full", message_handler, priority);
523 eff140d53b83 mod_mam: Add experimental implementation of the Message Archive Management ProtoXEP Kim Alvefur <zash@zash.se> parents: diff changeset	422
2506 5941aac79f06 mod_mam: Add support for XEP-0313 v0.5 Kim Alvefur <zash@zash.se> parents: 2505 diff changeset	423 module:add_feature(xmlns_mam0); -- COMPAT with XEP-0313 v 0.1
558 66de25ffc8d9 mod_mam: Implement archiving preferences. Kim Alvefur <zash@zash.se> parents: 523 diff changeset	424
1678 9ee56cc1be2c mod_mam: Advertise feature in disco#info for account as per XEP-0313 >= 0.2 Kim Alvefur <zash@zash.se> parents: 1587 diff changeset	425 module:hook("account-disco-info", function(event)
2506 5941aac79f06 mod_mam: Add support for XEP-0313 v0.5 Kim Alvefur <zash@zash.se> parents: 2505 diff changeset	426 (event.reply or event.stanza):tag("feature", {var=xmlns_mam0}):up();
5941aac79f06 mod_mam: Add support for XEP-0313 v0.5 Kim Alvefur <zash@zash.se> parents: 2505 diff changeset	427 (event.reply or event.stanza):tag("feature", {var=xmlns_mam1}):up();
2516 248054199d0f mod_mam: Add support for XEP-0313 v0.6 Kim Alvefur <zash@zash.se> parents: 2515 diff changeset	428 (event.reply or event.stanza):tag("feature", {var=xmlns_mam2}):up();
248054199d0f mod_mam: Add support for XEP-0313 v0.6 Kim Alvefur <zash@zash.se> parents: 2515 diff changeset	429 (event.reply or event.stanza):tag("feature", {var=xmlns_st_id}):up();
1678 9ee56cc1be2c mod_mam: Advertise feature in disco#info for account as per XEP-0313 >= 0.2 Kim Alvefur <zash@zash.se> parents: 1587 diff changeset	430 end);
9ee56cc1be2c mod_mam: Advertise feature in disco#info for account as per XEP-0313 >= 0.2 Kim Alvefur <zash@zash.se> parents: 1587 diff changeset	431

Mercurial > prosody-modules

annotate mod_mam/mod_mam.lua @ 3568:6b3181fe5617