From 4bab1aea3094c2fa6cd8ba96a42eda905a933581 Mon Sep 17 00:00:00 2001
From: jesopo
Date: Fri, 9 Nov 2018 10:43:05 +0000
Subject: Add a way to more explicitly denote when a CAP handshake has ended
 early

---
 src/IRCServer.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

(limited to 'src/IRCServer.py')

diff --git a/src/IRCServer.py b/src/IRCServer.py
index 8d1a366f..ad5d7451 100644
--- a/src/IRCServer.py
+++ b/src/IRCServer.py
@@ -26,6 +26,7 @@ class Server(IRCObject.Object):
         self.capabilities = set([]) # type: typing.Set[str]
         self.server_capabilities = {} # type: typing.Dict[str, str]
         self.batches = {} # type: typing.Dict[str, utils.irc.IRCLine]
+        self.cap_started = False
 
         self.write_buffer = b""
         self.buffered_lines = [] # type: typing.List[bytes]
@@ -361,7 +362,7 @@ class Server(IRCObject.Object):
         self._capabilities_waiting.add(capability)
     def capability_done(self, capability: str):
         self._capabilities_waiting.remove(capability)
-        if not self._capabilities_waiting:
+        if self.cap_started and not self._capabilities_waiting:
             self.send_capability_end()
 
     def send_pass(self, password: str):
-- 
cgit v1.3.1-10-gc9f91