KtIrc/src/main/kotlin/com/dmdirc/ktirc/messages/PrivmsgProcessor.kt

package com.dmdirc.ktirc.messages

import com.dmdirc.ktirc.events.ActionReceived
import com.dmdirc.ktirc.events.CtcpReceived
import com.dmdirc.ktirc.events.IrcEvent
import com.dmdirc.ktirc.events.MessageReceived
import com.dmdirc.ktirc.model.IrcMessage
import com.dmdirc.ktirc.model.User

internal class PrivmsgProcessor : MessageProcessor {

    override val commands = arrayOf("PRIVMSG")

    override fun process(message: IrcMessage) = message.sourceUser?.let { user ->
        listOf(when {
            message.isCtcp() -> handleCtcp(message, user)
            else -> MessageReceived(message.metadata, user, String(message.params[0]), String(message.params[1]))
        })
    } ?: emptyList()

    private fun handleCtcp(message: IrcMessage, user: User): IrcEvent {
        val content = String(message.params[1]).substring(1 until message.params[1].size - 1)
        val parts = content.split(' ', limit=2)
        val body = if (parts.size == 2) parts[1] else ""
        return when (parts[0].toUpperCase()) {
            "ACTION" -> ActionReceived(message.metadata, user, String(message.params[0]), body)
            else -> CtcpReceived(message.metadata, user, String(message.params[0]), parts[0], body)
        }
    }

    private fun IrcMessage.isCtcp() = params[1].size > 2 && params[1][0] == CTCP_BYTE && params[1][params[1].size - 1] == CTCP_BYTE

}