Clean up TSStorageManager usage.

SignalServiceKit/src/Messages/OWSFailedAttachmentDownloadsJob.h

@@ -16,7 +16,7 @@ NS_ASSUME_NONNULL_BEGIN
 /**
  * Database extensions required for class to work.
  */
-- (void)asyncRegisterDatabaseExtensions;
++ (void)asyncRegisterDatabaseExtensionsWithStorageManager:(TSStorageManager *)storageManager;
 
 /**
  * Only use the sync version for testing, generally we'll want to register extensions async

SignalServiceKit/src/Messages/OWSFailedAttachmentDownloadsJob.m

@@ -96,7 +96,7 @@ static NSString *const OWSFailedAttachmentDownloadsJobAttachmentStateIndex = @"i
 
 #pragma mark - YapDatabaseExtension
 
-- (YapDatabaseSecondaryIndex *)indexDatabaseExtension
++ (YapDatabaseSecondaryIndex *)indexDatabaseExtension
 {
     YapDatabaseSecondaryIndexSetup *setup = [YapDatabaseSecondaryIndexSetup new];
     [setup addColumn:OWSFailedAttachmentDownloadsJobAttachmentStateColumn
@@ -121,13 +121,13 @@ static NSString *const OWSFailedAttachmentDownloadsJobAttachmentStateIndex = @"i
 // Useful for tests, don't use in app startup path because it's slow.
 - (void)blockingRegisterDatabaseExtensions
 {
-    [self.storageManager registerExtension:[self indexDatabaseExtension]
+    [self.storageManager registerExtension:[self.class indexDatabaseExtension]
                                   withName:OWSFailedAttachmentDownloadsJobAttachmentStateIndex];
 }
 
-- (void)asyncRegisterDatabaseExtensions
++ (void)asyncRegisterDatabaseExtensionsWithStorageManager:(TSStorageManager *)storageManager
 {
-    [self.storageManager asyncRegisterExtension:[self indexDatabaseExtension]
+    [storageManager asyncRegisterExtension:[self indexDatabaseExtension]
                                   withName:OWSFailedAttachmentDownloadsJobAttachmentStateIndex
                            completionBlock:^(BOOL ready) {
                                if (ready) {

SignalServiceKit/src/Messages/OWSFailedMessagesJob.h

@@ -16,7 +16,7 @@ NS_ASSUME_NONNULL_BEGIN
 /**
  * Database extensions required for class to work.
  */
-- (void)asyncRegisterDatabaseExtensions;
++ (void)asyncRegisterDatabaseExtensionsWithStorageManager:(TSStorageManager *)storageManager;
 
 /**
  * Only use the sync version for testing, generally we'll want to register extensions async

SignalServiceKit/src/Messages/OWSFailedMessagesJob.m

@@ -105,7 +105,7 @@ static NSString *const OWSFailedMessagesJobMessageStateIndex = @"index_outoing_m
 
 #pragma mark - YapDatabaseExtension
 
-- (YapDatabaseSecondaryIndex *)indexDatabaseExtension
++ (YapDatabaseSecondaryIndex *)indexDatabaseExtension
 {
     YapDatabaseSecondaryIndexSetup *setup = [YapDatabaseSecondaryIndexSetup new];
     [setup addColumn:OWSFailedMessagesJobMessageStateColumn withType:YapDatabaseSecondaryIndexTypeInteger];
@@ -130,13 +130,13 @@ static NSString *const OWSFailedMessagesJobMessageStateIndex = @"index_outoing_m
 // Useful for tests, don't use in app startup path because it's slow.
 - (void)blockingRegisterDatabaseExtensions
 {
-    [self.storageManager registerExtension:[self indexDatabaseExtension]
+    [self.storageManager registerExtension:[self.class indexDatabaseExtension]
                                   withName:OWSFailedMessagesJobMessageStateIndex];
 }
 
-- (void)asyncRegisterDatabaseExtensions
++ (void)asyncRegisterDatabaseExtensionsWithStorageManager:(TSStorageManager *)storageManager
 {
-    [self.storageManager asyncRegisterExtension:[self indexDatabaseExtension]
+    [storageManager asyncRegisterExtension:[self indexDatabaseExtension]
                                   withName:OWSFailedMessagesJobMessageStateIndex
                            completionBlock:^(BOOL ready) {
                                if (ready) {

SignalServiceKit/src/Storage/AxolotlStore/TSStorageManager+PreKeyStore.m

@@ -2,6 +2,7 @@
 //  Copyright (c) 2017 Open Whisper Systems. All rights reserved.
 //
 
+#import "TSStorageKeys.h"
 #import "TSStorageManager+PreKeyStore.h"
 #import "TSStorageManager+keyFromIntLong.h"
 #import "YapDatabaseConnection+OWS.h"

SignalServiceKit/src/Storage/OWSIncomingMessageFinder.h

@@ -14,7 +14,7 @@ NS_ASSUME_NONNULL_BEGIN
 /**
  * Must be called before using this finder.
  */
-- (void)asyncRegisterExtension;
++ (void)asyncRegisterExtensionWithStorageManager:(TSStorageManager *)storageManager;
 
 /**
  * Detects existance of a duplicate incoming message.

SignalServiceKit/src/Storage/OWSIncomingMessageFinder.m

@@ -62,7 +62,7 @@ NSString *const OWSIncomingMessageFinderColumnSourceDeviceId = @"OWSIncomingMess
 
 #pragma mark - YAP integration
 
-- (YapDatabaseSecondaryIndex *)indexExtension
++ (YapDatabaseSecondaryIndex *)indexExtension
 {
     YapDatabaseSecondaryIndexSetup *setup = [YapDatabaseSecondaryIndexSetup new];
 
@@ -92,10 +92,10 @@ NSString *const OWSIncomingMessageFinderColumnSourceDeviceId = @"OWSIncomingMess
     return [[YapDatabaseSecondaryIndex alloc] initWithSetup:setup handler:handler];
 }
 
-- (void)asyncRegisterExtension
++ (void)asyncRegisterExtensionWithStorageManager:(TSStorageManager *)storageManager
 {
     DDLogInfo(@"%@ registering async.", self.logTag);
-    [self.storageManager asyncRegisterExtension:self.indexExtension
+    [storageManager asyncRegisterExtension:self.indexExtension
                                   withName:OWSIncomingMessageFinderExtensionName
                            completionBlock:^(BOOL ready) {
                                DDLogInfo(@"%@ finished registering async.", self.logTag);
@@ -106,7 +106,7 @@ NSString *const OWSIncomingMessageFinderColumnSourceDeviceId = @"OWSIncomingMess
 - (void)registerExtension
 {
     DDLogError(@"%@ registering SYNC. We should prefer async when possible.", self.logTag);
-    [self.storageManager registerExtension:self.indexExtension withName:OWSIncomingMessageFinderExtensionName];
+    [self.storageManager registerExtension:self.class.indexExtension withName:OWSIncomingMessageFinderExtensionName];
 }
 
 #pragma mark - instance methods

SignalServiceKit/src/Storage/TSStorageManager.h

@@ -3,7 +3,6 @@
 //
 
 #import "OWSStorage.h"
-#import "TSStorageKeys.h"
 
 NS_ASSUME_NONNULL_BEGIN
 

SignalServiceKit/src/Storage/TSStorageManager.m

@@ -11,13 +11,8 @@
 #import "OWSFileSystem.h"
 #import "OWSIncomingMessageFinder.h"
 #import "OWSMessageReceiver.h"
-#import "SignalRecipient.h"
-#import "TSAttachmentStream.h"
 #import "TSDatabaseSecondaryIndexes.h"
 #import "TSDatabaseView.h"
-#import "TSInteraction.h"
-#import "TSThread.h"
-#import <YapDatabase/YapDatabaseRelationship.h>
 
 NS_ASSUME_NONNULL_BEGIN
 
@@ -110,14 +105,11 @@ NSString *const TSStorageManagerExceptionName_CouldNotCreateDatabaseDirectory
     [TSDatabaseView asyncRegisterThreadSpecialMessagesDatabaseView];
 
     // Register extensions which aren't essential for rendering threads async.
-    [[OWSIncomingMessageFinder new] asyncRegisterExtension];
+    [OWSIncomingMessageFinder asyncRegisterExtensionWithStorageManager:self];
     [TSDatabaseView asyncRegisterSecondaryDevicesDatabaseView];
     [OWSDisappearingMessagesFinder asyncRegisterDatabaseExtensions:self];
-    OWSFailedMessagesJob *failedMessagesJob = [[OWSFailedMessagesJob alloc] initWithStorageManager:self];
-    [failedMessagesJob asyncRegisterDatabaseExtensions];
-    OWSFailedAttachmentDownloadsJob *failedAttachmentDownloadsMessagesJob =
-        [[OWSFailedAttachmentDownloadsJob alloc] initWithStorageManager:self];
-    [failedAttachmentDownloadsMessagesJob asyncRegisterDatabaseExtensions];
+    [OWSFailedMessagesJob asyncRegisterDatabaseExtensionsWithStorageManager:self];
+    [OWSFailedAttachmentDownloadsJob asyncRegisterDatabaseExtensionsWithStorageManager:self];
 
     // NOTE: [TSDatabaseView asyncRegistrationCompletion] ensures that
     // DatabaseViewRegistrationCompleteNotification is not fired until all