-- Copyright (C) 2016-2017 Thilo Molitor

--

-- This project is MIT/X11 licensed. Please see the

-- COPYING file in the source package for more information.

--

local add_filter = require "util.filters".add_filter;

local remove_filter = require "util.filters".remove_filter;

local datetime = require "util.datetime";

local xmlns_delay = "urn:xmpp:delay";

-- Raise an error if the modules has been loaded as a component in prosody's config

if module:get_host_type() == "component" then

	error(module.name.." should NOT be loaded as a component, check out http://prosody.im/doc/components", 0);

end

local add_delay = function(stanza, session)

	if stanza and stanza.name == "message" and stanza:get_child("delay", xmlns_delay) == nil then

		-- only add delay tag to chat or groupchat messages (should we add a delay to anything else, too???)

		if stanza.attr.type == "chat" or stanza.attr.type == "groupchat" then

			-- session.log("debug", "adding delay to message %s", tostring(stanza));

			stanza = stanza:tag("delay", { xmlns = xmlns_delay, from = session.host, stamp = datetime.datetime()});

		end

	end

	return stanza;

end

module:hook("resource-bind", function(event)

	add_filter(event.session, "stanzas/in", add_delay, 1);

end);

module:hook("smacks-hibernation-end", function(event)

	-- older smacks module versions send only the "intermediate" session in event.session and no session.resumed one

	if event.resumed then

		add_filter(event.resumed, "stanzas/in", add_delay, 1);

	end

end);

module:hook("pre-resource-unbind", function (event)

	remove_filter(event.session, "stanzas/in", add_delay);

end);