182 lines
11 KiB
Objective-C
182 lines
11 KiB
Objective-C
//
|
|
// Copyright (c) 2019 Open Whisper Systems. All rights reserved.
|
|
//
|
|
|
|
#import "AppSetup.h"
|
|
#import "Environment.h"
|
|
#import "Theme.h"
|
|
#import "VersionMigrations.h"
|
|
#import <AxolotlKit/SessionCipher.h>
|
|
#import <SignalMessaging/OWSDatabaseMigration.h>
|
|
#import <SignalMessaging/OWSProfileManager.h>
|
|
#import <SignalMessaging/SignalMessaging-Swift.h>
|
|
#import <SignalMetadataKit/SignalMetadataKit-Swift.h>
|
|
#import <SignalServiceKit/OWS2FAManager.h>
|
|
#import <SignalServiceKit/OWSAttachmentDownloads.h>
|
|
#import <SignalServiceKit/OWSBackgroundTask.h>
|
|
#import <SignalServiceKit/OWSBatchMessageProcessor.h>
|
|
#import <SignalServiceKit/OWSBlockingManager.h>
|
|
#import <SignalServiceKit/OWSDisappearingMessagesJob.h>
|
|
#import <SignalServiceKit/OWSIdentityManager.h>
|
|
#import <SignalServiceKit/OWSMessageDecrypter.h>
|
|
#import <SignalServiceKit/OWSMessageManager.h>
|
|
#import <SignalServiceKit/OWSMessageReceiver.h>
|
|
#import <SignalServiceKit/OWSOutgoingReceiptManager.h>
|
|
#import <SignalServiceKit/OWSReadReceiptManager.h>
|
|
#import <SignalServiceKit/OWSStorage.h>
|
|
#import <SignalServiceKit/SSKEnvironment.h>
|
|
#import <SignalServiceKit/SignalServiceKit-Swift.h>
|
|
#import <SignalServiceKit/TSSocketManager.h>
|
|
|
|
NS_ASSUME_NONNULL_BEGIN
|
|
|
|
@implementation AppSetup
|
|
|
|
+ (void)setupEnvironmentWithAppSpecificSingletonBlock:(dispatch_block_t)appSpecificSingletonBlock
|
|
migrationCompletion:(dispatch_block_t)migrationCompletion
|
|
{
|
|
OWSAssertDebug(appSpecificSingletonBlock);
|
|
OWSAssertDebug(migrationCompletion);
|
|
|
|
__block OWSBackgroundTask *_Nullable backgroundTask =
|
|
[OWSBackgroundTask backgroundTaskWithLabelStr:__PRETTY_FUNCTION__];
|
|
|
|
static dispatch_once_t onceToken;
|
|
dispatch_once(&onceToken, ^{
|
|
// Order matters here.
|
|
//
|
|
// All of these "singletons" should have any dependencies used in their
|
|
// initializers injected.
|
|
[[OWSBackgroundTaskManager sharedManager] observeNotifications];
|
|
|
|
SDSDatabaseStorage *databaseStorage = [SDSDatabaseStorage new];
|
|
OWSPrimaryStorage *primaryStorage = databaseStorage.yapPrimaryStorage;
|
|
[OWSPrimaryStorage protectFiles];
|
|
|
|
// AFNetworking (via CFNetworking) spools it's attachments to NSTemporaryDirectory().
|
|
// If you receive a media message while the device is locked, the download will fail if the temporary directory
|
|
// is NSFileProtectionComplete
|
|
BOOL success = [OWSFileSystem protectFileOrFolderAtPath:NSTemporaryDirectory()
|
|
fileProtectionType:NSFileProtectionCompleteUntilFirstUserAuthentication];
|
|
OWSAssert(success);
|
|
|
|
OWSPreferences *preferences = [OWSPreferences new];
|
|
|
|
TSNetworkManager *networkManager = [[TSNetworkManager alloc] initDefault];
|
|
OWSContactsManager *contactsManager = [OWSContactsManager new];
|
|
OWSLinkPreviewManager *linkPreviewManager = [OWSLinkPreviewManager new];
|
|
ContactsUpdater *contactsUpdater = [ContactsUpdater new];
|
|
OWSMessageSender *messageSender = [OWSMessageSender new];
|
|
MessageSenderJobQueue *messageSenderJobQueue = [MessageSenderJobQueue new];
|
|
OWSProfileManager *profileManager = [[OWSProfileManager alloc] initWithDatabaseStorage:databaseStorage];
|
|
OWSMessageManager *messageManager = [OWSMessageManager new];
|
|
OWSBlockingManager *blockingManager = [OWSBlockingManager new];
|
|
OWSIdentityManager *identityManager = [[OWSIdentityManager alloc] initWithDatabaseStorage:databaseStorage];
|
|
SSKSessionStore *sessionStore = [SSKSessionStore new];
|
|
SSKSignedPreKeyStore *signedPreKeyStore = [SSKSignedPreKeyStore new];
|
|
SSKPreKeyStore *preKeyStore = [SSKPreKeyStore new];
|
|
id<OWSUDManager> udManager = [OWSUDManagerImpl new];
|
|
OWSMessageDecrypter *messageDecrypter = [OWSMessageDecrypter new];
|
|
SSKMessageDecryptJobQueue *messageDecryptJobQueue = [SSKMessageDecryptJobQueue new];
|
|
OWSBatchMessageProcessor *batchMessageProcessor = [OWSBatchMessageProcessor new];
|
|
OWSMessageReceiver *messageReceiver = [OWSMessageReceiver new];
|
|
TSSocketManager *socketManager = [[TSSocketManager alloc] init];
|
|
TSAccountManager *tsAccountManager = [TSAccountManager new];
|
|
OWS2FAManager *ows2FAManager = [OWS2FAManager new];
|
|
OWSDisappearingMessagesJob *disappearingMessagesJob = [OWSDisappearingMessagesJob new];
|
|
OWSReadReceiptManager *readReceiptManager = [OWSReadReceiptManager new];
|
|
OWSOutgoingReceiptManager *outgoingReceiptManager = [OWSOutgoingReceiptManager new];
|
|
OWSSyncManager *syncManager = [[OWSSyncManager alloc] initDefault];
|
|
id<SSKReachabilityManager> reachabilityManager = [SSKReachabilityManagerImpl new];
|
|
id<OWSTypingIndicators> typingIndicators = [[OWSTypingIndicatorsImpl alloc] init];
|
|
OWSAttachmentDownloads *attachmentDownloads = [[OWSAttachmentDownloads alloc] init];
|
|
StickerManager *stickerManager = [[StickerManager alloc] init];
|
|
SignalServiceAddressCache *signalServiceAddressCache = [SignalServiceAddressCache new];
|
|
AccountServiceClient *accountServiceClient = [AccountServiceClient new];
|
|
OWSStorageServiceManager *storageServiceManager = OWSStorageServiceManager.shared;
|
|
|
|
OWSAudioSession *audioSession = [OWSAudioSession new];
|
|
OWSSounds *sounds = [OWSSounds new];
|
|
id<OWSProximityMonitoringManager> proximityMonitoringManager = [OWSProximityMonitoringManagerImpl new];
|
|
OWSWindowManager *windowManager = [[OWSWindowManager alloc] initDefault];
|
|
|
|
[Environment setShared:[[Environment alloc] initWithAudioSession:audioSession
|
|
preferences:preferences
|
|
proximityMonitoringManager:proximityMonitoringManager
|
|
sounds:sounds
|
|
windowManager:windowManager]];
|
|
|
|
[SMKEnvironment setShared:[[SMKEnvironment alloc] initWithAccountIdFinder:[OWSAccountIdFinder new]]];
|
|
|
|
[SSKEnvironment setShared:[[SSKEnvironment alloc] initWithContactsManager:contactsManager
|
|
linkPreviewManager:linkPreviewManager
|
|
messageSender:messageSender
|
|
messageSenderJobQueue:messageSenderJobQueue
|
|
profileManager:profileManager
|
|
primaryStorage:primaryStorage
|
|
contactsUpdater:contactsUpdater
|
|
networkManager:networkManager
|
|
messageManager:messageManager
|
|
blockingManager:blockingManager
|
|
identityManager:identityManager
|
|
sessionStore:sessionStore
|
|
signedPreKeyStore:signedPreKeyStore
|
|
preKeyStore:preKeyStore
|
|
udManager:udManager
|
|
messageDecrypter:messageDecrypter
|
|
messageDecryptJobQueue:messageDecryptJobQueue
|
|
batchMessageProcessor:batchMessageProcessor
|
|
messageReceiver:messageReceiver
|
|
socketManager:socketManager
|
|
tsAccountManager:tsAccountManager
|
|
ows2FAManager:ows2FAManager
|
|
disappearingMessagesJob:disappearingMessagesJob
|
|
readReceiptManager:readReceiptManager
|
|
outgoingReceiptManager:outgoingReceiptManager
|
|
reachabilityManager:reachabilityManager
|
|
syncManager:syncManager
|
|
typingIndicators:typingIndicators
|
|
attachmentDownloads:attachmentDownloads
|
|
stickerManager:stickerManager
|
|
databaseStorage:databaseStorage
|
|
signalServiceAddressCache:signalServiceAddressCache
|
|
accountServiceClient:accountServiceClient
|
|
storageServiceManager:storageServiceManager]];
|
|
|
|
appSpecificSingletonBlock();
|
|
|
|
OWSAssertDebug(SSKEnvironment.shared.isComplete);
|
|
|
|
// Register renamed classes.
|
|
[NSKeyedUnarchiver setClass:[OWSUserProfile class] forClassName:[OWSUserProfile collection]];
|
|
[NSKeyedUnarchiver setClass:[OWSDatabaseMigration class] forClassName:[OWSDatabaseMigration collection]];
|
|
|
|
[OWSStorage registerExtensionsWithMigrationBlock:^() {
|
|
// Pre-heat caches to avoid sneaky transactions during the migrations.
|
|
// We need to warm these caches _before_ the migrations run.
|
|
//
|
|
// We need to do as few writes as possible here, to avoid conflicts
|
|
// with the migrations which haven't run yet.
|
|
[blockingManager warmCaches];
|
|
[profileManager warmCaches];
|
|
[tsAccountManager warmCaches];
|
|
|
|
dispatch_async(dispatch_get_main_queue(), ^{
|
|
// Don't start database migrations until storage is ready.
|
|
[VersionMigrations performUpdateCheckWithCompletion:^() {
|
|
OWSAssertIsOnMainThread();
|
|
|
|
migrationCompletion();
|
|
|
|
OWSAssertDebug(backgroundTask);
|
|
backgroundTask = nil;
|
|
}];
|
|
});
|
|
}];
|
|
});
|
|
}
|
|
|
|
@end
|
|
|
|
NS_ASSUME_NONNULL_END
|