chirpnest/client.go

package main

import (
	"errors"
	"fmt"
	"strings"
	"time"
)

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

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 _, ok := clientsQueue[m.Msg.UserID64]; ok {
		if v, ok := clientsCW.Load(m.Msg.UserID64); ok {
			c := v.(*ChatWarsClient)
			if c.LastUpdate.Before(m.Msg.Date) {
				c.GuildID64 = m.GuildID64
				c.State = m.State
				c.LastUpdate = m.Msg.Date
				if getObjGuildID(``) != m.GuildID64 && strings.Compare(c.Role, ``) == 0 {
					clientSendCWMsg(m.Msg.UserID64, "/g_roles")
				}
			}
		} else {
			c := ChatWarsClient{
				UserID64:     m.UserID64,
				TelegramID64: m.Msg.UserID64,
				GuildID64:    m.GuildID64,
				State:        m.State,
				LastUpdate:   m.Msg.Date,
			}
			clientsCW.Store(m.Msg.UserID64, &c)
			if getObjGuildID(``) != m.GuildID64 {
				clientSendCWMsg(m.Msg.UserID64, "/g_roles")
			}
		}
	}
}

func clientMsgGoQuestAck(m *ChatWarsMessageGoQuestAck) {
	if _, ok := clientsQueue[m.Msg.UserID64]; ok {
		if v, ok := clientsCW.Load(m.Msg.UserID64); ok {
			c := v.(*ChatWarsClient)
			if c.LastUpdate.Before(m.Msg.Date) {
				c.LastUpdate = m.Msg.Date
				c.BusyUntil = m.Msg.Date.Add(m.Duration)
			}
		}
	}
}

func clientMsgGRolesAck(m *ChatWarsMessageGRolesAck) {
	if _, ok := clientsQueue[m.Msg.UserID64]; ok {
		if v, ok := clientsCW.Load(m.Msg.UserID64); ok {
			c := v.(*ChatWarsClient)
			if c.LastUpdate.Before(m.Msg.Date) {
				if m.CommanderID64 == c.UserID64 {
					c.Role = `commander`
				} else if m.BartenderID64 == c.UserID64 {
					c.Role = `bartender`
				} else if m.SquireID64 == c.UserID64 {
					c.Role = `squire`
				} else if m.TreasurerID64 == c.UserID64 {
					c.Role = `treasurer`
				} else {
					c.Role = `none`
				}
				c.LastUpdate = m.Msg.Date
			}
		}
	}
}

func clientGetUserID64(tgUserID64 int64) (int64, error) {
	if v, ok := clientsCW.Load(tgUserID64); ok {
		c := v.(*ChatWarsClient)
		return c.UserID64, nil
	}
	return 0, errors.New("Unknown user_id.")
}