irc-go/client/handlers.go

// written by Daniel Oaks <daniel@danieloaks.net>
// released under the ISC license

package gircclient

import (
	"strings"

	"github.com/DanielOaks/girc-go/eventmgr"
)

// welcomeHandler sets the nick to the first parameter of the 001 message.
// This ensures that when we connect to IRCds that silently truncate the
// nickname, we keep the correct one.
func welcomeHandler(event string, info eventmgr.InfoMap) {
	sc := info["server"].(*ServerConnection)
	sc.Nick = info["params"].([]string)[0]
}

func capHandler(event string, info eventmgr.InfoMap) {
	sc := info["server"].(*ServerConnection)
	params := info["params"].([]string)
	subcommand := strings.ToUpper(params[1])
	if !sc.Registered && (subcommand == "ACK" || subcommand == "NAK") {
		sendRegistration(sc)
	} else if subcommand == "LS" {
		if len(params) > 3 {
			sc.Caps.AddCaps(strings.Split(params[3], " "))
		} else {
			sc.Caps.AddCaps(strings.Split(params[2], " "))
			capsToRequest := sc.Caps.ToRequestLine()

			if len(capsToRequest) > 0 {
				sc.Send(nil, "", "CAP", "REQ", capsToRequest)
			}

			if !sc.Registered {
				sc.Send(nil, "", "CAP", "END")
			}
		}
	}
}

func sendRegistration(sc *ServerConnection) {
	sc.Nick = sc.InitialNick
	sc.Send(nil, "", "NICK", sc.InitialNick)
	sc.Send(nil, "", "USER", sc.InitialUser, "0", "*", sc.InitialRealName)
}