mirror of
https://github.com/ChuckNorrison/LightningTipBot.git
synced 2026-08-14 12:43:14 +02:00
bf643f103d
* 🚧 minor changes to all handlers... * 🚧 passing callback to tryEditStack * 🚧 edit and delete callback messages * 🚧 refactor again * 🚧 uncomment this * 🚧 using ctx.Message().Text * 🚧 refactor * 🚧 refactor * 🚧 uncomment * context not queruy * 🚧 update forked lightningtipbot * 🚧 reset ArticleResult * fix inline receive * readd shop * remove chat from shopView Co-authored-by: callebtc <93376500+callebtc@users.noreply.github.com>
84 lines
2.8 KiB
Go
84 lines
2.8 KiB
Go
package telegram
|
|
|
|
import (
|
|
"strings"
|
|
"time"
|
|
|
|
cmap "github.com/orcaman/concurrent-map"
|
|
log "github.com/sirupsen/logrus"
|
|
tb "gopkg.in/lightningtipbot/telebot.v3"
|
|
)
|
|
|
|
var editStack cmap.ConcurrentMap
|
|
|
|
type edit struct {
|
|
to tb.Editable
|
|
key string
|
|
what interface{}
|
|
options []interface{}
|
|
lastEdit time.Time
|
|
edited bool
|
|
}
|
|
|
|
func init() {
|
|
editStack = cmap.New()
|
|
}
|
|
|
|
const resultTrueError = "telebot: result is True"
|
|
const editSameStringError = "specified new message content and reply markup are exactly the same as a current content and reply markup of the message"
|
|
const retryAfterError = "retry after"
|
|
|
|
// startEditWorker will loop through the editStack and run tryEditMessage on not edited messages.
|
|
// if editFromStack is older than 5 seconds, editFromStack will be removed.
|
|
func (bot TipBot) startEditWorker() {
|
|
go func() {
|
|
for {
|
|
for _, k := range editStack.Keys() {
|
|
if e, ok := editStack.Get(k); ok {
|
|
editFromStack := e.(edit)
|
|
if !editFromStack.edited {
|
|
_, err := bot.tryEditMessage(editFromStack.to, editFromStack.what, editFromStack.options...)
|
|
if err != nil && strings.Contains(err.Error(), retryAfterError) {
|
|
// ignore any other error than retry after
|
|
log.Errorf("[startEditWorker] Edit error: %s. len(editStack)=%d", err.Error(), len(editStack.Keys()))
|
|
|
|
} else {
|
|
if err != nil {
|
|
log.Errorf("[startEditWorker] Ignoring edit error: %s. len(editStack)=%d", err.Error(), len(editStack.Keys()))
|
|
}
|
|
log.Tracef("[startEditWorker] message from stack edited %+v. len(editStack)=%d", editFromStack, len(editStack.Keys()))
|
|
editFromStack.lastEdit = time.Now()
|
|
editFromStack.edited = true
|
|
editStack.Set(k, editFromStack)
|
|
}
|
|
} else {
|
|
if editFromStack.lastEdit.Before(time.Now().Add(-(time.Duration(5) * time.Second))) {
|
|
log.Tracef("[startEditWorker] removing message edit from stack %+v. len(editStack)=%d", editFromStack, len(editStack.Keys()))
|
|
editStack.Remove(k)
|
|
}
|
|
}
|
|
}
|
|
}
|
|
time.Sleep(time.Millisecond * 1000)
|
|
}
|
|
}()
|
|
|
|
}
|
|
|
|
// tryEditStack will add the editable to the edit stack, if what (message) changed.
|
|
func (bot TipBot) tryEditStack(to tb.Editable, key string, what interface{}, options ...interface{}) {
|
|
sig, chat := to.MessageSig()
|
|
log.Tracef("[tryEditStack] sig=%s, chat=%d, key=%s, what=%+v, options=%+v", sig, chat, key, what, options)
|
|
// var sig = fmt.Sprintf("%s-%d", msgSig, chat)
|
|
if e, ok := editStack.Get(key); ok {
|
|
editFromStack := e.(edit)
|
|
if editFromStack.what == what.(string) {
|
|
log.Tracef("[tryEditStack] Message already in edit stack. Skipping")
|
|
return
|
|
}
|
|
}
|
|
e := edit{options: options, key: key, what: what, to: to}
|
|
|
|
editStack.Set(key, e)
|
|
log.Tracef("[tryEditStack] Added message %s to edit stack. len(editStack)=%d", key, len(editStack.Keys()))
|
|
}
|