gottdad/bot.go

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(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<CompanyID>[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<Username>[a-zA-Z0-9]+) (?P<Duration>[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<Username>[a-zA-Z0-9]+) (?P<Duration>[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 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<Passwd>[^ ]+)$")
	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<Company>[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_(?<Ref>[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)
}