Add replace command

1 files changed, 45 insertions, 0 deletions

diff --git a/telegram/commands.go b/telegram/commands.go
index 5553580..3682535 100644
--- a/telegram/commands.go
+++ b/telegram/commands.go
@@ -3,6 +3,7 @@ package telegram
 import (
 	"fmt"
 	"github.com/pkg/errors"
+	"regexp"
 	"strconv"
 	"strings"
 
@@ -289,6 +290,50 @@ func (c *Client) ProcessChatCommand(chatID int64, cmdline string) (string, bool)
 		if err != nil {
 			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":
 		return helpString(helpTypeChat), true
 	default: