delete/take/give

This commit is contained in:
shoopea 2021-11-13 10:40:43 +08:00
parent 24bf4dc386
commit aa2a229ad2
2 changed files with 147 additions and 13 deletions

152
bot.go
View File

@ -1,6 +1,8 @@
package main
import (
"crypto/rand"
"encoding/hex"
"encoding/json"
"fmt"
"regexp"
@ -15,6 +17,14 @@ type Bot struct {
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{
@ -24,6 +34,8 @@ func (b *Bot) Start() {
})
failError(err, "Bot.Start() : registering bot")
companyDeleteMap = make(map[string]*CompanyDeleteEntry)
b.BotHandlers()
}
@ -118,17 +130,50 @@ func botUnpause(m *tb.Message) {
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)
if m.Sender.ID == int(bot.Config.AdminID) {
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 _, ok := srv.Status.Companies[uint8(ID64)]; ok {
srv.DeleteCompany(uint8(ID64))
bot.SendChat(m.Chat.ID, "Deleting")
} else {
bot.SendChat(m.Chat.ID, "Company doesn't exist")
}
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
}
@ -179,11 +224,79 @@ func botPlayers(m *tb.Message) {
}
func botGive(m *tb.Message) {
PrintText(m)
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) {
PrintText(m)
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 botPasswd(m *tb.Message) {
@ -362,5 +475,26 @@ func botQuery(q *tb.Query) {
}
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)
}

View File

@ -1,6 +1,6 @@
// Code generated by version.sh (@generated) DO NOT EDIT.
package main
var githash = "caba808"
var buildstamp = "2021-11-12_01:17:14"
var commits = "188"
var version = "caba808-b188 - 2021-11-12_01:17:14"
var githash = "24bf4dc"
var buildstamp = "2021-11-13_02:40:31"
var commits = "189"
var version = "24bf4dc-b189 - 2021-11-13_02:40:31"