fix(Albums): Fix extra count of messages for image albums

Fixes: #9958
status-im · Apr 3, 2023 · 0302b79 · 0302b79
1 parent 5dc60f5
commit 0302b79
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/src/app/modules/shared_models/message_model.nim b/src/app/modules/shared_models/message_model.nim
@@ -319,9 +319,15 @@ QtObject:
     if messageId.len == 0:
       return
     for i in 0 ..< self.items.len:
-      if(self.items[i].id == messageId):
+      let item = self.items[i]
+      if(item.id == messageId):
         result = i
         return
+      elif item.albumId != "":
+        for j in 0 ..< item.albumMessageIds.len:
+          if(item.albumMessageIds[j] == messageId):
+            result = i
+            return
 
   proc findIdsOfTheMessagesWhichRespondedToMessageWithId*(self: Model, messageId: string): seq[string] =
     for i in 0 ..< self.items.len: