diff options
| -rw-r--r-- | telegram/commands.go | 51 | ||||
| -rw-r--r-- | telegram/utils.go | 2 |
2 files changed, 48 insertions, 5 deletions
diff --git a/telegram/commands.go b/telegram/commands.go index d3a8ea7..86477cc 100644 --- a/telegram/commands.go +++ b/telegram/commands.go @@ -8,6 +8,7 @@ import ( "dev.narayana.im/narayana/telegabber/xmpp/gateway" + log "github.com/sirupsen/logrus" "github.com/zelenin/go-tdlib/client" ) @@ -27,7 +28,7 @@ var transportCommands = map[string]command{ } var chatCommands = map[string]command{ - //"d": command{"[n]", "delete your last message(s)"}, + "d": command{"[n]", "delete your last message(s)"}, //"s": command{"regex replace", "edit your last message"}, //"add": command{"@username", "add @username to your chat list"}, //"join": command{"https://t.me/invite_link", "join to chat via invite link"}, @@ -242,12 +243,54 @@ func (c *Client) ProcessTransportCommand(cmdline string) string { // ProcessChatCommand executes a command sent in a mapped chat // and returns a response and the status of command support -func (c *Client) ProcessChatCommand(cmdline string) (string, bool) { - cmd, _ := parseCommand(cmdline) +func (c *Client) ProcessChatCommand(chatID int64, cmdline string) (string, bool) { + cmd, args := parseCommand(cmdline) switch cmd { + // delete last message(s) + case "d": + if c.me == nil { + return "@me is not initialized", true + } + var limit int32 + if len(args) > 0 { + limit64, err := strconv.ParseInt(args[0], 10, 32) + if err != nil { + return err.Error(), true + } + limit = int32(limit64) + } else { + limit = 1 + } + + messages, err := c.client.SearchChatMessages(&client.SearchChatMessagesRequest{ + ChatId: chatID, + Limit: limit, + SenderUserId: c.me.Id, + Filter: &client.SearchMessagesFilterEmpty{}, + }) + if err != nil { + return err.Error(), true + } + log.Debugf("pre-deletion query: %#v %#v", messages, messages.Messages) + + var messageIds []int64 + for _, message := range messages.Messages { + messageIds = append(messageIds, message.Id) + } + + _, err = c.client.DeleteMessages(&client.DeleteMessagesRequest{ + ChatId: chatID, + MessageIds: messageIds, + Revoke: true, + }) + if err != nil { + return err.Error(), true + } case "help": return helpString(helpTypeChat), true + default: + return "", false } - return "", false + return "", true } diff --git a/telegram/utils.go b/telegram/utils.go index a708c28..b088069 100644 --- a/telegram/utils.go +++ b/telegram/utils.go @@ -413,7 +413,7 @@ func (c *Client) messageToPrefix(message *client.Message, fileString string) str func (c *Client) ProcessOutgoingMessage(chatID int64, text string, messageID int64, returnJid string) { if strings.HasPrefix(text, "/") { // try to execute a command - response, isCommand := c.ProcessChatCommand(text) + response, isCommand := c.ProcessChatCommand(chatID, text) if response != "" { gateway.SendMessage(returnJid, strconv.FormatInt(chatID, 10), response, c.xmpp) } |