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
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
|
from src import ModuleManager, utils
@utils.export("channelset", {"setting": "automode",
"help": "Disable/Enable automode", "validate": utils.bool_or_none})
class Module(ModuleManager.BaseModule):
_name = "AutoMode"
def _get_modes(self, channel, user):
identified_account = user.get_identified_account()
if identified_account and channel.get_setting("automode", False):
modes = channel.get_user_setting(user.get_id(), "automodes", [])
return modes
return []
def _check_modes(self, channel, user):
modes = self._get_modes(channel, user)
if modes:
channel.send_mode("+%s" % "".join(modes),
" ".join([user.nickname for mode in modes]))
@utils.hook("received.join")
def on_join(self, event):
self._check_modes(event["channel"], event["user"])
@utils.hook("received.account")
@utils.hook("internal.identified")
def on_account(self, event):
for channel in event["user"].channels:
self._check_modes(channel, event["user"])
def _check_channel(self, channel):
modes = []
for user in channel.users:
user_modes = self._get_modes(channel, user)
for user_mode in user_modes:
if not channel.has_mode(user, user_mode):
modes.append([user_mode, user.nickname])
# break up in to chunks of (maximum) 3
# https://tools.ietf.org/html/rfc2812.html#section-3.2.3
mode_chunks = [modes[i:i+3] for i in range(0, len(modes), 3)]
for chunk in mode_chunks:
modes = [item[0] for item in chunk]
nicknames = [item[1] for item in chunk]
channel.send_mode(
"+%s" % "".join(modes), " ".join(nicknames))
@utils.hook("received.command.syncmodes", channel_only=True)
def sync_modes(self, event):
"""
:help: Check/sync user modes
:require_mode: o
"""
self._check_channel(event["target"])
@utils.hook("set.channelset.automode")
def on_automode_set(self, event):
if event["value"]:
self._check_channel(event["target"])
@utils.hook("self.join")
def self_join(self, event):
self._check_channel(event["channel"])
def _add_mode(self, event, mode, mode_name):
target_user = event["server"].get_user(event["args_split"][0])
automodes = event["target"].get_user_setting(target_user.get_id(),
"automodes", [])
if mode in automodes:
event["stderr"].write("'%s' already has automode %s" % (
target_user.nickname, mode_name))
else:
automodes.append(mode)
event["target"].set_user_setting(target_user.get_id(), "automodes",
automodes)
if event["target"] in target_user.channels:
self._check_modes(event["target"], target_user)
event["stdout"].write("Added automode %s for '%s'" % (
mode_name, target_user.nickname))
def _remove_mode(self, event, mode, mode_name):
target_user = event["server"].get_user(event["args_split"][0])
automodes = event["target"].get_user_setting(target_user.get_id(),
"automodes", [])
if not mode in automodes:
event["stderr"].write("'%s' doesn't have automode %s" % (
target_user.nickname, mode_name))
else:
automodes.remove(mode)
event["target"].set_user_setting(target_user.get_id(), "automodes",
automodes)
event["stdout"].write("Removed automode %s from '%s'" % (
mode_name, target_user.nickname))
@utils.hook("received.command.addop", min_args=1, channel_only=True)
def add_op(self, event):
"""
:help: Add a user to the auto-mode list as an op
:usage: <nickname>
:require_mode: o
"""
self._add_mode(event, "o", "op")
@utils.hook("received.command.removeop", min_args=1, channel_only=True)
def remove_op(self, event):
"""
:help: Remove a user from the auto-mode list as an op
:usage: <nickname>
:require_mode: o
"""
self._remove_mode(event, "o", "op")
@utils.hook("received.command.addvoice", min_args=1, channel_only=True)
def add_voice(self, event):
"""
:help: Add a user to the auto-mode list as a voice
:usage: <nickname>
:require_mode: o
"""
self._add_mode(event, "v", "voice")
@utils.hook("received.command.removevoice", min_args=1, channel_only=True)
def remove_voice(self, event):
"""
:help: Remove a user from the auto-mode list as a voice
:usage: <nickname>
:require_mode: o
"""
self._remove_mode(event, "v", "voice")
|
