From d0ef4432b349b230bdf4c84bfd4f71dd38ad47c5 Mon Sep 17 00:00:00 2001
From: jesopo
Date: Mon, 16 Jul 2018 21:41:39 +0100
Subject: move sending CAP sasl request to sasl.py

---
 modules/sasl.py | 12 +++++++++---
 1 file changed, 9 insertions(+), 3 deletions(-)

(limited to 'modules')

diff --git a/modules/sasl.py b/modules/sasl.py
index afe48522..7889640e 100644
--- a/modules/sasl.py
+++ b/modules/sasl.py
@@ -3,11 +3,17 @@ import base64
 class Module(object):
     def __init__(self, bot):
         self.bot = bot
-        bot.events.on("received").on("cap").hook(self.on_cap)
-        bot.events.on("received").on("authenticate").hook(self.on_authenticate)
-        bot.events.on("received").on("numeric").on(
+        bot.events.on("preprocess.connect").hook(self.preprocess_connect)
+        bot.events.on("received.cap").hook(self.on_cap)
+        bot.events.on("received.authenticate").hook(self.on_authenticate)
+        bot.events.on("received.numeric").on(
             "902", "903", "904", "905", "906", "907", "908").hook(self.on_90x)
 
+    def preprocess_connect(self, event):
+        sasl = event["server"].get_setting("sasl")
+        if sasl:
+            event["server"].send_capability_request("sasl")
+
     def on_cap(self, event):
         if event["subcommand"] == "NAK":
             event["server"].send_capability_end()
-- 
cgit v1.3.1-10-gc9f91