database migrate for unread mention

Session.xcodeproj/project.pbxproj

@@ -136,6 +136,7 @@
 		7B1581E2271E743B00848B49 /* OWSSounds.swift in Sources */ = {isa = PBXBuildFile; fileRef = 7B1581E1271E743B00848B49 /* OWSSounds.swift */; };
 		7B1D74AA27BCC16E0030B423 /* NSENotificationPresenter.swift in Sources */ = {isa = PBXBuildFile; fileRef = 7B1D74A927BCC16E0030B423 /* NSENotificationPresenter.swift */; };
 		7B1D74AC27BDE7510030B423 /* Promise+Timeout.swift in Sources */ = {isa = PBXBuildFile; fileRef = 7B1D74AB27BDE7510030B423 /* Promise+Timeout.swift */; };
+		7B1D74AE27C346220030B423 /* UnreadMentionMigtation.swift in Sources */ = {isa = PBXBuildFile; fileRef = 7B1D74AD27C346220030B423 /* UnreadMentionMigtation.swift */; };
 		7B4C75CB26B37E0F0000AC89 /* UnsendRequest.swift in Sources */ = {isa = PBXBuildFile; fileRef = 7B4C75CA26B37E0F0000AC89 /* UnsendRequest.swift */; };
 		7B4C75CD26BB92060000AC89 /* DeletedMessageView.swift in Sources */ = {isa = PBXBuildFile; fileRef = 7B4C75CC26BB92060000AC89 /* DeletedMessageView.swift */; };
 		7B7CB18B270591630079FF93 /* ShareLogsModal.swift in Sources */ = {isa = PBXBuildFile; fileRef = 7B7CB18A270591630079FF93 /* ShareLogsModal.swift */; };
@@ -1117,6 +1118,7 @@
 		7B1581E1271E743B00848B49 /* OWSSounds.swift */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.swift; path = OWSSounds.swift; sourceTree = "<group>"; };
 		7B1D74A927BCC16E0030B423 /* NSENotificationPresenter.swift */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.swift; path = NSENotificationPresenter.swift; sourceTree = "<group>"; };
 		7B1D74AB27BDE7510030B423 /* Promise+Timeout.swift */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.swift; path = "Promise+Timeout.swift"; sourceTree = "<group>"; };
+		7B1D74AD27C346220030B423 /* UnreadMentionMigtation.swift */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.swift; path = UnreadMentionMigtation.swift; sourceTree = "<group>"; };
 		7B2DB2AD26F1B0FF0035B509 /* si */ = {isa = PBXFileReference; lastKnownFileType = text.plist.strings; name = si; path = si.lproj/Localizable.strings; sourceTree = "<group>"; };
 		7B4C75CA26B37E0F0000AC89 /* UnsendRequest.swift */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.swift; path = UnsendRequest.swift; sourceTree = "<group>"; };
 		7B4C75CC26BB92060000AC89 /* DeletedMessageView.swift */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.swift; path = DeletedMessageView.swift; sourceTree = "<group>"; };
@@ -3044,6 +3046,7 @@
 			isa = PBXGroup;
 			children = (
 				B8B32044258C117C0020074B /* ContactsMigration.swift */,
+				7B1D74AD27C346220030B423 /* UnreadMentionMigtation.swift */,
 				C38EF271255B6D79007E1867 /* OWSDatabaseMigration.h */,
 				C38EF270255B6D79007E1867 /* OWSDatabaseMigration.m */,
 				C38EF26F255B6D79007E1867 /* OWSDatabaseMigrationRunner.h */,
@@ -4571,6 +4574,7 @@
 				C38EF31D255B6DBF007E1867 /* UIImage+OWS.swift in Sources */,
 				C3AAFFE825AE975D0089E6DD /* ConfigurationMessage+Convenience.swift in Sources */,
 				C38EF359255B6DCC007E1867 /* SheetViewController.swift in Sources */,
+				7B1D74AE27C346220030B423 /* UnreadMentionMigtation.swift in Sources */,
 				B8F5F52925EC4F8A003BF8D4 /* BlockListUIUtils.m in Sources */,
 				C38EF386255B6DD2007E1867 /* AttachmentApprovalInputAccessoryView.swift in Sources */,
 				B8C2B2C82563685C00551B4D /* CircleView.swift in Sources */,

SignalUtilitiesKit/Database/Migrations/OWSDatabaseMigrationRunner.m

@@ -26,7 +26,8 @@ NS_ASSUME_NONNULL_BEGIN
 - (NSArray<OWSDatabaseMigration *> *)allMigrations
 {
     return @[
-        [SNContactsMigration new]
+        [SNContactsMigration new],
+        [SNUnreadMentionMigration new]
     ];
 }
 

SignalUtilitiesKit/Database/Migrations/UnreadMentionMigtation.swift

@@ -0,0 +1,41 @@
+// Copyright © 2022 Rangeproof Pty Ltd. All rights reserved.
+
+@objc(SNUnreadMentionMigration)
+public class UnreadMentionMigration : OWSDatabaseMigration {
+    
+    @objc
+    class func migrationId() -> String {
+        return "003" // leave "002" for message request migration
+    }
+
+    override public func runUp(completion: @escaping OWSDatabaseMigrationCompletion) {
+        self.doMigrationAsync(completion: completion)
+    }
+    
+    private func doMigrationAsync(completion: @escaping OWSDatabaseMigrationCompletion) {
+        var threads: [TSThread] = []
+        Storage.read { transaction in
+            TSThread.enumerateCollectionObjects(with: transaction) { object, _ in
+                guard let thread = object as? TSThread, let threadID = thread.uniqueId else { return }
+                let unreadMessages = transaction.ext(TSUnreadDatabaseViewExtensionName) as! YapDatabaseViewTransaction
+                unreadMessages.enumerateKeysAndObjects(inGroup: threadID) { collection, key, object, index, stop in
+                    guard let unreadMessage = object as? TSIncomingMessage else { return }
+                    if unreadMessage.wasRead { return }
+                    if unreadMessage.isUserMentioned {
+                        thread.hasUnreadMentionMessage = true
+                        stop.pointee = true
+                    }
+                }
+                threads.append(thread)
+            }
+        }
+        Storage.write(with: { transaction in
+            threads.forEach { thread in
+                thread.save(with: transaction)
+            }
+            self.save(with: transaction) // Intentionally capture self
+        }, completion: {
+            completion()
+        })
+    }
+}