chirpnest/client.go

package main

import (
	"errors"
	"strings"
)

func getLockedClient(id int64, createMissing bool) (*ChirpClient, bool) {
	muxClients.RLock()
	if c, ok := clients[id]; ok {
		c.Mux.Lock()
		muxClients.RUnlock()
		return c, true
	} else if createMissing {
		c := new(ChirpClient)
		c.TGUserID64 = id
		c.Active = false
		c.Mux.Lock()
		muxClients.RUnlock()
		muxClients.Lock()
		clients[id] = c
		muxClients.Unlock()
		return c, true
	} else {
		muxClients.RUnlock()
		c := new(ChirpClient)
		return c, false
	}
}

func clientSendCWMsg(userID64 int64, s string) {
	c := TGCommand{
		Type:         commandSendMsg,
		Text:         s,
		FromUserID64: userID64,
		ToChatID64:   userID64ChtWrsBot,
	}
	MQTGCmdQueue <- c
}

func clientRefreshCWMsg(userID64 int64, chatID64 int64, msgID64 int64) {
	c := TGCommand{
		Type:         commandRefreshMsg,
		FromUserID64: userID64,
		FromChatID64: chatID64,
		FromMsgID64:  msgID64,
	}
	MQTGCmdQueue <- c
}

func clientMsgMeAck(m *ChatWarsMessageMeAck) {
	if clt, ok := getLockedClient(m.Msg.UserID64, false); ok {
		if clt.Active {
			if clt.CWLastUpdate.Before(m.Msg.Date) {
				clt.CWGuildID64 = m.GuildID64
				clt.CWUserID64 = m.UserID64
				clt.CWState = m.State
				clt.CWLastUpdate = m.Msg.Date
				if getObjGuildID(``) != m.GuildID64 && strings.Compare(clt.CWRole, ``) == 0 {
					clientSendCWMsg(m.Msg.UserID64, "/g_roles")
				}
			}
		}
		clt.Mux.Unlock()
	}
}

func clientMsgGoQuestAck(m *ChatWarsMessageGoQuestAck) {
	if clt, ok := getLockedClient(m.Msg.UserID64, false); ok {
		if clt.Active {
			if clt.CWLastUpdate.Before(m.Msg.Date) {
				clt.CWLastUpdate = m.Msg.Date
				clt.CWBusyUntil = m.Msg.Date.Add(m.Duration)
			}
		}
		clt.Mux.Unlock()
	}
}

func clientMsgGRolesAck(m *ChatWarsMessageGRolesAck) {
	if clt, ok := getLockedClient(m.Msg.UserID64, false); ok {
		if clt.Active {
			if clt.CWLastUpdate.Before(m.Msg.Date) {
				if m.CommanderID64 == clt.CWUserID64 {
					clt.CWRole = `commander`
				} else if m.BartenderID64 == clt.CWUserID64 {
					clt.CWRole = `bartender`
				} else if m.SquireID64 == clt.CWUserID64 {
					clt.CWRole = `squire`
				} else if m.TreasurerID64 == clt.CWUserID64 {
					clt.CWRole = `treasurer`
				} else {
					clt.CWRole = `none`
				}
				clt.CWLastUpdate = m.Msg.Date
			}
		}
		clt.Mux.Unlock()
	}
}

func clientGetCWUserID64(tgUserID64 int64) (int64, error) {
	if clt, ok := getLockedClient(tgUserID64, false); ok {
		i := clt.CWUserID64
		clt.Mux.Unlock()
		return i, nil
	}
	return 0, errors.New("Unknown user_id.")
}