Changes on sendMessage structure

Auties00 · May 20, 2024 · 10002c5 · 10002c5
1 parent 1316682
commit 10002c5
Showing 1 changed file with 3 additions and 6 deletions.
diff --git a/src/main/java/it/auties/whatsapp/api/Whatsapp.java b/src/main/java/it/auties/whatsapp/api/Whatsapp.java
@@ -503,8 +503,7 @@ public CompletableFuture<? extends MessageInfo> sendNewsletterMessage(JidProvide
      * @return a CompletableFuture
      */
     public CompletableFuture<? extends MessageInfo> sendMessage(JidProvider chat, ContextualMessage<?> message, MessageInfo quotedMessage) {
-        var contextInfo = ContextInfo.of(quotedMessage);
-        message.setContextInfo(contextInfo);
+        message.contextInfo().ifPresentOrElse(contextInfo -> message.setContextInfo(ContextInfo.of(contextInfo, quotedMessage)), () -> message.setContextInfo(ContextInfo.of(quotedMessage)));
         return sendMessage(chat, MessageContainer.of(message));
     }
 
@@ -517,8 +516,7 @@ public CompletableFuture<? extends MessageInfo> sendMessage(JidProvider chat, Co
      * @return a CompletableFuture
      */
     public CompletableFuture<ChatMessageInfo> sendChatMessage(JidProvider chat, ContextualMessage<?> message, MessageInfo quotedMessage) {
-        var contextInfo = ContextInfo.of(quotedMessage);
-        message.setContextInfo(contextInfo);
+        message.contextInfo().ifPresentOrElse(contextInfo -> message.setContextInfo(ContextInfo.of(contextInfo, quotedMessage)), () -> message.setContextInfo(ContextInfo.of(quotedMessage)));
         return sendChatMessage(chat, MessageContainer.of(message));
     }
 
@@ -532,8 +530,7 @@ public CompletableFuture<ChatMessageInfo> sendChatMessage(JidProvider chat, Cont
      * @return a CompletableFuture
      */
     public CompletableFuture<NewsletterMessageInfo> sendNewsletterMessage(JidProvider chat, ContextualMessage<?> message, MessageInfo quotedMessage) {
-        var contextInfo = ContextInfo.of(quotedMessage);
-        message.setContextInfo(contextInfo);
+        message.contextInfo().ifPresentOrElse(contextInfo -> message.setContextInfo(ContextInfo.of(contextInfo, quotedMessage)), () -> message.setContextInfo(ContextInfo.of(quotedMessage)));
         return sendNewsletterMessage(chat, MessageContainer.of(message));
     }