Page MenuHomePhabricator (Chris)
Files F626494

whitelistbot.lua
No OneTemporary
Actions

Authored By
Unknown
Size
2 KB
Referenced Files
None
Subscribers
None

whitelistbot.lua
View Options

--[[
Copyright 2020 Matthew Hesketh <matthew@matthewhesketh.com>
This code is licensed under the MIT. See LICENSE for details.
]]
local allowbot = {}
local mattata = require('mattata')
local redis = require('libs.redis')
function allowbot:init()
allowbot.commands = mattata.commands(self.info.username):command('allowbot'):command('allowbots'):command('wb').table
allowbot.help = '/allowbot <bots> - Allowlists the given bots in the current chat. Requires administrative privileges. Aliases: /allowbots, /wb.'
allowbot.example_bots = { 'gif', 'imdb', 'wiki', 'music', 'youtube', 'bold', 'sticker', 'vote', 'like', 'gamee', 'coub', 'pic', 'vid', 'bing' }
end
function allowbot:on_new_message(message, configuration, language)
if message.chat.type ~= 'supergroup' then
return false
elseif mattata.get_setting(message.chat.id, 'prevent inline bots') and message.via_bot and not mattata.is_group_admin(message.chat.id, message.from.id) then
return mattata.delete_message(message.chat.id, message.message_id)
end
end
function allowbot:on_message(message, configuration, language)
if not mattata.is_group_admin(message.chat.id, message.from.id) then
return mattata.send_reply(message, language.errors.admin)
end
local input = mattata.input(message.text)
if not input then
return mattata.send_reply(message, 'Please specify the @usernames of the bots you\'d like to allowlist.')
elseif not input:match('@?[%w_]') then
return mattata.send_reply(message, 'Please make sure you\'re specifying valid bot usernames!')
end
local bots = {}
for bot in input:gmatch('@?([%w_]+bot)') do
table.insert(bots, bot)
end
for _, bot in pairs(allowbot.example_bots) do
if input:match('@?' .. bot) then
table.insert(bots, bot)
end
end
if #bots == 0 then
return mattata.send_reply(message, 'Please make sure you\'re specifying valid bot usernames!')
end
for _, bot in pairs(bots) do
redis:sadd('allowlisted_bots:' .. message.chat.id, bot)
end
local output = string.format('Successfully allowlisted the following bots in this chat: %s', table.concat(bots, ', '))
return mattata.send_reply(message, output)
end
return allowbot

File Metadata

Mime Type
text/plain
Expires
Sun, Nov 15, 12:47 AM (1 d, 16 h)
Storage Engine
blob
Storage Format
Raw Data
Storage Handle
140541
Default Alt Text
whitelistbot.lua (2 KB)

Event Timeline