chirpnest/client.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
		if cx.Config.Wartime == `` {
			cx.Config.Wartime = `🛡Defend`
		}

		/*
			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.CWIdle = 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.CWIdle = true
			clt.CWLastUpdate = from
		} else {
		}
		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.CWIdle {
			ids = append(ids, c.TGUserID64)
		}
	}
	muxClients.RUnlock()
	if len(ids) == 0 {
		return nil, errors.New("No idle client.")
	}

	RndMux.Lock()
	id := RndSrc.Intn(len(ids))
	RndMux.Unlock()

	clients[ids[id]].Mux.Lock()

	return clients[ids[id]], nil

}

func getLockedAllIdleClientID64() ([]int64, error) {
	muxClients.RLock()
	ids := make([]int64, 0)
	for _, c := range clients {
		if c.CWIdle {
			ids = append(ids, c.TGUserID64)
			clients[c.TGUserID64].Mux.Lock()
		}
	}
	muxClients.RUnlock()

	return ids, nil
}

func getAllIdleClientID64() ([]int64, error) {
	muxClients.RLock()
	ids := make([]int64, 0)
	for _, c := range clients {
		if c.CWIdle {
			ids = append(ids, c.TGUserID64)
		}
	}
	muxClients.RUnlock()

	return ids, 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,
			Config: &ChirpConfig{
				InterceptPillage: true,
				Wartime:          `🛡Defend`,
			},
		}
		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.CWIdle = 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
}