mod_smacks_offline/mod_smacks_offline.lua
author Kim Alvefur <zash@zash.se>
Wed, 06 May 2015 00:55:00 +0200
changeset 1706 725ebd494000
child 1735 b912cb8e0b3c
permissions -rw-r--r--
mod_smacks_offline: Monkeypatch mod_smacks to send unacked messages to offline storage IFF there was only one session
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
1706
725ebd494000 mod_smacks_offline: Monkeypatch mod_smacks to send unacked messages to offline storage IFF there was only one session
Kim Alvefur <zash@zash.se>
parents:
diff changeset
     1
local t_insert = table.insert;
725ebd494000 mod_smacks_offline: Monkeypatch mod_smacks to send unacked messages to offline storage IFF there was only one session
Kim Alvefur <zash@zash.se>
parents:
diff changeset
     2
725ebd494000 mod_smacks_offline: Monkeypatch mod_smacks to send unacked messages to offline storage IFF there was only one session
Kim Alvefur <zash@zash.se>
parents:
diff changeset
     3
local mod_smacks = module:depends"smacks"
725ebd494000 mod_smacks_offline: Monkeypatch mod_smacks to send unacked messages to offline storage IFF there was only one session
Kim Alvefur <zash@zash.se>
parents:
diff changeset
     4
725ebd494000 mod_smacks_offline: Monkeypatch mod_smacks to send unacked messages to offline storage IFF there was only one session
Kim Alvefur <zash@zash.se>
parents:
diff changeset
     5
local function store_unacked_stanzas(session)
725ebd494000 mod_smacks_offline: Monkeypatch mod_smacks to send unacked messages to offline storage IFF there was only one session
Kim Alvefur <zash@zash.se>
parents:
diff changeset
     6
	local queue = session.outgoing_stanza_queue;
725ebd494000 mod_smacks_offline: Monkeypatch mod_smacks to send unacked messages to offline storage IFF there was only one session
Kim Alvefur <zash@zash.se>
parents:
diff changeset
     7
	local replacement_queue = {};
725ebd494000 mod_smacks_offline: Monkeypatch mod_smacks to send unacked messages to offline storage IFF there was only one session
Kim Alvefur <zash@zash.se>
parents:
diff changeset
     8
	session.outgoing_stanza_queue = replacement_queue;
725ebd494000 mod_smacks_offline: Monkeypatch mod_smacks to send unacked messages to offline storage IFF there was only one session
Kim Alvefur <zash@zash.se>
parents:
diff changeset
     9
725ebd494000 mod_smacks_offline: Monkeypatch mod_smacks to send unacked messages to offline storage IFF there was only one session
Kim Alvefur <zash@zash.se>
parents:
diff changeset
    10
	for _, stanza in ipairs(queue) do
725ebd494000 mod_smacks_offline: Monkeypatch mod_smacks to send unacked messages to offline storage IFF there was only one session
Kim Alvefur <zash@zash.se>
parents:
diff changeset
    11
		if stanza.name == "message" and stanza.attr.xmlns == nil and
725ebd494000 mod_smacks_offline: Monkeypatch mod_smacks to send unacked messages to offline storage IFF there was only one session
Kim Alvefur <zash@zash.se>
parents:
diff changeset
    12
				( stanza.attr.type == "chat" or ( stanza.attr.type or "normal" ) == "normal" ) then
725ebd494000 mod_smacks_offline: Monkeypatch mod_smacks to send unacked messages to offline storage IFF there was only one session
Kim Alvefur <zash@zash.se>
parents:
diff changeset
    13
			module:fire_event("message/offline/handle", { origin = session, stanza = stanza } )
725ebd494000 mod_smacks_offline: Monkeypatch mod_smacks to send unacked messages to offline storage IFF there was only one session
Kim Alvefur <zash@zash.se>
parents:
diff changeset
    14
		else
725ebd494000 mod_smacks_offline: Monkeypatch mod_smacks to send unacked messages to offline storage IFF there was only one session
Kim Alvefur <zash@zash.se>
parents:
diff changeset
    15
			t_insert(replacement_queue, stanza);
725ebd494000 mod_smacks_offline: Monkeypatch mod_smacks to send unacked messages to offline storage IFF there was only one session
Kim Alvefur <zash@zash.se>
parents:
diff changeset
    16
		end
725ebd494000 mod_smacks_offline: Monkeypatch mod_smacks to send unacked messages to offline storage IFF there was only one session
Kim Alvefur <zash@zash.se>
parents:
diff changeset
    17
	end
725ebd494000 mod_smacks_offline: Monkeypatch mod_smacks to send unacked messages to offline storage IFF there was only one session
Kim Alvefur <zash@zash.se>
parents:
diff changeset
    18
end
725ebd494000 mod_smacks_offline: Monkeypatch mod_smacks to send unacked messages to offline storage IFF there was only one session
Kim Alvefur <zash@zash.se>
parents:
diff changeset
    19
725ebd494000 mod_smacks_offline: Monkeypatch mod_smacks to send unacked messages to offline storage IFF there was only one session
Kim Alvefur <zash@zash.se>
parents:
diff changeset
    20
local handle_unacked_stanzas = mod_smacks.handle_unacked_stanzas;
725ebd494000 mod_smacks_offline: Monkeypatch mod_smacks to send unacked messages to offline storage IFF there was only one session
Kim Alvefur <zash@zash.se>
parents:
diff changeset
    21
725ebd494000 mod_smacks_offline: Monkeypatch mod_smacks to send unacked messages to offline storage IFF there was only one session
Kim Alvefur <zash@zash.se>
parents:
diff changeset
    22
local host_sessions = prosody.hosts[module.host].sessions;
725ebd494000 mod_smacks_offline: Monkeypatch mod_smacks to send unacked messages to offline storage IFF there was only one session
Kim Alvefur <zash@zash.se>
parents:
diff changeset
    23
mod_smacks.handle_unacked_stanzas = function (session)
725ebd494000 mod_smacks_offline: Monkeypatch mod_smacks to send unacked messages to offline storage IFF there was only one session
Kim Alvefur <zash@zash.se>
parents:
diff changeset
    24
	local sessions = host_sessions[session.username].sessions;
725ebd494000 mod_smacks_offline: Monkeypatch mod_smacks to send unacked messages to offline storage IFF there was only one session
Kim Alvefur <zash@zash.se>
parents:
diff changeset
    25
	if next(sessions) == session.resource and next(sessions, session.resource) == nil then
725ebd494000 mod_smacks_offline: Monkeypatch mod_smacks to send unacked messages to offline storage IFF there was only one session
Kim Alvefur <zash@zash.se>
parents:
diff changeset
    26
		store_unacked_stanzas(session)
725ebd494000 mod_smacks_offline: Monkeypatch mod_smacks to send unacked messages to offline storage IFF there was only one session
Kim Alvefur <zash@zash.se>
parents:
diff changeset
    27
	end
725ebd494000 mod_smacks_offline: Monkeypatch mod_smacks to send unacked messages to offline storage IFF there was only one session
Kim Alvefur <zash@zash.se>
parents:
diff changeset
    28
	return handle_unacked_stanzas(session);
725ebd494000 mod_smacks_offline: Monkeypatch mod_smacks to send unacked messages to offline storage IFF there was only one session
Kim Alvefur <zash@zash.se>
parents:
diff changeset
    29
end
725ebd494000 mod_smacks_offline: Monkeypatch mod_smacks to send unacked messages to offline storage IFF there was only one session
Kim Alvefur <zash@zash.se>
parents:
diff changeset
    30
725ebd494000 mod_smacks_offline: Monkeypatch mod_smacks to send unacked messages to offline storage IFF there was only one session
Kim Alvefur <zash@zash.se>
parents:
diff changeset
    31
function module.unload()
725ebd494000 mod_smacks_offline: Monkeypatch mod_smacks to send unacked messages to offline storage IFF there was only one session
Kim Alvefur <zash@zash.se>
parents:
diff changeset
    32
	mod_smacks.handle_unacked_stanzas = handle_unacked_stanzas;
725ebd494000 mod_smacks_offline: Monkeypatch mod_smacks to send unacked messages to offline storage IFF there was only one session
Kim Alvefur <zash@zash.se>
parents:
diff changeset
    33
end