Add replace command

This commit is contained in:
bodqhrohro 2019-12-07 18:37:14 +02:00
parent 3e791db201
commit 589876eef5
2 changed files with 91 additions and 36 deletions

View file

@ -3,6 +3,7 @@ package telegram
import ( import (
"fmt" "fmt"
"github.com/pkg/errors" "github.com/pkg/errors"
"regexp"
"strconv" "strconv"
"strings" "strings"
@ -289,6 +290,50 @@ func (c *Client) ProcessChatCommand(chatID int64, cmdline string) (string, bool)
if err != nil { if err != nil {
return err.Error(), true return err.Error(), true
} }
// edit last message
case "s":
if c.me == nil {
return "@me is not initialized", true
}
if len(args) < 2 {
return "Not enough arguments", true
}
regex, err := regexp.Compile(args[0])
if err != nil {
return err.Error(), true
}
messages, err := c.client.SearchChatMessages(&client.SearchChatMessagesRequest{
ChatId: chatID,
Limit: 1,
SenderUserId: c.me.Id,
Filter: &client.SearchMessagesFilterEmpty{},
})
log.Debugf("%#v", client.SearchChatMessagesRequest{
ChatId: chatID,
Limit: 1,
SenderUserId: c.me.Id,
Filter: &client.SearchMessagesFilterEmpty{},
})
if err != nil {
return err.Error(), true
}
if len(messages.Messages) == 0 {
return "No last message", true
}
message := messages.Messages[0]
if message == nil {
return "Last message is empty", true
}
messageText, ok := message.Content.(*client.MessageText)
if !ok {
return "Last message is not a text!", true
}
text := regex.ReplaceAllString(messageText.Text.Text, strings.Join(args[1:], " "))
c.ProcessOutgoingMessage(chatID, text, message.Id, "")
case "help": case "help":
return helpString(helpTypeChat), true return helpString(helpTypeChat), true
default: default:

View file

@ -411,7 +411,7 @@ func (c *Client) messageToPrefix(message *client.Message, fileString string) str
// ProcessOutgoingMessage executes commands or sends messages to mapped chats // ProcessOutgoingMessage executes commands or sends messages to mapped chats
func (c *Client) ProcessOutgoingMessage(chatID int64, text string, messageID int64, returnJid string) { func (c *Client) ProcessOutgoingMessage(chatID int64, text string, messageID int64, returnJid string) {
if strings.HasPrefix(text, "/") { if messageID == 0 && strings.HasPrefix(text, "/") {
// try to execute a command // try to execute a command
response, isCommand := c.ProcessChatCommand(chatID, text) response, isCommand := c.ProcessChatCommand(chatID, text)
if response != "" { if response != "" {
@ -430,6 +430,22 @@ func (c *Client) ProcessOutgoingMessage(chatID int64, text string, messageID int
log.Warnf("Send message to chat %v", chatID) log.Warnf("Send message to chat %v", chatID)
if messageID != 0 {
formattedText := &client.FormattedText{
Text: text,
}
// compile our message
message := &client.InputMessageText{
Text: formattedText,
}
c.client.EditMessageText(&client.EditMessageTextRequest{
ChatId: chatID,
MessageId: messageID,
InputMessageContent: message,
})
} else {
// quotations // quotations
var reply int64 var reply int64
replySlice := replyRegex.FindStringSubmatch(text) replySlice := replyRegex.FindStringSubmatch(text)
@ -452,6 +468,7 @@ func (c *Client) ProcessOutgoingMessage(chatID int64, text string, messageID int
text = text[newlinePos+1:] text = text[newlinePos+1:]
} }
} }
formattedText := &client.FormattedText{ formattedText := &client.FormattedText{
Text: text, Text: text,
} }
@ -470,13 +487,6 @@ func (c *Client) ProcessOutgoingMessage(chatID int64, text string, messageID int
} }
} }
if messageID != 0 {
c.client.EditMessageText(&client.EditMessageTextRequest{
ChatId: chatID,
MessageId: messageID,
InputMessageContent: message,
})
} else {
c.client.SendMessage(&client.SendMessageRequest{ c.client.SendMessage(&client.SendMessageRequest{
ChatId: chatID, ChatId: chatID,
ReplyToMessageId: reply, ReplyToMessageId: reply,