mirror of https://github.com/oxen-io/session-ios
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
121 lines
3.2 KiB
Objective-C
121 lines
3.2 KiB
Objective-C
//
|
|
// Copyright (c) 2018 Open Whisper Systems. All rights reserved.
|
|
//
|
|
|
|
#import "OWSMessageUtils.h"
|
|
#import "AppContext.h"
|
|
#import "MIMETypeUtil.h"
|
|
|
|
#import "OWSPrimaryStorage.h"
|
|
#import "TSAccountManager.h"
|
|
#import "TSAttachment.h"
|
|
#import "TSAttachmentStream.h"
|
|
#import "TSDatabaseView.h"
|
|
#import "TSIncomingMessage.h"
|
|
#import "TSMessage.h"
|
|
#import "TSOutgoingMessage.h"
|
|
#import "TSQuotedMessage.h"
|
|
#import "TSThread.h"
|
|
#import "UIImage+OWS.h"
|
|
#import <YapDatabase/YapDatabase.h>
|
|
#import "SSKAsserts.h"
|
|
#import <SignalUtilitiesKit/SignalUtilitiesKit-Swift.h>
|
|
|
|
NS_ASSUME_NONNULL_BEGIN
|
|
|
|
@interface OWSMessageUtils ()
|
|
|
|
@property (nonatomic, readonly) YapDatabaseConnection *dbConnection;
|
|
|
|
@end
|
|
|
|
#pragma mark -
|
|
|
|
@implementation OWSMessageUtils
|
|
|
|
+ (instancetype)sharedManager
|
|
{
|
|
static OWSMessageUtils *sharedMyManager = nil;
|
|
static dispatch_once_t onceToken;
|
|
dispatch_once(&onceToken, ^{
|
|
sharedMyManager = [[self alloc] initDefault];
|
|
});
|
|
return sharedMyManager;
|
|
}
|
|
|
|
- (instancetype)initDefault
|
|
{
|
|
OWSPrimaryStorage *primaryStorage = [OWSPrimaryStorage sharedManager];
|
|
|
|
return [self initWithPrimaryStorage:primaryStorage];
|
|
}
|
|
|
|
- (instancetype)initWithPrimaryStorage:(OWSPrimaryStorage *)primaryStorage
|
|
{
|
|
self = [super init];
|
|
|
|
if (!self) {
|
|
return self;
|
|
}
|
|
|
|
_dbConnection = primaryStorage.newDatabaseConnection;
|
|
|
|
OWSSingletonAssert();
|
|
|
|
return self;
|
|
}
|
|
|
|
- (NSUInteger)unreadMessagesCount
|
|
{
|
|
__block NSUInteger count = 0;
|
|
|
|
[LKStorage readWithBlock:^(YapDatabaseReadTransaction *transaction) {
|
|
YapDatabaseViewTransaction *unreadMessages = [transaction ext:TSUnreadDatabaseViewExtensionName];
|
|
NSArray<NSString *> *allGroups = [unreadMessages allGroups];
|
|
// FIXME: Confusingly, `allGroups` includes contact threads as well
|
|
for (NSString *groupID in allGroups) {
|
|
TSThread *thread = [TSThread fetchObjectWithUniqueID:groupID transaction:transaction];
|
|
|
|
// Don't increase the count for muted threads or message requests
|
|
if (thread.isMuted || thread.isMessageRequest) { continue; }
|
|
|
|
BOOL isGroupThread = thread.isGroupThread;
|
|
|
|
// For groups that only notifiy for mentions
|
|
if (isGroupThread && ((TSGroupThread *)thread).isOnlyNotifyingForMentions) {
|
|
count += [thread unreadMentionMessageCountWithTransaction:transaction];
|
|
} else {
|
|
count += [thread unreadMessageCountWithTransaction:transaction];
|
|
}
|
|
}
|
|
}];
|
|
|
|
return count;
|
|
}
|
|
|
|
- (NSUInteger)unreadMessagesCountExcept:(TSThread *)thread
|
|
{
|
|
__block NSUInteger numberOfItems;
|
|
[self.dbConnection readWithBlock:^(YapDatabaseReadTransaction *transaction) {
|
|
id databaseView = [transaction ext:TSUnreadDatabaseViewExtensionName];
|
|
OWSAssertDebug(databaseView);
|
|
numberOfItems = ([databaseView numberOfItemsInAllGroups] - [databaseView numberOfItemsInGroup:thread.uniqueId]);
|
|
}];
|
|
|
|
return numberOfItems;
|
|
}
|
|
|
|
- (void)updateApplicationBadgeCount
|
|
{
|
|
if (!CurrentAppContext().isMainApp) {
|
|
return;
|
|
}
|
|
|
|
NSUInteger numberOfItems = [self unreadMessagesCount];
|
|
[CurrentAppContext() setMainAppBadgeNumber:numberOfItems];
|
|
}
|
|
|
|
@end
|
|
|
|
NS_ASSUME_NONNULL_END
|