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

package gircclient

import (
	"strings"

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

// 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 featuresHandler(event string, info eventmgr.InfoMap) {
	sc := info["server"].(*ServerConnection)

	// parse features into our internal list
	tags := info["params"].([]string)
	tags = tags[1 : len(tags)-1] // remove first and last params
	sc.Features.Parse(tags...)

	if sc.Casemapping == ircmap.NONE {
		name, exists := sc.Features["CASEMAPPING"]
		if exists {
			sc.Casemapping = ircmap.Mappings[name.(string)]
		}
	}
}

func capHandler(event string, info eventmgr.InfoMap) {
	sc := info["server"].(*ServerConnection)
	params := info["params"].([]string)
	subcommand := strings.ToUpper(params[1])

	if subcommand == "ACK" {
		sc.Caps.EnableCaps(strings.Split(params[2], " ")...)
	} 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")
			}
		}
	} else if subcommand == "NEW" {
		sc.Caps.AddCaps(strings.Split(params[2], " ")...)
		capsToRequest := sc.Caps.ToRequestLine()

		if len(capsToRequest) > 0 {
			sc.Send(nil, "", "CAP", "REQ", capsToRequest)
		}
	} else if subcommand == "DEL" {
		sc.Caps.DelCaps(strings.Split(params[2], " ")...)
	}

	if !sc.Registered && (subcommand == "ACK" || subcommand == "NAK") {
		sendRegistration(sc)
	}
}

func pingHandler(event string, info eventmgr.InfoMap) {
	sc := info["server"].(*ServerConnection)
	sc.Send(nil, "", "PONG", info["params"].([]string)...)
}

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

	sc.Registered = true
}