Quit handling

src/main/kotlin/com/dmdirc/ktirc/events/ChannelStateHandler.kt

@@ -16,6 +16,7 @@ class ChannelStateHandler : EventHandler {
             is ChannelParted -> handlePart(client, event)
             is ChannelNamesReceived -> handleNamesReceived(client, event)
             is ChannelNamesFinished -> handleNamesFinished(client, event)
+            is UserQuit -> handleQuit(client, event)
         }
     }
 
@@ -63,4 +64,8 @@ class ChannelStateHandler : EventHandler {
         }
     }
 
+    private fun handleQuit(client: IrcClient, event: UserQuit) {
+        client.channelState.forEach { it.users -= event.user.nickname }
+    }
+
 }

src/main/kotlin/com/dmdirc/ktirc/events/Events.kt

@@ -34,3 +34,6 @@ data class ChannelNamesFinished(val channel: String) : IrcEvent()
 
 /** Raised when a message is received. */
 data class MessageReceived(val user: User, val target: String, val message: String) : IrcEvent()
+
+/** Raised when a user quits. */
+data class UserQuit(val user: User, val reason: String = "") : IrcEvent()

src/main/kotlin/com/dmdirc/ktirc/messages/MessageProcessor.kt

@@ -24,5 +24,6 @@ val messageProcessors = setOf(
         PartProcessor(),
         PingProcessor(),
         PrivmsgProcessor(),
+        QuitProcessor(),
         WelcomeProcessor()
 )

src/main/kotlin/com/dmdirc/ktirc/messages/QuitProcessor.kt

@@ -0,0 +1,18 @@
+package com.dmdirc.ktirc.messages
+
+import com.dmdirc.ktirc.events.UserQuit
+import com.dmdirc.ktirc.io.IrcMessage
+import com.dmdirc.ktirc.model.asUser
+
+internal class QuitProcessor : MessageProcessor {
+
+    override val commands = arrayOf("QUIT")
+
+    override fun process(message: IrcMessage) = message.prefix?.let {
+        listOf(UserQuit(it.asUser(), message.reason))
+    } ?: emptyList()
+
+    private val IrcMessage.reason
+        get() = if (params.isNotEmpty()) String(params[0]) else ""
+
+}

src/test/kotlin/com/dmdirc/ktirc/events/ChannelStateHandlerTest.kt

@@ -121,4 +121,29 @@ internal class ChannelStateHandlerTest {
         assertFalse("zerocool" in channel.users)
     }
 
+    @Test
+    fun `ChannelStateHandler removes user from all channel member lists for quits`() = runBlocking {
+        with (ChannelState("#thegibson") { CaseMapping.Rfc }) {
+            users += ChannelUser("ZeroCool")
+            channelStateMap += this
+        }
+
+        with (ChannelState("#dumpsterdiving") { CaseMapping.Rfc }) {
+            users += ChannelUser("ZeroCool")
+            channelStateMap += this
+        }
+
+        with (ChannelState("#chat") { CaseMapping.Rfc }) {
+            users += ChannelUser("AcidBurn")
+            channelStateMap += this
+        }
+
+        handler.processEvent(ircClient, UserQuit(User("zerocool", "dade", "root.localhost")))
+
+        assertFalse("zerocool" in channelStateMap["#thegibson"]!!.users)
+        assertFalse("zerocool" in channelStateMap["#dumpsterdiving"]!!.users)
+        assertFalse("zerocool" in channelStateMap["#chat"]!!.users)
+        assertTrue("acidburn" in channelStateMap["#chat"]!!.users)
+    }
+
 }

src/test/kotlin/com/dmdirc/ktirc/messages/QuitProcessorTest.kt

@@ -0,0 +1,34 @@
+package com.dmdirc.ktirc.messages
+
+import com.dmdirc.ktirc.events.UserQuit
+import com.dmdirc.ktirc.io.IrcMessage
+import com.dmdirc.ktirc.model.User
+import org.junit.jupiter.api.Assertions
+import org.junit.jupiter.api.Test
+
+internal class QuitProcessorTest {
+
+    @Test
+    fun `QuitProcessor raises quit event without message`() {
+        val events = QuitProcessor().process(
+                IrcMessage(null, "acidburn!libby@root.localhost".toByteArray(), "QUIT", emptyList()))
+        Assertions.assertEquals(1, events.size)
+        Assertions.assertEquals(UserQuit(User("acidburn", "libby", "root.localhost")), events[0])
+    }
+
+    @Test
+    fun `QuitProcessor raises quit event with message`() {
+        val events = QuitProcessor().process(
+                IrcMessage(null, "acidburn!libby@root.localhost".toByteArray(), "QUIT", listOf("Hack the planet!".toByteArray())))
+        Assertions.assertEquals(1, events.size)
+        Assertions.assertEquals(UserQuit(User("acidburn", "libby", "root.localhost"), "Hack the planet!"), events[0])
+    }
+
+    @Test
+    fun `QuitProcessor does nothing if prefix missing`() {
+        val events = QuitProcessor().process(
+                IrcMessage(null, null, "QUIT", listOf("Hack the planet!".toByteArray())))
+        Assertions.assertEquals(0, events.size)
+    }
+
+}