From a13211e0dbae51798a8bda17c78244f9ac161b21 Mon Sep 17 00:00:00 2001
From: James Lu <GLolol@overdrivenetworks.com>
Date: Sat, 16 Apr 2016 16:46:30 -0700
Subject: [PATCH] nefarious: implement KILL, fix bugs in outgoing kick &
 incoming join

---
 protocols/nefarious.py | 37 ++++++++++++++++++++++++++++++-------
 1 file changed, 30 insertions(+), 7 deletions(-)

diff --git a/protocols/nefarious.py b/protocols/nefarious.py
index fa2e569..b8f837f 100644
--- a/protocols/nefarious.py
+++ b/protocols/nefarious.py
@@ -263,9 +263,6 @@ class P10Protocol(Protocol):
         if not reason:
             reason = 'No reason given'
 
-        # Mangle kick targets for IRCds that require it.
-        target = self._getOutgoingNick(target)
-
         self._send(numeric, 'K %s %s :%s' % (channel, target, reason))
 
         # We can pretend the target left by its own will; all we really care about
@@ -273,6 +270,17 @@ class P10Protocol(Protocol):
         # handle_part() does that just fine.
         self.handle_part(target, 'KICK', [channel])
 
+    def kill(self, numeric, target, reason):
+        """Sends a kill from a PyLink client/server."""
+        # <- ABAAA D AyAAA :nefarious.midnight.vpn!GL (test)
+
+        if (not self.irc.isInternalClient(numeric)) and \
+                (not self.irc.isInternalServer(numeric)):
+            raise LookupError('No such PyLink client/server exists.')
+
+        self._send(numeric, 'D %s :Killed (%s)' % (target, reason))
+        self.removeClient(target)
+
     def message(self, numeric, target, text):
         """Sends a PRIVMSG from a PyLink client."""
         if not self.irc.isInternalClient(numeric):
@@ -651,7 +659,13 @@ class P10Protocol(Protocol):
         """Handles incoming JOINs and channel creations."""
         # <- ABAAA C #test3 1460744371
         # <- ABAAB J #test3 1460744371
-        ts = int(args[1])
+        # <- ABAAB J #test3
+        try:
+            # TS is optional
+            ts = int(args[1])
+        except IndexError:
+            ts = None
+
         if args[0] == '0' and command == 'JOIN':
             # /join 0; part the user from all channels
             oldchans = self.irc.users[numeric].channels.copy()
@@ -663,10 +677,11 @@ class P10Protocol(Protocol):
             return {'channels': oldchans, 'text': 'Left all channels.', 'parse_as': 'PART'}
         else:
             channel = utils.toLower(self.irc, args[0])
-            self.updateTS(channel, ts)
+            if ts:  # Only update TS if one was sent.
+                self.updateTS(channel, ts)
 
         return {'channel': channel, 'users': [source], 'modes':
-                self.irc.channels[channel].modes, 'ts': ts}
+                self.irc.channels[channel].modes, 'ts': ts or int(time.time())}
 
     handle_create = handle_join
 
@@ -747,9 +762,17 @@ class P10Protocol(Protocol):
         return {'channel': channel, 'target': kicked, 'text': args[2]}
 
     def handle_quit(self, numeric, command, args):
-        """Handles incoming QUIT commands."""
+        """Handles incoming QUITs."""
         # <- ABAAB Q :Killed (GL_ (bangbang))
         self.removeClient(numeric)
         return {'text': args[0]}
 
+    def handle_kill(self, numeric, command, args):
+        """Handles incoming KILLs."""
+        # <- ABAAA D AyAAA :nefarious.midnight.vpn!GL (test)
+        killed = args[0]
+        if killed in self.irc.users:
+            self.removeClient(killed)
+        return {'target': killed, 'text': args[1], 'userdata': self.irc.users.get(killed)}
+
 Class = P10Protocol