Fix editMessage() method

Setting additional protocolMessage to editedMessage
Auties00 · Apr 6, 2024 · 33bb07a · 33bb07a
1 parent 00f8b3f
commit 33bb07a
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/src/main/java/it/auties/whatsapp/api/Whatsapp.java b/src/main/java/it/auties/whatsapp/api/Whatsapp.java
@@ -656,11 +656,16 @@ public <T extends MessageInfo> CompletableFuture<T> editMessage(T oldMessage, Me
                         .fromMe(true)
                         .senderJid(jidOrThrowError())
                         .build();
+                var protocol = new ProtocolMessageBuilder()
+                        .protocolType(ProtocolMessage.Type.MESSAGE_EDIT)
+                        .key(key)
+                        .editedMessage(MessageContainer.of(newMessage))
+                        .build();
                 var info = new ChatMessageInfoBuilder()
                         .status(MessageStatus.PENDING)
                         .senderJid(jidOrThrowError())
                         .key(key)
-                        .message(MessageContainer.ofEditedMessage(newMessage))
+                        .message(MessageContainer.ofEditedMessage(protocol))
                         .timestampSeconds(Clock.nowSeconds())
                         .broadcast(oldChatInfo.chatJid().hasServer(JidServer.BROADCAST))
                         .build();