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
|
from src import ModuleManager, utils
BATCH_MAX = 10
class Module(ModuleManager.BaseModule):
@utils.hook("received.001")
def on_connect(self, event):
channels = event["server"].get_setting("autojoin", [])
if channels:
channel_batches = [channels[i:i+BATCH_MAX] for i in range(
0, len(channels), BATCH_MAX)]
for channel_batch in channel_batches:
event["server"].send_joins(channel_batch)
@utils.hook("self.join")
def on_join(self, event):
channels = event["server"].get_setting("autojoin", [])
channel_name = event["server"].irc_lower(event["channel"].name)
if not channel_name in channels:
channels.append(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)
return True
else:
return False
@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)
@utils.hook("raw.received.470")
def on_linkchannel(self, event):
initial = event["line"].args[1]
initial_lower = event["server"].irc_lower(initial)
linked = event["line"].args[2]
if self._remove_channel(event["server"], initial_lower):
self.log.warn(f"{str(event['server'])} "
f"channel {initial} linked to {linked} "
"- removed from autojoin")
|
