<commit_before>local st, jid, set = require "util.stanza", require "util.jid", require "util.set";

local admins = set.new(config.get(module:get_host(), "core", "admins"));

function handle_announcement(data)
	local origin, stanza = data.origin, data.stanza;
	local host, resource = select(2, jid.split(stanza.attr.to));
	
	if resource ~= "announce/online" then
		return; -- Not an announcement
	end
	
	if not admins:contains(jid.bare(origin.full_jid)) then
		-- Not an admin? Not allowed!
		module:log("warn", "Non-admin %s tried to send server announcement", tostring(jid.bare(origin.full_jid)));
		origin.send(st.error_reply(stanza, "cancel", "service-unavailable"));
		return;
	end
	
	module:log("info", "Sending server announcement to all online users");
	local host_session = hosts[host];
	local message = st.clone(stanza);
	message.attr.type = "headline";
	message.attr.from = host;
	
	local c = 0;
	for user in pairs(host_session.sessions) do
		c = c + 1;
		message.attr.to = user.."@"..host;
		core_post_stanza(host_session, message);
	end
	
	module:log("info", "Announcement sent to %d online users", c);
	return true;
end

module:hook("message/host", handle_announcement);<commit_msg>mod_announce: Use usermanager.is_admin to verify admin status<commit_after>local st, jid, set = require "util.stanza", require "util.jid", require "util.set";

local is_admin = require "core.usermanager".is_admin;
local admins = set.new(config.get(module:get_host(), "core", "admins"));

function handle_announcement(data)
	local origin, stanza = data.origin, data.stanza;
	local host, resource = select(2, jid.split(stanza.attr.to));
	
	if resource ~= "announce/online" then
		return; -- Not an announcement
	end
	
	if not is_admin(origin.full_jid) then
		-- Not an admin? Not allowed!
		module:log("warn", "Non-admin %s tried to send server announcement", tostring(jid.bare(origin.full_jid)));
		origin.send(st.error_reply(stanza, "cancel", "service-unavailable"));
		return;
	end
	
	module:log("info", "Sending server announcement to all online users");
	local host_session = hosts[host];
	local message = st.clone(stanza);
	message.attr.type = "headline";
	message.attr.from = host;
	
	local c = 0;
	for user in pairs(host_session.sessions) do
		c = c + 1;
		message.attr.to = user.."@"..host;
		core_post_stanza(host_session, message);
	end
	
	module:log("info", "Announcement sent to %d online users", c);
	return true;
end

module:hook("message/host", handle_announcement);
{'fraction_non_alphanumeric': 0.14464646464646463, 'fraction_numerical': 0.0024242424242424242, 'mean_word_length': 4.951923076923077, 'pattern_counts': {'":': 0, '<': 3, '<?xml version=': 0, '>': 3, 'https://': 0, 'lorem ipsum': 0, 'www.': 0, 'xml': 0}, 'pii_count': 0, 'substrings_counts': 0, 'word_list_counts': {'cursed_substrings.json': 0, 'profanity_word_list.json': 0, 'sexual_word_list.json': 0, 'zh_pornsignals.json': 0}}
{'dir': 'git-commits-cleaned', 'id': '5685691', 'n_tokens_mistral': 885, 'n_tokens_neox': 838, 'n_words': 221}