gorc/irc/handler/handler.go

// gorc project
// Copyright (C) 2022 IllusionMan1212
//
// This program is free software; you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation; either version 3 of the License, or
// (at your option) any later version.
//
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
// GNU General Public License for more details.
//
// You should have received a copy of the GNU General Public License along
// with this program; if not, see https://www.gnu.org/licenses.

package handler

import (
	"bufio"
	"fmt"
	"io"
	"log"
	"strconv"
	"strings"
	"time"

	"github.com/illusionman1212/gorc/cmds"
	"github.com/illusionman1212/gorc/irc"
	"github.com/illusionman1212/gorc/irc/commands"
	"github.com/illusionman1212/gorc/irc/parser"
)

func ReadLoop(client *irc.Client) {
	// 512 bytes as a base + 8192 additional bytes for tags
	// as specified here: https://modern.ircdocs.horse/#message-format
	r := bufio.NewReaderSize(client.TCPConn, 8192+512)

	for {
		msg, err := r.ReadString('\n')
		if err != nil {
			if err != io.EOF {
				log.Println(err)
			}
			client.TCPConn.Close()
			return
		}

		msg = strings.Replace(msg, "\r\n", "", 1)
		if ircMessage, valid := parser.ParseIRCMessage(msg); valid {
			HandleCommand(ircMessage, client)
		}
	}
}

func handlePing(msg irc.Message, client *irc.Client) {
	token := msg.Parameters[0]
	client.SendCommand(commands.PONG, token)
}

func handlePrivMsg(msg irc.Message, client *irc.Client) {
	source := strings.ToLower(strings.SplitN(msg.Source, "!", 2)[0])
	targets := strings.Split(msg.Parameters[0], ",")
	msgContent := msg.Parameters[1]
	privMsg := fmt.Sprintf("%s: %s", source, msgContent)

	msgOpts := irc.MsgFmtOpts{
		WithTimestamp: true,
	}

targetsLoop:
	for _, target := range targets {
		current := client.RootChannel
		for {
			if current.Value.Name == target || current.Value.Name == source {
				current.Value.AppendMsg(msg.Timestamp, privMsg, msgOpts)
				continue targetsLoop
			}

			current = current.Next
			if current == client.RootChannel {
				break
			}
		}

		if target == client.Nickname {
			newChannel := irc.Channel{
				Name:  source,
				Users: map[string]irc.User{source: {}},
			}

			newChannel.AppendMsg(msg.Timestamp, privMsg, msgOpts)
			client.AppendChannel(newChannel)
		}
	}
}

func handleNotice(msg irc.Message, client *irc.Client) {
	source := strings.SplitN(msg.Source, "!", 2)[0]
	targets := strings.Split(msg.Parameters[0], ",")
	msgContent := msg.Parameters[1]
	notice := fmt.Sprintf("%s: %s", source, msgContent)

	msgOpts := irc.MsgFmtOpts{
		WithTimestamp: true,
	}

	for _, target := range targets {
		if target == "*" || target == client.Nickname {
			client.RootChannel.Value.AppendMsg(msg.Timestamp, notice, msgOpts)
			continue
		}

		current := client.RootChannel
		for {
			if current.Value.Name == targets[0] {
				current.Value.AppendMsg(msg.Timestamp, notice, msgOpts)
			}

			current = current.Next
			if current == client.RootChannel {
				break
			}
		}
	}

}

func handleJoin(msg irc.Message, client *irc.Client) {
	nick := strings.SplitN(msg.Source, "!", 2)[0]
	channel := msg.Parameters[0]

	joinMsg := fmt.Sprintf("%s has joined", nick)

	msgOpts := irc.MsgFmtOpts{
		WithTimestamp: true,
		AsServerMsg:   true,
	}

	if nick == client.Nickname {
		newChannel := irc.Channel{
			Name:  channel,
			Users: make(map[string]irc.User),
		}

		client.ActiveChannel = client.AppendChannel(newChannel)
		client.ActiveChannel.Value.AppendMsg(msg.Timestamp, joinMsg, msgOpts)

		if _, exists := client.ActiveChannel.Value.Users[nick]; !exists {
			client.ActiveChannel.Value.Users[nick] = irc.User{}
		}
		client.Tea.Send(cmds.UpdateTabBar())
	} else {
		current := client.RootChannel
		for {
			if current.Value.Name == channel {
				current.Value.AppendMsg(msg.Timestamp, joinMsg, msgOpts)
				if _, exists := current.Value.Users[nick]; !exists {
					current.Value.Users[nick] = irc.User{}
				}
			}

			current = current.Next
			if current == client.RootChannel {
				break
			}
		}
	}

	if channel == client.ActiveChannel.Value.Name {
		client.Tea.Send(cmds.SwitchChannels())
	}
}

// Sent from server to client to acknowledge a nickname change, or to inform of a nickname change of another user.
func handleNick(msg irc.Message, client *irc.Client) {
	oldNick := strings.SplitN(msg.Source, "!", 2)[0]
	newNick := msg.Parameters[0]

	// TODO: the actual case-sensitivity is determined by the CASEMAPPING feature
	isMe := strings.EqualFold(oldNick, client.Nickname)

	msgOpts := irc.MsgFmtOpts{
		WithTimestamp: true,
		AsServerMsg:   true,
	}

	message := ""

	if isMe {
		message = fmt.Sprintf("Your nickname was changed to %v", newNick)
	} else {
		message = fmt.Sprintf("%v changed their nickname to %v", oldNick, newNick)
	}

	current := client.RootChannel
	for {
		// Rename this user in every channel we can find
		if user, ok := current.Value.Users[oldNick]; ok {
			delete(current.Value.Users, oldNick)
			current.Value.Users[newNick] = user
			current.Value.AppendMsg(msg.Timestamp, message, msgOpts)
		}

		// If we have a private channel open with this user, rename it as well.
		if strings.EqualFold(current.Value.Name, oldNick) {
			current.Value.Name = newNick

			if current == client.ActiveChannel {
				current.Value.Name = newNick
			}

			// if client.ActiveChannelIndex == i {
			// 	client.ActiveChannel = newNick
			// }
			break
		}

		current = current.Next
		if current == client.RootChannel {
			break
		}
	}

	if isMe {
		client.RootChannel.Value.AppendMsg(msg.Timestamp, message, msgOpts)
		client.Nickname = newNick
	}

	client.Tea.Send(cmds.UpdateNicks())
}

func handleKick(msg irc.Message, client *irc.Client) {
	channel := msg.Parameters[0]
	nicks := strings.Split(msg.Parameters[1], ",")

	// Reason is optional
	reason := ""

	if len(msg.Parameters) > 2 {
		reason = msg.Parameters[2]
	}

	msgOpts := irc.MsgFmtOpts{
		WithTimestamp: true,
		AsServerMsg:   true,
	}

	for _, nick := range nicks {
		kicker := strings.SplitN(msg.Source, "!", 2)[0]
		message := fmt.Sprintf("%v kicked %v from %v (%v)", kicker, nick, channel, reason)

		current := client.RootChannel
		for {
			if current.Value.Name == channel {
				if nick == client.Nickname {
					if current == client.ActiveChannel {
						client.ActiveChannel = current.Prev
					}

					client.RemoveChannel(current)

					message = fmt.Sprintf("You were kicked by %v from %v (%v)", kicker, channel, reason)
					client.RootChannel.Value.AppendMsg(msg.Timestamp, message, msgOpts)

					return
				}
				delete(current.Value.Users, nick)

				current.Value.AppendMsg(msg.Timestamp, message, msgOpts)
			}

			current = current.Next
			if current == client.RootChannel {
				break
			}
		}
	}

	client.Tea.Send(cmds.SwitchChannels())
}

func handleQuit(msg irc.Message, client *irc.Client) {
	nick := strings.SplitN(msg.Source, "!", 2)[0]
	reason := msg.Parameters[0]
	quitMsg := fmt.Sprintf("%s has quit (%s)", nick, reason)

	msgOpts := irc.MsgFmtOpts{
		WithTimestamp: true,
		AsServerMsg:   true,
	}

	// skip the server "channel"
	current := client.RootChannel.Next
	for {
		// if i == 0 {
		// 	continue
		// }
		current.Value.AppendMsg(msg.Timestamp, quitMsg, msgOpts)
		delete(current.Value.Users, nick)

		current = current.Next
		if current == client.RootChannel {
			break
		}
	}

	client.Tea.Send(cmds.SwitchChannels())
}

func handlePart(msg irc.Message, client *irc.Client) {
	nick := strings.SplitN(msg.Source, "!", 2)[0]
	channel := msg.Parameters[0]
	reason := ""
	if len(msg.Parameters) > 1 {
		reason = msg.Parameters[1]
	}

	partMsg := fmt.Sprintf("%s has left %s (%s)", nick, channel, reason)

	msgOpts := irc.MsgFmtOpts{
		WithTimestamp: true,
		AsServerMsg:   true,
	}

	current := client.RootChannel
	for {
		if current.Value.Name == channel {
			if nick == client.Nickname {
				if current == client.ActiveChannel {
					client.ActiveChannel = current.Prev
				}

				client.RemoveChannel(current)
			} else {
				current.Value.AppendMsg(msg.Timestamp, partMsg, msgOpts)
				delete(current.Value.Users, nick)
			}

			break
		}

		current = current.Next
		if current == client.RootChannel {
			break
		}
	}

	client.Tea.Send(cmds.SwitchChannels())
}

func handleTopic(msg irc.Message, client *irc.Client) {
	channel := msg.Parameters[0]
	topic := msg.Parameters[1]

	msgOpts := irc.MsgFmtOpts{
		AsServerMsg: true,
	}

	current := client.RootChannel
	for {
		if current.Value.Name == channel {
			current.Value.Topic = topic
			current.Value.AppendMsg(msg.Timestamp, fmt.Sprintf("Topic changed: %v", topic), msgOpts)
			break
		}

		current = current.Next
		if current == client.RootChannel {
			break
		}
	}
}

func handleCAP(msg irc.Message, client *irc.Client) {
	switch msg.Parameters[1] {
	case "LIST":
		fallthrough
	case "NEW":
		fallthrough
	case "LS":
		caps := strings.Split(msg.Parameters[2], " ")
		recognizedCaps := make([]irc.Capabilities, 0)
		capabilities := make(irc.Capabilities, 0)
		length := 0

		for _, capability := range caps {
			// NOTE: not sure if I should store the cap value the server advertises or use my own
			parts := strings.Split(capability, "=")
			key := parts[0]
			value := ""
			if len(parts) > 1 {
				value = parts[1]
			}

			if capEnabled, ok := commands.Capabilities[key]; ok {
				if capEnabled {
					// When the final parameter approaches 510 bytes,
					// we send multiple REQ commands
					// (https://ircv3.net/specs/extensions/capability-negotiation.html#the-cap-req-subcommand)
					if length+len(capability) >= 500 {
						recognizedCaps = append(recognizedCaps, capabilities)
						capabilities = make(irc.Capabilities, 0)
						length = 0
					}

					capabilities[key] = value
					length += len(capability)
				} else {
					log.Println("Unsupported capability:", capability)
				}
			} else {
				log.Println("Unknown capability:", capability)
			}
		}
		recognizedCaps = append(recognizedCaps, capabilities)

		for _, capsList := range recognizedCaps {
			list := make([]string, 0)
			for key, value := range capsList {
				if len(value) > 0 {
					list = append(list, key+"="+value)
				} else {
					list = append(list, key)
				}
			}
			client.SendCommand(commands.CAP, "REQ", strings.Join(list, " "))
		}
		client.SendCommand(commands.CAP, "END")
	case "ACK":
		if msg.Parameters[2] == "" {
			return
		}
		caps := strings.Split(msg.Parameters[2], " ")
		for _, capability := range caps {
			if capability[0] == '-' {
				delete(client.EnabledCapabilities, capability[1:])
			} else {
				parts := strings.Split(capability, "=")
				key := parts[0]
				value := ""
				if len(parts) > 1 {
					value = parts[1]
				}
				client.EnabledCapabilities[key] = value
			}
		}
	case "NAK":
		caps := strings.Split(msg.Parameters[2], " ")
		msgOpts := irc.MsgFmtOpts{
			WithTimestamp: true,
			AsServerMsg:   true,
		}
		client.RootChannel.Value.AppendMsg(msg.Timestamp, "Unrecognized capabilities: "+strings.Join(caps, " "), msgOpts)
	case "DEL":
		caps := strings.Split(msg.Parameters[2], " ")
		for _, capability := range caps {
			delete(client.EnabledCapabilities, capability)
		}
	}
}

func handleWELCOME(msg irc.Message, client *irc.Client) {
	nick := msg.Parameters[0]
	welcomeMsg := msg.Parameters[1]

	msgOpts := irc.MsgFmtOpts{
		WithTimestamp: true,
		AsServerMsg:   true,
	}

	// set server-registered nickname because the server MAY return a different nickname than
	// the one the user chose because of length restrictions or otherwise.
	client.Nickname = nick
	client.RootChannel.Value.AppendMsg(msg.Timestamp, welcomeMsg, msgOpts)

	// Only join the user-requested channel AFTER registration is complete.
	if client.InitialChannel != "" {
		client.SendCommand(commands.JOIN, client.InitialChannel)
	}
}

func handleYOURHOST(msg irc.Message, client *irc.Client) {
	host := msg.Parameters[1]

	msgOpts := irc.MsgFmtOpts{
		WithTimestamp: true,
		AsServerMsg:   true,
	}

	client.RootChannel.Value.AppendMsg(msg.Timestamp, host, msgOpts)
}

func handleCREATED(msg irc.Message, client *irc.Client) {
	created := msg.Parameters[1]

	msgOpts := irc.MsgFmtOpts{
		WithTimestamp: true,
		AsServerMsg:   true,
	}

	client.RootChannel.Value.AppendMsg(msg.Timestamp, created, msgOpts)
}

func handleMYINFO(msg irc.Message, client *irc.Client) {
	info := strings.Join(msg.Parameters[1:], " ")

	msgOpts := irc.MsgFmtOpts{
		WithTimestamp: true,
		AsServerMsg:   true,
	}

	client.RootChannel.Value.AppendMsg(msg.Timestamp, info, msgOpts)
}

func handleISUPPORT(msg irc.Message, client *irc.Client) {
	msgOpts := irc.MsgFmtOpts{
		WithTimestamp: true,
		AsServerMsg:   true,
	}

	for _, token := range msg.Parameters[1 : len(msg.Parameters)-1] {
		splits := strings.Split(token, "=")
		key := splits[0]
		value := ""
		if len(splits) > 1 {
			value = splits[1]
		}

		if supported, known := commands.Features[key]; known {
			if supported {
				client.EnabledFeatures[key] = value
			} else {
				log.Printf("Unsupported feature: \"%v\" with value: \"%v\"\n", key, value)
			}
		} else {
			log.Printf("Unknown feature: \"%v\" with value: \"%v\"\n", key, value)
		}

		client.RootChannel.Value.AppendMsg(msg.Timestamp, token, msgOpts)
	}
}

func handleLUSERCLIENT(msg irc.Message, client *irc.Client) {
	message := msg.Parameters[1]

	msgOpts := irc.MsgFmtOpts{
		WithTimestamp: true,
		AsServerMsg:   true,
	}

	client.RootChannel.Value.AppendMsg(msg.Timestamp, message, msgOpts)
}

func handleLUSEROP(msg irc.Message, client *irc.Client) {
	message := strings.Join(msg.Parameters[1:], " ")

	msgOpts := irc.MsgFmtOpts{
		WithTimestamp: true,
		AsServerMsg:   true,
	}

	client.RootChannel.Value.AppendMsg(msg.Timestamp, message, msgOpts)
}

func handleLUSERUNKNOWN(msg irc.Message, client *irc.Client) {
	message := strings.Join(msg.Parameters[1:], " ")

	msgOpts := irc.MsgFmtOpts{
		WithTimestamp: true,
		AsServerMsg:   true,
	}

	client.RootChannel.Value.AppendMsg(msg.Timestamp, message, msgOpts)
}

func handleLUSERCHANNELS(msg irc.Message, client *irc.Client) {
	message := strings.Join(msg.Parameters[1:], " ")

	msgOpts := irc.MsgFmtOpts{
		WithTimestamp: true,
		AsServerMsg:   true,
	}

	client.RootChannel.Value.AppendMsg(msg.Timestamp, message, msgOpts)
}

func handleLUSERME(msg irc.Message, client *irc.Client) {
	message := msg.Parameters[1]

	msgOpts := irc.MsgFmtOpts{
		WithTimestamp: true,
		AsServerMsg:   true,
	}

	client.RootChannel.Value.AppendMsg(msg.Timestamp, message, msgOpts)
}

func handleLOCALUSERS(msg irc.Message, client *irc.Client) {
	message := ""

	if len(msg.Parameters) > 2 {
		message = msg.Parameters[3]
	} else {
		message = msg.Parameters[1]
	}

	msgOpts := irc.MsgFmtOpts{
		WithTimestamp: true,
		AsServerMsg:   true,
	}

	client.RootChannel.Value.AppendMsg(msg.Timestamp, message, msgOpts)
}

func handleGLOBALUSERS(msg irc.Message, client *irc.Client) {
	message := ""

	if len(msg.Parameters) > 2 {
		message = msg.Parameters[3]
	} else {
		message = msg.Parameters[1]
	}

	msgOpts := irc.MsgFmtOpts{
		WithTimestamp: true,
		AsServerMsg:   true,
	}

	client.RootChannel.Value.AppendMsg(msg.Timestamp, message, msgOpts)
}

func handleAWAY(msg irc.Message, client *irc.Client) {
	nick := strings.SplitN(msg.Parameters[1], "!", 2)[0]
	reason := msg.Parameters[2]
	awayMsg := fmt.Sprintf("%s is away (%s)", nick, reason)

	msgOpts := irc.MsgFmtOpts{
		WithTimestamp: true,
		AsServerMsg:   true,
	}

	client.RootChannel.Value.AppendMsg(msg.Timestamp, awayMsg, msgOpts)
}

func handleUNAWAY(msg irc.Message, client *irc.Client) {
	message := msg.Parameters[1]

	msgOpts := irc.MsgFmtOpts{
		WithTimestamp: true,
		AsServerMsg:   true,
	}

	client.RootChannel.Value.AppendMsg(msg.Timestamp, message, msgOpts)
}

func handleNOWAWAY(msg irc.Message, client *irc.Client) {
	message := msg.Parameters[1]

	msgOpts := irc.MsgFmtOpts{
		WithTimestamp: true,
		AsServerMsg:   true,
	}

	client.RootChannel.Value.AppendMsg(msg.Timestamp, message, msgOpts)
}

func handleWHOISUSER(msg irc.Message, client *irc.Client) {
	nick := msg.Parameters[1]
	user := msg.Parameters[2]
	host := msg.Parameters[3]
	realName := msg.Parameters[5]

	msgOpts := irc.MsgFmtOpts{
		WithTimestamp: true,
		AsServerMsg:   true,
	}

	message := fmt.Sprintf(
		"Nick: %s | User: %s | Host: %s | Real name: %s",
		nick,
		user,
		host,
		realName,
	)

	client.RootChannel.Value.AppendMsg(msg.Timestamp, "WHOIS Information", msgOpts)
	client.RootChannel.Value.AppendMsg(msg.Timestamp, message, msgOpts)
}

func handleWHOISSERVER(msg irc.Message, client *irc.Client) {
	server := msg.Parameters[2]
	serverInfo := msg.Parameters[3]

	msgOpts := irc.MsgFmtOpts{
		WithTimestamp: true,
		AsServerMsg:   true,
	}

	message := fmt.Sprintf(
		"server: %s [%s]",
		server,
		serverInfo,
	)

	client.RootChannel.Value.AppendMsg(msg.Timestamp, message, msgOpts)
}

func handleWHOISIDLE(msg irc.Message, client *irc.Client) {
	idleSeconds := msg.Parameters[2]
	connectedTimestamp, err := strconv.ParseInt(msg.Parameters[3], 10, 64)

	if err != nil {
		// TODO: return this err and handle it in the parent
		log.Println(err)
	}

	msgOpts := irc.MsgFmtOpts{
		WithTimestamp: true,
		AsServerMsg:   true,
	}

	since := time.Unix(connectedTimestamp, 0)

	message := fmt.Sprintf(
		"idle for: %s seconds, connected since: %s",
		idleSeconds,
		since.Format(time.ANSIC),
	)
	client.RootChannel.Value.AppendMsg(msg.Timestamp, message, msgOpts)
}

func handleENDOFWHOIS(msg irc.Message, client *irc.Client) {
	message := msg.Parameters[2]

	msgOpts := irc.MsgFmtOpts{
		WithTimestamp: true,
		AsServerMsg:   true,
	}

	client.RootChannel.Value.AppendMsg(msg.Timestamp, message, msgOpts)
}

func handleWHOISCHANNELS(msg irc.Message, client *irc.Client) {
	chans := msg.Parameters[2]

	msgOpts := irc.MsgFmtOpts{
		WithTimestamp: true,
		AsServerMsg:   true,
	}

	message := fmt.Sprintf("channels: %s", chans)

	client.RootChannel.Value.AppendMsg(msg.Timestamp, message, msgOpts)
}

func handleNOTOPIC(msg irc.Message, client *irc.Client) {
	channel := msg.Parameters[1]
	msgStr := msg.Parameters[2]

	msgOpts := irc.MsgFmtOpts{
		AsServerMsg: true,
	}

	current := client.RootChannel
	for {
		if current.Value.Name == channel {
			current.Value.AppendMsg(msg.Timestamp, msgStr, msgOpts)
			break
		}

		current = current.Next
		if current == client.RootChannel {
			break
		}
	}
}

func handleTOPIC(msg irc.Message, client *irc.Client) {
	channel := msg.Parameters[1]
	topic := msg.Parameters[2]

	msgOpts := irc.MsgFmtOpts{
		AsServerMsg: true,
	}

	current := client.RootChannel
	for {
		if current.Value.Name == channel {
			current.Value.Topic = topic
			current.Value.AppendMsg(msg.Timestamp, fmt.Sprintf("TOPIC: %v", topic), msgOpts)
			break
		}

		current = current.Next
		if current == client.RootChannel {
			break
		}
	}
}

func handleVERSION(msg irc.Message, client *irc.Client) {
	version := msg.Parameters[1]
	server := msg.Parameters[2]
	versionMsg := fmt.Sprintf("Server: %v | Version: %v", server, version)

	if len(msg.Parameters) > 3 && msg.Parameters[3] != "" {
		comments := msg.Parameters[3]
		versionMsg = fmt.Sprintf("Server: %v | Version: %v | (%s)", server, version, comments)
	}

	msgOpts := irc.MsgFmtOpts{
		WithTimestamp: true,
		AsServerMsg:   true,
	}

	client.RootChannel.Value.AppendMsg(msg.Timestamp, versionMsg, msgOpts)
}

func handleNAMREPLY(msg irc.Message, client *irc.Client) {
	// TODO: do i need these
	// client := msg.Parameters[0]
	// chanSymbol := msg.Parameters[1]
	channel := msg.Parameters[2]
	nicks := strings.Split(msg.Parameters[3], " ")

	current := client.RootChannel
	for {
		if current.Value.Name == channel {
			for _, nick := range nicks {
				prefix := ""
				_nick := nick

				if commands.UserPrefixes[string(nick[0])] {
					prefix = string(nick[0])
					_nick = nick[1:]
				}

				current.Value.Users[_nick] = irc.User{
					Prefix: prefix,
				}
			}

			break
		}

		current = current.Next
		if current == client.RootChannel {
			break
		}
	}

	if channel == client.ActiveChannel.Value.Name {
		client.Tea.Send(cmds.SwitchChannels())
	}
}

func handleINFO(msg irc.Message, client *irc.Client) {
	message := msg.Parameters[1]

	msgOpts := irc.MsgFmtOpts{
		WithTimestamp: true,
		AsServerMsg:   true,
	}

	client.RootChannel.Value.AppendMsg(msg.Timestamp, message, msgOpts)
}

func handleENDOFINFO(msg irc.Message, client *irc.Client) {
	message := msg.Parameters[1]

	msgOpts := irc.MsgFmtOpts{
		WithTimestamp: true,
		AsServerMsg:   true,
	}

	client.RootChannel.Value.AppendMsg(msg.Timestamp, message, msgOpts)
}

func handleMOTDStart(msg irc.Message, client *irc.Client) {
	message := strings.Join(msg.Parameters[1:], " ")

	msgOpts := irc.MsgFmtOpts{
		WithTimestamp: true,
		AsServerMsg:   true,
	}

	client.RootChannel.Value.AppendMsg(msg.Timestamp, message, msgOpts)
}

func handleMOTD(msg irc.Message, client *irc.Client) {
	messageLine := msg.Parameters[1]

	msgOpts := irc.MsgFmtOpts{
		WithTimestamp: true,
		AsServerMsg:   true,
	}

	client.RootChannel.Value.AppendMsg(msg.Timestamp, messageLine, msgOpts)
}

func handleWHOISHOST(msg irc.Message, client *irc.Client) {
	message := msg.Parameters[1] + " " + msg.Parameters[2]

	msgOpts := irc.MsgFmtOpts{
		WithTimestamp: true,
		AsServerMsg:   true,
	}

	client.RootChannel.Value.AppendMsg(msg.Timestamp, message, msgOpts)
}

func handleWHOISMODES(msg irc.Message, client *irc.Client) {
	message := msg.Parameters[1] + " " + msg.Parameters[2]

	msgOpts := irc.MsgFmtOpts{
		WithTimestamp: true,
		AsServerMsg:   true,
	}

	client.RootChannel.Value.AppendMsg(msg.Timestamp, message, msgOpts)
}

func handleNOSUCHNICK(msg irc.Message, client *irc.Client) {
	message := msg.Parameters[1] + ": " + msg.Parameters[2]

	msgOpts := irc.MsgFmtOpts{
		WithTimestamp: true,
		AsErrorMsg:    true,
	}

	client.RootChannel.Value.AppendMsg(msg.Timestamp, message, msgOpts)
}

func handleNOSUCHSERVER(msg irc.Message, client *irc.Client) {
	message := msg.Parameters[1] + ": " + msg.Parameters[2]

	msgOpts := irc.MsgFmtOpts{
		WithTimestamp: true,
		AsErrorMsg:    true,
	}

	client.RootChannel.Value.AppendMsg(msg.Timestamp, message, msgOpts)
}

func handleNOSUCHCHANNEL(msg irc.Message, client *irc.Client) {
	channel := msg.Parameters[1]
	message := msg.Parameters[2]
	message = fmt.Sprintf("%v: %v", channel, message)

	msgOpts := irc.MsgFmtOpts{
		WithTimestamp: true,
		AsErrorMsg:    true,
	}

	client.RootChannel.Value.AppendMsg(msg.Timestamp, message, msgOpts)
}

func handleCANNOTSENDTOCHAN(msg irc.Message, client *irc.Client) {
	channel := msg.Parameters[1]
	message := msg.Parameters[2]

	msgOpts := irc.MsgFmtOpts{
		WithTimestamp: true,
		AsErrorMsg:    true,
	}

	current := client.RootChannel
	for {
		if current.Value.Name == channel {
			current.Value.AppendMsg(msg.Timestamp, message, msgOpts)
			return
		}

		current = current.Next
		if current == client.RootChannel {
			break
		}
	}

	message = fmt.Sprintf("%v: %v", channel, message)
	client.RootChannel.Value.AppendMsg(msg.Timestamp, message, msgOpts)
}

func handleTOOMANYCHANNELS(msg irc.Message, client *irc.Client) {
	channel := msg.Parameters[1]
	message := msg.Parameters[2]

	msgOpts := irc.MsgFmtOpts{
		WithTimestamp: true,
		AsErrorMsg:    true,
	}

	client.RootChannel.Value.AppendMsg(msg.Timestamp, fmt.Sprintf("%v: %v", channel, message), msgOpts)
}

func handleUNKNOWNCOMMAND(msg irc.Message, client *irc.Client) {
	message := msg.Parameters[1] + " " + msg.Parameters[2]

	msgOpts := irc.MsgFmtOpts{
		WithTimestamp: true,
		AsErrorMsg:    true,
	}

	client.RootChannel.Value.AppendMsg(msg.Timestamp, message, msgOpts)
}

func handleNOMOTD(msg irc.Message, client *irc.Client) {
	message := msg.Parameters[1]

	msgOpts := irc.MsgFmtOpts{
		WithTimestamp: true,
		AsServerMsg:   true,
	}

	client.RootChannel.Value.AppendMsg(msg.Timestamp, message, msgOpts)
}

func handleNONICKNAMEGIVEN(msg irc.Message, client *irc.Client) {
	message := msg.Parameters[1]

	msgOpts := irc.MsgFmtOpts{
		WithTimestamp: true,
		AsServerMsg:   true,
	}

	client.RootChannel.Value.AppendMsg(msg.Timestamp, message, msgOpts)
}

func handleNEEDMOREPARAMS(msg irc.Message, client *irc.Client) {
	message := msg.Parameters[1] + ": " + msg.Parameters[2]

	msgOpts := irc.MsgFmtOpts{
		WithTimestamp: true,
		AsErrorMsg:    true,
	}

	client.RootChannel.Value.AppendMsg(msg.Timestamp, message, msgOpts)
}

func handleALREADYREGISTERED(msg irc.Message, client *irc.Client) {
	message := msg.Parameters[1]

	msgOpts := irc.MsgFmtOpts{
		WithTimestamp: true,
		AsServerMsg:   true,
	}

	client.RootChannel.Value.AppendMsg(msg.Timestamp, message, msgOpts)
}

func handleBADCHANMASK(msg irc.Message, client *irc.Client) {
	channel := msg.Parameters[1]
	message := msg.Parameters[2]

	msgOpts := irc.MsgFmtOpts{
		WithTimestamp: true,
		AsErrorMsg:    true,
	}

	client.RootChannel.Value.AppendMsg(msg.Timestamp, fmt.Sprintf("%v: %v", channel, message), msgOpts)
}

func handleCHANOPRIVSNEEDED(msg irc.Message, client *irc.Client) {
	channel := msg.Parameters[1]
	message := msg.Parameters[2]

	msgOpts := irc.MsgFmtOpts{
		WithTimestamp: true,
		AsErrorMsg:    true,
	}

	current := client.RootChannel
	for {
		if current.Value.Name == channel {
			current.Value.AppendMsg(msg.Timestamp, message, msgOpts)
			break
		}

		current = current.Next
		if current == client.RootChannel {
			break
		}
	}
}

func HandleCommand(msg irc.Message, client *irc.Client) {
	// TODO: handle different commands
	switch msg.Command {
	case commands.PING:
		handlePing(msg, client)
	case commands.PRIVMSG:
		handlePrivMsg(msg, client)
	case commands.NOTICE:
		handleNotice(msg, client)
	case commands.JOIN:
		handleJoin(msg, client)
	case commands.NICK:
		handleNick(msg, client)
	case commands.KICK:
		handleKick(msg, client)
	case commands.QUIT:
		handleQuit(msg, client)
	case commands.PART:
		handlePart(msg, client)
	case commands.TOPIC:
		handleTopic(msg, client)
	case commands.CAP:
		handleCAP(msg, client)
	case commands.RPL_WELCOME:
		handleWELCOME(msg, client)
	case commands.RPL_YOURHOST:
		handleYOURHOST(msg, client)
	case commands.RPL_CREATED:
		handleCREATED(msg, client)
	case commands.RPL_MYINFO:
		handleMYINFO(msg, client)
	case commands.RPL_ISUPPORT:
		handleISUPPORT(msg, client)
	case commands.RPL_LUSERCLIENT:
		handleLUSERCLIENT(msg, client)
	case commands.RPL_LUSEROP:
		handleLUSEROP(msg, client)
	case commands.RPL_LUSERUNKNOWN:
		handleLUSERUNKNOWN(msg, client)
	case commands.RPL_LUSERCHANNELS:
		handleLUSERCHANNELS(msg, client)
	case commands.RPL_LUSERME:
		handleLUSERME(msg, client)
	case commands.RPL_LOCALUSERS:
		handleLOCALUSERS(msg, client)
	case commands.RPL_GLOBALUSERS:
		handleGLOBALUSERS(msg, client)
	case commands.RPL_AWAY:
		handleAWAY(msg, client)
	case commands.RPL_UNAWAY:
		handleUNAWAY(msg, client)
	case commands.RPL_NOWAWAY:
		handleNOWAWAY(msg, client)
	case commands.RPL_WHOISUSER:
		handleWHOISUSER(msg, client)
	case commands.RPL_WHOISSERVER:
		handleWHOISSERVER(msg, client)
	case commands.RPL_WHOISIDLE:
		handleWHOISIDLE(msg, client)
	case commands.RPL_ENDOFWHOIS:
		handleENDOFWHOIS(msg, client)
	case commands.RPL_WHOISCHANNELS:
		handleWHOISCHANNELS(msg, client)
	case commands.RPL_NOTOPIC:
		handleNOTOPIC(msg, client)
	case commands.RPL_TOPIC:
		handleTOPIC(msg, client)
	case commands.RPL_VERSION:
		handleVERSION(msg, client)
	case commands.RPL_NAMREPLY:
		handleNAMREPLY(msg, client)
	case commands.RPL_ENDOFNAMES:
		// TODO: toggle a flag on the channel to indicate
		// it received all the names correctly.

		// start a timeout and update said timeout on every RPL_NAMREPLY
		// and log an error if timeout ends without receiving this command.
	case commands.RPL_INFO:
		handleINFO(msg, client)
	case commands.RPL_ENDOFINFO:
		handleENDOFINFO(msg, client)
	case commands.RPL_MOTDSTART:
		handleMOTDStart(msg, client)
	case commands.RPL_MOTD:
		handleMOTD(msg, client)
	case commands.RPL_ENDOFMOTD:
		// TODO: toggle a flag on the client/server to indicate
		// it received all the MOTD correctly

		// start a timeout and update said timeout on every RPL_MOTD
		// and log an error if timeout ends without receiving this command.
	case commands.RPL_WHOISHOST:
		handleWHOISHOST(msg, client)
	case commands.RPL_WHOISMODES:
		handleWHOISMODES(msg, client)
	case commands.ERR_NOSUCHNICK:
		handleNOSUCHNICK(msg, client)
	case commands.ERR_NOSUCHSERVER:
		handleNOSUCHSERVER(msg, client)
	case commands.ERR_NOSUCHCHANNEL:
		handleNOSUCHCHANNEL(msg, client)
	case commands.ERR_CANNOTSENDTOCHAN:
		handleCANNOTSENDTOCHAN(msg, client)
	case commands.ERR_TOOMANYCHANNELS:
		handleTOOMANYCHANNELS(msg, client)
	case commands.ERR_UNKNOWNCOMMAND:
		handleUNKNOWNCOMMAND(msg, client)
	case commands.ERR_NOMOTD:
		handleNOMOTD(msg, client)
	case commands.ERR_NONICKNAMEGIVEN:
		handleNONICKNAMEGIVEN(msg, client)
	case commands.ERR_NEEDMOREPARAMS:
		handleNEEDMOREPARAMS(msg, client)
	case commands.ERR_ALREADYREGISTERED:
		handleALREADYREGISTERED(msg, client)
	case commands.ERR_BADCHANMASK:
		handleBADCHANMASK(msg, client)
	case commands.ERR_CHANOPRIVSNEEDED:
		handleCHANOPRIVSNEEDED(msg, client)
	default:
		fullMsg := fmt.Sprintf(
			"%s %s %s %s",
			msg.Tags,
			msg.Source,
			msg.Command,
			strings.Join(msg.Parameters, " "),
		)

		msgOpts := irc.MsgFmtOpts{
			WithTimestamp: true,
			NotImpl:       true,
		}

		client.RootChannel.Value.AppendMsg(msg.Timestamp, fullMsg, msgOpts)
	}

	// send a receivedIRCmsg tea message so the ui can update
	// we also use this tea message to scroll the viewport down
	client.Tea.Send(cmds.ReceivedIRCMsg())
}