package main import ( "crypto/rand" "encoding/hex" "encoding/json" "fmt" "regexp" "strconv" "time" tb "gopkg.in/tucnak/telebot.v2" ) type Bot struct { bot *tb.Bot Config *TelegramConfig } type CompanyDeleteEntry struct { CompanyID uint8 UserID int Time time.Time } var companyDeleteMap map[string]*CompanyDeleteEntry func (b *Bot) Start() { var err error b.bot, err = tb.NewBot(tb.Settings{ Token: b.Config.Token, URL: b.Config.URL, Poller: &tb.LongPoller{Timeout: 10 * time.Second}, }) failError(err, "Bot.Start() : registering bot") companyDeleteMap = make(map[string]*CompanyDeleteEntry) b.BotHandlers() } func (b *Bot) SendUser(id int64, msg string) { u := tb.User{ ID: int(id), } _, err := b.bot.Send(&u, msg) logErrorDebug(err, "Bot.SendUser()") } func (b *Bot) SendChat(chatID int64, text string) { opt := tb.SendOptions{ ParseMode: tb.ModeDefault, } ch := tb.Chat{ ID: chatID, } _, err := b.bot.Send(&ch, text, &opt) logErrorDebug(err, "Bot.SendChat()") } func (b *Bot) BotHandlers() { b.bot.Handle("/pause", botPause) b.bot.Handle("/unpause", botUnpause) b.bot.Handle("/register", botRegister) b.bot.Handle("/deregister", botDeregister) b.bot.Handle("/delete", botDelete) b.bot.Handle("/companies", botCompanies) b.bot.Handle("/clients", botClients) b.bot.Handle("/players", botPlayers) b.bot.Handle("/give", botGive) b.bot.Handle("/take", botTake) b.bot.Handle("/passwd", botPasswd) b.bot.Handle("/say", botSay) b.bot.Handle(tb.OnPhoto, botPhoto) b.bot.Handle(tb.OnChannelPost, botChannelPost) b.bot.Handle(tb.OnQuery, botQuery) b.bot.Handle(tb.OnText, botText) b.bot.Handle(tb.OnDocument, botDocument) go func() { time.Sleep(time.Second) b.SendUser(b.Config.AdminID, fmt.Sprintf("Started (%s)", version)) }() b.bot.Start() } func botPause(m *tb.Message) { for userID, cc := range cfg.Clients { if userID == m.Sender.ID { if co, ok := srv.Status.Companies[cc.CompanyID]; ok { if clt, ok2 := srv.Status.Clients[co.ClientID]; ok2 { clt.Paused = true if !srv.Status.Paused { srv.Pause() } else { bot.SendChat(bot.Config.ChatID, "Game already paused.") } return } } } } bot.SendChat(bot.Config.ChatID, "You are not playing and cannot pause the game.") return } func botUnpause(m *tb.Message) { PrintText(m) for userID, cc := range cfg.Clients { if userID == m.Sender.ID { if co, ok := srv.Status.Companies[cc.CompanyID]; ok { if clt, ok2 := srv.Status.Clients[co.ClientID]; ok2 { clt.Paused = false if !srv.NeedPause() { srv.Unpause() } else { bot.SendChat(bot.Config.ChatID, fmt.Sprintf("Cannot unpause : %s", srv.NeedPauseReason())) } return } } } } bot.SendChat(bot.Config.ChatID, "You are not playing and cannot unpause the game.") return } func botDelete(m *tb.Message) { logInfoDebug("[%d] %s(%d) | %s(%d) : delete : %s\n", m.ID, m.Chat.Title, m.Chat.ID, m.Sender.Username, m.Sender.ID, m.Text) r := regexp.MustCompile("/delete (?P[0-9]+)") id := uint8(255) if r.MatchString(m.Text) { ID64, _ := strconv.ParseInt(r.ReplaceAllString(m.Text, "${CompanyID}"), 10, 64) if m.Sender.ID == int(bot.Config.AdminID) { id = uint8(ID64) } else if cc, ok := cfg.Clients[m.Sender.ID]; ok && cc.CompanyID == uint8(ID64) { id = cc.CompanyID } else { bot.SendChat(m.Chat.ID, "Not authorized to delete") return } } else if cc, ok := cfg.Clients[m.Sender.ID]; ok { id = cc.CompanyID } else { bot.SendChat(m.Chat.ID, "User not registered") return } if id == 255 { bot.SendChat(m.Chat.ID, "No company registered") return } if co, ok := srv.Status.Companies[id]; ok { b := make([]byte, 8) _, err := rand.Read(b) logErrorDebug(err, "botDelete : rand.Read") if err != nil { bot.SendChat(m.Chat.ID, "internal error") return } c := hex.EncodeToString(b) d := &CompanyDeleteEntry{ CompanyID: id, UserID: m.Sender.ID, Time: time.Now(), } companyDeleteMap[c] = d bot.SendChat(m.Chat.ID, fmt.Sprintf("Press /delete_%s to delete '%s'", c, co.Name)) } else { bot.SendChat(m.Chat.ID, "Company doesn't exist") } return } func botCompanies(m *tb.Message) { str := "Companies :" for k, v := range srv.Status.Companies { str = str + "\r\n" + fmt.Sprintf(" - %s (%d - %t)", v.Name, k, v.Protected) } bot.SendChat(m.Chat.ID, str) } func botClients(m *tb.Message) { str := "Clients :" for k, v := range srv.Status.Clients { str = str + "\r\n" + fmt.Sprintf(" - %s (%d) : company #%d", v.Name, k, v.CompanyID) } bot.SendChat(m.Chat.ID, str) } func botPlayers(m *tb.Message) { d1 := time.Now().Sub(cfg.Game.StartDate) days := int(time.Duration(d1.Hours()) / 24) d2 := time.Duration(days+1)*(time.Hour)*24 - d1 str := fmt.Sprintf("Update in %s\r\n", d2) online := "" for _, cc := range cfg.Clients { if cc.Online { online = online + fmt.Sprintf(" - %s (%s) : %s", cc.Username, cc.TimeLeft, srv.Status.Companies[cc.CompanyID].Name) + "\r\n" } } offline := "" for _, cc := range cfg.Clients { if _, ok := srv.Status.Companies[cc.CompanyID]; ok && !cc.Online { offline = offline + fmt.Sprintf(" - %s (%s) : %s", cc.Username, cc.TimeLeft, srv.Status.Companies[cc.CompanyID].Name) + "\r\n" } } if len(online) > 0 { str = str + "Players online :\r\n" + online } if len(offline) > 0 { str = str + "Players offline :\r\n" + offline } bot.SendChat(m.Chat.ID, str) } func botGive(m *tb.Message) { if m.Sender.ID != int(cfg.Telegram.AdminID) { bot.SendChat(m.Chat.ID, "Only the admin can use this command.") return } r := regexp.MustCompile("^/give @(?P[a-zA-Z0-9]+) (?P[a-z0-9]+)") if !r.MatchString(m.Text) { bot.SendChat(m.Chat.ID, "Wrong format.") return } uStr := r.ReplaceAllString(m.Text, "${Username}") var u int for ccID, cc := range cfg.Clients { if cc.Username == uStr { u = ccID } } if u == 0 { bot.SendChat(m.Chat.ID, "No such user found.") return } dStr := r.ReplaceAllString(m.Text, "${Duration}") d, err := time.ParseDuration(dStr) logErrorDebug(err, "botGive : time.ParseDuration") if err != nil { bot.SendChat(m.Chat.ID, "Cannot parse duration.") } cc := cfg.Clients[u] cc.TimeLeft += d bot.SendChat(m.Chat.ID, fmt.Sprintf("@%s now has %s left.", uStr, cc.TimeLeft)) } func botTake(m *tb.Message) { if m.Sender.ID != int(cfg.Telegram.AdminID) { bot.SendChat(m.Chat.ID, "Only the admin can use this command.") return } r := regexp.MustCompile("^/take @(?P[a-zA-Z0-9]+) (?P[a-z0-9]+)") if !r.MatchString(m.Text) { bot.SendChat(m.Chat.ID, "Wrong format.") return } uStr := r.ReplaceAllString(m.Text, "${Username}") var u int for ccID, cc := range cfg.Clients { if cc.Username == uStr { u = ccID } } if u == 0 { bot.SendChat(m.Chat.ID, "No such user found.") return } dStr := r.ReplaceAllString(m.Text, "${Duration}") d, err := time.ParseDuration(dStr) logErrorDebug(err, "botGive : time.ParseDuration") if err != nil { bot.SendChat(m.Chat.ID, "Cannot parse duration.") } cc := cfg.Clients[u] cc.TimeLeft -= d bot.SendChat(m.Chat.ID, fmt.Sprintf("@%s now has %s left.", uStr, cc.TimeLeft)) } func botSay(m *tb.Message) { r := regexp.MustCompile("^\\/say( )+(?P.+)$") if !r.MatchString(m.Text) { bot.SendChat(m.Chat.ID, "No message provided") return } msg := r.ReplaceAllString(m.Text, "${Message}") px := PacketServerChat{ Packet: Packet{PType: AdminPacketServerChat}, ActionID: AdminNetworkActionChat, DestinationID: AdminDestinationTypeBroadcast, ClientID: 0, Message: msg, Amount: 0, } srv.Send(px.Bytes()) return } func botPasswd(m *tb.Message) { cc, ok := cfg.Clients[m.Sender.ID] if !ok { bot.SendChat(m.Chat.ID, "User not registered.") return } if cc.CompanyID == 255 { bot.SendChat(m.Chat.ID, "No company registered.") return } co, ok := srv.Status.Companies[cc.CompanyID] if !ok { bot.SendChat(m.Chat.ID, "Registered company doesn't exist.") cc.CompanyID = 255 return } r := regexp.MustCompile("^\\/passwd( )+(?P[^ ]+)$") if !r.MatchString(m.Text) { bot.SendChat(m.Chat.ID, "No passwd provided") return } // we have a parameter cc.Passwd = r.ReplaceAllString(m.Text, "${Passwd}") err := bot.bot.Delete(m) logErrorDebug(err, "botPasswd : Delete") srv.SetPasswd(co.CompanyExtlID, cc.Passwd) bot.SendUser(int64(m.Sender.ID), fmt.Sprintf("Passwd set to \"%s\"", cc.Passwd)) return } func botDeregister(m *tb.Message) { cc, ok := cfg.Clients[m.Sender.ID] if !ok { cc = &ClientConfig{ UserID: m.Sender.ID, Username: m.Sender.Username, CompanyID: 255, TimeLeft: 0, } cfg.Clients[m.Sender.ID] = cc bot.SendChat(m.Chat.ID, "User isn't registered.") return } if cc.CompanyID != 255 { for coID, co := range srv.Status.Companies { if coID == cc.CompanyID { cc.CompanyID = 255 bot.SendChat(m.Chat.ID, fmt.Sprintf("Deregistered from %s. %s playable left.", co.Name, cc.TimeLeft)) return } } logInfoAlert("botRegister : %s : no such CompanyID : %d", cc.Username, cc.CompanyID) cc.CompanyID = 255 bot.SendChat(m.Chat.ID, fmt.Sprintf("Registered company didn't exist anymore. %s playable left.", cc.TimeLeft)) return } return } func botRegister(m *tb.Message) { cc, ok := cfg.Clients[m.Sender.ID] if !ok { cc = &ClientConfig{ UserID: m.Sender.ID, Username: m.Sender.Username, CompanyID: 255, TimeLeft: 0, } cfg.Clients[m.Sender.ID] = cc } else { if cc.CompanyID != 255 { for coID, co := range srv.Status.Companies { if coID == cc.CompanyID { bot.SendChat(m.Chat.ID, fmt.Sprintf("Already registered %s. Please /deregister first.", co.Name)) return } } logInfoAlert("botRegister : %s : no such CompanyID : %d", cc.Username, cc.CompanyID) cc.CompanyID = 255 } } coList := make(map[uint8]struct{}) for coID, _ := range srv.Status.Companies { coList[coID] = struct{}{} } for _, c := range cfg.Clients { if c.CompanyID != 255 { if _, ok := coList[c.CompanyID]; !ok { logInfoAlert("botRegister : %s : no such CompanyID : %d", c.Username, c.CompanyID) c.CompanyID = 255 } else { delete(coList, c.CompanyID) } } } if len(coList) == 0 { bot.SendChat(m.Chat.ID, "No company to register") return } r := regexp.MustCompile("^\\/register( )+(?P[a-zA-Z\\.\\-0-9 ]+)$") if r.MatchString(m.Text) { // we have a parameter coName := r.ReplaceAllString(m.Text, "${Company}") for coID, co := range srv.Status.Companies { if co.Name == coName { for _, c := range cfg.Clients { if c.CompanyID == coID { bot.SendChat(m.Chat.ID, fmt.Sprintf("Company %s is already registered to @%s", coName, c.Username)) return } } cc.CompanyID = coID if cc.TimeLeft == 0 { days := int(time.Now().Sub(cfg.Game.StartDate).Hours() / 24) cc.TimeLeft = cfg.Game.StartingAllotment + cfg.Game.DailyAllotment*time.Duration(days) } bot.SendChat(m.Chat.ID, fmt.Sprintf("@%s registered %s (with %s playable)", cc.Username, srv.Status.Companies[cc.CompanyID].Name, cc.TimeLeft)) return } } bot.SendChat(m.Chat.ID, fmt.Sprintf("Can't find company %s", coName)) return } if len(coList) == 1 { for id, _ := range coList { cc.CompanyID = id if cc.TimeLeft == 0 { days := int(time.Now().Sub(cfg.Game.StartDate).Hours() / 24) cc.TimeLeft = cfg.Game.StartingAllotment + cfg.Game.DailyAllotment*time.Duration(days) } bot.SendChat(m.Chat.ID, fmt.Sprintf("@%s registered %s (with %s playable)", cc.Username, srv.Status.Companies[cc.CompanyID].Name, cc.TimeLeft)) return } } bot.SendChat(m.Chat.ID, "More than one company unregistered. Wait for bot update (poke @tiennou)") return } func PrintText(m *tb.Message) { logInfoDebug("[%d] %s(%d) | %s(%d) : %s\n", m.ID, m.Chat.Title, m.Chat.ID, m.Sender.Username, m.Sender.ID, m.Text) return } func botPhoto(m *tb.Message) { logInfoDebug("botPhoto :", m.Text) // photos only } func botDocument(m *tb.Message) { logInfoDebug("botDocument : %s (%d bytes)\n", m.Document.FileName, m.Document.File.FileSize) // documents only } func botChannelPost(m *tb.Message) { PrintText(m) b, _ := json.Marshal(m) logInfoDebug("botChannelPost : %s\n", string(b)) // channel posts only } func botQuery(q *tb.Query) { logInfoDebug("botQuery") // incoming inline queries } func botText(m *tb.Message) { r := regexp.MustCompile("^/delete_(?P[a-f0-9]{16})$") if r.MatchString(m.Text) { ref := r.ReplaceAllString(m.Text, "${Ref}") d, ok := companyDeleteMap[ref] if !ok { bot.SendChat(m.Chat.ID, "No corresponding deletion request.") return } if d.UserID != m.Sender.ID { bot.SendChat(m.Chat.ID, "Requesting user has to confirm himself.") return } if time.Now().Sub(d.Time) > time.Minute { bot.SendChat(m.Chat.ID, "Request expired.") return } srv.DeleteCompany(d.CompanyID) bot.SendChat(m.Chat.ID, "Company deleted.") delete(companyDeleteMap, ref) return } PrintText(m) }