From b09310791cc677d1a06a1940c263ea05edacd99a Mon Sep 17 00:00:00 2001
From: rhunk <101876869+rhunk@users.noreply.github.com>
Date: Thu, 18 May 2023 16:47:10 +0200
Subject: [PATCH] feat(messagelogger): message state indicator

---
 .../kotlin/me/rhunk/snapenhance/data/wrapper/impl/Message.kt  | 3 ++-
 .../me/rhunk/snapenhance/features/impl/spy/MessageLogger.kt   | 4 ++++
 2 files changed, 6 insertions(+), 1 deletion(-)

diff --git a/app/src/main/kotlin/me/rhunk/snapenhance/data/wrapper/impl/Message.kt b/app/src/main/kotlin/me/rhunk/snapenhance/data/wrapper/impl/Message.kt
index a92b741e..87be84ed 100644
--- a/app/src/main/kotlin/me/rhunk/snapenhance/data/wrapper/impl/Message.kt
+++ b/app/src/main/kotlin/me/rhunk/snapenhance/data/wrapper/impl/Message.kt
@@ -10,5 +10,6 @@ class Message(obj: Any?) : AbstractWrapper(obj) {
     val messageContent get() = MessageContent(instanceNonNull().getObjectField("mMessageContent"))
     val messageDescriptor get() = MessageDescriptor(instanceNonNull().getObjectField("mDescriptor"))
     val messageMetadata get() = MessageMetadata(instanceNonNull().getObjectField("mMetadata"))
-    val messageState get() = getEnumValue("mState", MessageState.COMMITTED)
+    var messageState get() = getEnumValue("mState", MessageState.COMMITTED)
+    set(value) = setEnumValue("mState", value)
 }
\ No newline at end of file
diff --git a/app/src/main/kotlin/me/rhunk/snapenhance/features/impl/spy/MessageLogger.kt b/app/src/main/kotlin/me/rhunk/snapenhance/features/impl/spy/MessageLogger.kt
index 4562b8b6..86fb514a 100644
--- a/app/src/main/kotlin/me/rhunk/snapenhance/features/impl/spy/MessageLogger.kt
+++ b/app/src/main/kotlin/me/rhunk/snapenhance/features/impl/spy/MessageLogger.kt
@@ -51,6 +51,10 @@ class MessageLogger : Feature("MessageLogger", loadParams = FeatureLoadParams.IN
                     }
                 }
 
+                //set the message state to CANCELING for visibility
+                if (message.messageContent.contentType != ContentType.SNAP && message.messageContent.contentType != ContentType.EXTERNAL_MEDIA) {
+                    message.messageState = MessageState.CANCELING
+                }
                 removedMessages.add(messageId)
                 return@hookConstructor
             }