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
|
from src import ModuleManager, Utils
class Module(ModuleManager.BaseModule):
@Utils.hook("received.numeric.001")
def on_connect(self, event):
channels = event["server"].get_setting("autojoin", [])
chan_keys = event["server"].get_setting("channel_keys", {})
channels_sorted = sorted(channels,
key=lambda x: 0 if x in chan_keys else 1)
keys_sorted = list(map(lambda x: x[1],
sorted(chan_keys.items(),
key=lambda x: channels_sorted.index(x[0]))))
for i in range(len(channels_sorted)):
channel = channels_sorted[i]
key = None if len(keys_sorted) <= i else keys_sorted[i]
event["server"].attempted_join[channel] = key
event["server"].send_join(
",".join(channels_sorted), ",".join(keys_sorted))
@Utils.hook("self.join")
def on_join(self, event):
channels = event["server"].get_setting("autojoin", [])
if not event["channel"].name in channels:
channels.append(event["channel"].name)
event["server"].set_setting("autojoin", channels)
def _remove_channel(self, server, channel_name):
channels = server.get_setting("autojoin", [])
if channel_name in channels:
channels.remove(channel_name)
server.set_setting("autojoin", channels)
@Utils.hook("self.part")
def on_part(self, event):
self._remove_channel(event["server"], event["channel"].name)
@Utils.hook("self.kick")
def on_kick(self, event):
self._remove_channel(event["server"], event["channel"].name)
|
