mirror of
https://github.com/SpiceBot/SpiceBot.git
synced 2026-11-10 21:50:59 -05:00
294 lines
12 KiB
Python
294 lines
12 KiB
Python
# coding=utf8
|
|
from __future__ import unicode_literals, absolute_import, division, print_function
|
|
"""
|
|
This is the SpiceBot Prerun system.
|
|
"""
|
|
|
|
from sopel.tools import Identifier
|
|
|
|
import functools
|
|
import copy
|
|
import datetime
|
|
import spicemanip
|
|
|
|
from .Tools import command_permissions_check
|
|
from .Commands import commands
|
|
from .Database import db as spicedb
|
|
|
|
|
|
class BotPrerun():
|
|
def __init__(self):
|
|
self.dict = {}
|
|
|
|
"""Command Splitting"""
|
|
|
|
def prerun(self, trigger_command_type='module'):
|
|
def actual_decorator(function):
|
|
|
|
@functools.wraps(function)
|
|
def internal_prerun(bot, trigger, *args, **kwargs):
|
|
|
|
# Primary command used for trigger, and a list of all words
|
|
trigger_args, trigger_command = self.trigger_args(trigger.args[1], trigger_command_type)
|
|
|
|
# Argsdict Defaults
|
|
argsdict_default = {}
|
|
argsdict_default["type"] = trigger_command_type
|
|
argsdict_default["com"] = trigger_command
|
|
|
|
realcom = commands.get_realcom(argsdict_default["com"], trigger_command_type)
|
|
argsdict_default["realcom"] = realcom
|
|
|
|
# split into && groupings
|
|
and_split = self.trigger_and_split(trigger_args)
|
|
|
|
# Create dict listings for trigger.sb
|
|
argsdict_list = self.trigger_argsdict_list(argsdict_default, and_split)
|
|
|
|
# Run the function for all splits
|
|
for argsdict in argsdict_list:
|
|
trigger.sb = copy.deepcopy(argsdict)
|
|
|
|
trigger.sb["args"], trigger.sb["hyphen_arg"] = self.trigger_hyphen_args(trigger.sb["args"])
|
|
if not trigger.sb["hyphen_arg"]:
|
|
# check if anything prohibits the nick from running the command
|
|
if self.trigger_runstatus(bot, trigger):
|
|
function(bot, trigger, *args, **kwargs)
|
|
else:
|
|
self.trigger_hyphen_arg_handler(bot, trigger)
|
|
return
|
|
|
|
return internal_prerun
|
|
return actual_decorator
|
|
|
|
def trigger_args(self, triggerargs_one, trigger_command_type='module'):
|
|
trigger_args = spicemanip.main(triggerargs_one, 'create')
|
|
if trigger_command_type in ['nickname']:
|
|
trigger_command = spicemanip.main(trigger_args, 2).lower()
|
|
trigger_args = spicemanip.main(trigger_args, '3+', 'list')
|
|
else:
|
|
trigger_command = spicemanip.main(trigger_args, 1).lower()[1:]
|
|
trigger_args = spicemanip.main(trigger_args, '2+', 'list')
|
|
return trigger_args, trigger_command
|
|
|
|
def trigger_and_split(self, trigger_args):
|
|
trigger_args_list_split = spicemanip.main(trigger_args, "split_&&")
|
|
if not len(trigger_args_list_split):
|
|
trigger_args_list_split.append([])
|
|
return trigger_args_list_split
|
|
|
|
def trigger_argsdict_list(self, argsdict_default, and_split):
|
|
prerun_split = []
|
|
for trigger_args_part in and_split:
|
|
argsdict_part = copy.deepcopy(argsdict_default)
|
|
argsdict_part["args"] = spicemanip.main(trigger_args_part, 'create')
|
|
if len(argsdict_part["args"]) and argsdict_part["args"][0] == "-a":
|
|
argsdict_part["adminswitch"] = True
|
|
argsdict_part["args"] = spicemanip.main(argsdict_part["args"], '2+', 'list')
|
|
else:
|
|
argsdict_part["adminswitch"] = False
|
|
prerun_split.append(argsdict_part)
|
|
return prerun_split
|
|
|
|
def trigger_runstatus(self, bot, trigger):
|
|
|
|
# Bots can't run commands
|
|
if Identifier(trigger.nick) == bot.nick:
|
|
return False
|
|
|
|
# Allow permissions for enabling and disabling commands via hyphenargs
|
|
if trigger.sb["adminswitch"]:
|
|
if command_permissions_check(bot, trigger, ['admins', 'owner', 'OP', 'ADMIN', 'OWNER']):
|
|
return True
|
|
else:
|
|
bot.notice("The admin switch (-a) is for use by authorized nicks ONLY.")
|
|
return False
|
|
|
|
# don't run commands that are disabled in channels
|
|
if not trigger.is_privmsg:
|
|
disabled_list = spicedb.get_channel_value(bot, trigger.sender, 'disabled_commands', 'commands') or {}
|
|
if trigger.sb["realcom"] in disabled_list.keys():
|
|
reason = disabled_list[trigger.sb["realcom"]]["reason"]
|
|
timestamp = disabled_list[trigger.sb["realcom"]]["timestamp"]
|
|
bywhom = disabled_list[trigger.sb["realcom"]]["disabledby"]
|
|
message = "The " + str(trigger.sb["com"]) + " command was disabled by " + bywhom + " in " + str(trigger.sender) + " at " + str(timestamp) + " for the following reason: " + str(reason)
|
|
return self.trigger_cant_run(bot, trigger, message)
|
|
|
|
return True
|
|
|
|
def trigger_cant_run(self, bot, trigger, message=None):
|
|
if message:
|
|
bot.notice(message, trigger.nick)
|
|
if command_permissions_check(bot, trigger, ['admins', 'owner', 'OP', 'ADMIN', 'OWNER']):
|
|
bot.notice("You however are authorized to bypass this warning with the (-a) admin switch.", trigger.nick)
|
|
return False
|
|
|
|
def trigger_hyphen_args(self, trigger_args_part):
|
|
valid_hyphen_args = [
|
|
'check',
|
|
'enable', 'disable'
|
|
]
|
|
hyphen_args = []
|
|
trigger_args_unhyphend = []
|
|
for worditem in trigger_args_part:
|
|
if str(worditem).startswith("--") and worditem[2:] in valid_hyphen_args:
|
|
hyphen_args.append(worditem[2:])
|
|
else:
|
|
trigger_args_unhyphend.append(worditem)
|
|
if len(hyphen_args):
|
|
hyphenarg = hyphen_args[0]
|
|
else:
|
|
hyphenarg = None
|
|
return trigger_args_unhyphend, hyphenarg
|
|
|
|
def trigger_hyphen_arg_handler(self, bot, trigger):
|
|
|
|
# Commands that cannot run via privmsg
|
|
if trigger.sb["hyphen_arg"] in ['check']:
|
|
if trigger.sb["com"].lower() != trigger.sb["realcom"]:
|
|
bot.say(trigger.sb["com"] + " is a valid alias command for " + trigger.sb["realcom"])
|
|
else:
|
|
bot.say(trigger.sb["com"] + " is a valid command")
|
|
return
|
|
|
|
elif trigger.sb["hyphen_arg"] in ['enable']:
|
|
|
|
if not command_permissions_check(bot, trigger, ['admins', 'owner', 'OP', 'ADMIN', 'OWNER']):
|
|
bot.say("I was unable to disable this command due to privilege issues.")
|
|
return
|
|
|
|
if trigger.is_privmsg:
|
|
bot.notice("This command must be run in a channel you which to enable it in.", trigger.nick)
|
|
return
|
|
|
|
if not commands.check_disabled_commands(bot, trigger.sb["realcom"], trigger.sender):
|
|
bot.notice(trigger.sb["com"] + " is already enabled in " + str(trigger.sender), trigger.nick)
|
|
return
|
|
|
|
commands.unset_command_disabled(bot, trigger.sb["realcom"], trigger.sender)
|
|
bot.say(trigger.sb["com"] + " is now enabled in " + str(trigger.sender))
|
|
return
|
|
|
|
elif trigger.sb["hyphen_arg"] in ['disable']:
|
|
|
|
if not command_permissions_check(bot, trigger, ['admins', 'owner', 'OP', 'ADMIN', 'OWNER']):
|
|
bot.say("I was unable to disable this command due to privilege issues.")
|
|
return
|
|
|
|
if trigger.is_privmsg:
|
|
bot.notice("This command must be run in a channel you which to disable it in.", trigger.nick)
|
|
return
|
|
|
|
if commands.check_disabled_commands(bot, trigger.sb["realcom"], trigger.sender):
|
|
bot.notice(trigger.sb["com"] + " is already disabled in " + str(trigger.sender), trigger.nick)
|
|
return
|
|
|
|
trailingmessage = spicemanip.main(trigger.sb["args"], 0) or "No reason given."
|
|
timestamp = str(datetime.datetime.utcnow())
|
|
|
|
commands.set_command_disabled(bot, trigger.sb["realcom"], trigger.sender, timestamp, trailingmessage, trigger.nick)
|
|
bot.say(trigger.sb["com"] + " is now disabled in " + str(trigger.sender))
|
|
return
|
|
|
|
return
|
|
|
|
|
|
prerun = BotPrerun()
|
|
|
|
|
|
class defunct():
|
|
"""Hyphenargs"""
|
|
|
|
def hyphenargs(self):
|
|
def actual_decorator(function):
|
|
@functools.wraps(function)
|
|
def internal_prerun(bot, trigger, *args, **kwargs):
|
|
trigger.sb["args"], trigger.sb["hyphen_arg"] = self.trigger_hyphen_args(trigger.sb["args"])
|
|
if not trigger.sb["hyphen_arg"]:
|
|
return function(bot, trigger, *args, **kwargs)
|
|
else:
|
|
return self.trigger_hyphen_arg_handler(bot, trigger)
|
|
return internal_prerun
|
|
return actual_decorator
|
|
|
|
def trigger_hyphen_args(self, trigger_args_part):
|
|
valid_hyphen_args = [
|
|
'check',
|
|
'enable', 'disable'
|
|
]
|
|
hyphen_args = []
|
|
trigger_args_unhyphend = []
|
|
for worditem in trigger_args_part:
|
|
if str(worditem).startswith("--") and worditem[2:] in valid_hyphen_args:
|
|
hyphen_args.append(worditem[2:])
|
|
else:
|
|
trigger_args_unhyphend.append(worditem)
|
|
if len(hyphen_args):
|
|
hyphenarg = hyphen_args[0]
|
|
else:
|
|
hyphenarg = None
|
|
return trigger_args_unhyphend, hyphenarg
|
|
|
|
def trigger_hyphen_arg_handler(self, bot, trigger):
|
|
|
|
# Commands that cannot run via privmsg
|
|
if trigger.sb["hyphen_arg"] in ['check']:
|
|
if trigger.sb["com"].lower() != trigger.sb["realcom"]:
|
|
bot.say(trigger.sb["com"] + " is a valid alias command for " + trigger.sb["realcom"])
|
|
else:
|
|
bot.say(trigger.sb["com"] + " is a valid command")
|
|
return
|
|
|
|
elif trigger.sb["hyphen_arg"] in ['enable']:
|
|
|
|
if not command_permissions_check(bot, trigger, ['admins', 'owner', 'OP', 'ADMIN', 'OWNER']):
|
|
bot.say("I was unable to disable this command due to privilege issues.")
|
|
return
|
|
|
|
if trigger.is_privmsg:
|
|
bot.notice("This command must be run in a channel you which to enable it in.", trigger.nick)
|
|
return
|
|
|
|
if not commands.check_disabled_commands(bot, trigger.sb["realcom"], trigger.sender):
|
|
bot.notice(trigger.sb["com"] + " is already enabled in " + str(trigger.sender), trigger.nick)
|
|
return
|
|
|
|
commands.unset_command_disabled(bot, trigger.sb["realcom"], trigger.sender)
|
|
bot.say(trigger.sb["com"] + " is now enabled in " + str(trigger.sender))
|
|
return
|
|
|
|
elif trigger.sb["hyphen_arg"] in ['disable']:
|
|
|
|
if not command_permissions_check(bot, trigger, ['admins', 'owner', 'OP', 'ADMIN', 'OWNER']):
|
|
bot.say("I was unable to disable this command due to privilege issues.")
|
|
return
|
|
|
|
if trigger.is_privmsg:
|
|
bot.notice("This command must be run in a channel you which to disable it in.", trigger.nick)
|
|
return
|
|
|
|
if commands.check_disabled_commands(bot, trigger.sb["realcom"], trigger.sender):
|
|
bot.notice(trigger.sb["com"] + " is already disabled in " + str(trigger.sender), trigger.nick)
|
|
return
|
|
|
|
trailingmessage = spicemanip.main(trigger.sb["args"], 0) or "No reason given."
|
|
timestamp = str(datetime.datetime.utcnow())
|
|
|
|
commands.set_command_disabled(bot, trigger.sb["realcom"], trigger.sender, timestamp, trailingmessage, trigger.nick)
|
|
bot.say(trigger.sb["com"] + " is now disabled in " + str(trigger.sender))
|
|
return
|
|
|
|
return
|
|
|
|
|
|
def require_owner():
|
|
def actual_decorator(function):
|
|
@functools.wraps(function)
|
|
def guarded(bot, trigger, *args, **kwargs):
|
|
if not trigger.owner:
|
|
bot.say(trigger.nick + " is not the owner")
|
|
else:
|
|
return function(bot, trigger, *args, **kwargs)
|
|
return guarded
|
|
return actual_decorator
|