Handle away numerics

Closes #15

CHANGELOG

@@ -3,7 +3,7 @@ vNEXT (in development)
  * Away support:
    * Added sendAway() method
    * Added UserAway event and fanned-out ChannelAway
-   * away message is now updated in the UserState
+   * Away message is now updated in the UserState
  * (Internal) improved the way byte buffers are used to
    reduce array copying and clean up code
 

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

@@ -11,6 +11,8 @@ internal const val RPL_ISUPPORT = "005"
 
 internal const val RPL_UMODEIS = "221"
 
+internal const val RPL_UNAWAY = "305"
+internal const val RPL_NOWAWAY = "306"
 internal const val RPL_CHANNELMODEIS = "324"
 internal const val RPL_NOTOPIC = "331"
 internal const val RPL_TOPIC = "332"

src/main/kotlin/com/dmdirc/ktirc/messages/processors/AwayProcessor.kt

@@ -1,16 +1,21 @@
 package com.dmdirc.ktirc.messages.processors
 
 import com.dmdirc.ktirc.events.UserAway
+import com.dmdirc.ktirc.messages.RPL_NOWAWAY
+import com.dmdirc.ktirc.messages.RPL_UNAWAY
 import com.dmdirc.ktirc.model.IrcMessage
+import com.dmdirc.ktirc.model.User
 import com.dmdirc.ktirc.model.asUser
 
 internal class AwayProcessor : MessageProcessor {
 
-    override val commands = arrayOf("AWAY")
+    override val commands = arrayOf("AWAY", RPL_UNAWAY, RPL_NOWAWAY)
 
     override fun process(message: IrcMessage) = sequence {
         when (message.command) {
             "AWAY" -> message.prefix?.let { yield(UserAway(message.metadata, it.asUser(), message.awayMessage)) }
+            RPL_NOWAWAY -> yield(UserAway(message.metadata, User(String(message.params[0])), ""))
+            RPL_UNAWAY -> yield(UserAway(message.metadata, User(String(message.params[0])), null))
         }
     }.toList()
 

src/test/kotlin/com/dmdirc/ktirc/messages/processors/AwayProcessorTest.kt

@@ -42,6 +42,30 @@ internal class AwayProcessorTest {
         assertNull(event.message)
     }
 
+    @Test
+    fun `raises away changed event for local user on NOWAWAY`() {
+        val events = AwayProcessor().process(
+                IrcMessage(emptyMap(), ":the.server".toByteArray(), "306", params("acidBurn", "You have been marked as being away")))
+        assertEquals(1, events.size)
+
+        val event = events[0]
+        assertEquals(TestConstants.time, event.metadata.time)
+        assertEquals(User("acidBurn"), event.user)
+        assertEquals("", event.message)
+    }
+
+    @Test
+    fun `raises away changed event for local user on UNAWAY`() {
+        val events = AwayProcessor().process(
+                IrcMessage(emptyMap(), ":the.server".toByteArray(), "305", params("acidBurn", "You are no longer marked as being away")))
+        assertEquals(1, events.size)
+
+        val event = events[0]
+        assertEquals(TestConstants.time, event.metadata.time)
+        assertEquals(User("acidBurn"), event.user)
+        assertNull(event.message)
+    }
+
     @Test
     fun `does nothing on away if prefix missing`() {
         val events = AwayProcessor().process(IrcMessage(emptyMap(), null, "AWAY", params("*")))