From 696d24432dc7ade1e6b312131265f02107480131 Mon Sep 17 00:00:00 2001
From: Ben Kurtovic <ben.kurtovic@gmail.com>
Date: Mon, 16 Nov 2015 05:05:48 -0600
Subject: [PATCH] Improve detection of maximum IRC message length.

---
 CHANGELOG                   |  1 +
 earwigbot/irc/connection.py | 33 +++++++++++++++++++++++++++++----
 2 files changed, 30 insertions(+), 4 deletions(-)

diff --git a/CHANGELOG b/CHANGELOG
index e09b882..53d36fa 100644
--- a/CHANGELOG
+++ b/CHANGELOG
@@ -1,6 +1,7 @@
 v0.3 (unreleased):
 
 - Added !remind all.
+- Improved detection of maximum IRC message length.
 
 v0.2 (released November 8, 2015):
 
diff --git a/earwigbot/irc/connection.py b/earwigbot/irc/connection.py
index 5fe15b0..2d10341 100644
--- a/earwigbot/irc/connection.py
+++ b/earwigbot/irc/connection.py
@@ -45,6 +45,7 @@ class IRCConnection(object):
         self._last_recv = time()
         self._last_send = 0
         self._last_ping = 0
+        self._myhost = "." * 63  # default: longest possible hostname
 
     def __repr__(self):
         """Return the canonical string representation of the IRCConnection."""
@@ -100,8 +101,19 @@ class IRCConnection(object):
                     self.logger.debug(msg)
                 self._last_send = time()
 
-    def _split(self, msgs, maxlen, maxsplits=3):
-        """Split a large message into multiple messages smaller than maxlen."""
+    def _get_maxlen(self, extra):
+        """Return our best guess of the maximum length of a standard message.
+
+        This applies mainly to PRIVMSGs and NOTICEs.
+        """
+        base_max = 512
+        userhost = len(self.nick) + len(self.ident) + len(self._myhost) + 2
+        padding = 4  # "\r\n" at end, ":" at beginning, and " " after userhost
+        return base_max - userhost - padding - extra
+
+    def _split(self, msgs, extralen, maxsplits=3):
+        """Split a large message into multiple messages."""
+        maxlen = self._get_maxlen(extralen)
         words = msgs.split(" ")
         splits = 0
         while words and splits < maxsplits:
@@ -128,6 +140,19 @@ class IRCConnection(object):
         self._last_recv = time()
         if line[0] == "PING":  # If we are pinged, pong back
             self.pong(line[1][1:])
+        elif line[1] == "001":  # Update nickname on startup
+            if line[2] != self.nick:
+                self.logger.warn("Nickname changed from {0} to {1}".format(
+                    self.nick, line[2]))
+                self._nick = line[2]
+        elif line[1] == "376":  # After sign-on, get our userhost
+            self._send("WHOIS {0}".format(self.nick))
+        elif line[1] == "311":  # Receiving WHOIS result
+            if line[2] == self.nick:
+                self._ident = line[4]
+                self._myhost = line[5]
+        elif line[1] == "396":  # Hostname change
+            self._myhost = line[3]
 
     def _process_message(self, line):
         """To be overridden in subclasses."""
@@ -163,7 +188,7 @@ class IRCConnection(object):
 
     def say(self, target, msg, hidelog=False):
         """Send a private message to a target on the server."""
-        for msg in self._split(msg, 400):
+        for msg in self._split(msg, len(target) + 10):
             msg = "PRIVMSG {0} :{1}".format(target, msg)
             self._send(msg, hidelog)
 
@@ -182,7 +207,7 @@ class IRCConnection(object):
 
     def notice(self, target, msg, hidelog=False):
         """Send a notice to a target on the server."""
-        for msg in self._split(msg, 400):
+        for msg in self._split(msg, len(target) + 9):
             msg = "NOTICE {0} :{1}".format(target, msg)
             self._send(msg, hidelog)