File

mod_muc_dicebot/mod_muc_dicebot.lua @ 5160:8474a3b80200

mod_firewall: Fix 'is_admin' internal dependency rule #1797 (thanks diane) Looks like the boolean logic was inverted here. Instead, for now, simply check if is_admin is there. It is deprecated in trunk and was briefly removed before being brought back with a 'deprecated' warning as part of the new roles and permissions work. Making this dependency conditioned on the existence of the underlying function should make it work until it actually goes away for real.
author Kim Alvefur <zash@zash.se>
date Fri, 27 Jan 2023 23:06:25 +0100
parent 4580:785389a2d2b3
line wrap: on
line source

local muc = module:depends("muc");
local rand = require"util.random";

local s_match = string.match;
local s_gmatch = string.gmatch;
local t_insert = table.insert;
local t_concat = table.concat;

local rooms = module:get_option_set("muc_dicebot_rooms", nil);
local xmlns_nick = "http://jabber.org/protocol/nick";

local function is_room_affected(roomjid)
	return not rooms or rooms:contains(roomjid)
end

local function roll(sides)
	if sides > 256 then
		return nil, "too many sides"
	end
	local factor = math.floor(256 / sides);
	local cutoff = sides * factor;
	module:log("error", "%d -> %d %d %d", sides, max, factor, cutoff);
	for i=1,10 do
		local randomness = string.byte(rand.bytes(1), 1);
		module:log("error", "%d", randomness);
		if randomness < cutoff then
			return (randomness % sides) + 1
		end
	end
	return nil, "failed to find valid number"
end

local function muc_broadcast_message(event)
	if not is_room_affected(event.room.jid) then
		return
	end

	local stanza = event.stanza;
	local body = stanza:get_child("body");
	if not body then
		return
	end

	local text = body:get_text();
	module:log("error", "%q %q %q", stanza, body, text);
	local dice = s_match(text, "^[%.!]r%s(.+)$");
	if not dice or dice == "" then
		return
	end

	local results = {};
	local count = 0;
	local sum = 0;
	for ndice, sep, sides in s_gmatch(dice, "(%d*)([wd]?)(%d+)") do
		if not sep or sep == "" then
			sides = ndice .. sides
			ndice = "1"
		end
		local ndice = tonumber(ndice);
		count = count + ndice;
		if count > 100 then
			return true
		end
		local sides = tonumber(sides);
		for i=1,ndice do
			local value = roll(sides);
			t_insert(results, tostring(value));
			sum = sum + value;
		end
	end
	body:text("\n⇒ "..t_concat(results, " ").." (sum: "..sum..")");
end

module:hook("muc-broadcast-message", muc_broadcast_message);