mirror of
https://github.com/SpiceBot/SpiceBot.git
synced 2026-11-10 21:50:59 -05:00
460 lines
22 KiB
Python
460 lines
22 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 as botcommands
|
|
from .Database import db as botdb
|
|
from .Channels import channels as botchannels
|
|
from .MessageLog import messagelog as botmessagelog
|
|
from .Config import config as botconfig
|
|
|
|
|
|
def prerun(t_command_type='module', t_command_subtype=None):
|
|
|
|
def actual_decorator(function):
|
|
|
|
@functools.wraps(function)
|
|
def internal_prerun(bot, trigger, *args, **kwargs):
|
|
|
|
if t_command_type == "nickname":
|
|
check_nick = spicemanip.main(trigger.args[1], 1).lower()
|
|
if check_nick != str(bot.nick).lower():
|
|
return
|
|
else:
|
|
if not str(trigger.args[1]).startswith(tuple(botconfig.core.prefix_list)):
|
|
return
|
|
|
|
trigger_command_type = str(t_command_type)
|
|
|
|
# Primary command used for trigger, and a list of all words
|
|
trigger_args, trigger_command, trigger_prefix = make_trigger_args(trigger.args[1], trigger_command_type)
|
|
|
|
if trigger_prefix == botconfig.SpiceBot_Commands.query_prefix:
|
|
return
|
|
|
|
trigger_command_type = botcommands.find_command_type(trigger_command)
|
|
if not trigger_command_type:
|
|
return
|
|
|
|
if t_command_subtype:
|
|
if trigger_command_type != t_command_subtype:
|
|
return
|
|
|
|
# Argsdict Defaults
|
|
argsdict_default = {}
|
|
argsdict_default["type"] = trigger_command_type
|
|
argsdict_default["com"] = trigger_command
|
|
|
|
# messagelog ID
|
|
argsdict_default["log_id"] = botmessagelog.messagelog_assign()
|
|
|
|
argsdict_default["realcom"] = botcommands.get_realcom(argsdict_default["com"], trigger_command_type)
|
|
|
|
if argsdict_default["type"] == 'nickname':
|
|
argsdict_default["comtext"] = "'" + bot.nick + " " + argsdict_default["com"] + "'"
|
|
argsdict_default["realcomtext"] = "'" + bot.nick + " " + argsdict_default["realcom"] + "'"
|
|
else:
|
|
argsdict_default["comtext"] = "'" + argsdict_default["com"] + "'"
|
|
argsdict_default["realcomtext"] = "'" + argsdict_default["realcom"] + "'"
|
|
|
|
argsdict_default["realcomref"] = argsdict_default["type"] + "_" + argsdict_default["realcom"]
|
|
|
|
argsdict_default["dict"] = botcommands.get_command_dict(argsdict_default["realcom"], trigger_command_type)
|
|
|
|
# split into && groupings
|
|
and_split = trigger_and_split(trigger_args)
|
|
|
|
# Create dict listings for trigger.sb
|
|
argsdict_list = trigger_argsdict_list(argsdict_default, and_split)
|
|
|
|
# Run the function for all splits
|
|
botmessagelog.messagelog_start(bot, trigger, argsdict_default["log_id"])
|
|
runcount = 0
|
|
for argsdict in argsdict_list:
|
|
runcount += 1
|
|
trigger.sb = copy.deepcopy(argsdict)
|
|
|
|
trigger.sb["runcount"] = runcount
|
|
|
|
trigger.sb["args"], trigger.sb["hyphen_arg"] = trigger_hyphen_args(trigger.sb["args"])
|
|
if not trigger.sb["hyphen_arg"]:
|
|
# check if anything prohibits the nick from running the command
|
|
if trigger_runstatus(bot, trigger):
|
|
function(bot, trigger, *args, **kwargs)
|
|
else:
|
|
trigger_hyphen_arg_handler(bot, trigger)
|
|
botmessagelog.messagelog_exit(bot, argsdict_default["log_id"])
|
|
|
|
return internal_prerun
|
|
return actual_decorator
|
|
|
|
|
|
def prerun_query(t_command_type='module', t_command_subtype=None):
|
|
|
|
def actual_decorator(function):
|
|
|
|
@functools.wraps(function)
|
|
def internal_prerun(bot, trigger, *args, **kwargs):
|
|
|
|
if t_command_type == "nickname":
|
|
check_nick = spicemanip.main(trigger.args[1], 1).lower()
|
|
if check_nick != str(bot.nick).lower():
|
|
return
|
|
else:
|
|
if not str(trigger.args[1]).startswith(tuple(botconfig.core.prefix_list)):
|
|
return
|
|
|
|
if t_command_subtype != t_command_type:
|
|
return
|
|
|
|
trigger_command_type = str(t_command_type)
|
|
|
|
# Primary command used for trigger, and a list of all words
|
|
trigger_args, trigger_command, trigger_prefix = make_trigger_args(trigger.args[1], trigger_command_type)
|
|
|
|
if trigger_prefix != botconfig.SpiceBot_Commands.query_prefix:
|
|
return
|
|
|
|
# Argsdict Defaults
|
|
argsdict_default = {}
|
|
argsdict_default["type"] = trigger_command_type
|
|
argsdict_default["com"] = trigger_command
|
|
|
|
# messagelog ID
|
|
argsdict_default["log_id"] = botmessagelog.messagelog_assign()
|
|
|
|
argsdict_default["realcom"] = "query_command"
|
|
|
|
argsdict_default["comtext"] = "'" + bot.nick + " query'"
|
|
argsdict_default["realcomtext"] = "'" + bot.nick + " query'"
|
|
|
|
argsdict_default["realcomref"] = "nickname_query"
|
|
|
|
argsdict_default["dict"] = botcommands.get_command_dict("query", 'nickname')
|
|
|
|
# split into && groupings
|
|
and_split = trigger_and_split(trigger_args)
|
|
|
|
# Create dict listings for trigger.sb
|
|
argsdict_list = trigger_argsdict_list(argsdict_default, and_split)
|
|
|
|
# Run the function for all splits
|
|
botmessagelog.messagelog_start(bot, trigger, argsdict_default["log_id"])
|
|
runcount = 0
|
|
for argsdict in argsdict_list:
|
|
runcount += 1
|
|
trigger.sb = copy.deepcopy(argsdict)
|
|
|
|
trigger.sb["runcount"] = runcount
|
|
|
|
trigger.sb["args"], trigger.sb["hyphen_arg"] = trigger_hyphen_args(trigger.sb["args"])
|
|
if not trigger.sb["hyphen_arg"]:
|
|
# check if anything prohibits the nick from running the command
|
|
if trigger_runstatus(bot, trigger):
|
|
function(bot, trigger, *args, **kwargs)
|
|
else:
|
|
trigger_hyphen_arg_handler(bot, trigger)
|
|
botmessagelog.messagelog_exit(bot, argsdict_default["log_id"])
|
|
|
|
return internal_prerun
|
|
return actual_decorator
|
|
|
|
|
|
def make_trigger_args(triggerargs_one, trigger_command_type='module'):
|
|
trigger_args = spicemanip.main(triggerargs_one, 'create')
|
|
if trigger_command_type in ['nickname']:
|
|
trigger_prefix = spicemanip.main(trigger_args, 2).lower()[0]
|
|
if trigger_prefix.isupper() or trigger_prefix.islower():
|
|
trigger_prefix = None
|
|
trigger_command = spicemanip.main(trigger_args, 2).lower()
|
|
trigger_args = spicemanip.main(trigger_args, '3+', 'list')
|
|
else:
|
|
trigger_prefix = spicemanip.main(trigger_args, 1).lower()[0]
|
|
trigger_command = spicemanip.main(trigger_args, 1).lower()[1:]
|
|
trigger_args = spicemanip.main(trigger_args, '2+', 'list')
|
|
return trigger_args, trigger_command, trigger_prefix
|
|
|
|
|
|
def trigger_and_split(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(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_argsdict_single(argsdict_default, trigger_args_part):
|
|
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
|
|
return argsdict_part
|
|
|
|
|
|
def trigger_runstatus(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:
|
|
botmessagelog.messagelog_error(trigger.sb["log_id"], "The admin switch (-a) is for use by authorized nicks ONLY.")
|
|
return False
|
|
|
|
if trigger.sb["hyphen_arg"]:
|
|
return False
|
|
|
|
if trigger.sb["runcount"] > 1:
|
|
# check channel multirun blocks
|
|
if not trigger.is_privmsg:
|
|
channel_disabled_list = botcommands.get_commands_disabled(str(trigger.sender), "multirun")
|
|
if trigger.sb["realcom"] in list(channel_disabled_list.keys()):
|
|
reason = channel_disabled_list[trigger.sb["realcom"]]["reason"]
|
|
timestamp = channel_disabled_list[trigger.sb["realcom"]]["timestamp"]
|
|
bywhom = channel_disabled_list[trigger.sb["realcom"]]["disabledby"]
|
|
message = "The " + str(trigger.sb["comtext"]) + " command multirun usage was disabled by " + bywhom + " for " + str(trigger.sender) + " at " + str(timestamp) + " for the following reason: " + str(reason)
|
|
return trigger_cant_run(bot, trigger, message)
|
|
|
|
# don't run commands that are disabled for specific users
|
|
nick_disabled_list = botcommands.get_commands_disabled(str(trigger.nick), "multirun")
|
|
if trigger.sb["realcom"] in list(nick_disabled_list.keys()):
|
|
bywhom = nick_disabled_list[trigger.sb["realcom"]]["disabledby"]
|
|
if bywhom != trigger.nick:
|
|
reason = nick_disabled_list[trigger.sb["realcom"]]["reason"]
|
|
timestamp = nick_disabled_list[trigger.sb["realcom"]]["timestamp"]
|
|
message = "The " + str(trigger.sb["comtext"]) + " command was multirun unsage disabled by " + bywhom + " for " + str(trigger.sender) + " at " + str(timestamp) + " for the following reason: " + str(reason)
|
|
return trigger_cant_run(bot, trigger, message)
|
|
else:
|
|
botcommands.unset_command_disabled(trigger.sb["realcomref"], trigger.nick, "multirun")
|
|
botmessagelog.messagelog_error(trigger.sb["log_id"], trigger.sb["comtext"] + " multirun is now enabled for " + str(trigger.nick))
|
|
|
|
# don't run commands that are disabled in channels
|
|
if not trigger.is_privmsg:
|
|
channel_disabled_list = botcommands.get_commands_disabled(str(trigger.sender), "fully")
|
|
if trigger.sb["realcom"] in list(channel_disabled_list.keys()):
|
|
reason = channel_disabled_list[trigger.sb["realcom"]]["reason"]
|
|
timestamp = channel_disabled_list[trigger.sb["realcom"]]["timestamp"]
|
|
bywhom = channel_disabled_list[trigger.sb["realcom"]]["disabledby"]
|
|
message = "The " + str(trigger.sb["comtext"]) + " command was disabled by " + bywhom + " for " + str(trigger.sender) + " at " + str(timestamp) + " for the following reason: " + str(reason)
|
|
return trigger_cant_run(bot, trigger, message)
|
|
|
|
# don't run commands that are disabled for specific users
|
|
nick_disabled_list = botcommands.get_commands_disabled(str(trigger.nick), "fully")
|
|
if trigger.sb["realcom"] in list(nick_disabled_list.keys()):
|
|
bywhom = nick_disabled_list[trigger.sb["realcom"]]["disabledby"]
|
|
if bywhom != trigger.nick:
|
|
reason = nick_disabled_list[trigger.sb["realcom"]]["reason"]
|
|
timestamp = nick_disabled_list[trigger.sb["realcom"]]["timestamp"]
|
|
message = "The " + str(trigger.sb["comtext"]) + " command was disabled by " + bywhom + " for " + str(trigger.sender) + " at " + str(timestamp) + " for the following reason: " + str(reason)
|
|
return trigger_cant_run(bot, trigger, message)
|
|
else:
|
|
botcommands.unset_command_disabled(trigger.sb["realcomref"], trigger.nick, "fully")
|
|
botmessagelog.messagelog(trigger.sb["log_id"], trigger.sb["comtext"] + " is now enabled for " + str(trigger.nick))
|
|
|
|
return True
|
|
|
|
|
|
def trigger_cant_run(bot, trigger, message=None):
|
|
if message:
|
|
botmessagelog.messagelog_error(trigger.sb["log_id"], message)
|
|
if command_permissions_check(bot, trigger, ['admins', 'owner', 'OP', 'ADMIN', 'OWNER']):
|
|
botmessagelog.messagelog_error(trigger.sb["log_id"], "You however are authorized to bypass this warning with the (-a) admin switch.")
|
|
return False
|
|
|
|
|
|
def trigger_hyphen_args(trigger_args_part):
|
|
valid_hyphen_args = [
|
|
'check',
|
|
'enable', 'disable',
|
|
'block', 'unblock',
|
|
"activate", "deactivate",
|
|
"on", "off",
|
|
'multirun', 'multiruns',
|
|
'example', 'usage',
|
|
'filename', 'filepath',
|
|
'foldername', 'folderpath',
|
|
"author",
|
|
'contribs', 'contrib', "contributors",
|
|
'alias', 'aliases'
|
|
]
|
|
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(bot, trigger):
|
|
|
|
# Commands that cannot run via privmsg
|
|
if trigger.sb["hyphen_arg"] in ['check']:
|
|
if trigger.sb["com"].lower() != trigger.sb["realcom"]:
|
|
botmessagelog.messagelog(trigger.sb["log_id"], trigger.sb["comtext"] + " is a valid alias command for " + trigger.sb["realcomtext"])
|
|
else:
|
|
botmessagelog.messagelog(trigger.sb["log_id"], trigger.sb["comtext"] + " is a valid command")
|
|
return
|
|
|
|
elif trigger.sb["hyphen_arg"] in [
|
|
'enable', 'disable',
|
|
'block', 'unblock',
|
|
"activate", "deactivate",
|
|
"on", "off"
|
|
]:
|
|
|
|
target = spicemanip.main(trigger.sb["args"], 1) or trigger.nick
|
|
if not target:
|
|
if trigger.sb["hyphen_arg"] in ['enable', 'unblock', "activate", "on"]:
|
|
botmessagelog.messagelog_error(trigger.sb["log_id"], "Who/Where am I enabling " + str(trigger.sb["comtext"]) + " usage for?")
|
|
else:
|
|
botmessagelog.messagelog_error(trigger.sb["log_id"], "Who/Where am I disabling " + str(trigger.sb["comtext"]) + " usage for?")
|
|
return
|
|
|
|
if not botdb.check_nick_id(target) and not botchannels.check_channel_bot(target, True):
|
|
botmessagelog.messagelog_error(trigger.sb["log_id"], "I don't know who/what " + str(target) + " is.")
|
|
return
|
|
|
|
if not command_permissions_check(bot, trigger, ['admins', 'owner', 'OP', 'ADMIN', 'OWNER']):
|
|
if target != trigger.nick:
|
|
if trigger.sb["hyphen_arg"] in ['enable', 'unblock', "activate", "on"]:
|
|
botmessagelog.messagelog_error(trigger.sb["log_id"], "I was unable to enable this command for " + str(target) + " due to privilege issues.")
|
|
else:
|
|
botmessagelog.messagelog_error(trigger.sb["log_id"], "I was unable to disable this command for " + str(target) + " due to privilege issues.")
|
|
return
|
|
|
|
if trigger.sb["hyphen_arg"] in ['enable', 'unblock', "activate", "on"]:
|
|
|
|
if not botcommands.check_commands_disabled(trigger.sb["realcomref"], target, "fully"):
|
|
botmessagelog.messagelog_error(trigger.sb["log_id"], trigger.sb["comtext"] + " is already enabled for " + str(target))
|
|
return
|
|
|
|
botcommands.unset_command_disabled(trigger.sb["realcomref"], target, "fully")
|
|
botmessagelog.messagelog(trigger.sb["log_id"], trigger.sb["comtext"] + " is now enabled for " + str(target))
|
|
return
|
|
|
|
else:
|
|
|
|
if botcommands.check_commands_disabled(trigger.sb["realcomref"], target, "fully"):
|
|
botmessagelog.messagelog_error(trigger.sb["log_id"], trigger.sb["comtext"] + " is already disabled for " + str(target))
|
|
return
|
|
|
|
trailingmessage = spicemanip.main(trigger.sb["args"], 0) or "No reason given."
|
|
timestamp = str(datetime.datetime.utcnow())
|
|
|
|
botcommands.set_command_disabled(trigger.sb["realcomref"], target, timestamp, trailingmessage, trigger.nick, "fully")
|
|
botmessagelog.messagelog(trigger.sb["log_id"], trigger.sb["comtext"] + " is now disabled for " + str(target))
|
|
return
|
|
|
|
if trigger.sb["hyphen_arg"] in ['multirun', 'multiruns']:
|
|
|
|
onoff = spicemanip.main(trigger.sb["args"], 1) or None
|
|
if not onoff or onoff not in ['enable', 'unblock', "activate", "on"]:
|
|
botmessagelog.messagelog_error(trigger.sb["log_id"], "Do you want to enable or disable " + str(trigger.sb["comtext"]) + " multirun usage?")
|
|
return
|
|
|
|
target = spicemanip.main(trigger.sb["args"], 1) or trigger.nick
|
|
if not target:
|
|
if onoff in ['enable', 'unblock', "activate", "on"]:
|
|
botmessagelog.messagelog_error(trigger.sb["log_id"], "Who/Where am I enabling " + str(trigger.sb["comtext"]) + " multirun usage for?")
|
|
else:
|
|
botmessagelog.messagelog_error(trigger.sb["log_id"], "Who/Where am I disabling " + str(trigger.sb["comtext"]) + " multirun usage for?")
|
|
return
|
|
|
|
if not botdb.check_nick_id(target) and not botchannels.check_channel_bot(target, True):
|
|
botmessagelog.messagelog_error(trigger.sb["log_id"], "I don't know who/what " + str(target) + " is.")
|
|
return
|
|
|
|
if not command_permissions_check(bot, trigger, ['admins', 'owner', 'OP', 'ADMIN', 'OWNER']):
|
|
if target != trigger.nick:
|
|
if onoff in ['enable', 'unblock', "activate", "on"]:
|
|
botmessagelog.messagelog_error(trigger.sb["log_id"], "I was unable to enable multirun usage on " + str(trigger.sb["comtext"]) + " for " + str(target) + " due to privilege issues.")
|
|
else:
|
|
botmessagelog.messagelog_error(trigger.sb["log_id"], "I was unable to enable multirun usage on " + str(trigger.sb["comtext"]) + " for " + str(target) + " due to privilege issues.")
|
|
return
|
|
|
|
if onoff in ['enable', 'unblock', "activate", "on"]:
|
|
|
|
if not botcommands.check_commands_disabled(trigger.sb["realcomref"], target, "multirun"):
|
|
botmessagelog.messagelog_error(trigger.sb["log_id"], trigger.sb["comtext"] + " multirun is already enabled for " + str(target))
|
|
return
|
|
|
|
botcommands.unset_command_disabled(trigger.sb["realcomref"], target, "multirun")
|
|
botmessagelog.messagelog(trigger.sb["log_id"], trigger.sb["comtext"] + " multirun is now enabled for " + str(target))
|
|
return
|
|
|
|
else:
|
|
|
|
if botcommands.check_commands_disabled(trigger.sb["realcomref"], target, "multirun"):
|
|
botmessagelog.messagelog_error(trigger.sb["log_id"], trigger.sb["comtext"] + " multirun is already disabled for " + str(target))
|
|
return
|
|
|
|
trailingmessage = spicemanip.main(trigger.sb["args"], 0) or "No reason given."
|
|
timestamp = str(datetime.datetime.utcnow())
|
|
|
|
botcommands.set_command_disabled(trigger.sb["realcomref"], target, timestamp, trailingmessage, trigger.nick, "multirun")
|
|
botmessagelog.messagelog(trigger.sb["log_id"], trigger.sb["comtext"] + " multirun is now disabled for " + str(target))
|
|
return
|
|
|
|
elif trigger.sb["hyphen_arg"] in ['example', 'usage']:
|
|
botmessagelog.messagelog(trigger.sb["log_id"], trigger.sb["comtext"] + ": " + str(trigger.sb["dict"]["example"]))
|
|
return
|
|
|
|
elif trigger.sb["hyphen_arg"] in ['filename', 'filepath']:
|
|
botmessagelog.messagelog(trigger.sb["log_id"], "The " + str(trigger.sb["comtext"]) + " file is located at " + str(trigger.sb["dict"][trigger.sb["hyphen_arg"]]))
|
|
return
|
|
|
|
elif trigger.sb["hyphen_arg"] in ['foldername', 'folderpath']:
|
|
botmessagelog.messagelog(trigger.sb["log_id"], "The " + str(trigger.sb["comtext"]) + " folder is located at " + str(trigger.sb["dict"][trigger.sb["hyphen_arg"]]))
|
|
return
|
|
|
|
elif trigger.sb["hyphen_arg"] in ['author']:
|
|
botmessagelog.messagelog(trigger.sb["log_id"], "The author of the " + str(trigger.sb["comtext"]) + " command is " + trigger.sb["dict"]["author"] + ".")
|
|
return
|
|
|
|
elif trigger.sb["hyphen_arg"] in ['contribs', 'contrib', "contributors"]:
|
|
botmessagelog.messagelog(trigger.sb["log_id"], "The contributors of the " + str(trigger.sb["comtext"]) + " command are " + spicemanip.main(trigger.sb["dict"]["contributors"], "andlist") + ".")
|
|
return
|
|
|
|
elif trigger.sb["hyphen_arg"] in ['alias', 'aliases']:
|
|
botmessagelog.messagelog(trigger.sb["log_id"], "The alaises of the " + str(trigger.sb["comtext"]) + " command are " + spicemanip.main(trigger.sb["dict"]["validcoms"], "andlist") + ".")
|
|
return
|
|
|
|
return
|