117 lines
5.7 KiB
Objective-C
117 lines
5.7 KiB
Objective-C
//
|
|
// Copyright (c) 2018 Open Whisper Systems. All rights reserved.
|
|
//
|
|
|
|
#import "DebugUIDiskUsage.h"
|
|
#import "OWSOrphanDataCleaner.h"
|
|
#import "OWSTableViewController.h"
|
|
#import "Signal-Swift.h"
|
|
#import <SignalCoreKit/NSDate+OWS.h>
|
|
#import <SignalServiceKit/OWSPrimaryStorage.h>
|
|
#import <SignalServiceKit/TSDatabaseView.h>
|
|
#import <SignalServiceKit/TSInteraction.h>
|
|
|
|
NS_ASSUME_NONNULL_BEGIN
|
|
|
|
@implementation DebugUIDiskUsage
|
|
|
|
#pragma mark - Factory Methods
|
|
|
|
- (NSString *)name
|
|
{
|
|
return @"Orphans & Disk Usage";
|
|
}
|
|
|
|
- (nullable OWSTableSection *)sectionForThread:(nullable TSThread *)thread
|
|
{
|
|
return [OWSTableSection sectionWithTitle:self.name
|
|
items:@[
|
|
[OWSTableItem itemWithTitle:@"Audit & Log"
|
|
actionBlock:^{
|
|
[OWSOrphanDataCleaner auditAndCleanup:NO];
|
|
}],
|
|
[OWSTableItem itemWithTitle:@"Audit & Clean Up"
|
|
actionBlock:^{
|
|
[OWSOrphanDataCleaner auditAndCleanup:YES];
|
|
}],
|
|
[OWSTableItem itemWithTitle:@"Save All Attachments"
|
|
actionBlock:^{
|
|
[DebugUIDiskUsage saveAllAttachments];
|
|
}],
|
|
[OWSTableItem itemWithTitle:@"Delete Messages older than 3 Months"
|
|
actionBlock:^{
|
|
[DebugUIDiskUsage deleteOldMessages_3Months];
|
|
}],
|
|
]];
|
|
}
|
|
|
|
+ (void)saveAllAttachments
|
|
{
|
|
OWSPrimaryStorage *primaryStorage = [OWSPrimaryStorage sharedManager];
|
|
[primaryStorage.newDatabaseConnection readWriteWithBlock:^(YapDatabaseReadWriteTransaction *_Nonnull transaction) {
|
|
NSMutableArray<TSAttachmentStream *> *attachmentStreams = [NSMutableArray new];
|
|
[transaction enumerateKeysAndObjectsInCollection:TSAttachmentStream.collection
|
|
usingBlock:^(NSString *key, TSAttachment *attachment, BOOL *stop) {
|
|
if (![attachment isKindOfClass:[TSAttachmentStream class]]) {
|
|
return;
|
|
}
|
|
TSAttachmentStream *attachmentStream
|
|
= (TSAttachmentStream *)attachment;
|
|
[attachmentStreams addObject:attachmentStream];
|
|
}];
|
|
|
|
OWSLogInfo(@"Saving %zd attachment streams.", attachmentStreams.count);
|
|
|
|
// Persist the new localRelativeFilePath property of TSAttachmentStream.
|
|
// For performance, we want to upgrade all existing attachment streams in
|
|
// a single transaction.
|
|
for (TSAttachmentStream *attachmentStream in attachmentStreams) {
|
|
[attachmentStream saveWithTransaction:transaction];
|
|
}
|
|
}];
|
|
}
|
|
|
|
+ (void)deleteOldMessages_3Months
|
|
{
|
|
[self deleteOldMessages:kMonthInterval * 3];
|
|
}
|
|
|
|
+ (void)deleteOldMessages:(NSTimeInterval)maxAgeSeconds
|
|
{
|
|
OWSPrimaryStorage *primaryStorage = [OWSPrimaryStorage sharedManager];
|
|
[primaryStorage.newDatabaseConnection readWriteWithBlock:^(YapDatabaseReadWriteTransaction *_Nonnull transaction) {
|
|
NSMutableArray<NSString *> *threadIds = [NSMutableArray new];
|
|
YapDatabaseViewTransaction *interactionsByThread = [transaction ext:TSMessageDatabaseViewExtensionName];
|
|
[interactionsByThread enumerateGroupsUsingBlock:^(NSString *group, BOOL *stop) {
|
|
[threadIds addObject:group];
|
|
}];
|
|
NSMutableArray<TSInteraction *> *interactionsToDelete = [NSMutableArray new];
|
|
for (NSString *threadId in threadIds) {
|
|
[interactionsByThread enumerateKeysAndObjectsInGroup:threadId
|
|
usingBlock:^(NSString *collection,
|
|
NSString *key,
|
|
TSInteraction *interaction,
|
|
NSUInteger index,
|
|
BOOL *stop) {
|
|
NSTimeInterval ageSeconds
|
|
= fabs(interaction.dateForSorting.timeIntervalSinceNow);
|
|
if (ageSeconds < maxAgeSeconds) {
|
|
*stop = YES;
|
|
return;
|
|
}
|
|
[interactionsToDelete addObject:interaction];
|
|
}];
|
|
}
|
|
|
|
OWSLogInfo(@"Deleting %zd interactions.", interactionsToDelete.count);
|
|
|
|
for (TSInteraction *interaction in interactionsToDelete) {
|
|
[interaction removeWithTransaction:transaction];
|
|
}
|
|
}];
|
|
}
|
|
|
|
@end
|
|
|
|
NS_ASSUME_NONNULL_END
|