Avoid webpage preview updates being sent as message edits

1 files changed, 33 insertions, 0 deletions

diff --git a/telegram/utils.go b/telegram/utils.go
index 9486349..da66189 100644
--- a/telegram/utils.go
+++ b/telegram/utils.go
@@ -24,6 +24,7 @@ import (
 	"github.com/zelenin/go-tdlib/client"
 )
 
+// VCardInfo contains intermediate data to produce a vCard
 type VCardInfo struct {
 	Fn       string
 	Photo    *client.File
@@ -34,6 +35,12 @@ type VCardInfo struct {
 	Info     string
 }
 
+// ChatMessageId uniquely identifies a Telegram message
+type ChatMessageId struct {
+	ChatId int64
+	MessageId int64
+}
+
 var errOffline = errors.New("TDlib instance is offline")
 
 var spaceRegex = regexp.MustCompile(`\s+`)
@@ -1384,3 +1391,29 @@ func (c *Client) getCarbonFullJids(isOutgoing bool, ignoredResource string) []st
 	}
 	return jids
 }
+
+func (c *Client) addToEditQueue(chatId, messageId int64) {
+	c.locks.editQueueLock.Lock()
+	defer c.locks.editQueueLock.Unlock()
+
+	c.editQueue[ChatMessageId{
+		ChatId:    chatId,
+		MessageId: messageId,
+	}] = true
+}
+
+func (c *Client) deleteFromEditQueue(chatId, messageId int64) bool {
+	c.locks.editQueueLock.Lock()
+	defer c.locks.editQueueLock.Unlock()
+
+	key := ChatMessageId{
+		ChatId:    chatId,
+		MessageId: messageId,
+	}
+	_, ok := c.editQueue[key]
+	if ok {
+		delete(c.editQueue, key)
+	}
+
+	return ok
+}