Merge branch 'charlesmchen/debugCreateGroups'

Signal/src/ViewControllers/DebugUI/DebugUIMessages.m

@@ -5,6 +5,7 @@
 #import "DebugUIMessages.h"
 #import "Environment.h"
 #import "OWSTableViewController.h"
+#import "SecurityUtils.h"
 #import "Signal-Swift.h"
 #import "ThreadUtil.h"
 #import <AFNetworking/AFNetworking.h>
@@ -44,9 +45,7 @@ NS_ASSUME_NONNULL_BEGIN
 {
     OWSAssert(thread);
 
-    return [OWSTableSection
-        sectionWithTitle:self.name
-                   items:@[
+    NSMutableArray<OWSTableItem *> *items = [@[
         [OWSTableItem itemWithTitle:@"Send 10 messages (1/sec.)"
                         actionBlock:^{
                             [DebugUIMessages sendTextMessages:10 thread:thread];
@@ -105,8 +104,7 @@ NS_ASSUME_NONNULL_BEGIN
                         }],
         [OWSTableItem itemWithTitle:@"Send unknown mimetype"
                         actionBlock:^{
-                                           [DebugUIMessages
-                                               sendRandomAttachment:thread
+                            [DebugUIMessages sendRandomAttachment:thread
                                                               uti:SignalAttachment.kUnknownTestAttachmentUTI];
                         }],
         [OWSTableItem itemWithTitle:@"Send pdf"
@@ -186,7 +184,24 @@ NS_ASSUME_NONNULL_BEGIN
                         actionBlock:^{
                             [DebugUIMessages sendTextAndSystemMessages:1000 thread:thread];
                         }],
-                   ]];
+    ] mutableCopy];
+    if ([thread isKindOfClass:[TSContactThread class]]) {
+        TSContactThread *contactThread = (TSContactThread *)thread;
+        NSString *recipientId = contactThread.contactIdentifier;
+        [items addObject:[OWSTableItem itemWithTitle:@"Create 10 new groups"
+                                         actionBlock:^{
+                                             [DebugUIMessages createNewGroups:10 recipientId:recipientId];
+                                         }]];
+        [items addObject:[OWSTableItem itemWithTitle:@"Create 100 new groups"
+                                         actionBlock:^{
+                                             [DebugUIMessages createNewGroups:100 recipientId:recipientId];
+                                         }]];
+        [items addObject:[OWSTableItem itemWithTitle:@"Create 1,000 new groups"
+                                         actionBlock:^{
+                                             [DebugUIMessages createNewGroups:1000 recipientId:recipientId];
+                                         }]];
+    }
+    return [OWSTableSection sectionWithTitle:self.name items:items];
 }
 
 + (void)sendTextMessageInThread:(TSThread *)thread counter:(int)counter
@@ -918,6 +933,45 @@ NS_ASSUME_NONNULL_BEGIN
     });
 }
 
++ (void)createNewGroups:(int)counter recipientId:(NSString *)recipientId
+{
+    if (counter < 1) {
+        return;
+    }
+
+    NSString *groupName = [NSUUID UUID].UUIDString;
+    NSMutableArray<NSString *> *recipientIds = [@[
+        recipientId,
+    ] mutableCopy];
+    NSData *groupId = [SecurityUtils generateRandomBytes:16];
+    TSGroupModel *groupModel =
+        [[TSGroupModel alloc] initWithTitle:groupName memberIds:recipientIds image:nil groupId:groupId];
+
+    __block TSGroupThread *thread;
+    [[TSStorageManager sharedManager].dbReadWriteConnection
+        readWriteWithBlock:^(YapDatabaseReadWriteTransaction *_Nonnull transaction) {
+            thread = [TSGroupThread getOrCreateThreadWithGroupModel:groupModel transaction:transaction];
+        }];
+    OWSAssert(thread);
+
+    TSOutgoingMessage *message = [[TSOutgoingMessage alloc] initWithTimestamp:[NSDate ows_millisecondTimeStamp]
+                                                                     inThread:thread
+                                                             groupMetaMessage:TSGroupMessageNew];
+    // This will save the message.
+    [message updateWithCustomMessage:NSLocalizedString(@"GROUP_CREATED", nil)];
+
+    OWSMessageSender *messageSender = [Environment getCurrent].messageSender;
+    void (^completion)() = ^{
+        dispatch_after(dispatch_time(DISPATCH_TIME_NOW, (int64_t)1.f * NSEC_PER_SEC), dispatch_get_main_queue(), ^{
+            [ThreadUtil sendMessageWithText:[@(counter) description] inThread:thread messageSender:messageSender];
+            dispatch_after(dispatch_time(DISPATCH_TIME_NOW, (int64_t)1.f * NSEC_PER_SEC), dispatch_get_main_queue(), ^{
+                [self createNewGroups:counter - 1 recipientId:recipientId];
+            });
+        });
+    };
+    [messageSender sendMessage:message success:completion failure:completion];
+}
+
 @end
 
 NS_ASSUME_NONNULL_END