user_messages.go

package demoinfocs

import (
	"fmt"

	"github.com/markus-wa/go-unassert"

	"github.com/markus-wa/demoinfocs-golang/events"
	"github.com/markus-wa/demoinfocs-golang/msg"
)

func (p *Parser) handleUserMessage(um *msg.CSVCMsg_UserMessage) {
	p.userMessageHandler.handler(msg.ECstrike15UserMessages(um.MsgType))(um)
}

type userMessageHandler struct {
	parser           *Parser
	msgTypeToHandler map[msg.ECstrike15UserMessages]userMessageHandlerFunc
}

func (umh userMessageHandler) handler(msgType msg.ECstrike15UserMessages) userMessageHandlerFunc {
	if handler, eventKnown := umh.msgTypeToHandler[msgType]; eventKnown {
		return handler
	}
	return func(*msg.CSVCMsg_UserMessage) { /* NOP */ }
}

func (umh userMessageHandler) dispatch(event interface{}) {
	umh.parser.eventDispatcher.Dispatch(event)
}

func (umh userMessageHandler) gameState() *GameState {
	return umh.parser.gameState
}

type userMessageHandlerFunc func(*msg.CSVCMsg_UserMessage)

func newUserMessageHandler(parser *Parser) userMessageHandler {
	umh := userMessageHandler{parser: parser}

	umh.msgTypeToHandler = map[msg.ECstrike15UserMessages]userMessageHandlerFunc{
		msg.ECstrike15UserMessages_CS_UM_SayText:          umh.sayText,
		msg.ECstrike15UserMessages_CS_UM_SayText2:         umh.sayText2,
		msg.ECstrike15UserMessages_CS_UM_ServerRankUpdate: umh.rankUpdate,
		// TODO: handle more user messages (if they are interesting)
		// Maybe msg.ECstrike15UserMessages_CS_UM_RadioText
	}

	return umh
}

func (umh userMessageHandler) sayText(um *msg.CSVCMsg_UserMessage) {
	st := new(msg.CCSUsrMsg_SayText)
	err := st.Unmarshal(um.MsgData)
	if err != nil {
		umh.dispatch(events.ParserWarn{Message: fmt.Sprintf("failed to decode SayText message: %s", err.Error())})
		unassert.Error("failed to decode SayText message: %s", err.Error())
	}

	umh.dispatch(events.SayText{
		EntIdx:    int(st.EntIdx),
		IsChat:    st.Chat,
		IsChatAll: st.Textallchat,
		Text:      st.Text,
	})
}

func (umh userMessageHandler) sayText2(um *msg.CSVCMsg_UserMessage) {
	st := new(msg.CCSUsrMsg_SayText2)
	err := st.Unmarshal(um.MsgData)
	if err != nil {
		umh.dispatch(events.ParserWarn{Message: fmt.Sprintf("failed to decode SayText2 message: %s", err.Error())})
		unassert.Error("failed to decode SayText2 message: %s", err.Error())
	}

	umh.dispatch(events.SayText2{
		EntIdx:    int(st.EntIdx),
		IsChat:    st.Chat,
		IsChatAll: st.Textallchat,
		MsgName:   st.MsgName,
		Params:    st.Params,
	})

	switch st.MsgName {
	case "Cstrike_Chat_All":
		fallthrough
	case "Cstrike_Chat_AllDead":
		sender := umh.gameState().playersByEntityID[int(st.EntIdx)]

		umh.dispatch(events.ChatMessage{
			Sender:    sender,
			Text:      st.Params[1],
			IsChatAll: st.Textallchat,
		})

	case "#CSGO_Coach_Join_T": // Ignore these
	case "#CSGO_Coach_Join_CT":
	case "#Cstrike_Name_Change":
	case "Cstrike_Chat_T_Loc":
	case "Cstrike_Chat_CT_Loc":
	case "Cstrike_Chat_T_Dead":
	case "Cstrike_Chat_CT_Dead":

	default:
		umh.dispatch(events.ParserWarn{Message: fmt.Sprintf("skipped sending ChatMessageEvent for SayText2 with unknown MsgName %q", st.MsgName)})
		unassert.Error("skipped sending ChatMessageEvent for SayText2 with unknown MsgName %q", st.MsgName)
	}
}

func (umh userMessageHandler) rankUpdate(um *msg.CSVCMsg_UserMessage) {
	st := new(msg.CCSUsrMsg_ServerRankUpdate)
	err := st.Unmarshal(um.MsgData)
	if err != nil {
		umh.dispatch(events.ParserWarn{Message: fmt.Sprintf("failed to decode ServerRankUpdate message: %s", err.Error())})
		unassert.Error("failed to decode ServerRankUpdate message: %s", err.Error())
	}

	for _, v := range st.RankUpdate {
		umh.dispatch(events.RankUpdate{
			SteamID:    int64(v.AccountId),
			SteamID32:  v.AccountId,
			RankOld:    int(v.RankOld),
			RankNew:    int(v.RankNew),
			WinCount:   int(v.NumWins),
			RankChange: v.RankChange,
		})
	}
}