package main import ( "encoding/json" "fmt" "log" "math/rand" "strconv" "strings" "time" tb "gopkg.in/tucnak/telebot.v2" ) func MQGetMsgWorker(id int, msgs chan<- ChatWarsMessage) { //log.Printf("MQGetMsgWorker[" + strconv.Itoa(id) + "] : Starting.") var x ChatWarsMessage s := MQSession{ User: cfg.Rabbit.User, Password: cfg.Rabbit.Password, Host: cfg.Rabbit.Host, Path: cfg.Rabbit.Path, SSL: false, Queue: "msg", isConnected: false, } for true { err := s.Open() for err != nil { logOnError(err, "MQGetMsgWorker["+strconv.Itoa(id)+"] : Connection to RabbitMQ failed.") time.Sleep(15) err = s.Open() } log.Printf("MQGetMsgWorker[" + strconv.Itoa(id) + "] : Connected to RabbitMQ") for d := range s.MQDelivery { // log.Printf("MQGetMsgWorker["+strconv.Itoa(id)+"] : Received a message: %s", string(d.Body)) err = json.Unmarshal(d.Body, &x) logOnError(err, "MQGetMsgWorker["+strconv.Itoa(id)+"] : Can't unmarshal.\n"+string(d.Body)) if err == nil { msgs <- x } } s.Close() } log.Printf("MQGetMsgWorker[" + strconv.Itoa(id) + "] : Closing.") } func SQLCWMsgWorker(id int, msgs <-chan ChatWarsMessage, objIds chan<- int64) { //log.Printf("SQLCWMsgWorker[" + strconv.Itoa(id) + "] : Starting.") for m := range msgs { objId, err := addObjMsg(m.ID64, m.ChatID64, m.UserID64, m.SenderUserID64, m.Date, m.Text) logOnError(err, "SQLCWMsgWorker["+strconv.Itoa(id)+"] : Inserting message.") if err == nil && objId != 0 { // log.Printf("SQLCWMsgWorker["+strconv.Itoa(id)+"] : Message inserted (%d).\n", objId) objIds <- objId } } log.Printf("SQLCWMsgWorker[" + strconv.Itoa(id) + "] : Closing.") } func SQLIdentifyMsgWorker(id int, objIds <-chan int64) { var i int //log.Printf("SQLIdentifyMsgWorker[" + strconv.Itoa(id) + "] : Starting.") for objId := range objIds { m, err := getObjMsg(objId) logOnError(err, "SQLIdentifyMsgWorker["+strconv.Itoa(id)+"] : Retrieving message.") if err == nil { //log.Printf("SQLIdentifyMsgWorker["+strconv.Itoa(id)+"] : Message retrieved (%d)\n%s\n", objId, m.Text) for i = 0; i < len(msgParsingRules) && msgParsingRules[i].re.FindStringSubmatch(m.Text) == nil; i++ { } if i == len(msgParsingRules) { log.Printf("SQLIdentifyMsgWorker["+strconv.Itoa(id)+"] : Can't find message type in rules(%d)\n%s\n", objId, m.Text) } else { r := msgParsingRules[i].re err = setObjSubTypeId(objId, msgParsingRules[i].MsgTypeID) logOnError(err, "SQLIdentifyMsgWorker["+strconv.Itoa(id)+"] : setObjSubTypeId") switch msgParsingRules[i].MsgTypeID { case objSubTypeMessageGRolesReq: case objSubTypeMessageGo: case objSubTypeMessageReportReq: case objSubTypeMessageGReportReq: case objSubTypeMessageHeroReq: case objSubTypeMessageMeReq: case objSubTypeMessageInventoryReq: case objSubTypeMessageTimeReq: case objSubTypeMessagePledge: case objSubTypeMessagePillageGo: case objSubTypeMessagePillageTimeout: case objSubTypeMessagePillageWin: case objSubTypeMessagePillageLoss: case objSubTypeMessageGoQuest: case objSubTypeMessageGoFastFight: case objSubTypeMessageGoArena: case objSubTypeMessageTop: case objSubTypeMessageMenu: case objSubTypeMessageBuyReq: case objSubTypeMessageSellReq: case objSubTypeMessageOrderbookReq: case objSubTypeMessageOrderbookAck: case objSubTypeMessageWithdrawReq: case objSubTypeMessageWithdrawCode: case objSubTypeMessageWithdrawRcv: case objSubTypeMessageDepositReq: case objSubTypeMessageDepositAck: case objSubTypeMessageStockReq: case objSubTypeMessageMiscReq: case objSubTypeMessageGStockAnyAck: _, err := parseSubTypeMessageGStockAnyAck(m, r) logOnError(err, "SQLIdentifyMsgWorker["+strconv.Itoa(id)+"] : Parsing objSubTypeMessageGStockAnyAck.") case objSubTypeMessageReportAck: cwm, err := parseSubTypeMessageReportAck(m, r) logOnError(err, "SQLIdentifyMsgWorker["+strconv.Itoa(id)+"] : Parsing objSubTypeMessageReportAck.") cwm.ObjID64 = objId case objSubTypeMessageMiniWar: cwm, err := parseSubTypeMessageMiniWar(m, r) logOnError(err, "SQLIdentifyMsgWorker["+strconv.Itoa(id)+"] : Parsing objSubTypeMessageMiniWar.") cwm.ObjID64 = objId err = insertMsgMiniWar(cwm) logOnError(err, "SQLIdentifyMsgWorker["+strconv.Itoa(id)+"] : insertMsgMiniWar") case objSubTypeMessagePillageInc: cwm, err := parseSubTypeMessagePillageInc(m, r) logOnError(err, "SQLIdentifyMsgWorker["+strconv.Itoa(id)+"] : Parsing objSubTypeMessagePillageInc.") cwm.ObjID64 = objId err = insertMsgPillageInc(cwm) logOnError(err, "SQLIdentifyMsgWorker["+strconv.Itoa(id)+"] : insertMsgPillageInc") // only catch live pillages if m.Date.Add(3 * time.Minute).After(time.Now()) { s := TGCommand{ Type: commandSendMsg, Text: fmt.Sprintf("Catching pillage (%s)", m.Date.Format(time.RFC3339)), ToUserID64: m.UserID64, } TGCmdQueue <- s p := JobPayloadPillage{ ObjID64: objId, } b, _ := json.Marshal(&p) _, err = createJob(objSubTypeJobPillage, objJobPriority, m.UserID64, m.Date.Add(time.Duration(25+rand.Intn(35))*time.Second), b) logOnError(err, "SQLIdentifyMsgWorker["+strconv.Itoa(id)+"] : createJob(JobPillage)") } case objSubTypeMessageMeAck: cwm, err := parseSubTypeMessageMeAck(m, r) logOnError(err, "SQLIdentifyMsgWorker["+strconv.Itoa(id)+"] : Parsing objSubTypeMessageMeAck.") cwm.Msg = m clientMsgMeAck(cwm) _, err = addObjXP(cwm.UserID64, cwm.ExpNow, cwm.ExpLvl, cwm.Level, m.Date) logOnError(err, "SQLIdentifyMsgWorker["+strconv.Itoa(id)+"] : addObjXP(MeAck)") case objSubTypeMessageGoQuestAck: cwm, err := parseSubTypeMessageGoQuestAck(m, r) logOnError(err, "SQLIdentifyMsgWorker["+strconv.Itoa(id)+"] : Parsing objSubTypeMessageMeAck.") cwm.ObjID64 = objId case objSubTypeMessageDuelFight: cwm, err := parseSubTypeMessageDuelFight(m, r) logOnError(err, "SQLIdentifyMsgWorker["+strconv.Itoa(id)+"] : Parsing objSubTypeMessageDuelFight.") cwm.ObjID64 = objId err = insertMsgDuelFight(cwm) logOnError(err, "SQLIdentifyMsgWorker["+strconv.Itoa(id)+"] : insertMsgDuelFight") case objSubTypeMessageUnionWar: _, err := parseSubTypeMessageUnionWar(m, r) logOnError(err, "SQLIdentifyMsgWorker["+strconv.Itoa(id)+"] : Parsing objSubTypeMessageUnionWar.") case objSubTypeMessageGRolesAck: cwm, err := parseSubTypeMessageGRolesAck(m, r) logOnError(err, "SQLIdentifyMsgWorker["+strconv.Itoa(id)+"] : Parsing objSubTypeMessageGRolesAck.") cwm.Msg = m clientMsgGRolesAck(cwm) case objSubTypeMessageAuctionAnnounce: cwm, err := parseSubTypeMessageAuctionAnnounce(m, r) logOnError(err, "SQLIdentifyMsgWorker["+strconv.Itoa(id)+"] : Parsing objSubTypeMessageAuctionAnnounce.") cwm.ObjID64 = objId err = insertMsgAuctionAnnounce(cwm) logOnError(err, "SQLIdentifyMsgWorker["+strconv.Itoa(id)+"] : insertMsgAuctionAnnounce") if cwm.End.After(time.Now().UTC()) || strings.Compare(cwm.Status, `#active`) == 0 { p := JobPayloadMsgRefresh{ ObjID64: m.ObjID64, } b, _ := json.Marshal(&p) _, err = createJob(objSubTypeJobMsgRefresh, objJobPriority, m.UserID64, cwm.End.Add(5*time.Minute).UTC(), b) } case objSubTypeMessageTimeAck: _, err := fromChatWarsDate(r.ReplaceAllString(m.Text, "${Day}") + " " + r.ReplaceAllString(m.Text, "${Month}") + " " + r.ReplaceAllString(m.Text, "${Year}") + " " + r.ReplaceAllString(m.Text, "${Hour}") + ":" + r.ReplaceAllString(m.Text, "${Minute}")) logOnError(err, "SQLIdentifyMsgWorker["+strconv.Itoa(id)+"] : fromChatWarsDate") default: log.Printf("SQLIdentifyMsgWorker["+strconv.Itoa(id)+"] : Unknwon message type in rule %d : %d (%d)\n%s\n", msgParsingRules[i].ID, msgParsingRules[i].MsgTypeID, objId, m.Text) } } } } log.Printf("SQLIdentifyMsgWorker[" + strconv.Itoa(id) + "] : Closing.") } func SQLJobWorker() { //log.Printf("SQLJobWorker : Starting.") for true { jobs, err := loadCurrentJobs() logOnError(err, "SQLJobWorker : loadCurrentJobs") /* if len(jobs) > 0 { log.Printf("SQLJobWorker : %d jobs.\n", len(jobs)) } */ for _, j := range jobs { JobQueue <- j } if len(jobs) < SQLJobSliceSize { time.Sleep(100 * time.Millisecond) } } log.Printf("SQLJobWorker : Closing.") } func JobWorker(id int, jobs <-chan Job) { //log.Printf("jobWorker[" + strconv.Itoa(id) + "] : Starting.") for j := range jobs { //log.Printf("JobWorker[%d] : Starting job %d.\n", id, j.ID64) switch j.JobTypeID { case objSubTypeJobRescanMsg: jobRescan(j) case objSubTypeJobSetJobDone: jobSetDone(j) case objSubTypeJobPillage: jobPillage(j) case objSubTypeJobMsgClient: jobMsgClient(j) case objSubTypeJobMsgRefresh: jobMsgRefresh(j) case objSubTypeJobBackupExport: jobBackupExport(j) case objSubTypeJobBackupImport: jobBackupImport(j) default: log.Printf("jobWorker["+strconv.Itoa(id)+"] : No handler for job type #%d.\n", j.JobTypeID) } } log.Printf("jobWorker[" + strconv.Itoa(id) + "] : Closing.") } func TGCmdWorker(id int, b *tb.Bot, cmds <-chan TGCommand) { //log.Printf("TGCmdWorker[" + strconv.Itoa(id) + "] : Starting.") for c := range cmds { //j, err := json.Marshal(c) //logOnError(err, "TGCmdWorker["+strconv.Itoa(id)+"] : Marshal(c)") //log.Printf("TGCmdWorker["+strconv.Itoa(id)+"] : new command.\n%s\n", string(j)) opt := tb.SendOptions{} switch c.ParseMode { case cmdParseModePlain: opt.ParseMode = tb.ModeDefault case cmdParseModeMarkDown: opt.ParseMode = tb.ModeMarkdown case cmdParseModeHTML: opt.ParseMode = tb.ModeHTML default: opt.ParseMode = tb.ModeDefault } switch c.Type { case commandSendMsg: if c.ToChatID64 != 0 { ch := tb.Chat{ ID: c.ToChatID64, } _, err := b.Send(&ch, c.Text, &opt) logOnError(err, "TGCmdWorker["+strconv.Itoa(id)+"] : SendMsg Chat") } else if c.ToUserID64 != 0 { u := tb.User{ ID: int(c.ToUserID64), } _, err := b.Send(&u, c.Text, &opt) logOnError(err, "TGCmdWorker["+strconv.Itoa(id)+"] : SendMsg User") } case commandReplyMsg: ch := tb.Chat{ ID: c.FromChatID64, } m := tb.Message{ ID: int(c.FromMsgID64), Chat: &ch, } _, err := b.Reply(&m, c.Text, &opt) logOnError(err, "TGCmdWorker["+strconv.Itoa(id)+"] : ReplyMsg") case commandSendDocument: if c.ToChatID64 != 0 { ch := tb.Chat{ ID: c.ToChatID64, } d := c.Document _, err := b.Send(&ch, &d) //_, err := c.Document.Send(b, ch, nil) logOnError(err, "TGCmdWorker["+strconv.Itoa(id)+"] : SendDocument Chat") } else if c.ToUserID64 != 0 { u := tb.User{ ID: int(c.ToUserID64), } d := c.Document _, err := b.Send(&u, &d) //_, err := c.Document.Send(b, &ch, nil) logOnError(err, "TGCmdWorker["+strconv.Itoa(id)+"] : SendDocument Chat") } default: } } log.Printf("TGCmdWorker[" + strconv.Itoa(id) + "] : Closing.") } func MQTGCmdWorker(id int, cmds <-chan TGCommand) { //log.Printf("MQTGCmdWorker[" + strconv.Itoa(id) + "] : Starting.") for c := range cmds { j, err := json.Marshal(c) logOnError(err, "MQTGCmdWorker["+strconv.Itoa(id)+"] : Marshal(c)") if clientIsAlive(c.FromUserID64) { err = clientsQueue[c.FromUserID64].Session.Publish("application/json", j) logOnError(err, "MQTGCmdWorker["+strconv.Itoa(id)+"] : Publishing message.") for err != nil && clientIsAlive(c.FromUserID64) { clientsQueue[c.FromUserID64].Session.Close() time.Sleep(5 * time.Second) clientsQueue[c.FromUserID64].Session.Open() err = clientsQueue[c.FromUserID64].Session.Publish("application/json", j) } } else { err = nil log.Printf("MQTGCmdWorker["+strconv.Itoa(id)+"] : client %d offline.\n", c.FromUserID64) } } log.Printf("MQTGCmdWorker[" + strconv.Itoa(id) + "] : Closing.") } func MQKeepAliveWorker() { s := MQSession{ User: cfg.Rabbit.User, Password: cfg.Rabbit.Password, Host: cfg.Rabbit.Host, Path: cfg.Rabbit.Path, SSL: false, Queue: "keepalive", isConnected: false, } //log.Printf("MQKeepAliveWorker : Starting.") for true { err := s.Open() for err != nil { logOnError(err, "MQKeepAliveWorker : Failed to connect to RabbitMQ") time.Sleep(15 * time.Second) err = s.Open() } for d := range m { // log.Printf("MQKeepAliveWorker : Received a message: %s", string(d.Body)) x := MQKeepAlive{} err = json.Unmarshal(d.Body, &x) logOnError(err, "MQKeepAliveWorker : Can't unmarshal.\n"+string(d.Body)) if err == nil { if x.Date.Add(10 * time.Second).Before(time.Now()) { // outdated keep-alive coming from client } else if v, ok := clientsKeepAlive.Load(x.UserID64); ok { k := v.(*MQKeepAlive) k.Date = x.Date } else { cs := MQSession{ User: cfg.Rabbit.User, Password: cfg.Rabbit.Password, Host: cfg.Rabbit.Host, Path: x.Queue, SSL: false, Queue: "msg", isConnected: false, } err = cs.Open() logOnError(err, "MQKeepAliveWorker : Failed to open MQ session") clientsKeepAlive.Store(x.UserID64, &x) clientsQueue[x.UserID64] = &cs c := TGCommand{ Type: commandSendMsg, ToUserID64: x.UserID64, Text: "Your client is connected.", } TGCmdQueue <- c c = TGCommand{ Type: commandSendMsg, ToUserID64: cfg.Bot.Admin, Text: fmt.Sprintf("Client `%s` is connected.", x.Nickname), } TGCmdQueue <- c clientSendCWMsg(x.UserID64, `🏅Me`) /* c = TGCommand{ Type: commandSendMsg, FromUserID64: x.UserID64, ToChatID64: userID64ChtWrsBot, Text: `/hero`, } MQTGCmdQueue <- c */ } } } } log.Printf("MQKeepAliveWorker : Closing.") } func MQTidyKeepAliveWorker() { //log.Printf("MQTidyKeepAliveWorker : Starting.") for true { t := time.Now() clientsKeepAlive.Range(clientKeepAlive) time.Sleep(time.Until(t.Add(time.Second))) } log.Printf("MQTidyKeepAliveWorker : Closing.") }