aboutsummaryrefslogtreecommitdiff
path: root/telegram
diff options
context:
space:
mode:
Diffstat (limited to 'telegram')
-rw-r--r--telegram/commands.go19
-rw-r--r--telegram/handlers.go92
-rw-r--r--telegram/utils.go60
-rw-r--r--telegram/utils_test.go40
4 files changed, 123 insertions, 88 deletions
diff --git a/telegram/commands.go b/telegram/commands.go
index 0200e05..039798f 100644
--- a/telegram/commands.go
+++ b/telegram/commands.go
@@ -362,16 +362,15 @@ func (c *Client) ProcessTransportCommand(cmdline string, resource string) (strin
return notOnline, false
}
- for _, id := range c.cache.ChatsKeys() {
- c.unsubscribe(id)
- }
-
_, err := c.client.LogOut()
if err != nil {
- c.forceClose()
return errors.Wrap(err, "Logout error").Error(), false
}
+ for _, id := range c.cache.ChatsKeys() {
+ c.unsubscribe(id)
+ }
+
c.Session.Login = ""
// cancel auth
case "cancelauth":
@@ -466,16 +465,6 @@ func (c *Client) ProcessTransportCommand(cmdline string, resource string) (strin
if gateway.MessageOutgoingPermissionVersion == 0 && args[0] == "carbons" && args[1] == "true" {
return "The server did not allow to enable carbons", false
}
- if !c.Session.RawMessages && args[0] == "nativeedits" && args[1] == "true" {
- return "nativeedits only works with rawmessages as of yet, enable it first", false
- }
- if c.Session.NativeEdits && args[0] == "rawmessages" && args[1] == "false" {
- _, err := c.Session.Set("nativeedits", "false")
- if err != nil {
- return err.Error(), false
- }
- msg = "Automatically disabling nativeedits too...\n"
- }
value, err := c.Session.Set(args[0], args[1])
if err != nil {
diff --git a/telegram/handlers.go b/telegram/handlers.go
index 425309e..6266292 100644
--- a/telegram/handlers.go
+++ b/telegram/handlers.go
@@ -56,30 +56,28 @@ func (c *Client) cleanTempFile(path string) {
}
func (c *Client) sendMarker(chatId, messageId int64, typ gateway.MarkerType) {
- if xmppId, err := gateway.IdsDB.GetByTgIds(c.Session.Login, c.jid, chatId, messageId); err == nil {
- resource := c.getFromOutbox(xmppId)
-
- var stringType string
- if typ == gateway.MarkerTypeReceived {
- stringType = "received"
- } else if typ == gateway.MarkerTypeDisplayed {
- stringType = "displayed"
- }
- log.WithFields(log.Fields{
- "xmppId": xmppId,
- "resource": resource,
- }).Debugf("marker: %s", stringType)
-
- if resource != "" {
- gateway.SendMessageMarker(
- c.jid+"/"+resource,
- strconv.FormatInt(chatId, 10),
- c.xmpp,
- typ,
- xmppId,
- )
- }
+ xmppId, err := gateway.IdsDB.GetByTgIds(c.Session.Login, c.jid, chatId, messageId)
+ if err != nil {
+ xmppId = strconv.FormatInt(messageId, 10)
+ }
+
+ var stringType string
+ if typ == gateway.MarkerTypeReceived {
+ stringType = "received"
+ } else if typ == gateway.MarkerTypeDisplayed {
+ stringType = "displayed"
}
+ log.WithFields(log.Fields{
+ "xmppId": xmppId,
+ }).Debugf("marker: %s", stringType)
+
+ gateway.SendMessageMarker(
+ c.jid,
+ strconv.FormatInt(chatId, 10),
+ c.xmpp,
+ typ,
+ xmppId,
+ )
}
func (c *Client) updateHandler() {
@@ -306,38 +304,48 @@ func (c *Client) updateMessageContent(update *client.UpdateMessageContent) {
}
if ok && lastXmppId == xmppId {
replaceId = xmppId
- message, err := c.client.GetMessage(&client.GetMessageRequest{
- ChatId: update.ChatId,
- MessageId: update.MessageId,
- })
- if err == nil {
- isCarbon = c.isCarbonsEnabled() && message.IsOutgoing
- } else {
- log.Errorf("No message %v/%v found, cannot reliably determine if it's a carbon", update.ChatId, update.MessageId)
- }
} else {
log.Infof("Mismatching message ids: %v %v, falling back to separate edit message", lastXmppId, xmppId)
}
}
- text := formatter.Format(
- textContent.Text.Text,
- textContent.Text.Entities,
- markupFunction,
- )
+ message, messageErr := c.client.GetMessage(&client.GetMessageRequest{
+ ChatId: update.ChatId,
+ MessageId: update.MessageId,
+ })
+ var prefix string
+ if messageErr == nil {
+ isCarbon = c.isCarbonsEnabled() && message.IsOutgoing
+ // reply correction support in clients is suboptimal yet, so cut them out for now
+ prefix, _ = c.messageToPrefix(message, "", "", true)
+ } else {
+ log.Errorf("No message %v/%v found, cannot reliably determine if it's a carbon", update.ChatId, update.MessageId)
+ }
+
+ var text strings.Builder
if replaceId == "" {
var editChar string
if c.Session.AsciiArrows {
- editChar = "e "
+ editChar = "e"
} else {
- editChar = "✎ "
+ editChar = "✎"
}
- text = editChar + fmt.Sprintf("%v | %s", update.MessageId, text)
+ text.WriteString(fmt.Sprintf("%s %v | ", editChar, update.MessageId))
+ } else if prefix != "" {
+ text.WriteString(prefix)
+ text.WriteString(c.getPrefixSeparator(update.ChatId))
}
+ text.WriteString(formatter.Format(
+ textContent.Text.Text,
+ textContent.Text.Entities,
+ markupFunction,
+ ))
+
+ sChatId := strconv.FormatInt(update.ChatId, 10)
for _, jid := range jids {
- gateway.SendMessage(jid, strconv.FormatInt(update.ChatId, 10), text, "e"+sId, c.xmpp, nil, replaceId, isCarbon, false)
+ gateway.SendMessage(jid, sChatId, text.String(), "e"+sId, c.xmpp, nil, replaceId, isCarbon, false)
}
}
}
@@ -373,6 +381,8 @@ func (c *Client) updateMessageSendSucceeded(update *client.UpdateMessageSendSucc
log.Errorf("failed to replace %v with %v: %v", update.OldMessageId, update.Message.Id, err.Error())
}
+ c.updateLastMessageHash(update.Message.ChatId, update.Message.Id, update.Message.Content)
+
c.sendMarker(update.Message.ChatId, update.Message.Id, gateway.MarkerTypeReceived)
// clean uploaded files
diff --git a/telegram/utils.go b/telegram/utils.go
index 5c26b8c..d656885 100644
--- a/telegram/utils.go
+++ b/telegram/utils.go
@@ -967,7 +967,7 @@ func (c *Client) isCarbonsEnabled() bool {
return gateway.MessageOutgoingPermissionVersion > 0 && c.Session.Carbons
}
-func (c *Client) messageToPrefix(message *client.Message, previewString string, fileString string) (string, *gateway.Reply) {
+func (c *Client) messageToPrefix(message *client.Message, previewString string, fileString string, suppressReply bool) (string, *gateway.Reply) {
isPM, err := c.IsPM(message.ChatId)
if err != nil {
log.Errorf("Could not determine if chat is PM: %v", err)
@@ -1005,27 +1005,32 @@ func (c *Client) messageToPrefix(message *client.Message, previewString string,
}
// reply to
- preview := true
- reply, tgReply := c.getMessageReply(message, preview, false)
+ var reply *gateway.Reply
+ if !suppressReply {
+ preview := true
+ gwReply, tgReply := c.getMessageReply(message, preview, false)
- if tgReply != nil {
- var replyStart, replyEnd int
+ if tgReply != nil {
+ reply = gwReply
- if len(prefix) > 0 {
- replyStart = c.countCharsInLines(&prefix) + (len(prefix)-1)*len(messageHeaderSeparator)
- }
+ var replyStart, replyEnd int
+
+ if len(prefix) > 0 {
+ replyStart = c.countCharsInLines(&prefix) + (len(prefix)-1)*len(messageHeaderSeparator)
+ }
- replyLine := "reply: " + c.formatMessageContent(preview, tgReply)
- prefix = append(prefix, replyLine)
+ replyLine := "reply: " + c.formatMessageContent(preview, tgReply)
+ prefix = append(prefix, replyLine)
- replyEnd = replyStart + utf8.RuneCountInString(replyLine)
- if len(prefix) > 0 {
- replyEnd += len(messageHeaderSeparator)
- }
+ replyEnd = replyStart + utf8.RuneCountInString(replyLine)
+ if len(prefix) > 0 {
+ replyEnd += len(messageHeaderSeparator)
+ }
- if reply != nil {
- reply.Start = uint64(replyStart)
- reply.End = uint64(replyEnd)
+ if reply != nil {
+ reply.Start = uint64(replyStart)
+ reply.End = uint64(replyEnd)
+ }
}
}
@@ -1060,6 +1065,17 @@ func (c *Client) ensureDownloadFile(file *client.File) *client.File {
return file
}
+// \n if it is groupchat and message is not empty
+func (c *Client) getPrefixSeparator(chatId int64) string {
+ var separator string
+ if chatId < 0 {
+ separator = "\n"
+ } else if chatId > 0 {
+ separator = " | "
+ }
+ return separator
+}
+
// ProcessIncomingMessage transfers a message to XMPP side and marks it as read on Telegram side
func (c *Client) ProcessIncomingMessage(chatId int64, message *client.Message) {
isCarbon := c.isCarbonsEnabled() && message.IsOutgoing
@@ -1103,21 +1119,15 @@ func (c *Client) ProcessIncomingMessage(chatId int64, message *client.Message) {
} else if !c.Session.RawMessages {
var newText strings.Builder
- prefix, prefixReply := c.messageToPrefix(message, previewName, fileName)
+ prefix, prefixReply := c.messageToPrefix(message, previewName, fileName, false)
reply = prefixReply
replyObtained = true
newText.WriteString(prefix)
if text != "" {
- // \n if it is groupchat and message is not empty
if prefix != "" {
- if chatId < 0 {
- newText.WriteString("\n")
- } else if chatId > 0 {
- newText.WriteString(" | ")
- }
+ newText.WriteString(c.getPrefixSeparator(chatId))
}
-
newText.WriteString(text)
}
text = newText.String()
diff --git a/telegram/utils_test.go b/telegram/utils_test.go
index a0939cd..e89077d 100644
--- a/telegram/utils_test.go
+++ b/telegram/utils_test.go
@@ -436,7 +436,7 @@ func TestMessageToPrefix1(t *testing.T) {
},
},
}
- prefix, gatewayReply := (&Client{Session: &persistence.Session{}}).messageToPrefix(&message, "", "")
+ prefix, gatewayReply := (&Client{Session: &persistence.Session{}}).messageToPrefix(&message, "", "", false)
if prefix != "➡ 42 | fwd: ziz" {
t.Errorf("Wrong prefix: %v", prefix)
}
@@ -454,7 +454,7 @@ func TestMessageToPrefix2(t *testing.T) {
},
},
}
- prefix, gatewayReply := (&Client{Session: &persistence.Session{}}).messageToPrefix(&message, "y.jpg", "")
+ prefix, gatewayReply := (&Client{Session: &persistence.Session{}}).messageToPrefix(&message, "y.jpg", "", false)
if prefix != "⬅ 56 | fwd: (zaz) | preview: y.jpg" {
t.Errorf("Wrong prefix: %v", prefix)
}
@@ -472,7 +472,7 @@ func TestMessageToPrefix3(t *testing.T) {
},
},
}
- prefix, gatewayReply := (&Client{Session: &persistence.Session{AsciiArrows: true}}).messageToPrefix(&message, "", "a.jpg")
+ prefix, gatewayReply := (&Client{Session: &persistence.Session{AsciiArrows: true}}).messageToPrefix(&message, "", "a.jpg", false)
if prefix != "< 56 | fwd: (zuz) | file: a.jpg" {
t.Errorf("Wrong prefix: %v", prefix)
}
@@ -486,7 +486,7 @@ func TestMessageToPrefix4(t *testing.T) {
Id: 23,
IsOutgoing: true,
}
- prefix, gatewayReply := (&Client{Session: &persistence.Session{AsciiArrows: true}}).messageToPrefix(&message, "", "")
+ prefix, gatewayReply := (&Client{Session: &persistence.Session{AsciiArrows: true}}).messageToPrefix(&message, "", "", false)
if prefix != "> 23" {
t.Errorf("Wrong prefix: %v", prefix)
}
@@ -504,7 +504,7 @@ func TestMessageToPrefix5(t *testing.T) {
},
},
}
- prefix, gatewayReply := (&Client{Session: &persistence.Session{AsciiArrows: true}}).messageToPrefix(&message, "h.jpg", "a.jpg")
+ prefix, gatewayReply := (&Client{Session: &persistence.Session{AsciiArrows: true}}).messageToPrefix(&message, "h.jpg", "a.jpg", false)
if prefix != "< 560 | fwd: (zyz) | preview: h.jpg | file: a.jpg" {
t.Errorf("Wrong prefix: %v", prefix)
}
@@ -530,7 +530,7 @@ func TestMessageToPrefix6(t *testing.T) {
},
},
}
- prefix, gatewayReply := (&Client{Session: &persistence.Session{AsciiArrows: true}}).messageToPrefix(&message, "", "")
+ prefix, gatewayReply := (&Client{Session: &persistence.Session{AsciiArrows: true}}).messageToPrefix(&message, "", "", false)
if prefix != "> 23 | reply: ziz @ unknown contact: TDlib instance is offline | tist uz iz" {
t.Errorf("Wrong prefix: %v", prefix)
}
@@ -556,7 +556,7 @@ func TestMessageToPrefix7(t *testing.T) {
},
},
}
- prefix, gatewayReply := (&Client{Session: &persistence.Session{AsciiArrows: true}}).messageToPrefix(&message, "", "")
+ prefix, gatewayReply := (&Client{Session: &persistence.Session{AsciiArrows: true}}).messageToPrefix(&message, "", "", false)
if prefix != "> 23 | reply: (zaz) @ unknown contact: TDlib instance is offline | tist" {
t.Errorf("Wrong prefix: %v", prefix)
}
@@ -565,6 +565,32 @@ func TestMessageToPrefix7(t *testing.T) {
}
}
+func TestMessageToPrefix8(t *testing.T) {
+ message := client.Message{
+ Id: 23,
+ ChatId: 42,
+ IsOutgoing: true,
+ ReplyTo: &client.MessageReplyToMessage{
+ ChatId: 41,
+ Content: &client.MessageText{
+ Text: &client.FormattedText{
+ Text: "tist",
+ },
+ },
+ Origin: &client.MessageOriginChannel{
+ AuthorSignature: "zuz",
+ },
+ },
+ }
+ prefix, gatewayReply := (&Client{Session: &persistence.Session{AsciiArrows: true}}).messageToPrefix(&message, "", "", true)
+ if prefix != "> 23" {
+ t.Errorf("Wrong prefix: %v", prefix)
+ }
+ if gatewayReply != nil {
+ t.Errorf("Reply is not nil: %v", gatewayReply)
+ }
+}
+
func GetSenderIdEmpty(t *testing.T) {
message := client.Message{}
senderId := (&Client{}).getMessageSenderId(&message)
generated by cgit v1.2.3 (git 2.39.1) at 2025-01-18 21:16:37 +0300