Firepup650/bitbot-3.11-fork
2
0
Fork 0
Code Issues Pull requests Activity Actions

Ask for our own modes when we get a 001

Browse source
This commit is contained in:
jesopo 2019-06-06 11:16:47 +01:00
parent eb7d90bb00
commit 4c6a398f53
2 changed files with 16 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
modules/line_handler/__init__.py
View file

@ -149,6 +149,10 @@ class Module(ModuleManager.BaseModule):
@utils.hook("raw.received.mode") @utils.hook("raw.received.mode")
def mode(self, event): def mode(self, event):
core.mode(self.events, event) core.mode(self.events, event)
# server telling us our own modes
@utils.hook("raw.received.221")
def umodeis(self, event):
core.handle_221(event)
# someone (maybe me!) has been invited somewhere # someone (maybe me!) has been invited somewhere
@utils.hook("raw.received.invite") @utils.hook("raw.received.invite")

17
modules/line_handler/core.py
View file

@ -11,6 +11,7 @@ def handle_001(event):
event["server"].name = event["source"].hostmask event["server"].name = event["source"].hostmask
event["server"].set_own_nickname(event["args"][0]) event["server"].set_own_nickname(event["args"][0])
event["server"].send_whois(event["server"].nickname) event["server"].send_whois(event["server"].nickname)
event["server"].send_mode(event["server"].nickname)
event["server"].connected = True event["server"].connected = True
def handle_005(events, event): def handle_005(events, event):
@ -64,6 +65,13 @@ def motd_start(event):
def motd_line(event): def motd_line(event):
event["server"].motd_lines.append(event["args"][1]) event["server"].motd_lines.append(event["args"][1])
def _own_modes(server, modes):
mode_chunks = RE_MODES.findall(modes)
for chunk in mode_chunks:
remove = chunk[0] == "-"
for mode in chunk[1:]:
server.change_own_mode(remove, mode)
def mode(events, event): def mode(events, event):
user = event["server"].get_user(event["source"].nickname) user = event["server"].get_user(event["source"].nickname)
target = event["args"][0] target = event["args"][0]
@ -88,14 +96,13 @@ def mode(events, event):
events.on("received.mode.channel").call(modes=modes, mode_args=_args, events.on("received.mode.channel").call(modes=modes, mode_args=_args,
channel=channel, server=event["server"], user=user) channel=channel, server=event["server"], user=user)
elif event["server"].is_own_nickname(target): elif event["server"].is_own_nickname(target):
modes = RE_MODES.findall(event["args"][1]) _own_modes(event["server"], event["args"][1])
for chunk in modes:
remove = chunk[0] == "-"
for mode in chunk[1:]:
event["server"].change_own_mode(remove, mode)
events.on("self.mode").call(modes=modes, server=event["server"]) events.on("self.mode").call(modes=modes, server=event["server"])
event["server"].send_who(event["server"].nickname) event["server"].send_who(event["server"].nickname)
def handle_221(event):
_own_modes(event["server"], event["args"][1])
def invite(events, event): def invite(events, event):
target_channel = event["args"][1] target_channel = event["args"][1]
user = event["server"].get_user(event["source"].nickname) user = event["server"].get_user(event["source"].nickname)