gather up device syncing classes

// FREEBIE

src/Messages/DeviceSyncing/OWSSyncContactsMessage.h

@@ -1,8 +1,12 @@
 //  Copyright © 2016 Open Whisper Systems. All rights reserved.
 
-#import "ContactsManagerProtocol.h"
 #import "OWSOutgoingSyncMessage.h"
 
+NS_ASSUME_NONNULL_BEGIN
+
+@class YapDatabaseReadWriteTransaction;
+@protocol ContactsManagerProtocol;
+
 @interface OWSSyncContactsMessage : OWSOutgoingSyncMessage
 
 - (instancetype)initWithContactsManager:(id<ContactsManagerProtocol>)contactsManager;
@@ -10,3 +14,5 @@
 - (NSData *)buildPlainTextAttachmentData;
 
 @end
+
+NS_ASSUME_NONNULL_END

src/Messages/DeviceSyncing/OWSSyncContactsMessage.m

@@ -2,12 +2,15 @@
 
 #import "OWSSyncContactsMessage.h"
 #import "Contact.h"
+#import "ContactsManagerProtocol.h"
 #import "NSDate+millisecondTimeStamp.h"
 #import "OWSSignalServiceProtos.pb.h"
 #import "TSAttachment.h"
 #import "TSAttachmentStream.h"
 #import <ProtocolBuffers/CodedOutputStream.h>
 
+NS_ASSUME_NONNULL_BEGIN
+
 @interface OWSSyncContactsMessage ()
 
 @property (nonatomic, readonly) id<ContactsManagerProtocol> contactsManager;
@@ -55,7 +58,7 @@
 
     OWSSignalServiceProtosSyncMessageContactsBuilder *contactsBuilder =
         [OWSSignalServiceProtosSyncMessageContactsBuilder new];
-    [contactsBuilder setBlob:[attachmentBuilder build]];
+    [contactsBuilder setBlobBuilder:attachmentBuilder];
 
     [syncMessageBuilder setContacts:[contactsBuilder build]];
 
@@ -122,3 +125,5 @@
 }
 
 @end
+
+NS_ASSUME_NONNULL_END

src/Messages/DeviceSyncing/OWSSyncGroupsMessage.h

@@ -0,0 +1,12 @@
+//  Copyright © 2016 Open Whisper Systems. All rights reserved.
+
+NS_ASSUME_NONNULL_BEGIN
+
+@interface OWSSyncGroupsMessage : OWSOutgoingSyncMessage
+
+- (void)saveWithTransaction:(YapDatabaseReadWriteTransaction *)transaction;
+- (NSData *)buildPlainTextAttachmentData;
+
+@end
+
+NS_ASSUME_NONNULL_END

src/Messages/DeviceSyncing/OWSSyncGroupsMessage.m

@@ -0,0 +1,105 @@
+//  Copyright © 2016 Open Whisper Systems. All rights reserved.
+
+#import "OWSSyncGroupsMessage.h"
+#import "OWSSignalServiceProtos.pb.h"
+
+NS_ASSUME_NONNULL_BEGIN
+
+@implementation OWSSyncGroupsMessage
+
+- (void)saveWithTransaction:(YapDatabaseReadWriteTransaction *)transaction
+{
+    // no-op
+
+    // There's no need to save this message, since it's not displayed to the user.
+    // Furthermore if we did save it, we probably don't want to save the conctactsManager property.
+}
+
+- (OWSSignalServiceProtosSyncMessage *)buildSyncMessage
+{
+    OWSSignalServiceProtosSyncMessageBuilder *syncMessageBuilder = [OWSSignalServiceProtosSyncMessageBuilder new];
+
+    if (self.attachmentIds.count != 1) {
+        DDLogError(@"expected sync contact message to have exactly one attachment, but found %lu",
+            (unsigned long)self.attachmentIds.count);
+    }
+    TSAttachment *attachment = [TSAttachmentStream fetchObjectWithUniqueID:self.attachmentIds[0]];
+
+    OWSSignalServiceProtosAttachmentPointerBuilder *attachmentBuilder =
+        [OWSSignalServiceProtosAttachmentPointerBuilder new];
+
+    [attachmentBuilder setId:[attachment.identifier unsignedLongLongValue]];
+    [attachmentBuilder setContentType:attachment.contentType];
+    [attachmentBuilder setKey:attachment.encryptionKey];
+
+    OWSSignalServiceProtosSyncMessageContactsBuilder *contactsBuilder =
+        [OWSSignalServiceProtosSyncMessageContactsBuilder new];
+    [contactsBuilder setBlobBuilder:attachmentBuilder];
+
+    [syncMessageBuilder setContacts:[contactsBuilder build]];
+
+    return [syncMessageBuilder build];
+}
+
+- (NSData *)buildPlainTextAttachmentData
+{
+    NSString *fileName =
+        [NSString stringWithFormat:@"%@_%@", [[NSProcessInfo processInfo] globallyUniqueString], @"contacts.dat"];
+    NSURL *fileURL = [NSURL fileURLWithPath:[NSTemporaryDirectory() stringByAppendingPathComponent:fileName]];
+    NSOutputStream *fileOutputStream = [NSOutputStream outputStreamWithURL:fileURL append:NO];
+    [fileOutputStream open];
+
+    PBCodedOutputStream *outputStream = [PBCodedOutputStream streamWithOutputStream:fileOutputStream];
+    DDLogInfo(@"Writing contacts data to %@", fileURL);
+    for (Contact *contact in self.contactsManager.signalContacts) {
+        OWSSignalServiceProtosContactDetailsBuilder *contactBuilder = [OWSSignalServiceProtosContactDetailsBuilder new];
+
+        [contactBuilder setName:contact.fullName];
+        [contactBuilder setNumber:contact.textSecureIdentifiers.firstObject];
+
+        NSData *avatarPng;
+        if (contact.image) {
+            OWSSignalServiceProtosContactDetailsAvatarBuilder *avatarBuilder =
+                [OWSSignalServiceProtosContactDetailsAvatarBuilder new];
+
+            [avatarBuilder setContentType:@"image/png"];
+            avatarPng = UIImagePNGRepresentation(contact.image);
+            // TODO check datasize and safely cast to int
+            [avatarBuilder setLength:(uint32_t)avatarPng.length];
+            [contactBuilder setAvatar:[avatarBuilder build]];
+        }
+
+        NSData *contactData = [[contactBuilder build] data];
+
+        uint32_t contactDataLength = (uint32_t)contactData.length;
+        [outputStream writeRawVarint32:contactDataLength];
+        [outputStream writeRawData:contactData];
+
+        if (contact.image) {
+            [outputStream writeRawData:avatarPng];
+        }
+    }
+    [outputStream flush];
+    [fileOutputStream close];
+
+    // TODO pass stream to builder rather than data as a singular hulk.
+    [NSInputStream inputStreamWithURL:fileURL];
+    NSError *error;
+    NSData *data = [NSData dataWithContentsOfURL:fileURL options:NSDataReadingMappedIfSafe error:&error];
+    if (error) {
+        DDLogError(@"Failed to read back contact data after writing it to %@ with error:%@", fileURL, error);
+    }
+    return data;
+
+    //    TODO delete contacts file.
+    //    NSError *error;
+    //    NSFileManager *manager = [NSFileManager defaultManager];
+    //    [manager removeItemAtURL:fileURL error:&error];
+    //    if (error) {
+    //        DDLogError(@"Failed removing temp file at url:%@ with error:%@", fileURL, error);
+    //    }
+}
+
+@end
+
+NS_ASSUME_NONNULL_END

src/Messages/TSMessagesManager.m

@@ -7,6 +7,7 @@
 #import "NSData+messagePadding.h"
 #import "OWSIncomingSentMessageTranscript.h"
 #import "OWSSyncContactsMessage.h"
+#import "OWSSyncGroupsMessage.h"
 #import "TSAccountManager.h"
 #import "TSAttachmentStream.h"
 #import "TSCall.h"
@@ -255,7 +256,19 @@
 
         } else if (syncMessage.request.type == OWSSignalServiceProtosSyncMessageRequestTypeGroups) {
             DDLogInfo(@"Received Contacts `groups` syncMessage.");
-            // TODO
+
+            OWSSyncGroupsMessage *syncGroupsMessage = [[OWSSyncGroupsMessage alloc] init];
+
+            [self sendAttachment:[syncGroupsMessage buildPlainTextAttachmentData]
+                contentType:OWSMimeTypeApplicationOctetStream
+                inMessage:syncGroupsMessage
+                thread:nil
+                success:^{
+                    DDLogInfo(@"Successfully sent Groups response syncMessage.");
+                }
+                failure:^{
+                    DDLogError(@"Failed to send Groups response syncMessage.");
+                }];
         }
     } else {
         DDLogWarn(@"Ignoring unsupported sync message.");