Handle contact and group sync messages

Author: Trolldemorted

Signal-Windows.Lib/IncomingMessages.cs

@@ -193,6 +193,7 @@ private void HandleMessage(SignalServiceEnvelope envelope)
                     var tmpFile = LibUtils.CreateTmpFile("groups_sync");
                     var plaintextStream = MessageReceiver.RetrieveAttachment(groups.AsPointer(), tmpFile, 10000, null);
                     var deviceGroupsStream = new DeviceGroupsInputStream(plaintextStream);
+                    var groupsList = new List<(SignalGroup, IList<string>)>();
                     DeviceGroup g;
                     while ((g = deviceGroupsStream.Read()) != null)
                     {
@@ -204,7 +205,18 @@ private void HandleMessage(SignalServiceEnvelope envelope)
 
                             }
                         }
+                        var group = new SignalGroup()
+                        {
+                            ThreadDisplayName = g.Name,
+                            ThreadId = Base64.EncodeBytes(g.Id),
+                            GroupMemberships = new List<GroupMembership>(),
+                            CanReceive = true,
+                            ExpiresInSeconds = g.ExpirationTimer != null ? g.ExpirationTimer.Value : 0
+                        };
+                        groupsList.Add((group, g.Members));
                     }
+                    List<SignalConversation> newConversations = SignalDBContext.InsertOrUpdateGroups(groupsList);
+                    SignalLibHandle.Instance.DispatchAddOrUpdateConversations(newConversations);
                 }
                 else if (content.SynchronizeMessage.Contacts != null && content.SynchronizeMessage.Contacts.Complete) //TODO incomplete updates
                 {
@@ -232,6 +244,7 @@ private void HandleMessage(SignalServiceEnvelope envelope)
                             ThreadDisplayName = c.Name,
                             ThreadId = c.Number,
                             Color = c.Color,
+                            CanReceive = true,
                             ExpiresInSeconds = c.ExpirationTimer != null ? c.ExpirationTimer.Value : 0
                         };
                         contactsList.Add(contact);

Signal-Windows.Lib/SignalLibHandle.cs

@@ -476,7 +476,7 @@ internal void DispatchHandleIdentityKeyChange(LinkedList<SignalMessage> messages
             Task.WaitAll(operations.ToArray());
         }
 
-        internal void DispatchAddOrUpdateConversations(List<SignalContact> newConversations)
+        internal void DispatchAddOrUpdateConversations(List<SignalConversation> newConversations)
         {
             List<Task> operations = new List<Task>();
             foreach (var dispatcher in Frames.Keys)

Signal-Windows.Lib/Storage/DB.cs

@@ -1086,14 +1086,53 @@ internal static SignalConversation UpdateMessageRead(long index, SignalConversat
             return dbConversation;
         }
 
-        internal static List<SignalContact> InsertOrUpdateContacts(List<SignalContact> contactsList)
+        internal static List<SignalConversation> InsertOrUpdateGroups(IList<(SignalGroup group, IList<string> members)> groups)
         {
-            List<SignalContact> refreshedContacts = new List<SignalContact>();
+            List<SignalConversation> refreshedGroups = new List<SignalConversation>();
             lock (DBLock)
             {
                 using (var ctx = new SignalDBContext())
                 {
-                    foreach (var contact in contactsList)
+                    foreach (var receivedGroup in groups)
+                    {
+                        var dbGroup = ctx.Groups
+                            .Where(g => g.ThreadId == receivedGroup.group.ThreadId)
+                            .Include(g => g.GroupMemberships)
+                            .SingleOrDefault();
+                        if (dbGroup != null)
+                        {
+                            dbGroup.GroupMemberships.Clear();
+                            dbGroup.ThreadDisplayName = receivedGroup.group.ThreadDisplayName;
+                            dbGroup.CanReceive = receivedGroup.group.CanReceive;
+                            dbGroup.ExpiresInSeconds = receivedGroup.group.ExpiresInSeconds;
+                        }
+                        else
+                        {
+                            dbGroup = receivedGroup.group;
+                            ctx.Groups.Add(dbGroup);
+                        }
+                        foreach (var member in receivedGroup.members)
+                        {
+                            dbGroup.GroupMemberships.Add(new GroupMembership()
+                            {
+                                Contact = GetOrCreateContact(ctx, member, 0),
+                                Group = dbGroup
+                            });
+                        }
+                    }
+                }
+            }
+            return refreshedGroups;
+        }
+
+        internal static List<SignalConversation> InsertOrUpdateContacts(IList<SignalContact> contacts)
+        {
+            List<SignalConversation> refreshedContacts = new List<SignalConversation>();
+            lock (DBLock)
+            {
+                using (var ctx = new SignalDBContext())
+                {
+                    foreach (var contact in contacts)
                     {
                         var dbContact = ctx.Contacts
                             .Where(c => c.ThreadId == contact.ThreadId)
@@ -1103,6 +1142,7 @@ internal static List<SignalContact> InsertOrUpdateContacts(List<SignalContact> c
                             refreshedContacts.Add(dbContact);
                             dbContact.ThreadDisplayName = contact.ThreadDisplayName;
                             dbContact.Color = contact.Color;
+                            dbContact.CanReceive = contact.CanReceive;
                             dbContact.ExpiresInSeconds = contact.ExpiresInSeconds;
                         }
                         else
@@ -1275,31 +1315,35 @@ public static List<SignalContact> GetAllContactsLocked()
 
         public static SignalContact GetOrCreateContactLocked(string username, long timestamp, bool notify = true)
         {
-            SignalContact contact;
-            bool createdNew = false;
             lock (DBLock)
             {
                 using (var ctx = new SignalDBContext())
                 {
-                    contact = ctx.Contacts
-                        .Where(c => c.ThreadId == username)
-                        .SingleOrDefault();
-                    if (contact == null)
-                    {
-                        contact = new SignalContact()
-                        {
-                            ThreadId = username,
-                            ThreadDisplayName = username,
-                            CanReceive = true,
-                            LastActiveTimestamp = timestamp,
-                            Color = null //Utils.CalculateDefaultColor(username)
-                        };
-                        ctx.Contacts.Add(contact);
-                        ctx.SaveChanges();
-                        createdNew = true;
-                    }
+                    return GetOrCreateContact(ctx, username, timestamp, notify);
                 }
             }
+        }
+
+        private static SignalContact GetOrCreateContact(SignalDBContext ctx, string username, long timestamp, bool notify = true)
+        {
+            bool createdNew = false;
+            SignalContact contact = contact = ctx.Contacts
+                .Where(c => c.ThreadId == username)
+                .SingleOrDefault();
+            if (contact == null)
+            {
+                contact = new SignalContact()
+                {
+                    ThreadId = username,
+                    ThreadDisplayName = username,
+                    CanReceive = true,
+                    LastActiveTimestamp = timestamp,
+                    Color = null //Utils.CalculateDefaultColor(username)
+                };
+                ctx.Contacts.Add(contact);
+                ctx.SaveChanges();
+                createdNew = true;
+            }
             if (createdNew && notify)
             {
                 SignalLibHandle.Instance.DispatchAddOrUpdateConversation(contact, null);