chirpnest/client.go
2020-02-05 11:29:30 +08:00

331 lines
7.7 KiB
Go

package main
import (
"encoding/json"
"errors"
"fmt"
"io/ioutil"
"log"
"strings"
"time"
"github.com/jinzhu/copier"
)
func loadClients() error {
var clts []ChirpClient
b, err := ioutil.ReadFile("./clients.json")
logOnError(err, "loadClients : ReadFile(./clients.json)")
if err != nil {
return err
}
err = json.Unmarshal(b, &clts)
logOnError(err, "loadClients : Unmarshal(./clients.json)")
if err != nil {
return err
}
muxClients.Lock()
for _, c := range clts {
cx := ChirpClient{}
copier.Copy(&cx, &c)
cx.Active = false
/*
cx.MQ.Connection, err = amqp.Dial("amqp://" + cx.MQ.User + ":" + cx.MQ.Password + "@" + cx.MQ.Host + "/" + cx.MQ.Path)
logOnError(err, "loadClients : Failed to connect to RabbitMQ")
*/
clients[cx.TGUserID64] = &cx
log.Printf("loadClients[%s] : done\n", cx.Login)
}
muxClients.Unlock()
return nil
}
func getLockedRoleClient(role string) (*ChirpClient, error) {
muxClients.RLock()
defer muxClients.RUnlock()
ids := make([]int64, 0)
for _, c := range clients {
if c.CWRole == role {
ids = append(ids, c.TGUserID64)
fmt.Printf("getLockedRoleClient(%s) : appending %s (%d).\n", role, c.Login, c.TGUserID64)
}
}
if len(ids) == 0 {
return nil, errors.New("No client for the role.")
}
RndMux.Lock()
id := RndSrc.Intn(len(ids))
RndMux.Unlock()
clients[ids[id]].Mux.Lock()
return clients[ids[id]], nil
}
func getLockedRandomClient() (*ChirpClient, error) {
muxClients.RLock()
ids := make([]int64, 0)
for _, c := range clients {
if c.Active {
ids = append(ids, c.TGUserID64)
}
}
muxClients.RUnlock()
if len(ids) == 0 {
return nil, errors.New("No active client.")
}
RndMux.Lock()
id := RndSrc.Intn(len(ids))
RndMux.Unlock()
clients[ids[id]].Mux.Lock()
return clients[ids[id]], nil
}
func setClientBusy(userID64 int64, from time.Time, duration time.Duration) error {
if clt, ok := getLockedClient(userID64, false); ok {
if from.UTC().Add(duration).After(time.Now().UTC()) {
clt.GameIdle = false
clt.CWBusyUntil = from.UTC().Add(duration)
log.Printf("setClientBusy[%s] : set for %s.\n", clt.Login, duration.String())
} else {
log.Printf("setClientBusy[%s] : not updated.\n", clt.Login)
}
clt.Mux.Unlock()
return nil
} else {
return errors.New("Client not found.")
}
}
func setClientIdle(userID64 int64, from time.Time) error {
if clt, ok := getLockedClient(userID64, false); ok {
if from.UTC().After(clt.CWLastUpdate.UTC()) {
clt.CWBusyUntil = from
clt.GameIdle = true
clt.CWLastUpdate = from
log.Printf("setClientIdle[%s] : updated.\n", clt.Login)
} else {
log.Printf("setClientIdle[%s] : not updated.\n", clt.Login)
}
clt.Mux.Unlock()
return nil
} else {
return errors.New("Client not found.")
}
}
func getLockedIdleClient() (*ChirpClient, error) {
muxClients.RLock()
ids := make([]int64, 0)
for _, c := range clients {
if c.GameIdle {
ids = append(ids, c.TGUserID64)
fmt.Printf("getLockedIdleClient : appending %s (%d).\n", c.Login, c.TGUserID64)
}
}
muxClients.RUnlock()
if len(ids) == 0 {
return nil, errors.New("No idle client.")
}
RndMux.Lock()
id := RndSrc.Intn(len(ids))
fmt.Printf("getLockedIdleClient : pulled %s.\n", clients[ids[id]].Login)
RndMux.Unlock()
clients[ids[id]].Mux.Lock()
return clients[ids[id]], nil
}
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 := ChirpClient{
TGUserID64: id,
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
return nil, false
}
}
func clientDelTGMsg(userID64 int64, fromMsgID64 int64, fromChatID64 int64) {
c := TGCommand{
Type: commandDeleteMsg,
FromUserID64: userID64,
FromMsgID64: fromMsgID64,
FromChatID64: fromChatID64,
}
MQTGCmdQueue <- c
}
func clientFwdCWMsg(userID64 int64, fromMsgID64 int64, fromChatID64 int64, toChatID64 int64) {
c := TGCommand{
Type: commandForwardMsg,
FromUserID64: userID64,
FromMsgID64: fromMsgID64,
FromChatID64: fromChatID64,
ToChatID64: toChatID64,
}
MQTGCmdQueue <- c
}
func clientSendTGMsgDelay(userID64 int64, chatID64 int64, s string, d time.Duration) {
c := TGCommand{
Type: commandSendMsg,
Text: s,
FromUserID64: userID64,
ToChatID64: chatID64,
Delay: d,
}
MQTGCmdQueue <- c
}
func clientSendTGMsg(userID64 int64, chatID64 int64, s string) {
clientSendTGMsgDelay(userID64, chatID64, s, 0)
}
func clientSendCWMsg(userID64 int64, s string) {
clientSendCWMsgDelay(userID64, s, 0)
}
func clientSendCWMsgDelay(userID64 int64, s string, d time.Duration) {
clientSendTGMsgDelay(userID64, userID64ChtWrsBot, s, d)
}
func clientRefreshCWMsg(userID64 int64, chatID64 int64, msgID64 int64) {
c := TGCommand{
Type: commandRefreshMsg,
FromUserID64: userID64,
FromChatID64: chatID64,
FromMsgID64: msgID64,
Delay: 0,
}
MQTGCmdQueue <- c
}
func clientCallbackDelay(userID64 int64, msgID64 int64, chatID64 int64, name string, data []byte, d time.Duration) {
c := TGCommand{
Type: commandCallback,
FromUserID64: userID64,
FromChatID64: chatID64,
FromMsgID64: msgID64,
Delay: d,
CallbackName: name,
CallbackData: data,
}
MQTGCmdQueue <- c
}
func clientCallback(userID64 int64, msgID64 int64, chatID64 int64, name string, data []byte) {
clientCallbackDelay(userID64, msgID64, chatID64, name, data, 0)
}
func clientMsgMeAck(m *ChatWarsMessageMeAck) {
if clt, ok := getLockedClient(m.Msg.TGUserID64, false); ok {
if clt.Active {
if clt.CWLastUpdate.Before(m.Msg.Date) {
clt.CWGuildID64 = m.CWGuildID64
clt.CWUserID64 = m.CWUserID64
clt.CWState = m.State
clt.CWClass = m.Class
clt.CWLastUpdate = m.Msg.Date
if getObjGuildID(``) != m.CWGuildID64 && strings.Compare(clt.CWRole, ``) == 0 {
clientSendCWMsg(m.Msg.TGUserID64, "/g_roles")
}
if m.State == `🛌Rest` {
clt.GameIdle = true
}
}
}
clt.Mux.Unlock()
}
}
func clientMsgGoQuestAck(m *ChatWarsMessageGoQuestAck) {
if clt, ok := getLockedClient(m.Msg.TGUserID64, 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.TGUserID64, 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.")
}
func clientSpreadQuestResultAmbush(cwm *ChatWarsMessageQuestResultAmbush) error {
/*
muxClients.RLock()
var ret string
for id, c := range clients {
if c.Active {
ret = fmt.Sprintf("%s%s | UserID : %d | TelegramID : %d (online)\n", ret, c.Login, c.CWUserID64, id)
} else {
ret = fmt.Sprintf("%s%s | UserID : %d | TelegramID : %d (offline)\n", ret, c.Login, c.CWUserID64, id)
}
}
muxClients.RUnlock()
*/
log.Printf("clientSpreadQuestResultAmbush : spreading.")
return nil
}