64 lines
2.4 KiB
Lua
64 lines
2.4 KiB
Lua
-- SPDX-FileCopyrightText: 2024 John Livingston <https://www.john-livingston.fr/>
|
|
-- SPDX-License-Identifier: AGPL-3.0-only
|
|
|
|
local st = require "util.stanza";
|
|
local path = require "util.paths";
|
|
local b64 = require "util.encodings".base64.encode;
|
|
local jid = require "util.jid";
|
|
|
|
module:add_feature("vcard-temp");
|
|
|
|
local avatars_dir = assert(module:get_option_string("peertubelivechat_random_vcard_avatars_path", nil), "'peertubelivechat_random_vcard_avatars_path' is a required option");
|
|
local avatars_files = assert(module:get_option_array("peertubelivechat_random_vcard_avatars_files", nil), "'peertubelivechat_random_vcard_avatars_files' is a required option");
|
|
local avatars = {};
|
|
local function load_avatar(filename)
|
|
local file = assert(io.open(path.join(avatars_dir, filename), "r"));
|
|
-- FIXME: check filetype, to only allow jpg or png. (for the day we would allow users to add custom avatars sets)
|
|
local filetype = filename.sub(filename, -3); -- jpg or png
|
|
local result = {
|
|
type = 'image/' .. filetype,
|
|
content = b64(file:read("*a"))
|
|
};
|
|
file:close();
|
|
return result;
|
|
end
|
|
local AVATARS_COUNT = 0;
|
|
for _, filename in pairs(avatars_files) do
|
|
AVATARS_COUNT = AVATARS_COUNT + 1;
|
|
avatars[AVATARS_COUNT] = load_avatar(filename);
|
|
end
|
|
module:log("info", "Loaded " .. AVATARS_COUNT .. ' avatars for host ' .. module:get_host() .. '.' );
|
|
|
|
module:hook("iq-get/bare/vcard-temp:vCard", function (event)
|
|
local origin, stanza = event.origin, event.stanza;
|
|
|
|
-- module:log("debug", "From '%s', To '%s', username '%s'.", stanza.attr.from, stanza.attr.to, origin.username);
|
|
|
|
local who = jid.split(stanza.attr.to) or origin.username;
|
|
if not who then
|
|
module:log("debug", "No who, will not generate random vCard");
|
|
origin.send(st.error_reply(stanza, "cancel", "item-not-found"));
|
|
return true;
|
|
end
|
|
|
|
module:log("debug", "random vCard request for %s", who);
|
|
|
|
local n = 1;
|
|
for c in who:gmatch"." do
|
|
n = math.fmod(n + c:byte(), AVATARS_COUNT) + 1;
|
|
end
|
|
|
|
module:log("debug", "%s will have the avatar number %s.", who, n);
|
|
|
|
local vcard_temp = st.stanza("vCard", { xmlns = "vcard-temp" });
|
|
-- vcard_temp:text_tag("FN", who);
|
|
-- vcard_temp:text_tag("NICKNAME", who);
|
|
vcard_temp:tag("PHOTO");
|
|
vcard_temp:text_tag("TYPE", avatars[n].type);
|
|
vcard_temp:text_tag("BINVAL", avatars[n].content);
|
|
vcard_temp:up();
|
|
|
|
origin.send(st.reply(stanza):add_child(vcard_temp));
|
|
return true;
|
|
end);
|