Software `/` code `/` prosody-modules

Annotate

mod_mam_muc/mod_mam_muc.lua @ 1311:27b2a357c73c

mod_mam_muc: porting archive checks from mod_mam (to avoid tracebacks about calling null functions)

author	Vadim Misbakh-Soloviov <mva@mva.name>
date	Wed, 19 Feb 2014 18:03:39 +0700
parent	1279:2118a2eeb1d5
child	1312:a48cf3ccdf9c

rev	line source
820 005037032d65 mod_mam_muc: MUC version of mod_mam Kim Alvefur <zash@zash.se> parents: diff changeset	1 -- XEP-0313: Message Archive Management for Prosody
1141 1091be1c3aba mod_mam_muc: Switch to new stanza storage API Kim Alvefur <zash@zash.se> parents: 1140 diff changeset	2 -- Copyright (C) 2011-2013 Kim Alvefur
820 005037032d65 mod_mam_muc: MUC version of mod_mam Kim Alvefur <zash@zash.se> parents: diff changeset	3 --
005037032d65 mod_mam_muc: MUC version of mod_mam Kim Alvefur <zash@zash.se> parents: diff changeset	4 -- This file is MIT/X11 licensed.
005037032d65 mod_mam_muc: MUC version of mod_mam Kim Alvefur <zash@zash.se> parents: diff changeset	5
005037032d65 mod_mam_muc: MUC version of mod_mam Kim Alvefur <zash@zash.se> parents: diff changeset	6 local xmlns_mam = "urn:xmpp:mam:tmp";
005037032d65 mod_mam_muc: MUC version of mod_mam Kim Alvefur <zash@zash.se> parents: diff changeset	7 local xmlns_delay = "urn:xmpp:delay";
005037032d65 mod_mam_muc: MUC version of mod_mam Kim Alvefur <zash@zash.se> parents: diff changeset	8 local xmlns_forward = "urn:xmpp:forward:0";
1143 8098683b6d6f mod_mam_muc: Allow archiving to be enabled trough in the room configuration Kim Alvefur <zash@zash.se> parents: 1142 diff changeset	9 local muc_form_config_option = "muc#roomconfig_enablelogging"
820 005037032d65 mod_mam_muc: MUC version of mod_mam Kim Alvefur <zash@zash.se> parents: diff changeset	10
005037032d65 mod_mam_muc: MUC version of mod_mam Kim Alvefur <zash@zash.se> parents: diff changeset	11 local st = require "util.stanza";
005037032d65 mod_mam_muc: MUC version of mod_mam Kim Alvefur <zash@zash.se> parents: diff changeset	12 local rsm = module:require "mod_mam/rsm";
005037032d65 mod_mam_muc: MUC version of mod_mam Kim Alvefur <zash@zash.se> parents: diff changeset	13 local jid_bare = require "util.jid".bare;
005037032d65 mod_mam_muc: MUC version of mod_mam Kim Alvefur <zash@zash.se> parents: diff changeset	14 local jid_split = require "util.jid".split;
1278 40f077b18dfe mod_mam_muc: Override sending of room history and use archives Kim Alvefur <zash@zash.se> parents: 1277 diff changeset	15 local room_mt = module:depends"muc".room_mt;
820 005037032d65 mod_mam_muc: MUC version of mod_mam Kim Alvefur <zash@zash.se> parents: diff changeset	16
1141 1091be1c3aba mod_mam_muc: Switch to new stanza storage API Kim Alvefur <zash@zash.se> parents: 1140 diff changeset	17 local getmetatable = getmetatable;
1091be1c3aba mod_mam_muc: Switch to new stanza storage API Kim Alvefur <zash@zash.se> parents: 1140 diff changeset	18 local function is_stanza(x)
1091be1c3aba mod_mam_muc: Switch to new stanza storage API Kim Alvefur <zash@zash.se> parents: 1140 diff changeset	19 return getmetatable(x) == st.stanza_mt;
1091be1c3aba mod_mam_muc: Switch to new stanza storage API Kim Alvefur <zash@zash.se> parents: 1140 diff changeset	20 end
820 005037032d65 mod_mam_muc: MUC version of mod_mam Kim Alvefur <zash@zash.se> parents: diff changeset	21
005037032d65 mod_mam_muc: MUC version of mod_mam Kim Alvefur <zash@zash.se> parents: diff changeset	22 local tostring = tostring;
005037032d65 mod_mam_muc: MUC version of mod_mam Kim Alvefur <zash@zash.se> parents: diff changeset	23 local time_now = os.time;
005037032d65 mod_mam_muc: MUC version of mod_mam Kim Alvefur <zash@zash.se> parents: diff changeset	24 local m_min = math.min;
005037032d65 mod_mam_muc: MUC version of mod_mam Kim Alvefur <zash@zash.se> parents: diff changeset	25 local timestamp, timestamp_parse = require "util.datetime".datetime, require "util.datetime".parse;
005037032d65 mod_mam_muc: MUC version of mod_mam Kim Alvefur <zash@zash.se> parents: diff changeset	26 local default_max_items, max_max_items = 20, module:get_option_number("max_archive_query_results", 50);
1278 40f077b18dfe mod_mam_muc: Override sending of room history and use archives Kim Alvefur <zash@zash.se> parents: 1277 diff changeset	27 local max_history_length = module:get_option_number("max_history_messages", 1000);
820 005037032d65 mod_mam_muc: MUC version of mod_mam Kim Alvefur <zash@zash.se> parents: diff changeset	28
1143 8098683b6d6f mod_mam_muc: Allow archiving to be enabled trough in the room configuration Kim Alvefur <zash@zash.se> parents: 1142 diff changeset	29 local log_all_rooms = module:get_option_boolean("muc_log_all_rooms", false);
8098683b6d6f mod_mam_muc: Allow archiving to be enabled trough in the room configuration Kim Alvefur <zash@zash.se> parents: 1142 diff changeset	30 local log_by_default = module:get_option_boolean("muc_log_by_default", true);
1142 fabdaa0d99e3 mod_mam_muc: Stap archived messages Kim Alvefur <zash@zash.se> parents: 1141 diff changeset	31 local advertise_archive = module:get_option_boolean("muc_log_advertise", true);
fabdaa0d99e3 mod_mam_muc: Stap archived messages Kim Alvefur <zash@zash.se> parents: 1141 diff changeset	32
1311 27b2a357c73c mod_mam_muc: porting archive checks from mod_mam (to avoid tracebacks about calling null functions) Vadim Misbakh-Soloviov <mva@mva.name> parents: 1279 diff changeset	33 local archive_store = "archive2";
27b2a357c73c mod_mam_muc: porting archive checks from mod_mam (to avoid tracebacks about calling null functions) Vadim Misbakh-Soloviov <mva@mva.name> parents: 1279 diff changeset	34 local archive = module:open_store(archive_store, "archive");
27b2a357c73c mod_mam_muc: porting archive checks from mod_mam (to avoid tracebacks about calling null functions) Vadim Misbakh-Soloviov <mva@mva.name> parents: 1279 diff changeset	35 if not archive then
27b2a357c73c mod_mam_muc: porting archive checks from mod_mam (to avoid tracebacks about calling null functions) Vadim Misbakh-Soloviov <mva@mva.name> parents: 1279 diff changeset	36 module:log("error", "Could not open archive storage");
27b2a357c73c mod_mam_muc: porting archive checks from mod_mam (to avoid tracebacks about calling null functions) Vadim Misbakh-Soloviov <mva@mva.name> parents: 1279 diff changeset	37 return
27b2a357c73c mod_mam_muc: porting archive checks from mod_mam (to avoid tracebacks about calling null functions) Vadim Misbakh-Soloviov <mva@mva.name> parents: 1279 diff changeset	38 elseif not archive.find then
27b2a357c73c mod_mam_muc: porting archive checks from mod_mam (to avoid tracebacks about calling null functions) Vadim Misbakh-Soloviov <mva@mva.name> parents: 1279 diff changeset	39 module:log("error", "mod_%s does not support archiving, switch to mod_storage_sql2", archive._provided_by);
27b2a357c73c mod_mam_muc: porting archive checks from mod_mam (to avoid tracebacks about calling null functions) Vadim Misbakh-Soloviov <mva@mva.name> parents: 1279 diff changeset	40 return
27b2a357c73c mod_mam_muc: porting archive checks from mod_mam (to avoid tracebacks about calling null functions) Vadim Misbakh-Soloviov <mva@mva.name> parents: 1279 diff changeset	41 end
27b2a357c73c mod_mam_muc: porting archive checks from mod_mam (to avoid tracebacks about calling null functions) Vadim Misbakh-Soloviov <mva@mva.name> parents: 1279 diff changeset	42
1139 b32d65e41755 mod_mam_muc: Get room objects in a less awkward fashion Kim Alvefur <zash@zash.se> parents: 1138 diff changeset	43 local rooms = hosts[module.host].modules.muc.rooms;
820 005037032d65 mod_mam_muc: MUC version of mod_mam Kim Alvefur <zash@zash.se> parents: diff changeset	44
1276 01dfaf2f2782 mod_mam_muc: Hide logging option from room configuration if set to log all rooms Kim Alvefur <zash@zash.se> parents: 1275 diff changeset	45 if not log_all_rooms then
01dfaf2f2782 mod_mam_muc: Hide logging option from room configuration if set to log all rooms Kim Alvefur <zash@zash.se> parents: 1275 diff changeset	46 module:hook("muc-config-form", function(event)
01dfaf2f2782 mod_mam_muc: Hide logging option from room configuration if set to log all rooms Kim Alvefur <zash@zash.se> parents: 1275 diff changeset	47 local room, form = event.room, event.form;
01dfaf2f2782 mod_mam_muc: Hide logging option from room configuration if set to log all rooms Kim Alvefur <zash@zash.se> parents: 1275 diff changeset	48 local logging_enabled = room._data.logging;
01dfaf2f2782 mod_mam_muc: Hide logging option from room configuration if set to log all rooms Kim Alvefur <zash@zash.se> parents: 1275 diff changeset	49 if logging_enabled == nil then
01dfaf2f2782 mod_mam_muc: Hide logging option from room configuration if set to log all rooms Kim Alvefur <zash@zash.se> parents: 1275 diff changeset	50 logging_enabled = log_by_default;
01dfaf2f2782 mod_mam_muc: Hide logging option from room configuration if set to log all rooms Kim Alvefur <zash@zash.se> parents: 1275 diff changeset	51 end
01dfaf2f2782 mod_mam_muc: Hide logging option from room configuration if set to log all rooms Kim Alvefur <zash@zash.se> parents: 1275 diff changeset	52 table.insert(form,
01dfaf2f2782 mod_mam_muc: Hide logging option from room configuration if set to log all rooms Kim Alvefur <zash@zash.se> parents: 1275 diff changeset	53 {
01dfaf2f2782 mod_mam_muc: Hide logging option from room configuration if set to log all rooms Kim Alvefur <zash@zash.se> parents: 1275 diff changeset	54 name = muc_form_config_option,
01dfaf2f2782 mod_mam_muc: Hide logging option from room configuration if set to log all rooms Kim Alvefur <zash@zash.se> parents: 1275 diff changeset	55 type = "boolean",
01dfaf2f2782 mod_mam_muc: Hide logging option from room configuration if set to log all rooms Kim Alvefur <zash@zash.se> parents: 1275 diff changeset	56 label = "Enable Logging?",
01dfaf2f2782 mod_mam_muc: Hide logging option from room configuration if set to log all rooms Kim Alvefur <zash@zash.se> parents: 1275 diff changeset	57 value = logging_enabled,
01dfaf2f2782 mod_mam_muc: Hide logging option from room configuration if set to log all rooms Kim Alvefur <zash@zash.se> parents: 1275 diff changeset	58 }
01dfaf2f2782 mod_mam_muc: Hide logging option from room configuration if set to log all rooms Kim Alvefur <zash@zash.se> parents: 1275 diff changeset	59 );
01dfaf2f2782 mod_mam_muc: Hide logging option from room configuration if set to log all rooms Kim Alvefur <zash@zash.se> parents: 1275 diff changeset	60 end);
1143 8098683b6d6f mod_mam_muc: Allow archiving to be enabled trough in the room configuration Kim Alvefur <zash@zash.se> parents: 1142 diff changeset	61
1276 01dfaf2f2782 mod_mam_muc: Hide logging option from room configuration if set to log all rooms Kim Alvefur <zash@zash.se> parents: 1275 diff changeset	62 module:hook("muc-config-submitted", function(event)
01dfaf2f2782 mod_mam_muc: Hide logging option from room configuration if set to log all rooms Kim Alvefur <zash@zash.se> parents: 1275 diff changeset	63 local room, fields, changed = event.room, event.fields, event.changed;
01dfaf2f2782 mod_mam_muc: Hide logging option from room configuration if set to log all rooms Kim Alvefur <zash@zash.se> parents: 1275 diff changeset	64 local new = fields[muc_form_config_option];
01dfaf2f2782 mod_mam_muc: Hide logging option from room configuration if set to log all rooms Kim Alvefur <zash@zash.se> parents: 1275 diff changeset	65 if new ~= room._data.logging then
01dfaf2f2782 mod_mam_muc: Hide logging option from room configuration if set to log all rooms Kim Alvefur <zash@zash.se> parents: 1275 diff changeset	66 room._data.logging = new;
01dfaf2f2782 mod_mam_muc: Hide logging option from room configuration if set to log all rooms Kim Alvefur <zash@zash.se> parents: 1275 diff changeset	67 if type(changed) == "table" then
01dfaf2f2782 mod_mam_muc: Hide logging option from room configuration if set to log all rooms Kim Alvefur <zash@zash.se> parents: 1275 diff changeset	68 changed[muc_form_config_option] = true;
01dfaf2f2782 mod_mam_muc: Hide logging option from room configuration if set to log all rooms Kim Alvefur <zash@zash.se> parents: 1275 diff changeset	69 else
01dfaf2f2782 mod_mam_muc: Hide logging option from room configuration if set to log all rooms Kim Alvefur <zash@zash.se> parents: 1275 diff changeset	70 event.changed = true;
01dfaf2f2782 mod_mam_muc: Hide logging option from room configuration if set to log all rooms Kim Alvefur <zash@zash.se> parents: 1275 diff changeset	71 end
1143 8098683b6d6f mod_mam_muc: Allow archiving to be enabled trough in the room configuration Kim Alvefur <zash@zash.se> parents: 1142 diff changeset	72 end
1276 01dfaf2f2782 mod_mam_muc: Hide logging option from room configuration if set to log all rooms Kim Alvefur <zash@zash.se> parents: 1275 diff changeset	73 end);
01dfaf2f2782 mod_mam_muc: Hide logging option from room configuration if set to log all rooms Kim Alvefur <zash@zash.se> parents: 1275 diff changeset	74 end
1143 8098683b6d6f mod_mam_muc: Allow archiving to be enabled trough in the room configuration Kim Alvefur <zash@zash.se> parents: 1142 diff changeset	75
1278 40f077b18dfe mod_mam_muc: Override sending of room history and use archives Kim Alvefur <zash@zash.se> parents: 1277 diff changeset	76 local _send_history = room_mt.send_history;
1279 2118a2eeb1d5 mod_mam_muc: Override method for storing history messages instead of hooking stanza events Kim Alvefur <zash@zash.se> parents: 1278 diff changeset	77 local _save_to_history = room_mt.save_to_history;
1278 40f077b18dfe mod_mam_muc: Override sending of room history and use archives Kim Alvefur <zash@zash.se> parents: 1277 diff changeset	78 function module.unload()
40f077b18dfe mod_mam_muc: Override sending of room history and use archives Kim Alvefur <zash@zash.se> parents: 1277 diff changeset	79 room_mt.send_history = _send_history;
1279 2118a2eeb1d5 mod_mam_muc: Override method for storing history messages instead of hooking stanza events Kim Alvefur <zash@zash.se> parents: 1278 diff changeset	80 room_mt.save_to_history = _save_to_history;
1278 40f077b18dfe mod_mam_muc: Override sending of room history and use archives Kim Alvefur <zash@zash.se> parents: 1277 diff changeset	81 end
1143 8098683b6d6f mod_mam_muc: Allow archiving to be enabled trough in the room configuration Kim Alvefur <zash@zash.se> parents: 1142 diff changeset	82
820 005037032d65 mod_mam_muc: MUC version of mod_mam Kim Alvefur <zash@zash.se> parents: diff changeset	83 -- Handle archive queries
1138 5c97ee75cadb mod_mam_muc: Switch to iq-get hook and drop some indentation Kim Alvefur <zash@zash.se> parents: 820 diff changeset	84 module:hook("iq-get/bare/"..xmlns_mam..":query", function(event)
820 005037032d65 mod_mam_muc: MUC version of mod_mam Kim Alvefur <zash@zash.se> parents: diff changeset	85 local origin, stanza = event.origin, event.stanza;
1144 ccb0c5afe658 mod_mam_muc: Fix room lookup, should be indexed by bare jid Kim Alvefur <zash@zash.se> parents: 1143 diff changeset	86 local room = stanza.attr.to;
1146 9fa89dc7a86f mod_mam_muc: Search the rooms archive correctly (copypaste error from mod_mam) Kim Alvefur <zash@zash.se> parents: 1145 diff changeset	87 local room_node = jid_split(room);
820 005037032d65 mod_mam_muc: MUC version of mod_mam Kim Alvefur <zash@zash.se> parents: diff changeset	88 local query = stanza.tags[1];
005037032d65 mod_mam_muc: MUC version of mod_mam Kim Alvefur <zash@zash.se> parents: diff changeset	89
1139 b32d65e41755 mod_mam_muc: Get room objects in a less awkward fashion Kim Alvefur <zash@zash.se> parents: 1138 diff changeset	90 local room_obj = rooms[room];
820 005037032d65 mod_mam_muc: MUC version of mod_mam Kim Alvefur <zash@zash.se> parents: diff changeset	91 if not room_obj then
1145 5a00f9bec6e7 mod_mam_muc: Send item-not-found if the requested room does not exist Kim Alvefur <zash@zash.se> parents: 1144 diff changeset	92 return origin.send(st.error_reply(stanza, "cancel", "item-not-found"))
820 005037032d65 mod_mam_muc: MUC version of mod_mam Kim Alvefur <zash@zash.se> parents: diff changeset	93 end
005037032d65 mod_mam_muc: MUC version of mod_mam Kim Alvefur <zash@zash.se> parents: diff changeset	94 local from = jid_bare(stanza.attr.from);
005037032d65 mod_mam_muc: MUC version of mod_mam Kim Alvefur <zash@zash.se> parents: diff changeset	95
1140 402cb9b604eb mod_mam_muc: Send proper error reply when one is not allowed to query archive Kim Alvefur <zash@zash.se> parents: 1139 diff changeset	96 -- Banned or not a member of a members-only room?
820 005037032d65 mod_mam_muc: MUC version of mod_mam Kim Alvefur <zash@zash.se> parents: diff changeset	97 if room_obj._affiliations[from] == "outcast"
005037032d65 mod_mam_muc: MUC version of mod_mam Kim Alvefur <zash@zash.se> parents: diff changeset	98 or room_obj._data.members_only and not room_obj._affiliations[from] then
1140 402cb9b604eb mod_mam_muc: Send proper error reply when one is not allowed to query archive Kim Alvefur <zash@zash.se> parents: 1139 diff changeset	99 return origin.send(st.error_reply(stanza, "auth", "forbidden"))
820 005037032d65 mod_mam_muc: MUC version of mod_mam Kim Alvefur <zash@zash.se> parents: diff changeset	100 end
005037032d65 mod_mam_muc: MUC version of mod_mam Kim Alvefur <zash@zash.se> parents: diff changeset	101
1138 5c97ee75cadb mod_mam_muc: Switch to iq-get hook and drop some indentation Kim Alvefur <zash@zash.se> parents: 820 diff changeset	102 local qid = query.attr.queryid;
5c97ee75cadb mod_mam_muc: Switch to iq-get hook and drop some indentation Kim Alvefur <zash@zash.se> parents: 820 diff changeset	103
5c97ee75cadb mod_mam_muc: Switch to iq-get hook and drop some indentation Kim Alvefur <zash@zash.se> parents: 820 diff changeset	104 -- Search query parameters
5c97ee75cadb mod_mam_muc: Switch to iq-get hook and drop some indentation Kim Alvefur <zash@zash.se> parents: 820 diff changeset	105 local qstart = query:get_child_text("start");
5c97ee75cadb mod_mam_muc: Switch to iq-get hook and drop some indentation Kim Alvefur <zash@zash.se> parents: 820 diff changeset	106 local qend = query:get_child_text("end");
1141 1091be1c3aba mod_mam_muc: Switch to new stanza storage API Kim Alvefur <zash@zash.se> parents: 1140 diff changeset	107 module:log("debug", "Archive query, id %s from %s until %s)",
1091be1c3aba mod_mam_muc: Switch to new stanza storage API Kim Alvefur <zash@zash.se> parents: 1140 diff changeset	108 tostring(qid), qstart or "the dawn of time", qend or "now");
820 005037032d65 mod_mam_muc: MUC version of mod_mam Kim Alvefur <zash@zash.se> parents: diff changeset	109
1138 5c97ee75cadb mod_mam_muc: Switch to iq-get hook and drop some indentation Kim Alvefur <zash@zash.se> parents: 820 diff changeset	110 if qstart or qend then -- Validate timestamps
5c97ee75cadb mod_mam_muc: Switch to iq-get hook and drop some indentation Kim Alvefur <zash@zash.se> parents: 820 diff changeset	111 local vstart, vend = (qstart and timestamp_parse(qstart)), (qend and timestamp_parse(qend))
5c97ee75cadb mod_mam_muc: Switch to iq-get hook and drop some indentation Kim Alvefur <zash@zash.se> parents: 820 diff changeset	112 if (qstart and not vstart) or (qend and not vend) then
5c97ee75cadb mod_mam_muc: Switch to iq-get hook and drop some indentation Kim Alvefur <zash@zash.se> parents: 820 diff changeset	113 origin.send(st.error_reply(stanza, "modify", "bad-request", "Invalid timestamp"))
5c97ee75cadb mod_mam_muc: Switch to iq-get hook and drop some indentation Kim Alvefur <zash@zash.se> parents: 820 diff changeset	114 return true
5c97ee75cadb mod_mam_muc: Switch to iq-get hook and drop some indentation Kim Alvefur <zash@zash.se> parents: 820 diff changeset	115 end
5c97ee75cadb mod_mam_muc: Switch to iq-get hook and drop some indentation Kim Alvefur <zash@zash.se> parents: 820 diff changeset	116 qstart, qend = vstart, vend;
5c97ee75cadb mod_mam_muc: Switch to iq-get hook and drop some indentation Kim Alvefur <zash@zash.se> parents: 820 diff changeset	117 end
820 005037032d65 mod_mam_muc: MUC version of mod_mam Kim Alvefur <zash@zash.se> parents: diff changeset	118
1141 1091be1c3aba mod_mam_muc: Switch to new stanza storage API Kim Alvefur <zash@zash.se> parents: 1140 diff changeset	119 -- RSM stuff
1091be1c3aba mod_mam_muc: Switch to new stanza storage API Kim Alvefur <zash@zash.se> parents: 1140 diff changeset	120 local qset = rsm.get(query);
1091be1c3aba mod_mam_muc: Switch to new stanza storage API Kim Alvefur <zash@zash.se> parents: 1140 diff changeset	121 local qmax = m_min(qset and qset.max or default_max_items, max_max_items);
1091be1c3aba mod_mam_muc: Switch to new stanza storage API Kim Alvefur <zash@zash.se> parents: 1140 diff changeset	122 local reverse = qset and qset.before or false;
1091be1c3aba mod_mam_muc: Switch to new stanza storage API Kim Alvefur <zash@zash.se> parents: 1140 diff changeset	123
1091be1c3aba mod_mam_muc: Switch to new stanza storage API Kim Alvefur <zash@zash.se> parents: 1140 diff changeset	124 local before, after = qset and qset.before, qset and qset.after;
1091be1c3aba mod_mam_muc: Switch to new stanza storage API Kim Alvefur <zash@zash.se> parents: 1140 diff changeset	125 if type(before) ~= "string" then before = nil; end
820 005037032d65 mod_mam_muc: MUC version of mod_mam Kim Alvefur <zash@zash.se> parents: diff changeset	126
1138 5c97ee75cadb mod_mam_muc: Switch to iq-get hook and drop some indentation Kim Alvefur <zash@zash.se> parents: 820 diff changeset	127 -- Load all the data!
1146 9fa89dc7a86f mod_mam_muc: Search the rooms archive correctly (copypaste error from mod_mam) Kim Alvefur <zash@zash.se> parents: 1145 diff changeset	128 local data, err = archive:find(room_node, {
1141 1091be1c3aba mod_mam_muc: Switch to new stanza storage API Kim Alvefur <zash@zash.se> parents: 1140 diff changeset	129 start = qstart; ["end"] = qend; -- Time range
1091be1c3aba mod_mam_muc: Switch to new stanza storage API Kim Alvefur <zash@zash.se> parents: 1140 diff changeset	130 limit = qmax;
1091be1c3aba mod_mam_muc: Switch to new stanza storage API Kim Alvefur <zash@zash.se> parents: 1140 diff changeset	131 before = before; after = after;
1091be1c3aba mod_mam_muc: Switch to new stanza storage API Kim Alvefur <zash@zash.se> parents: 1140 diff changeset	132 reverse = reverse;
1091be1c3aba mod_mam_muc: Switch to new stanza storage API Kim Alvefur <zash@zash.se> parents: 1140 diff changeset	133 total = true;
1091be1c3aba mod_mam_muc: Switch to new stanza storage API Kim Alvefur <zash@zash.se> parents: 1140 diff changeset	134 });
1138 5c97ee75cadb mod_mam_muc: Switch to iq-get hook and drop some indentation Kim Alvefur <zash@zash.se> parents: 820 diff changeset	135
1141 1091be1c3aba mod_mam_muc: Switch to new stanza storage API Kim Alvefur <zash@zash.se> parents: 1140 diff changeset	136 if not data then
1091be1c3aba mod_mam_muc: Switch to new stanza storage API Kim Alvefur <zash@zash.se> parents: 1140 diff changeset	137 return origin.send(st.error_reply(stanza, "cancel", "internal-server-error"));
1091be1c3aba mod_mam_muc: Switch to new stanza storage API Kim Alvefur <zash@zash.se> parents: 1140 diff changeset	138 end
1091be1c3aba mod_mam_muc: Switch to new stanza storage API Kim Alvefur <zash@zash.se> parents: 1140 diff changeset	139 local count = err;
1138 5c97ee75cadb mod_mam_muc: Switch to iq-get hook and drop some indentation Kim Alvefur <zash@zash.se> parents: 820 diff changeset	140
1141 1091be1c3aba mod_mam_muc: Switch to new stanza storage API Kim Alvefur <zash@zash.se> parents: 1140 diff changeset	141 -- Wrap it in stuff and deliver
1091be1c3aba mod_mam_muc: Switch to new stanza storage API Kim Alvefur <zash@zash.se> parents: 1140 diff changeset	142 local first, last;
1091be1c3aba mod_mam_muc: Switch to new stanza storage API Kim Alvefur <zash@zash.se> parents: 1140 diff changeset	143 for id, item, when in data do
1091be1c3aba mod_mam_muc: Switch to new stanza storage API Kim Alvefur <zash@zash.se> parents: 1140 diff changeset	144 local fwd_st = st.message{ to = origin.full_jid }
1091be1c3aba mod_mam_muc: Switch to new stanza storage API Kim Alvefur <zash@zash.se> parents: 1140 diff changeset	145 :tag("result", { xmlns = xmlns_mam, queryid = qid, id = id })
1138 5c97ee75cadb mod_mam_muc: Switch to iq-get hook and drop some indentation Kim Alvefur <zash@zash.se> parents: 820 diff changeset	146 :tag("forwarded", { xmlns = xmlns_forward })
5c97ee75cadb mod_mam_muc: Switch to iq-get hook and drop some indentation Kim Alvefur <zash@zash.se> parents: 820 diff changeset	147 :tag("delay", { xmlns = xmlns_delay, stamp = timestamp(when) }):up();
1141 1091be1c3aba mod_mam_muc: Switch to new stanza storage API Kim Alvefur <zash@zash.se> parents: 1140 diff changeset	148
1091be1c3aba mod_mam_muc: Switch to new stanza storage API Kim Alvefur <zash@zash.se> parents: 1140 diff changeset	149 if not is_stanza(item) then
1091be1c3aba mod_mam_muc: Switch to new stanza storage API Kim Alvefur <zash@zash.se> parents: 1140 diff changeset	150 item = st.deserialize(item);
1138 5c97ee75cadb mod_mam_muc: Switch to iq-get hook and drop some indentation Kim Alvefur <zash@zash.se> parents: 820 diff changeset	151 end
1141 1091be1c3aba mod_mam_muc: Switch to new stanza storage API Kim Alvefur <zash@zash.se> parents: 1140 diff changeset	152 item.attr.xmlns = "jabber:client";
1091be1c3aba mod_mam_muc: Switch to new stanza storage API Kim Alvefur <zash@zash.se> parents: 1140 diff changeset	153 fwd_st:add_child(item);
820 005037032d65 mod_mam_muc: MUC version of mod_mam Kim Alvefur <zash@zash.se> parents: diff changeset	154
1141 1091be1c3aba mod_mam_muc: Switch to new stanza storage API Kim Alvefur <zash@zash.se> parents: 1140 diff changeset	155 if not first then first = id; end
1091be1c3aba mod_mam_muc: Switch to new stanza storage API Kim Alvefur <zash@zash.se> parents: 1140 diff changeset	156 last = id;
1091be1c3aba mod_mam_muc: Switch to new stanza storage API Kim Alvefur <zash@zash.se> parents: 1140 diff changeset	157
1091be1c3aba mod_mam_muc: Switch to new stanza storage API Kim Alvefur <zash@zash.se> parents: 1140 diff changeset	158 origin.send(fwd_st);
1138 5c97ee75cadb mod_mam_muc: Switch to iq-get hook and drop some indentation Kim Alvefur <zash@zash.se> parents: 820 diff changeset	159 end
5c97ee75cadb mod_mam_muc: Switch to iq-get hook and drop some indentation Kim Alvefur <zash@zash.se> parents: 820 diff changeset	160 -- That's all folks!
5c97ee75cadb mod_mam_muc: Switch to iq-get hook and drop some indentation Kim Alvefur <zash@zash.se> parents: 820 diff changeset	161 module:log("debug", "Archive query %s completed", tostring(qid));
820 005037032d65 mod_mam_muc: MUC version of mod_mam Kim Alvefur <zash@zash.se> parents: diff changeset	162
1141 1091be1c3aba mod_mam_muc: Switch to new stanza storage API Kim Alvefur <zash@zash.se> parents: 1140 diff changeset	163 if reverse then first, last = last, first; end
1091be1c3aba mod_mam_muc: Switch to new stanza storage API Kim Alvefur <zash@zash.se> parents: 1140 diff changeset	164 return origin.send(st.reply(stanza)
1091be1c3aba mod_mam_muc: Switch to new stanza storage API Kim Alvefur <zash@zash.se> parents: 1140 diff changeset	165 :query(xmlns_mam):add_child(rsm.generate {
1091be1c3aba mod_mam_muc: Switch to new stanza storage API Kim Alvefur <zash@zash.se> parents: 1140 diff changeset	166 first = first, last = last, count = count }));
820 005037032d65 mod_mam_muc: MUC version of mod_mam Kim Alvefur <zash@zash.se> parents: diff changeset	167 end);
005037032d65 mod_mam_muc: MUC version of mod_mam Kim Alvefur <zash@zash.se> parents: diff changeset	168
1278 40f077b18dfe mod_mam_muc: Override sending of room history and use archives Kim Alvefur <zash@zash.se> parents: 1277 diff changeset	169 function room_mt:send_history(to, stanza)
40f077b18dfe mod_mam_muc: Override sending of room history and use archives Kim Alvefur <zash@zash.se> parents: 1277 diff changeset	170 local maxchars, maxstanzas, seconds, since;
40f077b18dfe mod_mam_muc: Override sending of room history and use archives Kim Alvefur <zash@zash.se> parents: 1277 diff changeset	171 local history_tag = stanza:find("{http://jabber.org/protocol/muc}x/history")
40f077b18dfe mod_mam_muc: Override sending of room history and use archives Kim Alvefur <zash@zash.se> parents: 1277 diff changeset	172 if history_tag then
40f077b18dfe mod_mam_muc: Override sending of room history and use archives Kim Alvefur <zash@zash.se> parents: 1277 diff changeset	173 module:log("debug", tostring(history_tag));
40f077b18dfe mod_mam_muc: Override sending of room history and use archives Kim Alvefur <zash@zash.se> parents: 1277 diff changeset	174 local history_attr = history_tag.attr;
40f077b18dfe mod_mam_muc: Override sending of room history and use archives Kim Alvefur <zash@zash.se> parents: 1277 diff changeset	175 maxchars = tonumber(history_attr.maxchars);
40f077b18dfe mod_mam_muc: Override sending of room history and use archives Kim Alvefur <zash@zash.se> parents: 1277 diff changeset	176 maxstanzas = tonumber(history_attr.maxstanzas);
40f077b18dfe mod_mam_muc: Override sending of room history and use archives Kim Alvefur <zash@zash.se> parents: 1277 diff changeset	177 seconds = tonumber(history_attr.seconds);
40f077b18dfe mod_mam_muc: Override sending of room history and use archives Kim Alvefur <zash@zash.se> parents: 1277 diff changeset	178 since = history_attr.since;
40f077b18dfe mod_mam_muc: Override sending of room history and use archives Kim Alvefur <zash@zash.se> parents: 1277 diff changeset	179 if since then
40f077b18dfe mod_mam_muc: Override sending of room history and use archives Kim Alvefur <zash@zash.se> parents: 1277 diff changeset	180 since = timestamp_parse(since);
40f077b18dfe mod_mam_muc: Override sending of room history and use archives Kim Alvefur <zash@zash.se> parents: 1277 diff changeset	181 end
40f077b18dfe mod_mam_muc: Override sending of room history and use archives Kim Alvefur <zash@zash.se> parents: 1277 diff changeset	182 if seconds then
40f077b18dfe mod_mam_muc: Override sending of room history and use archives Kim Alvefur <zash@zash.se> parents: 1277 diff changeset	183 since = math.max(os.time() - seconds, since or 0);
40f077b18dfe mod_mam_muc: Override sending of room history and use archives Kim Alvefur <zash@zash.se> parents: 1277 diff changeset	184 end
40f077b18dfe mod_mam_muc: Override sending of room history and use archives Kim Alvefur <zash@zash.se> parents: 1277 diff changeset	185 end
40f077b18dfe mod_mam_muc: Override sending of room history and use archives Kim Alvefur <zash@zash.se> parents: 1277 diff changeset	186
40f077b18dfe mod_mam_muc: Override sending of room history and use archives Kim Alvefur <zash@zash.se> parents: 1277 diff changeset	187 -- Load all the data!
40f077b18dfe mod_mam_muc: Override sending of room history and use archives Kim Alvefur <zash@zash.se> parents: 1277 diff changeset	188 local data, err = archive:find(jid_split(self.jid), {
40f077b18dfe mod_mam_muc: Override sending of room history and use archives Kim Alvefur <zash@zash.se> parents: 1277 diff changeset	189 limit = m_min(maxstanzas or 20, max_history_length);
40f077b18dfe mod_mam_muc: Override sending of room history and use archives Kim Alvefur <zash@zash.se> parents: 1277 diff changeset	190 after = since;
40f077b18dfe mod_mam_muc: Override sending of room history and use archives Kim Alvefur <zash@zash.se> parents: 1277 diff changeset	191 reverse = true;
40f077b18dfe mod_mam_muc: Override sending of room history and use archives Kim Alvefur <zash@zash.se> parents: 1277 diff changeset	192 });
40f077b18dfe mod_mam_muc: Override sending of room history and use archives Kim Alvefur <zash@zash.se> parents: 1277 diff changeset	193
40f077b18dfe mod_mam_muc: Override sending of room history and use archives Kim Alvefur <zash@zash.se> parents: 1277 diff changeset	194 if not data then
40f077b18dfe mod_mam_muc: Override sending of room history and use archives Kim Alvefur <zash@zash.se> parents: 1277 diff changeset	195 module:log("error", "Could not fetch history: %s", tostring(err));
40f077b18dfe mod_mam_muc: Override sending of room history and use archives Kim Alvefur <zash@zash.se> parents: 1277 diff changeset	196 return
40f077b18dfe mod_mam_muc: Override sending of room history and use archives Kim Alvefur <zash@zash.se> parents: 1277 diff changeset	197 end
40f077b18dfe mod_mam_muc: Override sending of room history and use archives Kim Alvefur <zash@zash.se> parents: 1277 diff changeset	198
40f077b18dfe mod_mam_muc: Override sending of room history and use archives Kim Alvefur <zash@zash.se> parents: 1277 diff changeset	199 local to_send = {};
40f077b18dfe mod_mam_muc: Override sending of room history and use archives Kim Alvefur <zash@zash.se> parents: 1277 diff changeset	200 local charcount = 0;
40f077b18dfe mod_mam_muc: Override sending of room history and use archives Kim Alvefur <zash@zash.se> parents: 1277 diff changeset	201 local chars;
40f077b18dfe mod_mam_muc: Override sending of room history and use archives Kim Alvefur <zash@zash.se> parents: 1277 diff changeset	202 for id, item, when in data do
40f077b18dfe mod_mam_muc: Override sending of room history and use archives Kim Alvefur <zash@zash.se> parents: 1277 diff changeset	203 item.attr.to = to;
40f077b18dfe mod_mam_muc: Override sending of room history and use archives Kim Alvefur <zash@zash.se> parents: 1277 diff changeset	204 item:tag("delay", { xmlns = "urn:xmpp:delay", from = self.jid, stamp = timestamp(when) }):up(); -- XEP-0203
40f077b18dfe mod_mam_muc: Override sending of room history and use archives Kim Alvefur <zash@zash.se> parents: 1277 diff changeset	205 if maxchars then
40f077b18dfe mod_mam_muc: Override sending of room history and use archives Kim Alvefur <zash@zash.se> parents: 1277 diff changeset	206 chars = #tostring(item);
40f077b18dfe mod_mam_muc: Override sending of room history and use archives Kim Alvefur <zash@zash.se> parents: 1277 diff changeset	207 if chars + charcount > maxchars then break end
40f077b18dfe mod_mam_muc: Override sending of room history and use archives Kim Alvefur <zash@zash.se> parents: 1277 diff changeset	208 charcount = charcount + chars;
40f077b18dfe mod_mam_muc: Override sending of room history and use archives Kim Alvefur <zash@zash.se> parents: 1277 diff changeset	209 end
40f077b18dfe mod_mam_muc: Override sending of room history and use archives Kim Alvefur <zash@zash.se> parents: 1277 diff changeset	210 to_send[1+#to_send] = item;
40f077b18dfe mod_mam_muc: Override sending of room history and use archives Kim Alvefur <zash@zash.se> parents: 1277 diff changeset	211 end
40f077b18dfe mod_mam_muc: Override sending of room history and use archives Kim Alvefur <zash@zash.se> parents: 1277 diff changeset	212 for i = #to_send,1,-1 do
40f077b18dfe mod_mam_muc: Override sending of room history and use archives Kim Alvefur <zash@zash.se> parents: 1277 diff changeset	213 self:_route_stanza(to_send[i]);
40f077b18dfe mod_mam_muc: Override sending of room history and use archives Kim Alvefur <zash@zash.se> parents: 1277 diff changeset	214 end
40f077b18dfe mod_mam_muc: Override sending of room history and use archives Kim Alvefur <zash@zash.se> parents: 1277 diff changeset	215 end
40f077b18dfe mod_mam_muc: Override sending of room history and use archives Kim Alvefur <zash@zash.se> parents: 1277 diff changeset	216
820 005037032d65 mod_mam_muc: MUC version of mod_mam Kim Alvefur <zash@zash.se> parents: diff changeset	217 -- Handle messages
1279 2118a2eeb1d5 mod_mam_muc: Override method for storing history messages instead of hooking stanza events Kim Alvefur <zash@zash.se> parents: 1278 diff changeset	218 function room_mt:save_to_history(stanza)
820 005037032d65 mod_mam_muc: MUC version of mod_mam Kim Alvefur <zash@zash.se> parents: diff changeset	219 local orig_to = stanza.attr.to;
1279 2118a2eeb1d5 mod_mam_muc: Override method for storing history messages instead of hooking stanza events Kim Alvefur <zash@zash.se> parents: 1278 diff changeset	220 local room = jid_split(self.jid);
820 005037032d65 mod_mam_muc: MUC version of mod_mam Kim Alvefur <zash@zash.se> parents: diff changeset	221
1279 2118a2eeb1d5 mod_mam_muc: Override method for storing history messages instead of hooking stanza events Kim Alvefur <zash@zash.se> parents: 1278 diff changeset	222 -- Policy check
1143 8098683b6d6f mod_mam_muc: Allow archiving to be enabled trough in the room configuration Kim Alvefur <zash@zash.se> parents: 1142 diff changeset	223 if not ( log_all_rooms == true -- Logging forced on all rooms
1279 2118a2eeb1d5 mod_mam_muc: Override method for storing history messages instead of hooking stanza events Kim Alvefur <zash@zash.se> parents: 1278 diff changeset	224 or (self._data.logging == nil and log_by_default == true)
2118a2eeb1d5 mod_mam_muc: Override method for storing history messages instead of hooking stanza events Kim Alvefur <zash@zash.se> parents: 1278 diff changeset	225 or self._data.logging ) then return end -- Don't log
1143 8098683b6d6f mod_mam_muc: Allow archiving to be enabled trough in the room configuration Kim Alvefur <zash@zash.se> parents: 1142 diff changeset	226
1279 2118a2eeb1d5 mod_mam_muc: Override method for storing history messages instead of hooking stanza events Kim Alvefur <zash@zash.se> parents: 1278 diff changeset	227 module:log("debug", "We're logging this")
820 005037032d65 mod_mam_muc: MUC version of mod_mam Kim Alvefur <zash@zash.se> parents: diff changeset	228 -- And stash it
1142 fabdaa0d99e3 mod_mam_muc: Stap archived messages Kim Alvefur <zash@zash.se> parents: 1141 diff changeset	229 local ok, id = archive:append(room, time_now(), "", stanza);
fabdaa0d99e3 mod_mam_muc: Stap archived messages Kim Alvefur <zash@zash.se> parents: 1141 diff changeset	230 if ok and advertise_archive then
fabdaa0d99e3 mod_mam_muc: Stap archived messages Kim Alvefur <zash@zash.se> parents: 1141 diff changeset	231 stanza:tag("archived", { xmlns = xmlns_mam, by = jid_bare(orig_to), id = id }):up();
fabdaa0d99e3 mod_mam_muc: Stap archived messages Kim Alvefur <zash@zash.se> parents: 1141 diff changeset	232 end
820 005037032d65 mod_mam_muc: MUC version of mod_mam Kim Alvefur <zash@zash.se> parents: diff changeset	233 end
005037032d65 mod_mam_muc: MUC version of mod_mam Kim Alvefur <zash@zash.se> parents: diff changeset	234
1277 999891a9ae5d mod_mam_muc: Remove archives when a room is destroyed Kim Alvefur <zash@zash.se> parents: 1276 diff changeset	235 module:hook("muc-room-destroyed", function(event)
999891a9ae5d mod_mam_muc: Remove archives when a room is destroyed Kim Alvefur <zash@zash.se> parents: 1276 diff changeset	236 archive:delete(jid_split(event.room.jid));
999891a9ae5d mod_mam_muc: Remove archives when a room is destroyed Kim Alvefur <zash@zash.se> parents: 1276 diff changeset	237 end);
999891a9ae5d mod_mam_muc: Remove archives when a room is destroyed Kim Alvefur <zash@zash.se> parents: 1276 diff changeset	238
1141 1091be1c3aba mod_mam_muc: Switch to new stanza storage API Kim Alvefur <zash@zash.se> parents: 1140 diff changeset	239 -- TODO should we perhaps log presence as well?
1091be1c3aba mod_mam_muc: Switch to new stanza storage API Kim Alvefur <zash@zash.se> parents: 1140 diff changeset	240 -- And role/affiliation changes?
1091be1c3aba mod_mam_muc: Switch to new stanza storage API Kim Alvefur <zash@zash.se> parents: 1140 diff changeset	241
820 005037032d65 mod_mam_muc: MUC version of mod_mam Kim Alvefur <zash@zash.se> parents: diff changeset	242 module:add_feature(xmlns_mam);

prosody-modules

27b2a357c73c

mod_mam_muc/mod_mam_muc.lua

Software / code / prosody-modules

Annotate

mod_mam_muc/mod_mam_muc.lua @ 1311:27b2a357c73c

Software `/` code `/` prosody-modules