313 lines
12 KiB
Objective-C
313 lines
12 KiB
Objective-C
//
|
|
// Copyright (c) 2018 Open Whisper Systems. All rights reserved.
|
|
//
|
|
|
|
#import "OWSBackupExportViewController.h"
|
|
#import "OWSBackup.h"
|
|
#import "OWSProgressView.h"
|
|
#import "Signal-Swift.h"
|
|
#import "ThreadUtil.h"
|
|
#import <SignalMessaging/AttachmentSharing.h>
|
|
#import <SignalMessaging/Environment.h>
|
|
#import <SignalMessaging/SignalMessaging-Swift.h>
|
|
#import <SignalMessaging/UIColor+OWS.h>
|
|
#import <SignalMessaging/UIFont+OWS.h>
|
|
#import <SignalMessaging/UIView+OWS.h>
|
|
#import <SignalServiceKit/MIMETypeUtil.h>
|
|
|
|
NS_ASSUME_NONNULL_BEGIN
|
|
|
|
@interface OWSBackupExportViewController () <OWSBackupDelegate>
|
|
|
|
@property (nonatomic) OWSBackup *backup;
|
|
|
|
@property (nonatomic, nullable) OWSProgressView *progressView;
|
|
|
|
@end
|
|
|
|
#pragma mark -
|
|
|
|
@implementation OWSBackupExportViewController
|
|
|
|
- (void)loadView
|
|
{
|
|
[super loadView];
|
|
|
|
self.view.backgroundColor = [UIColor whiteColor];
|
|
|
|
self.navigationItem.title = NSLocalizedString(@"BACKUP_EXPORT_VIEW_TITLE", @"Title for the 'backup export' view.");
|
|
self.navigationItem.leftBarButtonItem =
|
|
[[UIBarButtonItem alloc] initWithBarButtonSystemItem:UIBarButtonSystemItemStop
|
|
target:self
|
|
action:@selector(dismissWasPressed:)];
|
|
|
|
self.backup.delegate = self;
|
|
|
|
[self updateUI];
|
|
}
|
|
|
|
- (void)exportBackup:(TSThread *_Nullable)currentThread skipPassword:(BOOL)skipPassword
|
|
{
|
|
OWSAssertIsOnMainThread();
|
|
|
|
// We set ourselves as the delegate of the backup later,
|
|
// after we've loaded our view.
|
|
self.backup = [OWSBackup new];
|
|
[self.backup exportBackup:currentThread skipPassword:skipPassword];
|
|
}
|
|
|
|
- (void)updateUI
|
|
{
|
|
for (UIView *subview in self.view.subviews) {
|
|
[subview removeFromSuperview];
|
|
}
|
|
self.progressView = nil;
|
|
|
|
switch (self.backup.backupState) {
|
|
case OWSBackupState_InProgress:
|
|
[self showInProgressUI];
|
|
break;
|
|
case OWSBackupState_Cancelled:
|
|
[self showCancelledUI];
|
|
break;
|
|
case OWSBackupState_Complete:
|
|
[self showCompleteUI];
|
|
break;
|
|
case OWSBackupState_Failed:
|
|
[self showFailedUI];
|
|
break;
|
|
}
|
|
}
|
|
|
|
- (void)showInProgressUI
|
|
{
|
|
self.progressView = [OWSProgressView new];
|
|
self.progressView.color = [UIColor ows_materialBlueColor];
|
|
self.progressView.progress = self.backup.backupProgress;
|
|
[self.progressView autoSetDimension:ALDimensionWidth toSize:300];
|
|
[self.progressView autoSetDimension:ALDimensionHeight toSize:20];
|
|
|
|
UILabel *label = [UILabel new];
|
|
label.text = NSLocalizedString(
|
|
@"BACKUP_EXPORT_IN_PROGRESS_MESSAGE", @"Message indicating that backup export is in progress.");
|
|
label.textColor = [UIColor blackColor];
|
|
label.font = [UIFont ows_regularFontWithSize:18.f];
|
|
label.textAlignment = NSTextAlignmentCenter;
|
|
label.numberOfLines = 0;
|
|
label.lineBreakMode = NSLineBreakByWordWrapping;
|
|
|
|
UIView *container = [UIView verticalStackWithSubviews:@[
|
|
label,
|
|
self.progressView,
|
|
]
|
|
spacing:10];
|
|
[self.view addSubview:container];
|
|
[container autoVCenterInSuperview];
|
|
[container autoPinWidthToSuperviewWithMargin:25.f];
|
|
}
|
|
|
|
- (void)showCancelledUI
|
|
{
|
|
// Show nothing.
|
|
}
|
|
|
|
- (void)showCompleteUI
|
|
{
|
|
NSMutableArray<UIView *> *subviews = [NSMutableArray new];
|
|
|
|
{
|
|
NSString *message = NSLocalizedString(
|
|
@"BACKUP_EXPORT_COMPLETE_MESSAGE", @"Message indicating that backup export is complete.");
|
|
|
|
UILabel *label = [UILabel new];
|
|
label.text = message;
|
|
label.textColor = [UIColor blackColor];
|
|
label.font = [UIFont ows_regularFontWithSize:18.f];
|
|
label.textAlignment = NSTextAlignmentCenter;
|
|
label.numberOfLines = 0;
|
|
label.lineBreakMode = NSLineBreakByWordWrapping;
|
|
[subviews addObject:label];
|
|
}
|
|
|
|
if (self.backup.backupPassword) {
|
|
NSString *message = [NSString stringWithFormat:NSLocalizedString(@"BACKUP_EXPORT_PASSWORD_MESSAGE_FORMAT",
|
|
@"Format for message indicating that backup export "
|
|
@"is complete. Embeds: {{the backup password}}."),
|
|
self.backup.backupPassword];
|
|
|
|
UILabel *label = [UILabel new];
|
|
label.text = message;
|
|
label.textColor = [UIColor blackColor];
|
|
label.font = [UIFont ows_regularFontWithSize:14.f];
|
|
label.textAlignment = NSTextAlignmentCenter;
|
|
label.numberOfLines = 0;
|
|
label.lineBreakMode = NSLineBreakByWordWrapping;
|
|
[subviews addObject:label];
|
|
}
|
|
|
|
[subviews addObject:[UIView new]];
|
|
|
|
if (self.backup.backupPassword) {
|
|
[subviews
|
|
addObject:[self makeButtonWithTitle:NSLocalizedString(@"BACKUP_EXPORT_COPY_PASSWORD_BUTTON",
|
|
@"Label for button that copies backup password to the pasteboard.")
|
|
selector:@selector(copyPassword)]];
|
|
}
|
|
|
|
[subviews addObject:[self makeButtonWithTitle:NSLocalizedString(@"BACKUP_EXPORT_SHARE_BACKUP_BUTTON",
|
|
@"Label for button that opens share UI for backup.")
|
|
selector:@selector(shareBackup)]];
|
|
|
|
if (self.backup.currentThread) {
|
|
[subviews
|
|
addObject:[self makeButtonWithTitle:NSLocalizedString(@"BACKUP_EXPORT_SEND_BACKUP_BUTTON",
|
|
@"Label for button that 'send backup' in the current conversation.")
|
|
selector:@selector(sendBackup)]];
|
|
}
|
|
|
|
// TODO: We should offer the option to save the backup to "Files", iCloud, Dropbox, etc.
|
|
|
|
UIView *container = [UIView verticalStackWithSubviews:subviews spacing:10];
|
|
[self.view addSubview:container];
|
|
[container autoVCenterInSuperview];
|
|
[container autoPinWidthToSuperviewWithMargin:25.f];
|
|
}
|
|
|
|
- (void)showFailedUI
|
|
{
|
|
NSMutableArray<UIView *> *subviews = [NSMutableArray new];
|
|
|
|
{
|
|
NSString *message
|
|
= NSLocalizedString(@"BACKUP_EXPORT_FAILED_MESSAGE", @"Message indicating that backup export failed.");
|
|
|
|
UILabel *label = [UILabel new];
|
|
label.text = message;
|
|
label.textColor = [UIColor blackColor];
|
|
label.font = [UIFont ows_regularFontWithSize:18.f];
|
|
label.textAlignment = NSTextAlignmentCenter;
|
|
label.numberOfLines = 0;
|
|
label.lineBreakMode = NSLineBreakByWordWrapping;
|
|
[subviews addObject:label];
|
|
}
|
|
|
|
// TODO: We should offer the option to save the backup to "Files", iCloud, Dropbox, etc.
|
|
|
|
UIView *container = [UIView verticalStackWithSubviews:subviews spacing:10];
|
|
[self.view addSubview:container];
|
|
[container autoVCenterInSuperview];
|
|
[container autoPinWidthToSuperviewWithMargin:25.f];
|
|
}
|
|
|
|
- (UIView *)makeButtonWithTitle:(NSString *)title selector:(SEL)selector
|
|
{
|
|
const CGFloat kButtonHeight = 40;
|
|
OWSFlatButton *button = [OWSFlatButton buttonWithTitle:title
|
|
font:[OWSFlatButton fontForHeight:kButtonHeight]
|
|
titleColor:[UIColor whiteColor]
|
|
backgroundColor:[UIColor ows_materialBlueColor]
|
|
target:self
|
|
selector:selector];
|
|
[button autoSetDimension:ALDimensionWidth toSize:140];
|
|
[button autoSetDimension:ALDimensionHeight toSize:kButtonHeight];
|
|
return button;
|
|
}
|
|
|
|
- (void)copyPassword
|
|
{
|
|
OWSAssert(self.backup.backupPassword.length > 0);
|
|
|
|
// TODO: We could consider clearing the password from the pasteboard after a certain delay.
|
|
[UIPasteboard.generalPasteboard setString:self.backup.backupPassword];
|
|
}
|
|
|
|
- (void)shareBackup
|
|
{
|
|
OWSAssertIsOnMainThread();
|
|
OWSAssert(self.backup.backupZipPath.length > 0);
|
|
|
|
[AttachmentSharing showShareUIForURL:[NSURL fileURLWithPath:self.backup.backupZipPath]];
|
|
}
|
|
|
|
- (void)sendBackup
|
|
{
|
|
OWSAssertIsOnMainThread();
|
|
OWSAssert(self.backup.backupZipPath.length > 0);
|
|
OWSAssert(self.backup.currentThread);
|
|
|
|
[ModalActivityIndicatorViewController
|
|
presentFromViewController:self
|
|
canCancel:NO
|
|
backgroundBlock:^(ModalActivityIndicatorViewController *modalActivityIndicator) {
|
|
NSString *fileName = [self.backup.backupZipPath lastPathComponent];
|
|
|
|
OWSMessageSender *messageSender = [Environment current].messageSender;
|
|
NSString *utiType = [MIMETypeUtil utiTypeForFileExtension:fileName.pathExtension];
|
|
DataSource *_Nullable dataSource =
|
|
[DataSourcePath dataSourceWithFilePath:self.backup.backupZipPath];
|
|
[dataSource setSourceFilename:fileName];
|
|
SignalAttachment *attachment =
|
|
[SignalAttachment attachmentWithDataSource:dataSource dataUTI:utiType];
|
|
if (!attachment || [attachment hasError]) {
|
|
OWSFail(@"%@ attachment[%@]: %@",
|
|
self.logTag,
|
|
[attachment sourceFilename],
|
|
[attachment errorName]);
|
|
return;
|
|
}
|
|
dispatch_async(dispatch_get_main_queue(), ^{
|
|
[ThreadUtil
|
|
sendMessageWithAttachment:attachment
|
|
inThread:self.backup.currentThread
|
|
messageSender:messageSender
|
|
completion:^(NSError *_Nullable error) {
|
|
|
|
OWSAssertIsOnMainThread();
|
|
[modalActivityIndicator dismissWithCompletion:^{
|
|
if (error) {
|
|
DDLogError(@"%@ send backup failed: %@", self.logTag, error);
|
|
[OWSAlerts
|
|
showAlertWithTitle:NSLocalizedString(
|
|
@"BACKUP_EXPORT_SEND_BACKUP_FAILED",
|
|
@"Message indicating that sending "
|
|
@"the backup failed.")];
|
|
} else {
|
|
[OWSAlerts
|
|
showAlertWithTitle:
|
|
NSLocalizedString(@"BACKUP_EXPORT_SEND_BACKUP_SUCCESS",
|
|
@"Message indicating that sending the backup "
|
|
@"succeeded.")];
|
|
}
|
|
}];
|
|
}];
|
|
});
|
|
}];
|
|
}
|
|
|
|
- (void)dismissWasPressed:(id)sender
|
|
{
|
|
[self.backup cancel];
|
|
|
|
[self.navigationController popViewControllerAnimated:YES];
|
|
}
|
|
|
|
#pragma mark - OWSBackupDelegate
|
|
|
|
- (void)backupStateDidChange
|
|
{
|
|
DDLogInfo(@"%@ %s.", self.logTag, __PRETTY_FUNCTION__);
|
|
|
|
[self updateUI];
|
|
}
|
|
|
|
- (void)backupProgressDidChange
|
|
{
|
|
DDLogInfo(@"%@ %s.", self.logTag, __PRETTY_FUNCTION__);
|
|
|
|
self.progressView.progress = self.backup.backupProgress;
|
|
}
|
|
|
|
@end
|
|
|
|
NS_ASSUME_NONNULL_END
|