Revamp perform.py (including adding channel-performs)

This commit is contained in:
jesopo 2019-07-02 20:46:00 +01:00
parent 702f5c5972
commit a077e086cd

View file

@ -4,35 +4,69 @@
from src import EventManager, ModuleManager, utils
class Module(ModuleManager.BaseModule):
def _execute(self, server):
commands = server.get_setting("perform", [])
for i, command in enumerate(commands):
command = command.split("%%")
for j, part in enumerate(command[:]):
command[j] = part.replace("%nick%", server.nickname)
command = "%".join(command)
server.send_raw(command)
def _execute(self, server, commands, **kwargs):
for command in commands:
print(command)
server.send_raw(command.format(**kwargs))
@utils.hook("received.001", priority=EventManager.PRIORITY_URGENT)
def on_connect(self, event):
self._execute(event["server"])
commands = event["server"].get_setting("perform", [])
self._execute(event["server"], commands, NICK=event["server"].nickname)
@utils.hook("received.command.performadd", min_args=1)
def perform_add(self, event):
"""
:help: Add command to be executed on connect
:usage: <raw command>
:permission: perform
"""
perform = event["server"].get_setting("perform", [])
perform.append(event["args"])
event["server"].set_setting("perform", perform)
event["stdout"].write("Added command")
@utils.hook("self.join", priority=EventManager.PRIORITY_URGENT)
def on_join(self, event):
commands = event["channel"].get_setting("perform", [])
self._execute(event["server"], commands, NICK=event["server"].nickname,
CHAN=event["channel"].name)
@utils.hook("received.command.performexecute")
def perform_execute(self, event):
"""
:help: Execute all saved commands
:permission: perform
"""
self._execute(event["server"])
def _perform(self, target, args_split):
subcommand = args_split[0].lower()
current_perform = target.get_setting("perform", [])
if subcommand == "list":
return "Configured commands: %s" % ", ".join(current_perform)
message = None
if subcommand == "add":
if not len(args_split) > 1:
raise utils.EventError("Please provide a raw command to add")
current_perform.append(" ".join(args_split[1:]))
message = "Added command"
elif subcommand == "remove":
if not len(args_split) > 1:
raise utils.EventError("Please provide an index to remove")
if not args_split[1].isdigit():
raise utils.EventError("Please provide a number")
index = int(args_split[1])
if not index < len(current_perform):
raise utils.EventError("Index out of bounds")
current_perform.pop(index)
message = "Removed command"
else:
raise utils.EventError("Unknown subcommand '%s'" % subcommand)
target.set_setting("perform", current_perform)
return message
@utils.hook("received.command.perform", min_args=1)
@utils.kwarg("min_args", 1)
@utils.kwarg("help", "Edit on-connect command configuration")
@utils.kwarg("usage", "list")
@utils.kwarg("usage", "add <raw command>")
@utils.kwarg("usage", "remove <index>")
@utils.kwarg("permission", "perform")
def perform(self, event):
event["stdout"].write(self._perform(event["server"],
event["args_split"]))
@utils.hook("received.command.cperform", min_args=1)
@utils.kwarg("min_args", 1)
@utils.kwarg("channel_only", True)
@utils.kwarg("help", "Edit channel on-join command configuration")
@utils.kwarg("usage", "list")
@utils.kwarg("usage", "add <raw command>")
@utils.kwarg("usage", "remove <index>")
@utils.kwarg("permission", "cperform")
def cperform(self, event):
event["stdout"].write(self._perform(event["target"],
event["args_split"]))