module:depends("audit");
-- luacheck: read globals module.audit

local jid = require "util.jid";
local st = require "util.stanza";

module:hook("user-registered", function(event)
	local session = event.session;
	local custom = {};
	local invite = event.validated_invite or (event.session and event.session.validated_invite);
	if invite then
		table.insert(custom, st.stanza(
			"invite-used",
			{
				xmlns = "xmpp:prosody.im/audit",
				token = invite.token,
			}
		))
	end
	module:audit(jid.join(event.username, module.host), "user-registered", {
		session = session,
		custom = custom,
	});
end);