#--depends-on commands #--depends-on permissions from src import EventManager, IRCLine, ModuleManager, utils class Module(ModuleManager.BaseModule): def _execute(self, server, commands, **kwargs): for command in commands: line = command.format(**kwargs) if IRCLine.is_human(line): line = IRCLine.parse_human(line) else: line = IRCLine.parse_line(line) server.send(line) @utils.hook("received.001", priority=EventManager.PRIORITY_URGENT) def on_connect(self, event): commands = event["server"].get_setting("perform", []) self._execute(event["server"], commands, NICK=event["server"].nickname) @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) 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 ") @utils.kwarg("usage", "remove ") @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 ") @utils.kwarg("usage", "remove ") @utils.kwarg("permission", "cperform") def cperform(self, event): event["stdout"].write(self._perform(event["target"], event["args_split"]))