Merge pull request #190 from RyanRory/close-group-creation-out-of-sync

Fix Session Out of Sync Issue Upon Creating a Closed Group

SignalServiceKit/src/Loki/Protocol/Closed Groups/ClosedGroupsProtocol.swift

@@ -32,9 +32,8 @@ public final class ClosedGroupsProtocol : NSObject {
         return !thread.isUserAdmin(inGroup: hexEncodedPublicKey, transaction: transaction)
     }
 
-    @objc(establishSessionsIfNeededWithClosedGroupMembers:in:)
-    public static func establishSessionsIfNeeded(with closedGroupMembers: [String], in thread: TSGroupThread) {
-        storage.dbReadWriteConnection.readWrite { transaction in
+    @objc(establishSessionsIfNeededWithClosedGroupMembers:in:using:)
+    public static func establishSessionsIfNeeded(with closedGroupMembers: [String], in thread: TSGroupThread, using transaction: YapDatabaseReadWriteTransaction) {
         closedGroupMembers.forEach { hexEncodedPublicKey in
             guard hexEncodedPublicKey != getUserHexEncodedPublicKey() else { return }
             let hasSession = storage.containsSession(hexEncodedPublicKey, deviceId: Int32(OWSDevicePrimaryDeviceId), protocolContext: transaction)
@@ -49,5 +48,4 @@ public final class ClosedGroupsProtocol : NSObject {
         }
     }
 }
-}
 

SignalServiceKit/src/Loki/Protocol/Sync Messages/SyncMessagesProtocol.swift

@@ -136,7 +136,7 @@ public final class SyncMessagesProtocol : NSObject {
         newGroupThread.groupModel = newGroupModel // TODO: Should this use the setGroupModel method on TSGroupThread?
         newGroupThread.save(with: transaction)
         // Try to establish sessions with all members for which none exists yet when a group is created or updated
-        ClosedGroupsProtocol.establishSessionsIfNeeded(with: members, in: newGroupThread)
+        ClosedGroupsProtocol.establishSessionsIfNeeded(with: members, in: newGroupThread, using: transaction)
         OWSDisappearingMessagesJob.shared().becomeConsistent(withDisappearingDuration: transcript.dataMessage.expireTimer, thread: newGroupThread, createdByRemoteRecipientId: nil, createdInExistingGroup: true, transaction: transaction)
         let groupUpdatedMessage = TSInfoMessage(timestamp: NSDate.ows_millisecondTimeStamp(), in: newGroupThread, messageType: .typeGroupUpdate, customMessage: groupUpdatedMessageDescription)
         groupUpdatedMessage.save(with: transaction)
@@ -207,7 +207,7 @@ public final class SyncMessagesProtocol : NSObject {
                 thread = TSGroupThread.getOrCreateThread(with: groupModel, transaction: transaction)
                 thread.save(with: transaction)
             }
-            ClosedGroupsProtocol.establishSessionsIfNeeded(with: groupModel.groupMemberIds, in: thread)
+            ClosedGroupsProtocol.establishSessionsIfNeeded(with: groupModel.groupMemberIds, in: thread, using: transaction)
         }
     }
 

SignalServiceKit/src/Messages/OWSMessageManager.m

@@ -1342,7 +1342,7 @@ NS_ASSUME_NONNULL_BEGIN
                 BOOL wasCurrentUserRemovedFromGroup = [removedMemberIds containsObject:userMasterHexEncodedPublicKey];
                 if (!wasCurrentUserRemovedFromGroup) {
                     // Loki: Try to establish sessions with all members when a group is created or updated
-                    [LKClosedGroupsProtocol establishSessionsIfNeededWithClosedGroupMembers:newMemberIds.allObjects in:newGroupThread];
+                    [LKClosedGroupsProtocol establishSessionsIfNeededWithClosedGroupMembers:newMemberIds.allObjects in:newGroupThread using:transaction];
                 }
 
                 [[OWSDisappearingMessagesJob sharedJob] becomeConsistentWithDisappearingDuration:dataMessage.expireTimer