added manager plugin
This commit is contained in:
parent
c817c774b1
commit
f228fea149
|
|
@ -1,61 +0,0 @@
|
|||
-- Kick an user from the chat group.
|
||||
-- Use !kick name User_name or !kick id id_number
|
||||
-- The User_name is the print_name (there are no spaces but _)
|
||||
|
||||
--~ TODO:
|
||||
--~ - !kick all
|
||||
--~ Get a list of users and kick them all (Useful for delete groups)
|
||||
--~ - !kick new // !protect
|
||||
--~ Kick each new user who entered in the group
|
||||
--~ - !ban user#id
|
||||
--~ Blacklist an user to enter a group
|
||||
|
||||
do
|
||||
|
||||
function ban(usr,chat)
|
||||
print ("Trying to kick: "..usr.." to "..chat)
|
||||
local success = chat_del_user (chat, usr, ok_cb, false)
|
||||
if not success then
|
||||
return "An error happened"
|
||||
else
|
||||
local kicked = "Kicked user: "..usr.." from "..chat
|
||||
return kicked
|
||||
end
|
||||
end
|
||||
|
||||
function run(msg, matches)
|
||||
chat_ = 'chat#id'..msg.to.id
|
||||
-- The message must come from a chat group OR
|
||||
if msg.to.type ~= 'chat' then
|
||||
return 'This isn\'t a chat group!'
|
||||
end
|
||||
|
||||
-- User submitted a user name
|
||||
if matches[1] == "name" then
|
||||
user_ = matches[2]
|
||||
user_ = string.gsub(user_," ","_")
|
||||
ban(user_,chat_)
|
||||
-- User submitted an id
|
||||
elseif matches[1] == "id" then
|
||||
for i=2,#matches do
|
||||
user_ = 'user#id'..matches[i]
|
||||
ban(user_,chat_)
|
||||
end
|
||||
end
|
||||
user_ = nil
|
||||
chat_ = nil
|
||||
end
|
||||
|
||||
return {
|
||||
description = "Ban an user from the chat group. Credits: @Rutrus",
|
||||
usage = {
|
||||
"/kick name [user_name]",
|
||||
"/kick id [user_id]+" },
|
||||
patterns = {
|
||||
"^/kick (name) (.*)",
|
||||
"^/kick (id) (%d+)"
|
||||
},
|
||||
run = run,
|
||||
privileged = true
|
||||
}
|
||||
end
|
||||
|
|
@ -0,0 +1,212 @@
|
|||
local function is_user_whitelisted(id)
|
||||
local hash = 'whitelist:user#id'..id
|
||||
local white = redis:get(hash) or false
|
||||
return white
|
||||
end
|
||||
|
||||
local function is_chat_whitelisted(id)
|
||||
local hash = 'whitelist:chat#id'..id
|
||||
local white = redis:get(hash) or false
|
||||
return white
|
||||
end
|
||||
|
||||
local function kick_user(user_id, chat_id)
|
||||
local chat = 'chat#id'..chat_id
|
||||
local user = 'user#id'..user_id
|
||||
chat_del_user(chat, user, ok_cb, true)
|
||||
end
|
||||
|
||||
local function ban_user(user_id, chat_id)
|
||||
-- Save to redis
|
||||
local hash = 'banned:'..chat_id..':'..user_id
|
||||
redis:set(hash, true)
|
||||
-- Kick from chat
|
||||
kick_user(user_id, chat_id)
|
||||
end
|
||||
|
||||
local function is_banned(user_id, chat_id)
|
||||
local hash = 'banned:'..chat_id..':'..user_id
|
||||
local banned = redis:get(hash)
|
||||
return banned or false
|
||||
end
|
||||
|
||||
local function pre_process(msg)
|
||||
|
||||
-- SERVICE MESSAGE
|
||||
if msg.action and msg.action.type then
|
||||
local action = msg.action.type
|
||||
-- Check if banned user joins chat
|
||||
if action == 'chat_add_user' or action == 'chat_add_user_link' then
|
||||
local user_id
|
||||
if msg.action.link_issuer then
|
||||
user_id = msg.from.id
|
||||
else
|
||||
user_id = msg.action.user.id
|
||||
end
|
||||
print('Checking invited user '..user_id)
|
||||
local banned = is_banned(user_id, msg.to.id)
|
||||
if banned then
|
||||
print('User wurde gebannt!')
|
||||
kick_user(user_id, msg.to.id)
|
||||
end
|
||||
end
|
||||
-- No further checks
|
||||
return msg
|
||||
end
|
||||
|
||||
-- BANNED USER TALKING
|
||||
if msg.to.type == 'chat' then
|
||||
local user_id = msg.from.id
|
||||
local chat_id = msg.to.id
|
||||
local banned = is_banned(user_id, chat_id)
|
||||
if banned then
|
||||
print('Banned user talking!')
|
||||
ban_user(user_id, chat_id)
|
||||
msg.text = ''
|
||||
end
|
||||
end
|
||||
|
||||
-- WHITELIST
|
||||
local hash = 'whitelist:enabled'
|
||||
local whitelist = redis:get(hash)
|
||||
local issudo = is_sudo(msg)
|
||||
|
||||
-- Allow all sudo users even if whitelist is allowed
|
||||
if whitelist and not issudo then
|
||||
print('Whitelist enabled and not sudo')
|
||||
-- Check if user or chat is whitelisted
|
||||
local allowed = is_user_whitelisted(msg.from.id)
|
||||
|
||||
if not allowed then
|
||||
print('User '..msg.from.id..' not whitelisted')
|
||||
if msg.to.type == 'chat' then
|
||||
allowed = is_chat_whitelisted(msg.to.id)
|
||||
if not allowed then
|
||||
print ('Chat '..msg.to.id..' not whitelisted')
|
||||
else
|
||||
print ('Chat '..msg.to.id..' whitelisted :)')
|
||||
end
|
||||
end
|
||||
else
|
||||
print('User '..msg.from.id..' allowed :)')
|
||||
end
|
||||
|
||||
if not allowed then
|
||||
msg.text = ''
|
||||
end
|
||||
|
||||
else
|
||||
print('Whitelist not enabled or is sudo')
|
||||
end
|
||||
|
||||
return msg
|
||||
end
|
||||
|
||||
local function run(msg, matches)
|
||||
|
||||
-- Silent ignore
|
||||
if not is_sudo(msg) then
|
||||
return nil
|
||||
end
|
||||
|
||||
if matches[1] == 'ban' then
|
||||
local user_id = matches[3]
|
||||
local chat_id = msg.to.id
|
||||
|
||||
if msg.to.type == 'chat' then
|
||||
if matches[2] == 'user' then
|
||||
ban_user(user_id, chat_id)
|
||||
return 'User '..user_id..' banned'
|
||||
end
|
||||
if matches[2] == 'delete' then
|
||||
local hash = 'banned:'..chat_id..':'..user_id
|
||||
redis:del(hash)
|
||||
return 'User '..user_id..' unbanned'
|
||||
end
|
||||
else
|
||||
return 'This isn\'t a chat group'
|
||||
end
|
||||
end
|
||||
|
||||
if matches[1] == 'kick' then
|
||||
if msg.to.type == 'chat' then
|
||||
kick_user(matches[2], msg.to.id)
|
||||
else
|
||||
return 'This isn\'t a chat group'
|
||||
end
|
||||
end
|
||||
|
||||
if matches[1] == 'whitelist' then
|
||||
if matches[2] == 'enable' then
|
||||
local hash = 'whitelist:enabled'
|
||||
redis:set(hash, true)
|
||||
return 'Enabled whitelist'
|
||||
end
|
||||
|
||||
if matches[2] == 'disable' then
|
||||
local hash = 'whitelist:enabled'
|
||||
redis:del(hash)
|
||||
return 'Disabled whitelist'
|
||||
end
|
||||
|
||||
if matches[2] == 'user' then
|
||||
local hash = 'whitelist:user#id'..matches[3]
|
||||
redis:set(hash, true)
|
||||
return 'User "'..matches[3]..'" wurde freigeschalten'
|
||||
end
|
||||
|
||||
if matches[2] == 'chat' then
|
||||
if msg.to.type ~= 'chat' then
|
||||
return 'Das hier ist kein Chat!'
|
||||
end
|
||||
local hash = 'whitelist:chat#id'..msg.to.id
|
||||
redis:set(hash, true)
|
||||
return 'Chat "'..msg.to.id..'" wurde freigeschalten'
|
||||
end
|
||||
|
||||
if matches[2] == 'delete' and matches[3] == 'user' then
|
||||
local hash = 'whitelist:user#id'..matches[4]
|
||||
redis:del(hash)
|
||||
return 'User "'..matches[4]..'" removed from whitelist'
|
||||
end
|
||||
|
||||
if matches[2] == 'delete' and matches[3] == 'chat' then
|
||||
if msg.to.type ~= 'chat' then
|
||||
return 'This isn\'t a chat group'
|
||||
end
|
||||
local hash = 'whitelist:chat#id'..msg.to.id
|
||||
redis:del(hash)
|
||||
return 'Chat '..msg.to.id..' removed from whitelist'
|
||||
end
|
||||
|
||||
end
|
||||
end
|
||||
|
||||
return {
|
||||
description = "Plugin to manage bans, kicks and white/black lists.",
|
||||
usage = {
|
||||
"/whitelist <enable>/<disable>: Enable or disable whitelist mode",
|
||||
"/whitelist user <user_id>: Allow user to use the bot when whitelist mode is enabled",
|
||||
"/whitelist chat: Allow everybody on current chat to use the bot when whitelist mode is enabled",
|
||||
"/whitelist delete user <user_id>: Remove user from whitelist",
|
||||
"/whitelist delete chat: Remove chat from whitelist",
|
||||
"/ban user <user_id>: Kick user from chat and kicks it if joins chat again",
|
||||
"/ban delete <user_id>: Unban user",
|
||||
"/kick <user_id> Kick user from chat group"
|
||||
},
|
||||
patterns = {
|
||||
"^/(whitelist) (enable)$",
|
||||
"^/(whitelist) (disable)$",
|
||||
"^/(whitelist) (user) (%d+)$",
|
||||
"^/(whitelist) (chat)$",
|
||||
"^/(whitelist) (delete) (user) (%d+)$",
|
||||
"^/(whitelist) (delete) (chat)$",
|
||||
"^/(ban) (user) (%d+)$",
|
||||
"^/(ban) (delete) (%d+)$",
|
||||
"^/(kick) (%d+)$",
|
||||
"^//tgservice (.+)$",
|
||||
},
|
||||
run = run,
|
||||
pre_process = pre_process,
|
||||
privileged = true
|
||||
}
|
||||
Reference in New Issue