//==============================================================================
// This file is part of Master Password.
// Copyright (c) 2011-2017, Maarten Billemont.
//
// Master Password is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// Master Password is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU General Public License for more details.
//
// You can find a copy of the GNU General Public License in the
// LICENSE file.  Alternatively, see <http://www.gnu.org/licenses/>.
//==============================================================================

#import "MPiOSAppDelegate.h"
#import "MPAppDelegate_Key.h"
#import "MPAppDelegate_Store.h"
#import "MPStoreViewController.h"
#import "mpw-marshal.h"
#import "MPSecrets.h"

#import <Sentry/Sentry.h>
#import <Countly/Countly.h>

@interface MPiOSAppDelegate()<UIDocumentInteractionControllerDelegate>

@property(nonatomic, strong) UIDocumentInteractionController *interactionController;
@property(nonatomic, strong) PearlHangDetector *hangDetector;

@end

MPLogSink mpw_log_sink_pearl;
void mpw_log_sink_pearl(const MPLogEvent *record) {

    PearlLogLevel level = PearlLogLevelInfo;
    switch (record->level) {
        case LogLevelTrace:
            level = PearlLogLevelDebug;
            break;
        case LogLevelDebug:
            level = PearlLogLevelDebug;
            break;
        case LogLevelInfo:
            level = PearlLogLevelInfo;
            break;
        case LogLevelWarning:
            level = PearlLogLevelWarn;
            break;
        case LogLevelError:
            level = PearlLogLevelError;
            break;
        case LogLevelFatal:
            level = PearlLogLevelFatal;
            break;
    }

    [[PearlLogger get] inFile:[@(record->file) lastPathComponent] atLine:record->line fromFunction:@(record->function)
                    withLevel:level text:@(record->message)];
}

@implementation MPiOSAppDelegate

+ (void)initialize {

    static dispatch_once_t once = 0;
    dispatch_once( &once, ^{
        [PearlLogger get].printLevel = [[MPiOSConfig get].traceMode boolValue]? PearlLogLevelDebug: PearlLogLevelInfo;
        [PearlLogger get].historyLevel = [[MPiOSConfig get].traceMode boolValue]? PearlLogLevelTrace: PearlLogLevelInfo;
#ifdef DEBUG
        [PearlLogger get].printLevel = PearlLogLevelDebug;
#endif

        mpw_verbosity = LogLevelTrace;
        mpw_log_sink_register( &mpw_log_sink_pearl );
    } );
}

- (BOOL)application:(UIApplication *)application didFinishLaunchingWithOptions:(NSDictionary *)launchOptions {

    @try {
        // Sentry
        SentryClient.sharedClient = [[SentryClient alloc] initWithDsn:decrypt( sentryDSN ) didFailWithError:nil];
#ifdef DEBUG
        SentryClient.sharedClient.environment = @"Development";
#elif PUBLIC
        SentryClient.sharedClient.environment = @"Public";
#else
        SentryClient.sharedClient.environment = @"Private";
#endif
        SentryClient.sharedClient.enabled = [MPiOSConfig get].sendInfo;
        [SentryClient.sharedClient enableAutomaticBreadcrumbTracking];
        [SentryClient.sharedClient startCrashHandlerWithError:nil];
        [[PearlLogger get] registerListener:^BOOL(PearlLogMessage *message) {
            PearlLogLevel level = PearlLogLevelWarn;
            if ([[MPConfig get].sendInfo boolValue])
                level = PearlLogLevelDebug;

            if (message.level >= level) {
                SentrySeverity sentryLevel = kSentrySeverityInfo;
                switch (message.level) {
                    case PearlLogLevelTrace:
                        sentryLevel = kSentrySeverityDebug;
                        break;
                    case PearlLogLevelDebug:
                        sentryLevel = kSentrySeverityDebug;
                        break;
                    case PearlLogLevelInfo:
                        sentryLevel = kSentrySeverityInfo;
                        break;
                    case PearlLogLevelWarn:
                        sentryLevel = kSentrySeverityWarning;
                        break;
                    case PearlLogLevelError:
                        sentryLevel = kSentrySeverityError;
                        break;
                    case PearlLogLevelFatal:
                        sentryLevel = kSentrySeverityFatal;
                        break;
                }
                SentryBreadcrumb *breadcrumb = [[SentryBreadcrumb alloc] initWithLevel:sentryLevel category:@"Pearl"];
                breadcrumb.type = @"log";
                breadcrumb.message = message.message;
                breadcrumb.timestamp = message.occurrence;
                breadcrumb.data = @{ @"file": message.fileName, @"line": @(message.lineNumber), @"function": message.function };
                [SentryClient.sharedClient.breadcrumbs addBreadcrumb:breadcrumb];
            }

            return YES;
        }];

        // Countly
        CountlyConfig *countlyConfig = [CountlyConfig new];
        countlyConfig.host = @"https://countly.lyndir.com";
        countlyConfig.appKey = decrypt( countlyKey );
        countlyConfig.features = @[ CLYPushNotifications, CLYAutoViewTracking ];
        countlyConfig.requiresConsent = YES;
#if DEBUG
        countlyConfig.pushTestMode = CLYPushTestModeDevelopment;
#elif ! PUBLIC
        countlyConfig.pushTestMode = CLYPushTestModeTestFlightOrAdHoc;
#endif
        countlyConfig.alwaysUsePOST = YES;
        countlyConfig.deviceID = [PearlKeyChain deviceIdentifier];
        countlyConfig.secretSalt = decrypt( countlySalt );
        countlyConfig.enableDebug = YES;
        [Countly.sharedInstance startWithConfig:countlyConfig];

#if ! DEBUG
        [self.hangDetector = [[PearlHangDetector alloc] initWithHangAction:^(NSTimeInterval hangTime) {
            MPError( [NSError errorWithDomain:MPErrorDomain code:MPErrorHangCode userInfo:@{
                    @"time": @(hangTime)
            }], @"Timeout waiting for main thread after %fs.", hangTime );
        }] start];
#endif
    }
    @catch (id exception) {
        err( @"During Analytics Setup: %@", exception );
    }
    @try {
        PearlAddNotificationObserver( MPCheckConfigNotification, nil, [NSOperationQueue mainQueue], ^(id self, NSNotification *note) {
            [self updateConfigKey:note.object];
        } );
        PearlAddNotificationObserver( NSUserDefaultsDidChangeNotification, nil, nil, ^(id self, NSNotification *note) {
            [[NSNotificationCenter defaultCenter] postNotificationName:MPCheckConfigNotification object:nil];
        } );
    }
    @catch (id exception) {
        err( @"During Config Test: %@", exception );
    }
    @try {
        [super application:application didFinishLaunchingWithOptions:launchOptions];
    }
    @catch (id exception) {
        err( @"During Pearl Application Launch: %@", exception );
    }
    @try {
        inf( @"Started up with device identifier: %@", [PearlKeyChain deviceIdentifier] );

        PearlAddNotificationObserver(
                MPFoundInconsistenciesNotification, nil, [NSOperationQueue mainQueue], ^(id self, NSNotification *note) {
            switch ((MPFixableResult)[note.userInfo[MPInconsistenciesFixResultUserKey] unsignedIntegerValue]) {

                case MPFixableResultNoProblems:
                    break;
                case MPFixableResultProblemsFixed: {
                    UIAlertController *alert = [UIAlertController alertControllerWithTitle:@"Inconsistencies Fixed" message:
                                    @"Some inconsistencies were detected in your sites.\n"
                                    @"All issues were fixed."
                                                                            preferredStyle:UIAlertControllerStyleAlert];
                    [alert addAction:[UIAlertAction actionWithTitle:@"Okay" style:UIAlertActionStyleCancel handler:nil]];
                    [self presentViewController:alert animated:YES completion:nil];
                    break;
                }
                case MPFixableResultProblemsNotFixed: {
                    UIAlertController *alert = [UIAlertController alertControllerWithTitle:@"Inconsistencies Found" message:
                                    @"Some inconsistencies were detected in your sites.\n"
                                    @"Not all issues could be fixed.  Try signing in to each user or checking the logs."
                                                                            preferredStyle:UIAlertControllerStyleAlert];
                    [alert addAction:[UIAlertAction actionWithTitle:@"Okay" style:UIAlertActionStyleCancel handler:nil]];
                    [self presentViewController:alert animated:YES completion:nil];
                    break;
                }
            }
        } );

        if (@available( iOS 12, * )) {
            [Countly.sharedInstance askForNotificationPermissionWithOptions:UNAuthorizationOptionProvisional completionHandler:
                    ^(BOOL granted, NSError *error) {
                        inf( @"provisional: %d: %@", granted, error );
            }];
        }


        PearlMainQueueOperation( ^{
            if ([[MPiOSConfig get].showSetup boolValue])
                [self.navigationController performSegueWithIdentifier:@"setup" sender:self];

            if (![[NSUserDefaults standardUserDefaults] boolForKey:@"notificationsDecided"]) {
                UIAlertController *alert =  [UIAlertController alertControllerWithTitle:@"Coming Soon" message:
                                @"Master Password is rolling out a new modern personal security platform and we're excited to bring you along.\n\n"
                                @"When it's time, we'll send you a notification to help you make an effortless transition."
                                                                        preferredStyle:UIAlertControllerStyleAlert];
                [alert addAction:[UIAlertAction actionWithTitle:@"Thanks" style:UIAlertActionStyleDefault handler:^(UIAlertAction *action) {
                    [Countly.sharedInstance askForNotificationPermission];
                    [[NSUserDefaults standardUserDefaults] setBool:YES forKey:@"notificationsDecided"];
                }]];
                [(self.navigationController.presentedViewController?: (UIViewController *)self.navigationController)
                        presentViewController:alert animated:YES completion:nil];
            }
        } );
    }
    @catch (id exception) {
        err( @"During Post-Startup: %@", exception );
    }

    return YES;
}

- (BOOL)application:(UIApplication *)application openURL:(NSURL *)url
  sourceApplication:(NSString *)sourceApplication annotation:(id)annotation {

    // No URL?
    if (!url)
        return NO;

    // Arbitrary URL to mpsites data.
    [[[NSURLSession sharedSession] dataTaskWithURL:url completionHandler:
            ^(NSData *importedSitesData, NSURLResponse *response, NSError *error) {
                if (error)
                    MPError( error, @"While reading imported sites from %@.", url );

                if (!importedSitesData) {
                    PearlMainQueue( ^{
                        UIAlertController *alert = [UIAlertController alertControllerWithTitle:@"Error" message:
                                        strf( @"Master Password couldn't read the import sites.\n\n%@",
                                                (id)[error localizedDescription]?: error )
                                                                                preferredStyle:UIAlertControllerStyleAlert];
                        [alert addAction:[UIAlertAction actionWithTitle:@"Continue" style:UIAlertActionStyleCancel handler:nil]];
                        [self.navigationController presentViewController:alert animated:YES completion:nil];
                    } );
                    return;
                }

                NSString *importedSitesString = [[NSString alloc] initWithData:importedSitesData encoding:NSUTF8StringEncoding];
                if (!importedSitesString) {
                    PearlMainQueue( ^{
                        UIAlertController *alert = [UIAlertController alertControllerWithTitle:@"Error" message:
                                        @"Master Password couldn't understand the import file."
                                                                                preferredStyle:UIAlertControllerStyleAlert];
                        [alert addAction:[UIAlertAction actionWithTitle:@"Continue" style:UIAlertActionStyleCancel handler:nil]];
                        [self.navigationController presentViewController:alert animated:YES completion:nil];
                    } );
                    return;
                }

                [self importSites:importedSitesString];
            }] resume];

    return YES;
}

- (void)importSites:(NSString *)importData {

    if ([NSThread isMainThread]) {
        PearlNotMainQueue( ^{
            [self importSites:importData];
        } );
        return;
    }

    PearlOverlay *activityOverlay = [PearlOverlay showProgressOverlayWithTitle:@"Importing"];
    [self importSites:importData askImportPassword:^NSString *(NSString *userName) {
        return PearlAwait( ^(void (^setResult)(id)) {
            PearlMainQueue( ^{
                UIAlertController *alert = [UIAlertController alertControllerWithTitle:strf( @"Importing Sites For\n%@", userName ) message:
                                @"Enter the master password used to create this export file."
                                                                        preferredStyle:UIAlertControllerStyleAlert];
                [alert addTextFieldWithConfigurationHandler:^(UITextField *textField) {
                    textField.secureTextEntry = YES;
                }];
                [alert addAction:[UIAlertAction actionWithTitle:@"Import" style:UIAlertActionStyleDefault handler:^(UIAlertAction *action) {
                    setResult( alert.textFields.firstObject.text );
                }]];
                [alert addAction:[UIAlertAction actionWithTitle:@"Cancel" style:UIAlertActionStyleCancel handler:^(UIAlertAction *action) {
                    setResult( nil );
                }]];
                [self.navigationController presentViewController:alert animated:YES completion:nil];
            } );
        } );
    } askUserPassword:^NSString *(NSString *userName) {
        return PearlAwait( (id)^(void (^setResult)(id)) {
            PearlMainQueue( ^{
                UIAlertController *alert = [UIAlertController alertControllerWithTitle:strf( @"Master Password For\n%@", userName ) message:
                                @"Enter the current master password for this user."
                                                                        preferredStyle:UIAlertControllerStyleAlert];
                [alert addTextFieldWithConfigurationHandler:^(UITextField *textField) {
                    textField.secureTextEntry = YES;
                }];
                [alert addAction:[UIAlertAction actionWithTitle:@"Import" style:UIAlertActionStyleDefault handler:^(UIAlertAction *action) {
                    setResult( alert.textFields.firstObject.text );
                }]];
                [alert addAction:[UIAlertAction actionWithTitle:@"Cancel" style:UIAlertActionStyleCancel handler:^(UIAlertAction *action) {
                    setResult( nil );
                }]];
                [self.navigationController presentViewController:alert animated:YES completion:nil];
            } );
        } );
    }          result:^(NSError *error) {
        PearlMainQueue( ^{
            [activityOverlay cancelOverlayAnimated:YES];

            if (error && !(error.domain == NSCocoaErrorDomain && error.code == NSUserCancelledError)) {
                UIAlertController *controller = [UIAlertController alertControllerWithTitle:@"Error" message:[error localizedDescription]
                                                                             preferredStyle:UIAlertControllerStyleAlert];
                [controller addAction:[UIAlertAction actionWithTitle:@"Continue" style:UIAlertActionStyleCancel handler:nil]];
                [self.navigationController presentViewController:controller animated:YES completion:nil];
            }
        } );
    }];
}

- (void)applicationWillEnterForeground:(UIApplication *)application {

    inf( @"Will foreground" );

    [super applicationWillEnterForeground:application];

    [self.hangDetector start];
}

- (void)applicationDidBecomeActive:(UIApplication *)application {

    inf( @"Re-activated" );
    [[NSNotificationCenter defaultCenter] postNotificationName:MPCheckConfigNotification object:nil];

    PearlNotMainQueue( ^{
        NSString *importData = [UIPasteboard generalPasteboard].string;
        MPMarshalledFile *importFile = mpw_marshal_read( NULL, importData.UTF8String );
        if (importFile && importFile->error.type == MPMarshalSuccess && importFile->info->format != MPMarshalFormatNone) {
            PearlMainQueue( ^{
                UIAlertController *alert = [UIAlertController alertControllerWithTitle:@"Import Sites?" message:
                                @"We've detected Master Password import sites on your pasteboard, would you like to import them?"
                                                                        preferredStyle:UIAlertControllerStyleAlert];
                [alert addAction:[UIAlertAction actionWithTitle:@"Import Sites" style:UIAlertActionStyleDefault handler:
                        ^(UIAlertAction *action) {
                            [self importSites:importData];
                            [UIPasteboard generalPasteboard].string = @"";
                        }]];
                [alert addAction:[UIAlertAction actionWithTitle:@"No" style:UIAlertActionStyleCancel handler:nil]];
                [self.navigationController presentViewController:alert animated:YES completion:nil];
            } );
        }
        mpw_marshal_file_free( &importFile );
    } );

    [super applicationDidBecomeActive:application];
}

- (void)applicationDidReceiveMemoryWarning:(UIApplication *)application {

    inf( @"Received memory warning." );

    [super applicationDidReceiveMemoryWarning:application];
}

- (void)applicationDidEnterBackground:(UIApplication *)application {

    inf( @"Did background" );
    if (![[MPiOSConfig get].rememberLogin boolValue]) {
        [UIView setAnimationsEnabled:NO];
        [self signOut];
        [UIView setAnimationsEnabled:YES];
    }

    [self.hangDetector stop];

//    self.task = [application beginBackgroundTaskWithExpirationHandler:^{
//        [application endBackgroundTask:self.task];
//        dbg( @"background expiring" );
//    }];
//    PearlNotMainQueueOperation( ^{
//        NSString *pbstring = [UIPasteboard generalPasteboard].string;
//        while (YES) {
//            NSString *newString = [UIPasteboard generalPasteboard].string;
//            if (![newString isEqualToString:pbstring]) {
//                dbg( @"pasteboard changed to: %@", newString );
//                pbstring = newString;
//                NSURL *url = [NSURL URLWithString:pbstring];
//                if (url) {
//                    NSString *siteName = [url host];
//                }
//                MPKey *key = [MPiOSAppDelegate get].key;
//                if (key)
//                    [MPiOSAppDelegate managedObjectContextPerformBlock:^(NSManagedObjectContext *context) {
//                        NSFetchRequest<MPSiteEntity *>
//                                *fetchRequest = [NSFetchRequest fetchRequestWithEntityName:NSStringFromClass( [MPSiteEntity class] )];
//                        fetchRequest.sortDescriptors = @[
//                                [[NSSortDescriptor alloc] initWithKey:NSStringFromSelector( @selector( lastUsed ) ) ascending:NO]
//                        ];
//                        fetchRequest.fetchBatchSize = 2;
//                        fetchRequest.predicate = [NSPredicate predicateWithFormat:@"(name LIKE[cd] %@) AND user == %@", siteName,
//                                                                                  [[MPiOSAppDelegate get] activeUserOID]];
//                        NSError *error = nil;
//                        NSArray<MPSiteEntity *> *results = [fetchRequest execute:&error];
//                        dbg( @"site search, error: %@, results:\n%@", error, results );
//                        if ([results count]) {
//                            [UIPasteboard generalPasteboard].string = [[results firstObject] resolvePasswordUsingKey:key];
//                        }
//                    }];
//            }
//            [NSThread sleepForTimeInterval:5];
//        }
//    } );

    [super applicationDidEnterBackground:application];
}

#pragma mark - Behavior

- (void)showFeedbackWithLogs:(BOOL)logs forVC:(UIViewController *)viewController {

    if (![PearlEMail canSendMail]) {
        UIAlertController *alert = [UIAlertController alertControllerWithTitle:@"Feedback" message:
                        @"Have a question, comment, issue or just saying thanks?\n\n"
                        @"We'd love to hear what you think!\n"
                        @"help@masterpassword.app"
                                                                preferredStyle:UIAlertControllerStyleAlert];
        [alert addAction:[UIAlertAction actionWithTitle:@"Okay" style:UIAlertActionStyleCancel handler:nil]];
        [self.navigationController presentViewController:alert animated:YES completion:nil];
    }
    else if (logs) {
        UIAlertController *alert = [UIAlertController alertControllerWithTitle:@"Feedback" message:
                        @"Have a question, comment, issue or just saying thanks?\n\n"
                        @"If you're having trouble, it may help us if you can first reproduce the problem "
                        @"and then include log files in your message."
                                                                preferredStyle:UIAlertControllerStyleAlert];
        [alert addAction:[UIAlertAction actionWithTitle:@"Include Logs" style:UIAlertActionStyleDefault handler:^(UIAlertAction *action) {
            [self openFeedbackWithLogs:YES forVC:viewController];
        }]];
        [alert addAction:[UIAlertAction actionWithTitle:@"No Logs" style:UIAlertActionStyleDefault handler:^(UIAlertAction *action) {
            [self openFeedbackWithLogs:NO forVC:viewController];
        }]];
        [self.navigationController presentViewController:alert animated:YES completion:nil];
    }
    else
        [self openFeedbackWithLogs:NO forVC:viewController];
}

- (void)openFeedbackWithLogs:(BOOL)logs forVC:(UIViewController *)viewController {

    NSString *userName = [[MPiOSAppDelegate get] activeUserForMainThread].name;
    PearlLogLevel logLevel = PearlLogLevelInfo;
    if (logs && ([[MPConfig get].sendInfo boolValue] || [[MPiOSConfig get].traceMode boolValue]))
        logLevel = PearlLogLevelDebug;

    [[[PearlEMail alloc] initForEMailTo:@"Master Password Development <help@masterpassword.app"
                                subject:strf( @"Feedback for Master Password [%@]",
                                        [[PearlKeyChain deviceIdentifier] stringByDeletingMatchesOf:@"-.*"] )
                                   body:strf( @"\n\n\n"
                                              @"--\n"
                                              @"%@"
                                              @"Master Password %@, build %@",
                                           userName? ([userName stringByAppendingString:@"\n"]): @"",
                                           [PearlInfoPlist get].CFBundleShortVersionString,
                                           [PearlInfoPlist get].CFBundleVersion )

                            attachments:(logs
                                         ? [[PearlEMailAttachment alloc]
                                                 initWithContent:[[[PearlLogger get] formatMessagesWithLevel:logLevel]
                                                         dataUsingEncoding:NSUTF8StringEncoding]
                                                        mimeType:@"text/plain"
                                                        fileName:strf( @"%@-%@.log",
                                                                [[NSDateFormatter rfc3339DateFormatter] stringFromDate:[NSDate date]],
                                                                [PearlKeyChain deviceIdentifier] )]
                                         : nil), nil]
            showComposerForVC:viewController];
}

- (void)handleCoordinatorError:(NSError *)error {

    static dispatch_once_t once = 0;
    dispatch_once( &once, ^{
        PearlMainQueue( ^{
            UIAlertController *alert = [UIAlertController alertControllerWithTitle:@"Failed To Load Sites" message:
                            @"Master Password was unable to open your sites history.\n"
                            @"This may be due to corruption.  You can either reset Master Password and "
                            @"recreate your user, or E-Mail us your logs and leave your corrupt store as-is for now."
                                                                    preferredStyle:UIAlertControllerStyleAlert];
            [alert addAction:[UIAlertAction actionWithTitle:@"E-Mail Logs" style:UIAlertActionStyleDefault
                                                    handler:^(UIAlertAction *action) {
                                                        [self openFeedbackWithLogs:YES forVC:nil];
                                                    }]];
            [alert addAction:[UIAlertAction actionWithTitle:@"Reset" style:UIAlertActionStyleDefault handler:^(UIAlertAction *action) {
                [self deleteAndResetStore];
            }]];
            [alert addAction:[UIAlertAction actionWithTitle:@"Ignore" style:UIAlertActionStyleCancel handler:nil]];
            [self.navigationController presentViewController:alert animated:YES completion:nil];
        } );
    } );
}

- (void)showExportForVC:(UIViewController *)viewController {

    UIAlertController *alert = [UIAlertController alertControllerWithTitle:@"Exporting Your Sites" message:
                    @"An export is great for keeping a backup list of your accounts.\n\n"
                    @"When the file is ready, you will be able to mail it to yourself.\n"
                    @"You can open it with a text editor or with Master Password if you need to restore your list of sites."
                                                            preferredStyle:UIAlertControllerStyleAlert];
    [alert addAction:[UIAlertAction actionWithTitle:@"Export Sites" style:UIAlertActionStyleDefault handler:^(UIAlertAction *action) {
        UIAlertController *sheet = [UIAlertController alertControllerWithTitle:@"Show Passwords?" message:
                        @"Would you like to make all your passwords visible in the export file?\n\n"
                        @"A safe export will include all sites but make their passwords invisible.\n"
                        @"It is great as a backup and remains safe when fallen in the wrong hands."
                                                                preferredStyle:UIAlertControllerStyleActionSheet];
        [sheet addAction:[UIAlertAction actionWithTitle:@"Safe Export" style:UIAlertActionStyleDefault handler:^(UIAlertAction *action) {
            [self showExportRevealPasswords:NO forVC:viewController];
        }]];
        [sheet addAction:[UIAlertAction actionWithTitle:@"Show Passwords" style:UIAlertActionStyleDefault handler:^(UIAlertAction *action) {
            [self showExportRevealPasswords:YES forVC:viewController];
        }]];
        [sheet addAction:[UIAlertAction actionWithTitle:@"Cancel" style:UIAlertActionStyleCancel handler:nil]];
        [self.navigationController presentViewController:sheet animated:YES completion:nil];
    }]];
    [alert addAction:[UIAlertAction actionWithTitle:@"Cancel" style:UIAlertActionStyleCancel handler:nil]];
    [self.navigationController presentViewController:alert animated:YES completion:nil];
}

- (void)showExportRevealPasswords:(BOOL)revealPasswords forVC:(UIViewController *)viewController {

    if (![PearlEMail canSendMail]) {
        UIAlertController *alert = [UIAlertController alertControllerWithTitle:@"Cannot Send Mail" message:
                        @"Your device is not yet set up for sending mail.\n"
                        @"Close Master Password, go into Settings and add a Mail account."
                                                                preferredStyle:UIAlertControllerStyleAlert];
        [alert addAction:[UIAlertAction actionWithTitle:@"Okay" style:UIAlertActionStyleCancel handler:nil]];
        [self.navigationController presentViewController:alert animated:YES completion:nil];
        return;
    }

    [self exportSitesRevealPasswords:revealPasswords askExportPassword:^NSString *(NSString *userName) {
        return PearlAwait( ^(void (^setResult)(id)) {
            PearlMainQueue( ^{
                UIAlertController *alert = [UIAlertController alertControllerWithTitle:strf( @"Master Password For:\n%@", userName )
                                                                               message:@"Enter your master password to export the user."
                                                                        preferredStyle:UIAlertControllerStyleAlert];
                [alert addTextFieldWithConfigurationHandler:^(UITextField *textField) {
                    textField.secureTextEntry = YES;
                }];
                [alert addAction:[UIAlertAction actionWithTitle:@"Export" style:UIAlertActionStyleDefault handler:^(UIAlertAction *action) {
                    setResult( alert.textFields.firstObject.text );
                }]];
                [alert addAction:[UIAlertAction actionWithTitle:@"Cancel" style:UIAlertActionStyleCancel handler:^(UIAlertAction *action) {
                    setResult( nil );
                }]];
                [self.navigationController presentViewController:alert animated:YES completion:nil];
            } );
        } );
    }                         result:^(NSString *exportedUser, NSError *error) {
        if (!exportedUser || error) {
            MPError( error, @"Failed to export mpsites." );
            PearlMainQueue( ^{
                UIAlertController *alert = [UIAlertController alertControllerWithTitle:@"Export Error" message:[error localizedDescription]
                                                                        preferredStyle:UIAlertControllerStyleAlert];
                [alert addAction:[UIAlertAction actionWithTitle:@"Okay" style:UIAlertActionStyleCancel handler:nil]];
                [self.navigationController presentViewController:alert animated:YES completion:nil];
            } );
            return;
        }

        NSDateFormatter *exportDateFormatter = [NSDateFormatter new];
        [exportDateFormatter setDateFormat:@"yyyy'-'MM'-'dd"];
        NSString *exportFileName = strf( @"%@ (%@).mpsites",
                [self activeUserForMainThread].name, [exportDateFormatter stringFromDate:[NSDate date]] );

        UIAlertController *alert = [UIAlertController alertControllerWithTitle:@"Export Destination" message:nil
                                                                preferredStyle:UIAlertControllerStyleActionSheet];
        [alert addAction:[UIAlertAction actionWithTitle:@"Send As E-Mail" style:UIAlertActionStyleDefault handler:^(UIAlertAction *action) {
            NSString *message;
            if (revealPasswords)
                message = strf( @"Export of Master Password sites with passwords included.\n\n"
                                @"REMINDER: Make sure nobody else sees this file!  Passwords are visible!\n\n\n"
                                @"--\n"
                                @"%@\n"
                                @"Master Password %@, build %@",
                        [self activeUserForMainThread].name,
                        [PearlInfoPlist get].CFBundleShortVersionString,
                        [PearlInfoPlist get].CFBundleVersion );
            else
                message = strf( @"Backup of Master Password sites.\n\n\n"
                                @"--\n"
                                @"%@\n"
                                @"Master Password %@, build %@",
                        [self activeUserForMainThread].name,
                        [PearlInfoPlist get].CFBundleShortVersionString,
                        [PearlInfoPlist get].CFBundleVersion );

            [PearlEMail sendEMailTo:nil fromVC:viewController subject:@"Master Password Export" body:message
                        attachments:[[PearlEMailAttachment alloc] initWithContent:[exportedUser dataUsingEncoding:NSUTF8StringEncoding]
                                                                         mimeType:@"text/plain"
                                                                         fileName:exportFileName], nil];
            return;
        }]];
        [alert addAction:[UIAlertAction actionWithTitle:@"Share / Export" style:UIAlertActionStyleDefault handler:^(UIAlertAction *action) {
            NSURL *applicationSupportURL = [[[NSFileManager defaultManager] URLsForDirectory:NSApplicationSupportDirectory
                                                                                   inDomains:NSUserDomainMask] lastObject];
            NSURL *exportURL = [[applicationSupportURL
                    URLByAppendingPathComponent:[NSBundle mainBundle].bundleIdentifier isDirectory:YES]
                    URLByAppendingPathComponent:exportFileName isDirectory:NO];
            NSError *writeError = nil;
            if (![[exportedUser dataUsingEncoding:NSUTF8StringEncoding]
                    writeToURL:exportURL options:NSDataWritingFileProtectionComplete error:&writeError])
                MPError( writeError, @"Failed to write export data to URL %@.", exportURL );
            else {
                self.interactionController = [UIDocumentInteractionController interactionControllerWithURL:exportURL];
                self.interactionController.UTI = @"com.lyndir.masterpassword.sites";
                self.interactionController.delegate = self;
                [self.interactionController presentOpenInMenuFromRect:CGRectZero inView:viewController.view animated:YES];
            }
        }]];
        [alert addAction:[UIAlertAction actionWithTitle:@"Continue" style:UIAlertActionStyleCancel handler:nil]];
        [self.navigationController presentViewController:alert animated:YES completion:nil];
    }];
}

- (void)changeMasterPasswordFor:(MPUserEntity *)user saveInContext:(NSManagedObjectContext *)moc didResetBlock:(void ( ^ )(void))didReset {

    PearlMainQueue( ^{
        UIAlertController *alert = [UIAlertController alertControllerWithTitle:@"Changing Master Password" message:
                        @"If you continue, you'll be able to set a new master password.\n\n"
                        @"Changing your master password will cause all your generated passwords to change!\n"
                        @"Changing the master password back to the old one will cause your passwords to revert as well."
                                                                preferredStyle:UIAlertControllerStyleAlert];
        [alert addAction:[UIAlertAction actionWithTitle:@"Abort" style:UIAlertActionStyleDefault handler:^(UIAlertAction *action) {
            [moc performBlockAndWait:^{
                inf( @"Clearing keyID for user: %@.", user.userID );
                user.keyID = nil;
                [self forgetSavedKeyFor:user];
                [moc saveToStore];
            }];

            [self signOut];
            if (didReset)
                didReset();
        }]];
        [alert addAction:[UIAlertAction actionWithTitle:@"Abort" style:UIAlertActionStyleCancel handler:nil]];
        [self.navigationController presentViewController:alert animated:YES completion:nil];
    } );
}

#pragma mark - UIDocumentInteractionControllerDelegate

- (void)documentInteractionController:(UIDocumentInteractionController *)controller didEndSendingToApplication:(NSString *)application {

//    self.interactionController = nil;
}

#pragma mark - PearlConfigDelegate

- (void)didUpdateConfigForKey:(SEL)configKey fromValue:(id)value {

    [[NSNotificationCenter defaultCenter] postNotificationName:MPCheckConfigNotification object:NSStringFromSelector( configKey )];
}

- (void)updateConfigKey:(NSString *)key {

    // Trace mode
    [PearlLogger get].historyLevel = [[MPiOSConfig get].traceMode boolValue]? PearlLogLevelTrace: PearlLogLevelInfo;

    // Send info
    if ([[MPConfig get].sendInfo boolValue]) {
        [Countly.sharedInstance giveConsentForAllFeatures];

        if ([PearlLogger get].printLevel > PearlLogLevelInfo)
            [PearlLogger get].printLevel = PearlLogLevelInfo;

        NSMutableDictionary *prefs = [NSMutableDictionary new];
        prefs[@"rememberLogin"] = [MPConfig get].rememberLogin;
        prefs[@"sendInfo"] = [MPConfig get].sendInfo;
        prefs[@"helpHidden"] = [MPiOSConfig get].helpHidden;
        prefs[@"showQuickStart"] = [MPiOSConfig get].showSetup;
        prefs[@"firstRun"] = [PearlConfig get].firstRun;
        prefs[@"launchCount"] = [PearlConfig get].launchCount;
        prefs[@"askForReviews"] = [PearlConfig get].askForReviews;
        prefs[@"reviewAfterLaunches"] = [PearlConfig get].reviewAfterLaunches;
        prefs[@"reviewedVersion"] = [PearlConfig get].reviewedVersion;
        prefs[@"simulator"] = @([PearlDeviceUtils isSimulator]);
        prefs[@"encrypted"] = @([PearlDeviceUtils isAppEncrypted]);
        prefs[@"jailbroken"] = @([PearlDeviceUtils isJailbroken]);
        prefs[@"platform"] = [PearlDeviceUtils platform];
#ifdef APPSTORE
        prefs[@"reviewedVersion"] = @([PearlDeviceUtils isAppEncrypted]);
#else
        prefs[@"reviewedVersion"] = @(YES);
#endif
        PearlMainQueueOperation( ^{
            if (![[SentryClient.sharedClient.extra dictionaryWithValuesForKeys:prefs.allKeys] isEqualToDictionary:prefs]) {
                NSMutableDictionary *extra = [SentryClient.sharedClient.extra mutableCopy]?: [NSMutableDictionary dictionary];
                [extra addEntriesFromDictionary:prefs];
                SentryClient.sharedClient.extra = extra;
            }
        } );
    }
    else {
        [Countly.sharedInstance cancelConsentForAllFeatures];
    }
}

@end