2015-12-07 03:31:43 +01:00
|
|
|
//
|
2019-01-04 19:35:50 +01:00
|
|
|
// Copyright (c) 2019 Open Whisper Systems. All rights reserved.
|
2015-12-07 03:31:43 +01:00
|
|
|
//
|
|
|
|
|
|
|
|
#import "TSDatabaseView.h"
|
2016-09-03 00:53:47 +02:00
|
|
|
#import "OWSDevice.h"
|
2017-02-07 21:09:04 +01:00
|
|
|
#import "OWSReadTracking.h"
|
2018-03-20 22:22:19 +01:00
|
|
|
#import "TSAttachment.h"
|
2018-11-19 17:38:29 +01:00
|
|
|
#import "TSAttachmentPointer.h"
|
2015-12-07 03:31:43 +01:00
|
|
|
#import "TSIncomingMessage.h"
|
2017-05-26 16:32:13 +02:00
|
|
|
#import "TSInvalidIdentityKeyErrorMessage.h"
|
2017-02-15 23:09:55 +01:00
|
|
|
#import "TSOutgoingMessage.h"
|
2015-12-07 03:31:43 +01:00
|
|
|
#import "TSThread.h"
|
2017-12-12 16:31:05 +01:00
|
|
|
#import <YapDatabase/YapDatabaseAutoView.h>
|
2017-11-29 17:44:58 +01:00
|
|
|
#import <YapDatabase/YapDatabaseCrossProcessNotification.h>
|
2017-12-12 16:31:05 +01:00
|
|
|
#import <YapDatabase/YapDatabaseViewTypes.h>
|
2015-12-07 03:31:43 +01:00
|
|
|
|
2019-01-23 15:05:08 +01:00
|
|
|
NS_ASSUME_NONNULL_BEGIN
|
|
|
|
|
2017-06-15 19:37:10 +02:00
|
|
|
NSString *const TSInboxGroup = @"TSInboxGroup";
|
|
|
|
NSString *const TSArchiveGroup = @"TSArchiveGroup";
|
2015-12-07 03:31:43 +01:00
|
|
|
|
2017-06-15 19:37:10 +02:00
|
|
|
NSString *const TSUnreadIncomingMessagesGroup = @"TSUnreadIncomingMessagesGroup";
|
|
|
|
NSString *const TSSecondaryDevicesGroup = @"TSSecondaryDevicesGroup";
|
|
|
|
|
2017-07-10 20:37:24 +02:00
|
|
|
// YAPDB BUG: when changing from non-persistent to persistent view, we had to rename TSThreadDatabaseViewExtensionName
|
|
|
|
// -> TSThreadDatabaseViewExtensionName2 to work around https://github.com/yapstudios/YapDatabase/issues/324
|
|
|
|
NSString *const TSThreadDatabaseViewExtensionName = @"TSThreadDatabaseViewExtensionName2";
|
2018-10-03 23:41:43 +02:00
|
|
|
|
|
|
|
// We sort interactions by a monotonically increasing counter.
|
|
|
|
//
|
|
|
|
// Previously we sorted the interactions database by local timestamp, which was problematic if the local clock changed.
|
|
|
|
// We need to maintain the legacy extension for purposes of migration.
|
|
|
|
//
|
|
|
|
// The "Legacy" sorting extension name constant has the same value as always, so that it won't need to be rebuilt, while
|
2018-12-19 22:06:12 +01:00
|
|
|
// the "Modern" sorting extension name constant has the same symbol name that we've always used for sorting
|
|
|
|
// interactions, so that the callsites won't need to change.
|
2018-10-03 23:41:43 +02:00
|
|
|
NSString *const TSMessageDatabaseViewExtensionName = @"TSMessageDatabaseViewExtensionName_Monotonic";
|
|
|
|
NSString *const TSMessageDatabaseViewExtensionName_Legacy = @"TSMessageDatabaseViewExtensionName";
|
|
|
|
|
2017-06-15 19:37:10 +02:00
|
|
|
NSString *const TSThreadOutgoingMessageDatabaseViewExtensionName = @"TSThreadOutgoingMessageDatabaseViewExtensionName";
|
|
|
|
NSString *const TSUnreadDatabaseViewExtensionName = @"TSUnreadDatabaseViewExtensionName";
|
|
|
|
NSString *const TSUnseenDatabaseViewExtensionName = @"TSUnseenDatabaseViewExtensionName";
|
|
|
|
NSString *const TSThreadSpecialMessagesDatabaseViewExtensionName = @"TSThreadSpecialMessagesDatabaseViewExtensionName";
|
|
|
|
NSString *const TSSecondaryDevicesDatabaseViewExtensionName = @"TSSecondaryDevicesDatabaseViewExtensionName";
|
2018-03-20 22:22:19 +01:00
|
|
|
NSString *const TSLazyRestoreAttachmentsDatabaseViewExtensionName
|
|
|
|
= @"TSLazyRestoreAttachmentsDatabaseViewExtensionName";
|
|
|
|
NSString *const TSLazyRestoreAttachmentsGroup = @"TSLazyRestoreAttachmentsGroup";
|
2017-06-15 19:37:10 +02:00
|
|
|
|
2018-01-29 19:40:16 +01:00
|
|
|
@interface OWSStorage (TSDatabaseView)
|
|
|
|
|
|
|
|
- (BOOL)registerExtension:(YapDatabaseExtension *)extension withName:(NSString *)extensionName;
|
|
|
|
|
|
|
|
@end
|
|
|
|
|
|
|
|
#pragma mark -
|
|
|
|
|
2015-12-07 03:31:43 +01:00
|
|
|
@implementation TSDatabaseView
|
|
|
|
|
2018-01-11 15:56:38 +01:00
|
|
|
+ (void)registerCrossProcessNotifier:(OWSStorage *)storage
|
2017-11-29 17:44:58 +01:00
|
|
|
{
|
2018-09-06 19:01:24 +02:00
|
|
|
OWSAssertDebug(storage);
|
2018-01-11 15:56:38 +01:00
|
|
|
|
2017-11-29 17:44:58 +01:00
|
|
|
// I don't think the identifier and name of this extension matter for our purposes,
|
|
|
|
// so long as they don't conflict with any other extension names.
|
|
|
|
YapDatabaseExtension *extension =
|
|
|
|
[[YapDatabaseCrossProcessNotification alloc] initWithIdentifier:@"SignalCrossProcessNotifier"];
|
2018-01-11 15:56:38 +01:00
|
|
|
[storage registerExtension:extension withName:@"SignalCrossProcessNotifier"];
|
2017-11-29 17:44:58 +01:00
|
|
|
}
|
|
|
|
|
2017-06-15 19:37:10 +02:00
|
|
|
+ (void)registerMessageDatabaseViewWithName:(NSString *)viewName
|
2017-05-25 18:17:45 +02:00
|
|
|
viewGrouping:(YapDatabaseViewGrouping *)viewGrouping
|
|
|
|
version:(NSString *)version
|
2018-01-11 15:56:38 +01:00
|
|
|
storage:(OWSStorage *)storage
|
2017-05-24 15:45:34 +02:00
|
|
|
{
|
2017-12-19 17:38:25 +01:00
|
|
|
OWSAssertIsOnMainThread();
|
2018-09-06 19:01:24 +02:00
|
|
|
OWSAssertDebug(viewName.length > 0);
|
|
|
|
OWSAssertDebug((viewGrouping));
|
|
|
|
OWSAssertDebug(storage);
|
2017-05-24 15:45:34 +02:00
|
|
|
|
2018-01-11 15:56:38 +01:00
|
|
|
YapDatabaseView *existingView = [storage registeredExtension:viewName];
|
2017-05-24 15:45:34 +02:00
|
|
|
if (existingView) {
|
2018-08-27 16:29:51 +02:00
|
|
|
OWSFailDebug(@"Registered database view twice: %@", viewName);
|
2017-06-15 19:37:10 +02:00
|
|
|
return;
|
2015-12-07 03:31:43 +01:00
|
|
|
}
|
|
|
|
|
2017-05-24 15:45:34 +02:00
|
|
|
YapDatabaseViewSorting *viewSorting = [self messagesSorting];
|
|
|
|
|
|
|
|
YapDatabaseViewOptions *options = [[YapDatabaseViewOptions alloc] init];
|
|
|
|
options.isPersistent = YES;
|
|
|
|
options.allowedCollections =
|
|
|
|
[[YapWhitelistBlacklist alloc] initWithWhitelist:[NSSet setWithObject:[TSInteraction collection]]];
|
|
|
|
|
2017-12-12 16:31:05 +01:00
|
|
|
YapDatabaseView *view = [[YapDatabaseAutoView alloc] initWithGrouping:viewGrouping
|
|
|
|
sorting:viewSorting
|
|
|
|
versionTag:version
|
|
|
|
options:options];
|
2018-01-30 18:39:27 +01:00
|
|
|
[storage asyncRegisterExtension:view withName:viewName];
|
2017-05-24 15:45:34 +02:00
|
|
|
}
|
|
|
|
|
2018-01-26 21:53:26 +01:00
|
|
|
+ (void)asyncRegisterUnreadDatabaseView:(OWSStorage *)storage
|
2017-05-24 15:45:34 +02:00
|
|
|
{
|
2017-02-07 21:09:04 +01:00
|
|
|
YapDatabaseViewGrouping *viewGrouping = [YapDatabaseViewGrouping withObjectBlock:^NSString *(
|
|
|
|
YapDatabaseReadTransaction *transaction, NSString *collection, NSString *key, id object) {
|
|
|
|
if ([object conformsToProtocol:@protocol(OWSReadTracking)]) {
|
|
|
|
id<OWSReadTracking> possiblyRead = (id<OWSReadTracking>)object;
|
2017-05-25 18:17:45 +02:00
|
|
|
if (!possiblyRead.wasRead && possiblyRead.shouldAffectUnreadCounts) {
|
2017-02-07 21:09:04 +01:00
|
|
|
return possiblyRead.uniqueThreadId;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return nil;
|
|
|
|
}];
|
2015-12-07 03:31:43 +01:00
|
|
|
|
2017-06-15 19:37:10 +02:00
|
|
|
[self registerMessageDatabaseViewWithName:TSUnreadDatabaseViewExtensionName
|
|
|
|
viewGrouping:viewGrouping
|
2018-10-03 23:41:43 +02:00
|
|
|
version:@"2"
|
2018-01-11 15:56:38 +01:00
|
|
|
storage:storage];
|
2017-05-25 18:17:45 +02:00
|
|
|
}
|
|
|
|
|
2018-01-11 15:56:38 +01:00
|
|
|
+ (void)asyncRegisterUnseenDatabaseView:(OWSStorage *)storage
|
2017-05-25 18:17:45 +02:00
|
|
|
{
|
|
|
|
YapDatabaseViewGrouping *viewGrouping = [YapDatabaseViewGrouping withObjectBlock:^NSString *(
|
|
|
|
YapDatabaseReadTransaction *transaction, NSString *collection, NSString *key, id object) {
|
|
|
|
if ([object conformsToProtocol:@protocol(OWSReadTracking)]) {
|
|
|
|
id<OWSReadTracking> possiblyRead = (id<OWSReadTracking>)object;
|
|
|
|
if (!possiblyRead.wasRead) {
|
|
|
|
return possiblyRead.uniqueThreadId;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return nil;
|
|
|
|
}];
|
|
|
|
|
2017-06-15 19:37:10 +02:00
|
|
|
[self registerMessageDatabaseViewWithName:TSUnseenDatabaseViewExtensionName
|
|
|
|
viewGrouping:viewGrouping
|
2018-10-03 23:41:43 +02:00
|
|
|
version:@"2"
|
2018-01-11 15:56:38 +01:00
|
|
|
storage:storage];
|
2015-12-07 03:31:43 +01:00
|
|
|
}
|
|
|
|
|
2018-01-11 15:56:38 +01:00
|
|
|
+ (void)asyncRegisterThreadSpecialMessagesDatabaseView:(OWSStorage *)storage
|
2017-05-23 16:12:04 +02:00
|
|
|
{
|
|
|
|
YapDatabaseViewGrouping *viewGrouping = [YapDatabaseViewGrouping withObjectBlock:^NSString *(
|
|
|
|
YapDatabaseReadTransaction *transaction, NSString *collection, NSString *key, id object) {
|
2018-06-20 16:46:03 +02:00
|
|
|
if (![object isKindOfClass:[TSInteraction class]]) {
|
2018-08-27 18:51:32 +02:00
|
|
|
OWSFailDebug(@"Unexpected entity %@ in collection: %@", [object class], collection);
|
2018-06-20 16:46:03 +02:00
|
|
|
return nil;
|
|
|
|
}
|
2017-06-14 18:26:07 +02:00
|
|
|
TSInteraction *interaction = (TSInteraction *)object;
|
|
|
|
if ([interaction isDynamicInteraction]) {
|
|
|
|
return interaction.uniqueThreadId;
|
|
|
|
} else if ([object isKindOfClass:[TSInvalidIdentityKeyErrorMessage class]]) {
|
2017-05-26 17:00:37 +02:00
|
|
|
return interaction.uniqueThreadId;
|
2017-05-26 18:59:20 +02:00
|
|
|
} else if ([object isKindOfClass:[TSErrorMessage class]]) {
|
|
|
|
TSErrorMessage *errorMessage = (TSErrorMessage *)object;
|
|
|
|
if (errorMessage.errorType == TSErrorMessageNonBlockingIdentityChange) {
|
|
|
|
return errorMessage.uniqueThreadId;
|
|
|
|
}
|
2017-05-26 16:32:13 +02:00
|
|
|
}
|
|
|
|
return nil;
|
|
|
|
}];
|
|
|
|
|
2017-06-15 19:37:10 +02:00
|
|
|
[self registerMessageDatabaseViewWithName:TSThreadSpecialMessagesDatabaseViewExtensionName
|
|
|
|
viewGrouping:viewGrouping
|
2018-10-03 23:41:43 +02:00
|
|
|
version:@"2"
|
2018-01-11 15:56:38 +01:00
|
|
|
storage:storage];
|
2017-05-24 15:45:34 +02:00
|
|
|
}
|
2017-05-23 16:12:04 +02:00
|
|
|
|
2018-10-03 23:41:43 +02:00
|
|
|
+ (void)asyncRegisterLegacyThreadInteractionsDatabaseView:(OWSStorage *)storage
|
|
|
|
{
|
|
|
|
OWSAssertIsOnMainThread();
|
|
|
|
OWSAssert(storage);
|
|
|
|
|
|
|
|
YapDatabaseView *existingView = [storage registeredExtension:TSMessageDatabaseViewExtensionName_Legacy];
|
|
|
|
if (existingView) {
|
|
|
|
OWSFailDebug(@"Registered database view twice: %@", TSMessageDatabaseViewExtensionName_Legacy);
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
|
|
|
YapDatabaseViewGrouping *viewGrouping = [YapDatabaseViewGrouping withObjectBlock:^NSString *(
|
|
|
|
YapDatabaseReadTransaction *transaction, NSString *collection, NSString *key, id object) {
|
|
|
|
if (![object isKindOfClass:[TSInteraction class]]) {
|
|
|
|
OWSFailDebug(@"%@ Unexpected entity %@ in collection: %@", self.logTag, [object class], collection);
|
|
|
|
return nil;
|
|
|
|
}
|
|
|
|
TSInteraction *interaction = (TSInteraction *)object;
|
|
|
|
|
|
|
|
return interaction.uniqueThreadId;
|
|
|
|
}];
|
|
|
|
|
|
|
|
YapDatabaseViewSorting *viewSorting =
|
|
|
|
[YapDatabaseViewSorting withObjectBlock:^NSComparisonResult(YapDatabaseReadTransaction *transaction,
|
|
|
|
NSString *group,
|
|
|
|
NSString *collection1,
|
|
|
|
NSString *key1,
|
|
|
|
id object1,
|
|
|
|
NSString *collection2,
|
|
|
|
NSString *key2,
|
|
|
|
id object2) {
|
|
|
|
if (![object1 isKindOfClass:[TSInteraction class]]) {
|
|
|
|
OWSFailDebug(@"%@ Unexpected entity %@ in collection: %@", self.logTag, [object1 class], collection1);
|
|
|
|
return NSOrderedSame;
|
|
|
|
}
|
|
|
|
if (![object2 isKindOfClass:[TSInteraction class]]) {
|
|
|
|
OWSFailDebug(@"%@ Unexpected entity %@ in collection: %@", self.logTag, [object2 class], collection2);
|
|
|
|
return NSOrderedSame;
|
|
|
|
}
|
|
|
|
TSInteraction *interaction1 = (TSInteraction *)object1;
|
|
|
|
TSInteraction *interaction2 = (TSInteraction *)object2;
|
|
|
|
|
|
|
|
// Legit usage of timestampForLegacySorting since we're registering the
|
|
|
|
// legacy extension
|
|
|
|
uint64_t timestamp1 = interaction1.timestampForLegacySorting;
|
|
|
|
uint64_t timestamp2 = interaction2.timestampForLegacySorting;
|
|
|
|
|
|
|
|
if (timestamp1 > timestamp2) {
|
|
|
|
return NSOrderedDescending;
|
|
|
|
} else if (timestamp1 < timestamp2) {
|
|
|
|
return NSOrderedAscending;
|
|
|
|
} else {
|
|
|
|
return NSOrderedSame;
|
|
|
|
}
|
|
|
|
}];
|
|
|
|
|
|
|
|
YapDatabaseViewOptions *options = [YapDatabaseViewOptions new];
|
|
|
|
options.isPersistent = YES;
|
|
|
|
options.allowedCollections =
|
|
|
|
[[YapWhitelistBlacklist alloc] initWithWhitelist:[NSSet setWithObject:[TSInteraction collection]]];
|
|
|
|
|
|
|
|
YapDatabaseView *view =
|
|
|
|
[[YapDatabaseAutoView alloc] initWithGrouping:viewGrouping sorting:viewSorting versionTag:@"1" options:options];
|
|
|
|
|
|
|
|
[storage asyncRegisterExtension:view withName:TSMessageDatabaseViewExtensionName_Legacy];
|
|
|
|
}
|
|
|
|
|
2018-01-26 21:53:26 +01:00
|
|
|
+ (void)asyncRegisterThreadInteractionsDatabaseView:(OWSStorage *)storage
|
2017-05-24 15:45:34 +02:00
|
|
|
{
|
|
|
|
YapDatabaseViewGrouping *viewGrouping = [YapDatabaseViewGrouping withObjectBlock:^NSString *(
|
|
|
|
YapDatabaseReadTransaction *transaction, NSString *collection, NSString *key, id object) {
|
2018-06-20 16:46:03 +02:00
|
|
|
if (![object isKindOfClass:[TSInteraction class]]) {
|
2018-08-27 18:51:32 +02:00
|
|
|
OWSFailDebug(@"Unexpected entity %@ in collection: %@", [object class], collection);
|
2018-06-20 16:46:03 +02:00
|
|
|
return nil;
|
|
|
|
}
|
2017-06-14 18:26:07 +02:00
|
|
|
TSInteraction *interaction = (TSInteraction *)object;
|
2018-06-20 16:46:03 +02:00
|
|
|
|
2017-06-14 18:26:07 +02:00
|
|
|
return interaction.uniqueThreadId;
|
2017-05-30 19:03:56 +02:00
|
|
|
}];
|
|
|
|
|
2017-06-15 19:37:10 +02:00
|
|
|
[self registerMessageDatabaseViewWithName:TSMessageDatabaseViewExtensionName
|
|
|
|
viewGrouping:viewGrouping
|
2018-10-03 23:41:43 +02:00
|
|
|
version:@"2"
|
2018-01-11 15:56:38 +01:00
|
|
|
storage:storage];
|
2017-05-30 19:03:56 +02:00
|
|
|
}
|
|
|
|
|
2018-01-11 15:56:38 +01:00
|
|
|
+ (void)asyncRegisterThreadOutgoingMessagesDatabaseView:(OWSStorage *)storage
|
2017-05-30 19:03:56 +02:00
|
|
|
{
|
|
|
|
YapDatabaseViewGrouping *viewGrouping = [YapDatabaseViewGrouping withObjectBlock:^NSString *(
|
|
|
|
YapDatabaseReadTransaction *transaction, NSString *collection, NSString *key, id object) {
|
|
|
|
if ([object isKindOfClass:[TSOutgoingMessage class]]) {
|
|
|
|
return ((TSOutgoingMessage *)object).uniqueThreadId;
|
|
|
|
}
|
|
|
|
return nil;
|
|
|
|
}];
|
|
|
|
|
2017-06-15 19:37:10 +02:00
|
|
|
[self registerMessageDatabaseViewWithName:TSThreadOutgoingMessageDatabaseViewExtensionName
|
|
|
|
viewGrouping:viewGrouping
|
2018-10-03 23:41:43 +02:00
|
|
|
version:@"3"
|
2018-01-11 15:56:38 +01:00
|
|
|
storage:storage];
|
2017-05-30 19:03:56 +02:00
|
|
|
}
|
|
|
|
|
2018-01-26 21:53:26 +01:00
|
|
|
+ (void)asyncRegisterThreadDatabaseView:(OWSStorage *)storage
|
2017-06-15 19:37:10 +02:00
|
|
|
{
|
2018-01-11 15:56:38 +01:00
|
|
|
YapDatabaseView *threadView = [storage registeredExtension:TSThreadDatabaseViewExtensionName];
|
2015-12-07 03:31:43 +01:00
|
|
|
if (threadView) {
|
2018-08-27 16:29:51 +02:00
|
|
|
OWSFailDebug(@"Registered database view twice: %@", TSThreadDatabaseViewExtensionName);
|
2017-06-15 19:37:10 +02:00
|
|
|
return;
|
2015-12-07 03:31:43 +01:00
|
|
|
}
|
|
|
|
|
2017-07-27 17:51:36 +02:00
|
|
|
YapDatabaseViewGrouping *viewGrouping = [YapDatabaseViewGrouping withObjectBlock:^NSString *(
|
|
|
|
YapDatabaseReadTransaction *transaction, NSString *collection, NSString *key, id object) {
|
|
|
|
if (![object isKindOfClass:[TSThread class]]) {
|
2018-08-27 18:51:32 +02:00
|
|
|
OWSFailDebug(@"Unexpected entity %@ in collection: %@", [object class], collection);
|
2017-07-27 17:51:36 +02:00
|
|
|
return nil;
|
|
|
|
}
|
|
|
|
TSThread *thread = (TSThread *)object;
|
|
|
|
|
2019-10-04 06:52:59 +02:00
|
|
|
if (thread.shouldThreadBeVisible && !thread.isForceHidden) {
|
2017-08-02 20:18:27 +02:00
|
|
|
// Do nothing; we never hide threads that have ever had a message.
|
|
|
|
} else {
|
|
|
|
YapDatabaseViewTransaction *viewTransaction = [transaction ext:TSMessageDatabaseViewExtensionName];
|
2018-09-06 19:01:24 +02:00
|
|
|
OWSAssertDebug(viewTransaction);
|
2017-08-02 20:18:27 +02:00
|
|
|
NSUInteger threadMessageCount = [viewTransaction numberOfItemsInGroup:thread.uniqueId];
|
2019-10-04 06:52:59 +02:00
|
|
|
if (threadMessageCount < 1 || thread.isForceHidden) {
|
2017-08-02 20:18:27 +02:00
|
|
|
return nil;
|
|
|
|
}
|
2017-07-27 17:51:36 +02:00
|
|
|
}
|
|
|
|
|
2018-10-03 23:41:43 +02:00
|
|
|
return [thread isArchivedWithTransaction:transaction] ? TSArchiveGroup : TSInboxGroup;
|
2017-07-27 17:51:36 +02:00
|
|
|
}];
|
2015-12-07 03:31:43 +01:00
|
|
|
|
|
|
|
YapDatabaseViewSorting *viewSorting = [self threadSorting];
|
|
|
|
|
|
|
|
YapDatabaseViewOptions *options = [[YapDatabaseViewOptions alloc] init];
|
2017-07-06 04:31:58 +02:00
|
|
|
options.isPersistent = YES;
|
2015-12-07 03:31:43 +01:00
|
|
|
options.allowedCollections =
|
|
|
|
[[YapWhitelistBlacklist alloc] initWithWhitelist:[NSSet setWithObject:[TSThread collection]]];
|
|
|
|
|
|
|
|
YapDatabaseView *databaseView =
|
2018-12-11 23:07:50 +01:00
|
|
|
[[YapDatabaseAutoView alloc] initWithGrouping:viewGrouping sorting:viewSorting versionTag:@"4" options:options];
|
2017-07-27 17:51:36 +02:00
|
|
|
|
2018-01-26 21:53:26 +01:00
|
|
|
[storage asyncRegisterExtension:databaseView withName:TSThreadDatabaseViewExtensionName];
|
2015-12-07 03:31:43 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
+ (YapDatabaseViewSorting *)threadSorting {
|
|
|
|
return [YapDatabaseViewSorting withObjectBlock:^NSComparisonResult(YapDatabaseReadTransaction *transaction,
|
2018-06-20 16:46:03 +02:00
|
|
|
NSString *group,
|
|
|
|
NSString *collection1,
|
|
|
|
NSString *key1,
|
|
|
|
id object1,
|
|
|
|
NSString *collection2,
|
|
|
|
NSString *key2,
|
|
|
|
id object2) {
|
|
|
|
if (![object1 isKindOfClass:[TSThread class]]) {
|
2018-08-27 18:51:32 +02:00
|
|
|
OWSFailDebug(@"Unexpected entity %@ in collection: %@", [object1 class], collection1);
|
2018-06-20 16:46:03 +02:00
|
|
|
return NSOrderedSame;
|
|
|
|
}
|
|
|
|
if (![object2 isKindOfClass:[TSThread class]]) {
|
2018-08-27 18:51:32 +02:00
|
|
|
OWSFailDebug(@"Unexpected entity %@ in collection: %@", [object2 class], collection2);
|
2018-06-20 16:46:03 +02:00
|
|
|
return NSOrderedSame;
|
|
|
|
}
|
|
|
|
TSThread *thread1 = (TSThread *)object1;
|
|
|
|
TSThread *thread2 = (TSThread *)object2;
|
|
|
|
if ([group isEqualToString:TSArchiveGroup] || [group isEqualToString:TSInboxGroup]) {
|
2018-10-03 23:41:43 +02:00
|
|
|
|
|
|
|
TSInteraction *_Nullable lastInteractionForInbox1 =
|
|
|
|
[thread1 lastInteractionForInboxWithTransaction:transaction];
|
|
|
|
NSDate *date1 = lastInteractionForInbox1 ? lastInteractionForInbox1.receivedAtDate : thread1.creationDate;
|
|
|
|
|
|
|
|
TSInteraction *_Nullable lastInteractionForInbox2 =
|
|
|
|
[thread2 lastInteractionForInboxWithTransaction:transaction];
|
|
|
|
NSDate *date2 = lastInteractionForInbox2 ? lastInteractionForInbox2.receivedAtDate : thread2.creationDate;
|
|
|
|
|
|
|
|
return [date1 compare:date2];
|
2018-06-20 16:46:03 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
return NSOrderedSame;
|
2015-12-07 03:31:43 +01:00
|
|
|
}];
|
|
|
|
}
|
|
|
|
|
|
|
|
+ (YapDatabaseViewSorting *)messagesSorting {
|
|
|
|
return [YapDatabaseViewSorting withObjectBlock:^NSComparisonResult(YapDatabaseReadTransaction *transaction,
|
2018-06-20 16:46:03 +02:00
|
|
|
NSString *group,
|
|
|
|
NSString *collection1,
|
|
|
|
NSString *key1,
|
|
|
|
id object1,
|
|
|
|
NSString *collection2,
|
|
|
|
NSString *key2,
|
|
|
|
id object2) {
|
|
|
|
if (![object1 isKindOfClass:[TSInteraction class]]) {
|
2018-08-27 18:51:32 +02:00
|
|
|
OWSFailDebug(@"Unexpected entity %@ in collection: %@", [object1 class], collection1);
|
2018-06-20 16:46:03 +02:00
|
|
|
return NSOrderedSame;
|
|
|
|
}
|
|
|
|
if (![object2 isKindOfClass:[TSInteraction class]]) {
|
2018-08-27 18:51:32 +02:00
|
|
|
OWSFailDebug(@"Unexpected entity %@ in collection: %@", [object2 class], collection2);
|
2018-06-20 16:46:03 +02:00
|
|
|
return NSOrderedSame;
|
|
|
|
}
|
|
|
|
TSInteraction *message1 = (TSInteraction *)object1;
|
|
|
|
TSInteraction *message2 = (TSInteraction *)object2;
|
|
|
|
|
|
|
|
return [message1 compareForSorting:message2];
|
2015-12-07 03:31:43 +01:00
|
|
|
}];
|
|
|
|
}
|
|
|
|
|
2018-01-11 15:56:38 +01:00
|
|
|
+ (void)asyncRegisterSecondaryDevicesDatabaseView:(OWSStorage *)storage
|
2016-09-03 00:53:47 +02:00
|
|
|
{
|
2018-03-20 22:22:19 +01:00
|
|
|
YapDatabaseViewGrouping *viewGrouping = [YapDatabaseViewGrouping withObjectBlock:^NSString *_Nullable(
|
|
|
|
YapDatabaseReadTransaction *transaction, NSString *collection, NSString *key, id object) {
|
2018-06-20 16:46:03 +02:00
|
|
|
if (![object isKindOfClass:[OWSDevice class]]) {
|
2018-08-27 18:51:32 +02:00
|
|
|
OWSFailDebug(@"Unexpected entity %@ in collection: %@", [object class], collection);
|
2018-06-20 16:46:03 +02:00
|
|
|
return nil;
|
|
|
|
}
|
|
|
|
OWSDevice *device = (OWSDevice *)object;
|
|
|
|
if (![device isPrimaryDevice]) {
|
|
|
|
return TSSecondaryDevicesGroup;
|
2018-03-20 22:22:19 +01:00
|
|
|
}
|
|
|
|
return nil;
|
|
|
|
}];
|
2016-09-03 00:53:47 +02:00
|
|
|
|
2018-06-20 16:46:03 +02:00
|
|
|
YapDatabaseViewSorting *viewSorting = [YapDatabaseViewSorting withObjectBlock:^NSComparisonResult(
|
|
|
|
YapDatabaseReadTransaction *transaction,
|
|
|
|
NSString *group,
|
|
|
|
NSString *collection1,
|
|
|
|
NSString *key1,
|
|
|
|
id object1,
|
|
|
|
NSString *collection2,
|
|
|
|
NSString *key2,
|
|
|
|
id object2) {
|
|
|
|
if (![object1 isKindOfClass:[OWSDevice class]]) {
|
2018-08-27 18:51:32 +02:00
|
|
|
OWSFailDebug(@"Unexpected entity %@ in collection: %@", [object1 class], collection1);
|
2018-06-20 16:46:03 +02:00
|
|
|
return NSOrderedSame;
|
|
|
|
}
|
|
|
|
if (![object2 isKindOfClass:[OWSDevice class]]) {
|
2018-08-27 18:51:32 +02:00
|
|
|
OWSFailDebug(@"Unexpected entity %@ in collection: %@", [object2 class], collection2);
|
2016-09-03 00:53:47 +02:00
|
|
|
return NSOrderedSame;
|
2018-06-20 16:46:03 +02:00
|
|
|
}
|
|
|
|
OWSDevice *device1 = (OWSDevice *)object1;
|
|
|
|
OWSDevice *device2 = (OWSDevice *)object2;
|
|
|
|
|
|
|
|
return [device2.createdAt compare:device1.createdAt];
|
|
|
|
}];
|
2016-09-03 00:53:47 +02:00
|
|
|
|
|
|
|
YapDatabaseViewOptions *options = [YapDatabaseViewOptions new];
|
|
|
|
options.isPersistent = YES;
|
|
|
|
|
|
|
|
NSSet *deviceCollection = [NSSet setWithObject:[OWSDevice collection]];
|
|
|
|
options.allowedCollections = [[YapWhitelistBlacklist alloc] initWithWhitelist:deviceCollection];
|
|
|
|
|
|
|
|
YapDatabaseView *view =
|
2017-12-12 16:31:05 +01:00
|
|
|
[[YapDatabaseAutoView alloc] initWithGrouping:viewGrouping sorting:viewSorting versionTag:@"3" options:options];
|
2016-09-03 00:53:47 +02:00
|
|
|
|
2018-01-26 21:53:26 +01:00
|
|
|
[storage asyncRegisterExtension:view withName:TSSecondaryDevicesDatabaseViewExtensionName];
|
2016-09-03 00:53:47 +02:00
|
|
|
}
|
|
|
|
|
2018-03-20 22:22:19 +01:00
|
|
|
+ (void)asyncRegisterLazyRestoreAttachmentsDatabaseView:(OWSStorage *)storage
|
|
|
|
{
|
|
|
|
YapDatabaseViewGrouping *viewGrouping = [YapDatabaseViewGrouping withObjectBlock:^NSString *_Nullable(
|
|
|
|
YapDatabaseReadTransaction *transaction, NSString *collection, NSString *key, id object) {
|
|
|
|
if (![object isKindOfClass:[TSAttachment class]]) {
|
2018-08-27 18:51:32 +02:00
|
|
|
OWSFailDebug(@"Unexpected entity %@ in collection: %@", [object class], collection);
|
2018-03-20 22:22:19 +01:00
|
|
|
return nil;
|
|
|
|
}
|
2018-11-19 17:38:29 +01:00
|
|
|
if (![object isKindOfClass:[TSAttachmentPointer class]]) {
|
2018-03-20 22:22:19 +01:00
|
|
|
return nil;
|
|
|
|
}
|
2018-11-19 17:38:29 +01:00
|
|
|
TSAttachmentPointer *attachmentPointer = (TSAttachmentPointer *)object;
|
|
|
|
if (attachmentPointer.lazyRestoreFragment) {
|
2018-03-20 22:22:19 +01:00
|
|
|
return TSLazyRestoreAttachmentsGroup;
|
|
|
|
} else {
|
|
|
|
return nil;
|
|
|
|
}
|
|
|
|
}];
|
|
|
|
|
2018-11-19 17:38:29 +01:00
|
|
|
YapDatabaseViewSorting *viewSorting =
|
|
|
|
[YapDatabaseViewSorting withObjectBlock:^NSComparisonResult(YapDatabaseReadTransaction *transaction,
|
|
|
|
NSString *group,
|
|
|
|
NSString *collection1,
|
|
|
|
NSString *key1,
|
|
|
|
id object1,
|
|
|
|
NSString *collection2,
|
|
|
|
NSString *key2,
|
|
|
|
id object2) {
|
|
|
|
if (![object1 isKindOfClass:[TSAttachmentPointer class]]) {
|
|
|
|
OWSFailDebug(@"Unexpected entity %@ in collection: %@", [object1 class], collection1);
|
|
|
|
return NSOrderedSame;
|
|
|
|
}
|
|
|
|
if (![object2 isKindOfClass:[TSAttachmentPointer class]]) {
|
|
|
|
OWSFailDebug(@"Unexpected entity %@ in collection: %@", [object2 class], collection2);
|
|
|
|
return NSOrderedSame;
|
|
|
|
}
|
2018-03-20 22:22:19 +01:00
|
|
|
|
2018-11-19 17:38:29 +01:00
|
|
|
// Specific ordering doesn't matter; we just need a stable ordering.
|
|
|
|
TSAttachmentPointer *attachmentPointer1 = (TSAttachmentPointer *)object1;
|
|
|
|
TSAttachmentPointer *attachmentPointer2 = (TSAttachmentPointer *)object2;
|
|
|
|
return [attachmentPointer1.uniqueId compare:attachmentPointer2.uniqueId];
|
|
|
|
}];
|
2018-03-20 22:22:19 +01:00
|
|
|
|
|
|
|
YapDatabaseViewOptions *options = [YapDatabaseViewOptions new];
|
|
|
|
options.isPersistent = YES;
|
|
|
|
options.allowedCollections =
|
|
|
|
[[YapWhitelistBlacklist alloc] initWithWhitelist:[NSSet setWithObject:[TSAttachment collection]]];
|
|
|
|
YapDatabaseView *view =
|
2018-11-19 17:38:29 +01:00
|
|
|
[[YapDatabaseAutoView alloc] initWithGrouping:viewGrouping sorting:viewSorting versionTag:@"4" options:options];
|
2018-08-02 19:44:00 +02:00
|
|
|
[storage asyncRegisterExtension:view withName:TSLazyRestoreAttachmentsDatabaseViewExtensionName];
|
2018-03-20 22:22:19 +01:00
|
|
|
}
|
|
|
|
|
2017-06-15 19:37:10 +02:00
|
|
|
+ (id)unseenDatabaseViewExtension:(YapDatabaseReadTransaction *)transaction
|
|
|
|
{
|
2018-09-06 19:01:24 +02:00
|
|
|
OWSAssertDebug(transaction);
|
2017-06-15 19:37:10 +02:00
|
|
|
|
2019-01-04 19:35:50 +01:00
|
|
|
id _Nullable result = [transaction ext:TSUnseenDatabaseViewExtensionName];
|
|
|
|
OWSAssertDebug(result);
|
2017-06-15 19:37:10 +02:00
|
|
|
|
2019-01-04 19:35:50 +01:00
|
|
|
// TODO: I believe we can now safely remove this?
|
2017-06-15 19:37:10 +02:00
|
|
|
if (!result) {
|
|
|
|
result = [transaction ext:TSUnreadDatabaseViewExtensionName];
|
2018-09-06 19:01:24 +02:00
|
|
|
OWSAssertDebug(result);
|
2017-06-15 19:37:10 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
return result;
|
|
|
|
}
|
|
|
|
|
2018-10-03 23:41:43 +02:00
|
|
|
// MJK TODO - dynamic interactions
|
2017-06-15 19:37:10 +02:00
|
|
|
+ (id)threadOutgoingMessageDatabaseView:(YapDatabaseReadTransaction *)transaction
|
|
|
|
{
|
2018-09-06 19:01:24 +02:00
|
|
|
OWSAssertDebug(transaction);
|
2017-06-15 19:37:10 +02:00
|
|
|
|
|
|
|
id result = [transaction ext:TSThreadOutgoingMessageDatabaseViewExtensionName];
|
2018-09-06 19:01:24 +02:00
|
|
|
OWSAssertDebug(result);
|
2017-06-15 19:37:10 +02:00
|
|
|
|
|
|
|
return result;
|
|
|
|
}
|
|
|
|
|
|
|
|
+ (id)threadSpecialMessagesDatabaseView:(YapDatabaseReadTransaction *)transaction
|
|
|
|
{
|
2018-09-06 19:01:24 +02:00
|
|
|
OWSAssertDebug(transaction);
|
2017-06-15 19:37:10 +02:00
|
|
|
|
|
|
|
id result = [transaction ext:TSThreadSpecialMessagesDatabaseViewExtensionName];
|
2018-09-06 19:01:24 +02:00
|
|
|
OWSAssertDebug(result);
|
2017-06-15 19:37:10 +02:00
|
|
|
|
|
|
|
return result;
|
|
|
|
}
|
|
|
|
|
2015-12-07 03:31:43 +01:00
|
|
|
@end
|
2019-01-23 15:05:08 +01:00
|
|
|
|
|
|
|
NS_ASSUME_NONNULL_END
|