67 lines
1.8 KiB
Python
67 lines
1.8 KiB
Python
from util import hook, bucket
|
|
|
|
TOKENS = 10
|
|
RESTORE_RATE = 2
|
|
MESSAGE_COST = 5
|
|
|
|
|
|
buckets = {}
|
|
|
|
@hook.sieve
|
|
def sieve_suite(bot, input, func, kind, args):
|
|
conn = input.conn
|
|
if input.command == 'PRIVMSG' and \
|
|
input.nick.endswith('bot') and args.get('ignorebots', True):
|
|
return None
|
|
|
|
if kind == "command":
|
|
uid = (input.nick, input.chan)
|
|
|
|
if not uid in buckets:
|
|
_bucket = bucket.TokenBucket(TOKENS, RESTORE_RATE)
|
|
_bucket.consume(MESSAGE_COST)
|
|
buckets[uid] = _bucket
|
|
return input
|
|
|
|
_bucket = buckets[uid]
|
|
if _bucket.consume(MESSAGE_COST):
|
|
pass
|
|
else:
|
|
print("pong!")
|
|
return None
|
|
|
|
disabled_commands = conn.config.get('disabled_commands', [])
|
|
if input.trigger in disabled_commands:
|
|
return None
|
|
|
|
return input
|
|
|
|
|
|
acl = conn.config.get('acls', {}).get(func.__name__)
|
|
if acl:
|
|
if 'deny-except' in acl:
|
|
allowed_channels = list(map(str.lower, acl['deny-except']))
|
|
if input.chan.lower() not in allowed_channels:
|
|
return None
|
|
if 'allow-except' in acl:
|
|
denied_channels = list(map(str.lower, acl['allow-except']))
|
|
if input.chan.lower() in denied_channels:
|
|
return None
|
|
|
|
# shim so plugins using the old "adminonly" permissions format still work
|
|
if args.get('adminonly', False):
|
|
args["permissions"] = ["adminonly"]
|
|
|
|
if args.get('permissions', False):
|
|
mask = input.mask.lower()
|
|
|
|
allowed_permissions = args.get('permissions', [])
|
|
for perm in allowed_permissions:
|
|
if conn.permissions.has_perm_mask(mask, perm):
|
|
return input
|
|
|
|
input.notice("Sorry, you are not allowed to use this command.")
|
|
return None
|
|
|
|
return input
|