1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
|
#--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, spec):
subcommand = spec[0]
current_perform = target.get_setting("perform", [])
if subcommand == "list":
return "Configured commands: %s" % ", ".join(current_perform)
message = None
if subcommand == "add":
current_perform.append(spec[1])
message = "Added command"
elif subcommand == "remove":
index = spec[1]
if not index < len(current_perform):
raise utils.EventError("Index out of bounds")
command = current_perform.pop(index)
message = "Removed command %d (%s)" % (index, command)
target.set_setting("perform", current_perform)
return message
@utils.hook("received.command.perform", permission="perform",
help="Edit on-connect command configuration")
@utils.hook("received.command.cperform", permission="perform",
help="Edit channel on-join command configuration", channel_only=True)
@utils.kwarg("help", "Edit on-connect command configuration")
@utils.spec("!'list")
@utils.spec("!'add !<command>string")
@utils.spec("!'remove !<index>int")
def perform(self, event):
if event["command"] == "perform":
target = event["server"]
elif event["command"] == "cperform":
target = event["target"]
out = self._perform(target, event["spec"])
event["stdout"].write("%s: %s" % (event["user"].nickname, out))
|
